Fix the issue of Arch-Intent

This commit is contained in:
Shuguang Chen 2024-12-06 11:53:46 -08:00
parent 4fcfd83639
commit 818ec0a9db
4 changed files with 7 additions and 9 deletions

View file

@ -108,7 +108,9 @@ class ArchIntentHandler(ArchBaseHandler):
extra_body=self.generation_params,
)
model_response = Message(content=model_response, tool_calls=[])
model_response = Message(
content=model_response.choices[0].message.content, tool_calls=[]
)
chat_completion_response = ChatCompletionResponse(
choices=[Choice(message=model_response)], model=self.model_name