fix: convert_anthropic_tool_to_databricks_tool

This commit is contained in:
Ishaan Jaff
2025-09-11 20:12:01 -07:00
parent 32d87c242b
commit 698a2dd382
@@ -169,18 +169,20 @@ class DatabricksConfig(DatabricksBase, OpenAILikeChatConfig, AnthropicConfig):
if tool is None:
return None
kwags: dict = {
# Build DatabricksFunction explicitly to avoid parameter conflicts
function_params: DatabricksFunction = {
"name": tool["name"],
"parameters": cast(dict, tool.get("input_schema") or {})
}
# Only add description if it exists
description = tool.get("description")
if description is not None:
kwags["description"] = cast(Union[dict, str], description)
function_params["description"] = cast(Union[dict, str], description)
return DatabricksTool(
type="function",
function=DatabricksFunction(name=tool["name"], **kwags),
function=function_params,
)
def _map_openai_to_dbrx_tool(self, model: str, tools: List) -> List[DatabricksTool]: