Added logprobs handling in choice response and user can get the logit… #1236
+44
−2
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
This update addresses the need to obtain the model output logits when using the Camel framework with vLLM deployment. Since the original Camel framework did not provide an interface to retrieve logits, two properties,
logprobs
andtop_logprobs
, were added to thevllm_config
. Additionally, thechat_agent
response handling was modified to allow logging and retrieving the output logits, enabling more detailed scoring information.Motivation and Context
In the existing Camel framework, there was no direct way to access the logits from model outputs, which are crucial for deploy the prm agent in camel (some prm use the logits of the output as the score). This change was made to provide a better interface for obtaining these logits, improving transparency and control over the model’s output, especially useful for tasks like scoring.
Types of changes
What types of changes does your code introduce? Put an
x
in all the boxes that apply:Implemented Tasks
logprobs
andtop_logprobs
properties tovllm_config
.chat_agent
response handling to capture and log logits.Checklist
Go over all the following points, and put an
x
in all the boxes that apply. If you are unsure about any of these, don't hesitate to ask. We are here to help!