From d01492dce05c2a4c06c490e1b83f669f532e04ef Mon Sep 17 00:00:00 2001 From: Eric Huang Date: Tue, 11 Mar 2025 13:49:21 -0700 Subject: [PATCH] chore: remove litellm type conversion Summary: Test Plan: --- src/llama_stack_client/lib/agents/client_tool.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/llama_stack_client/lib/agents/client_tool.py b/src/llama_stack_client/lib/agents/client_tool.py index 71da2b6b..0d15dade 100644 --- a/src/llama_stack_client/lib/agents/client_tool.py +++ b/src/llama_stack_client/lib/agents/client_tool.py @@ -197,8 +197,7 @@ def get_params_definition(self) -> Dict[str, Parameter]: params[name] = Parameter( name=name, description=param_doc or f"Parameter {name}", - # Hack: litellm/openai expects "string" for str type - parameter_type=type_hint.__name__ if type_hint.__name__ != "str" else "string", + parameter_type=type_hint.__name__, default=(param.default if param.default != inspect.Parameter.empty else None), required=is_required, )