jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update VectorStoreToolkit imports

Browse source
This commit is contained in:
anovazzi1 2024-01-09 19:27:04 -03:00
commit 0ebf7d35e4
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/components/toolkits/VectorStoreToolkit.py
View file

@ -1,8 +1,8 @@
from langflow import CustomComponent 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 ( from langflow.field_typing import (
VectorStore,
Tool, Tool,
) )
@ -17,6 +17,6 @@ class VectorStoreToolkitComponent(CustomComponent):
def build( def build(
self, self,
vectorstore_info: VectorStore, vectorstore_info: VectorStoreInfo,
) -> Tool: ) -> Tool:
return VectorStoreToolkit(vectorstore_info=vectorstore_info) return VectorStoreToolkit(vectorstore_info=vectorstore_info)