github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-30 19:49:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

google-genai[patch]: no error for FunctionMessage (#17215)

Browse Source 
Both should eventually match this:
https://github.com/langchain-ai/langchain/blob/master/libs/partners/google-vertexai/langchain_google_vertexai/chat_models.py#L179

But seems undocumented / can't find types in genai package
This commit is contained in:
Erick Friis 2024-02-07 17:14:50 -08:00 committed by GitHub
parent 2ecf318218
commit 927ab77d6e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/partners/google-genai/langchain_google_genai/chat_models.py
View File

@ -39,6 +39,7 @@ from langchain_core.messages import (
BaseMessage,
ChatMessage,
ChatMessageChunk,
FunctionMessage,
HumanMessage,
HumanMessageChunk,
SystemMessage,
@ -326,14 +327,20 @@ llm = ChatGoogleGenerativeAI(model="gemini-pro", convert_system_message_to_human
continue
elif isinstance(message, AIMessage):
role = "model"
# TODO: Handle AImessage with function call
parts = _convert_to_parts(message.content)
elif isinstance(message, HumanMessage):
role = "user"
parts = _convert_to_parts(message.content)
elif isinstance(message, FunctionMessage):
role = "user"
# TODO: Handle FunctionMessage
parts = _convert_to_parts(message.content)
else:
raise ValueError(
f"Unexpected message with type {type(message)} at the position {i}."
)
parts = _convert_to_parts(message.content)
if raw_system_message:
if role == "model":
raise ValueError(