Skip to content

Commit

Permalink
google-genai[patch]: fix new core typing (#16988)
Browse files Browse the repository at this point in the history
  • Loading branch information
efriis authored Feb 4, 2024
1 parent 35446c8 commit 8490511
Showing 1 changed file with 4 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -341,15 +341,17 @@ 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[Dict[Any, Any], str]]]:
"""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
return parts[0].text
elif not parts:
logger.warning("Gemini produced an empty response.")
return ""
messages = []
messages: List[Union[Dict[Any, Any], str]] = []
for part in parts:
if part.text is not None:
messages.append(
Expand Down

0 comments on commit 8490511

Please sign in to comment.