[Gemini] InvalidArgument: 400 Request contains an invalid argument. During task with name 'extract' and id 'XXXXX'

[vikaskapur]

Code:

class Memory(BaseModel):
    content: str = Field(description="The main content of the memory. For example: User expressed interest in learning about French.")

model = ChatVertexAI(model="gemini-pro", temperature=0)

# Create the extractor
trustcall_extractor = create_extractor(
    model,
    tools=[Memory],
    tool_choice="Memory",
    enable_inserts=True,
)

# Instruction
instruction = """Extract memories from the following conversation:"""

# Conversation
conversation = [HumanMessage(content="Hi, I'm Lance."), 
                AIMessage(content="Nice to meet you, Lance."), 
                HumanMessage(content="This morning I had a nice bike ride in San Francisco.")]

# Invoke the extractor
result = trustcall_extractor.invoke({"messages": [SystemMessage(content=instruction)] + conversation})

Error:

---> 78     raise exceptions.from_grpc_error(exc) from exc

InvalidArgument: 400 Request contains an invalid argument.
During task with name 'extract' and id '3bd5381e-9d8a-8177-44c2-58bd27dfc961'

[QuinRiva]

The Json structure that trustcall uses is not compatible with Gemini. I've created a pull request to add Gemini support

[nikhilpi]

@hinthornw what would be the timeframe on reviewing this? I've been using Gemini with Trustcall, and it recently broke.