From 67e974a5e74b755ef9890887d17a3b41727efe0f Mon Sep 17 00:00:00 2001 From: Leonid Kuligin Date: Tue, 13 Feb 2024 21:18:43 +0100 Subject: [PATCH] fix linting --- libs/genai/langchain_google_genai/chat_models.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/libs/genai/langchain_google_genai/chat_models.py b/libs/genai/langchain_google_genai/chat_models.py index 37cc5f34..2980ffa2 100644 --- 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 @@ -349,7 +351,7 @@ def _parts_to_content(parts: List[genai.types.PartType]) -> Union[List[dict], st elif not parts: logger.warning("Gemini produced an empty response.") return "" - messages = [] + messages: List[Union[str, dict]] = [] for part in parts: if part.text is not None: messages.append(