File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
samples/langchain_on_vertexai Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -91,7 +91,7 @@ def similarity_search(query: str) -> list[Document]:
9191DISPLAY_NAME = os .getenv ("DISPLAY_NAME" ) or "PrebuiltAgent"
9292
9393remote_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
107107print (remote_app .query (input = "movies about engineers" )) # type: ignore[attr-defined]
You can’t perform that action at this time.
0 commit comments