docs: fix with_structured_output examples (#80)

langchain-ai · Mar 24, 2024 · 5f795a4 · 5f795a4
1 parent 51bf9ef
commit 5f795a4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/libs/vertexai/langchain_google_vertexai/chat_models.py b/libs/vertexai/langchain_google_vertexai/chat_models.py
@@ -646,7 +646,7 @@ class AnswerWithJustification(BaseModel):
                     answer: str
                     justification: str
 
-                llm = ChatVertexAI(model="gemini-pro", temperature=0)
+                llm = ChatVertexAI(model_name="gemini-pro", temperature=0)
                 structured_llm = llm.with_structured_output(AnswerWithJustification)
 
                 structured_llm.invoke("What weighs more a pound of bricks or a pound of feathers")
@@ -665,7 +665,7 @@ class AnswerWithJustification(BaseModel):
                     answer: str
                     justification: str
 
-                llm = ChatVertexAI(model="gemini-pro", temperature=0)
+                llm = ChatVertexAI(model_name="gemini-pro", temperature=0)
                 structured_llm = llm.with_structured_output(AnswerWithJustification, include_raw=True)
 
                 structured_llm.invoke("What weighs more a pound of bricks or a pound of feathers")
@@ -688,7 +688,7 @@ class AnswerWithJustification(BaseModel):
                     justification: str
 
                 dict_schema = convert_to_openai_tool(AnswerWithJustification)
-                llm = ChatVertexAI(model="gemini-pro", temperature=0)
+                llm = ChatVertexAI(model_name="gemini-pro", temperature=0)
                 structured_llm = llm.with_structured_output(dict_schema)
 
                 structured_llm.invoke("What weighs more a pound of bricks or a pound of feathers")