github/genai-toolbox
1
0
Fork 0
mirror of https://github.com/googleapis/genai-toolbox.git synced 2026-01-09 15:38:08 -05:00
Code Issues Packages Projects Releases 14 Wiki Activity

doc(langchain-sdk): Update LangChain SDK example to use Gemini Pro model for better tool selection. (#196)

Browse Source
This commit is contained in:
Anubhav Dhawan
2025-01-10 11:02:23 +05:30
committed by GitHub
parent cccc34d019
commit c58f0104f8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sdks/langchain/README.md
View File

@@ -90,7 +90,7 @@ toolkit.
```py
from langchain_google_vertexai import ChatVertexAI
model = ChatVertexAI(model="gemini-1.5-flash-001")
model = ChatVertexAI(model="gemini-1.5-pro-002")
# Initialize agent with tools
agent = model.bind_tools(tools)
@@ -123,7 +123,7 @@ def call_model(state: MessagesState):
# We return a list, because this will get added to the existing list
return {"messages": [response]}
model = ChatVertexAI(model="gemini-1.5-flash-001")
model = ChatVertexAI(model="gemini-1.5-pro-002")
builder = StateGraph(MessagesState)
tool_node = ToolNode(tools)