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

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
pull/17141/head
Erick Friis 8 months ago committed by GitHub
parent 2ecf318218
commit 927ab77d6e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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(

Loading…
Cancel
Save