ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix mypy linting

Browse Source
This commit is contained in:
Ishaan Jaff 2025-09-10 16:17:22 -07:00
parent 199c262e33
commit d78ed53cbb
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
litellm/llms/databricks/chat/transformation.py
View File

@ -169,13 +169,14 @@ class DatabricksConfig(DatabricksBase, OpenAILikeChatConfig, AnthropicConfig):
if tool is None:
return None
kwags = {
"name":tool["name"],
"parameters":cast(dict, tool.get("input_schema") or {})
kwags: dict = {
"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 description is not None:
kwags["description"] = cast(Union[dict, str], description)
return DatabricksTool(
type="function",
@ -336,8 +337,8 @@ class DatabricksConfig(DatabricksBase, OpenAILikeChatConfig, AnthropicConfig):
elif isinstance(content, list):
content_str = ""
for item in content:
if item["type"] == "text":
content_str += item["text"]
if item.get("type") == "text":
content_str += item.get("text", "")
return content_str
else:
raise Exception(f"Unsupported content type: {type(content)}")
@ -366,8 +367,9 @@ class DatabricksConfig(DatabricksBase, OpenAILikeChatConfig, AnthropicConfig):
reasoning_content: Optional[str] = None
if isinstance(content, list):
for item in content:
if item["type"] == "reasoning":
for sum in item["summary"]:
if item.get("type") == "reasoning":
summary_list = item.get("summary", [])
for sum in summary_list:
if reasoning_content is None:
reasoning_content = ""
reasoning_content += sum["text"]