Skip to content

Commit e4d7a47

Browse files
authored
Update prebuilt_langchain_agent_template.py
1 parent c2ca355 commit e4d7a47

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

samples/langchain_on_vertexai/prebuilt_langchain_agent_template.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def similarity_search(query: str) -> list[Document]:
9191
DISPLAY_NAME = os.getenv("DISPLAY_NAME") or "PrebuiltAgent"
9292

9393
remote_app = reasoning_engines.ReasoningEngine.create(
94-
reasoning_engines.LangchainAgent(
94+
reasoning_engines.LangchainAgent( # type: ignore[arg-type]
9595
model="gemini-2.0-flash-001",
9696
tools=[similarity_search], # type: ignore[list-item]
9797
model_kwargs={
@@ -102,6 +102,6 @@ def similarity_search(query: str) -> list[Document]:
102102
display_name=DISPLAY_NAME,
103103
sys_version="3.11",
104104
extra_packages=["config.py"],
105-
) # type: ignore[arg-type]
105+
)
106106

107107
print(remote_app.query(input="movies about engineers")) # type: ignore[attr-defined]

0 commit comments

Comments
 (0)