From cf676e7aeff459330c5df604a636955d494aff45 Mon Sep 17 00:00:00 2001 From: TomeHirata Date: Mon, 1 Sep 2025 17:51:24 +0900 Subject: [PATCH] fix mypy --- litellm/llms/databricks/chat/transformation.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/litellm/llms/databricks/chat/transformation.py b/litellm/llms/databricks/chat/transformation.py index 9330b01923..d3df5bbf36 100644 --- a/litellm/llms/databricks/chat/transformation.py +++ b/litellm/llms/databricks/chat/transformation.py @@ -382,20 +382,18 @@ class DatabricksConfig(DatabricksBase, OpenAILikeChatConfig, AnthropicConfig): ) -> Optional[List[Any]]: if content is None: return None - citations: Optional[List[Any]] = None + citations = [] if isinstance(content, list): for item in content: text = item.get("text", None) - if item.get("citations") is not None: - if citations is None: - citations = [] + if citations_item := item.get("citations"): citations.append( [ {**citation, "supported_text": text} - for citation in item["citations"] + for citation in citations_item ] ) - return citations + return citations or None def _transform_dbrx_choices( self, choices: List[DatabricksChoice], json_mode: Optional[bool] = None