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

Update VectorStoreRouterToolkit to use VectorStoreInfo

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

5
src/backend/langflow/components/toolkits/VectorStoreRouterToolkit.py
View file

@ -1,7 +1,8 @@
from langflow import CustomComponent from langflow import CustomComponent
from typing import List from typing import List
from langchain.vectorstores import VectorStore from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreRouterToolkit
from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreInfo
class VectorStoreRouterToolkitComponent(CustomComponent): class VectorStoreRouterToolkitComponent(CustomComponent):
display_name = "VectorStoreRouterToolkit" display_name = "VectorStoreRouterToolkit"
@ -14,7 +15,7 @@ class VectorStoreRouterToolkitComponent(CustomComponent):
def build( def build(
self, self,
vectorstores: List[VectorStore], vectorstores: List[VectorStoreInfo],
): ):
# Assuming the class `VectorStoreRouterToolkit` exists within a module, but since there # Assuming the class `VectorStoreRouterToolkit` exists within a module, but since there
# is no further information provided about the module structure, I will assume it is # is no further information provided about the module structure, I will assume it is