scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

fix: mypy error

Browse source
This commit is contained in:
swarnabhasinha 2025-09-10 14:01:31 +05:30
parent 550d6ad98f
commit 47efc7f846
No known key found for this signature in database
GPG key ID: 9B0C3EA24F21722E
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/llms/databricks/chat/transformation.py
View file

@ -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]: