Gemini with structured output using langchain

gemini_structured.py

from langchain_google_genai import ChatGoogleGenerativeAI
import os

llm = ChatGoogleGenerativeAI(
    model="gemini-2.0-flash-001",
    temperature=0,
    max_tokens=None,
    timeout=None,
    max_retries=2,
)

from pydantic import BaseModel

class Sentence(BaseModel):
  original: str
  translated: str
    
messages = [
    (
        "system",
        "You are a helpful assistant that translates English to French. Translate the user sentence.",
    ),
    ("human", "I love programming."),
]

structured_llm = llm.with_structured_output(Sentence)
ai_msg = structured_llm.invoke(messages)
ai_msg