apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

add else

Browse source
This commit is contained in:
cotran 2024-12-06 14:34:51 -08:00
parent 6dc92fbbc1
commit d5c034ea79
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
model_server/app/model_handler/function_calling.py
View file

@ -79,8 +79,10 @@ class ArchIntentHandler(ArchBaseHandler):
Returns:
bool: A boolean value to indicate if any intent match with prompts or not
"""
return content.choices[0].message.content == "Yes"
if hasattr(content.choices[0].message, "content"):
return content.choices[0].message.content == "Yes"
else:
return False
@override
async def chat_completion(self, req: ChatMessage) -> ChatCompletionResponse:
@ -318,7 +320,7 @@ class ArchFunctionHandler(ArchBaseHandler):
if required_param not in func_args:
is_valid = False
error_tool_call = tool_call
error_message = f"`{required_param}` is requried by the function `{func_name}` but not found in the tool call!"
error_message = f"`{required_param}` is requiried by the function `{func_name}` but not found in the tool call!"
return is_valid, error_tool_call, error_message
# Verify the data type of each parameter in the tool calls