fix linting

langchain-ai · Feb 13, 2024 · 6ac4cf9 · 6ac4cf9
1 parent f4efa8a
commit 6ac4cf9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/genai/langchain_google_genai/chat_models.py b/libs/genai/langchain_google_genai/chat_models.py
@@ -341,7 +341,9 @@ def _parse_chat_history(
     return messages
 
 
-def _parts_to_content(parts: List[genai.types.PartType]) -> Union[List[dict], str]:
+def _parts_to_content(
+    parts: List[genai.types.PartType],
+) -> Union[str, List[Union[str, dict]]]:
     """Converts a list of Gemini API Part objects into a list of LangChain messages."""
     if len(parts) == 1 and parts[0].text is not None and not parts[0].inline_data:
         # Simple text response. The typical response