diff --git a/litellm/llms/databricks/chat/transformation.py b/litellm/llms/databricks/chat/transformation.py index 0f3530f85da..3522b6006d7 100644 --- a/litellm/llms/databricks/chat/transformation.py +++ b/litellm/llms/databricks/chat/transformation.py @@ -169,17 +169,19 @@ class DatabricksConfig(DatabricksBase, OpenAILikeChatConfig, AnthropicConfig): if tool is None: return None - kwags = { - "name":tool["name"], - "parameters":cast(dict, tool.get("input_schema") or {}) + function_data: DatabricksFunction = { + "name": tool["name"], + "parameters": cast(dict, tool.get("input_schema") or {}), } if tool.get("description"): - kwags["description"] = tool.get("description") + description = tool.get("description") + if isinstance(description, (dict, str)): + function_data["description"] = description return DatabricksTool( type="function", - function=DatabricksFunction(**kwags), + function=function_data, ) def _map_openai_to_dbrx_tool(self, model: str, tools: List) -> List[DatabricksTool]: