From 0ebf7d35e478cbc9f1596f6bea4bf3151bfed41c Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Tue, 9 Jan 2024 19:27:04 -0300 Subject: [PATCH] Update VectorStoreToolkit imports --- .../langflow/components/toolkits/VectorStoreToolkit.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/components/toolkits/VectorStoreToolkit.py b/src/backend/langflow/components/toolkits/VectorStoreToolkit.py index b83246b07..236e2146f 100644 --- a/src/backend/langflow/components/toolkits/VectorStoreToolkit.py +++ b/src/backend/langflow/components/toolkits/VectorStoreToolkit.py @@ -1,8 +1,8 @@ from langflow import CustomComponent -from langchain.toolkits import VectorStoreToolkit +from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreToolkit +from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreInfo from langflow.field_typing import ( - VectorStore, Tool, ) @@ -17,6 +17,6 @@ class VectorStoreToolkitComponent(CustomComponent): def build( self, - vectorstore_info: VectorStore, + vectorstore_info: VectorStoreInfo, ) -> Tool: return VectorStoreToolkit(vectorstore_info=vectorstore_info)