From 102e0e0f932ab9699a95e6541ae9b0eeb335d827 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Tue, 9 Jan 2024 11:51:50 -0300 Subject: [PATCH] Update imports in VectorStoreRouterAgent.py --- .../langflow/components/agents/VectorStoreRouterAgent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/agents/VectorStoreRouterAgent.py b/src/backend/langflow/components/agents/VectorStoreRouterAgent.py index 6aec821c8..d613f3080 100644 --- a/src/backend/langflow/components/agents/VectorStoreRouterAgent.py +++ b/src/backend/langflow/components/agents/VectorStoreRouterAgent.py @@ -1,7 +1,7 @@ from langflow import CustomComponent -from langchain.llms import BaseLanguageModel -from langchain.vectorstores import VectorStoreRouterToolkit +from langchain_core.language_models.base import BaseLanguageModel +from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreRouterToolkit from langchain.agents import AgentExecutor from typing import Callable