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, )