fix test_anthropic_function_call_with_no_schema

This commit is contained in:
Ishaan Jaff 2024-11-12 18:02:24 -08:00
parent 186a679243
commit 3ccbd5bb7b

View file

@ -160,15 +160,17 @@ class AnthropicConfig:
returned_tool: Optional[AllAnthropicToolsValues] = None
if tool["type"] == "function" or tool["type"] == "custom":
_input_function_parameters: dict = (
tool["function"].get("parameters", None) or {}
)
_tool_input_schema: AnthropicInputSchema = AnthropicInputSchema(
**_input_function_parameters
_input_schema: dict = tool["function"].get(
"parameters",
{
"type": "object",
"properties": {},
},
)
input_schema: AnthropicInputSchema = AnthropicInputSchema(**_input_schema)
_tool = AnthropicMessagesTool(
name=tool["function"]["name"],
input_schema=_tool_input_schema,
input_schema=input_schema,
)
_description = tool["function"].get("description")