Skip to content

Commit f81314d

Browse files
authored
doc(langchain-sdk): Update LangChain SDK example to use Gemini Pro model for better tool selection. (#196)
1 parent 3c20539 commit f81314d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ toolkit.
9090
```py
9191
from langchain_google_vertexai import ChatVertexAI
9292

93-
model = ChatVertexAI(model="gemini-1.5-flash-001")
93+
model = ChatVertexAI(model="gemini-1.5-pro-002")
9494

9595
# Initialize agent with tools
9696
agent = model.bind_tools(tools)
@@ -123,7 +123,7 @@ def call_model(state: MessagesState):
123123
# We return a list, because this will get added to the existing list
124124
return {"messages": [response]}
125125

126-
model = ChatVertexAI(model="gemini-1.5-flash-001")
126+
model = ChatVertexAI(model="gemini-1.5-pro-002")
127127
builder = StateGraph(MessagesState)
128128
tool_node = ToolNode(tools)
129129

0 commit comments

Comments
 (0)