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

Add VectorStoreInfo import to VectorStoreAgent.py

Browse source
This commit is contained in:
anovazzi1 2024-01-09 19:16:55 -03:00
commit 5e94bfb8b2
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/agents/VectorStoreAgent.py
View file

@ -1,8 +1,9 @@
from langflow import CustomComponent
from langchain.agents import AgentExecutor
from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreInfo
from typing import Union, Callable
from langflow.field_typing import BaseLanguageModel, VectorStore
from langflow.field_typing import BaseLanguageModel
class VectorStoreAgentComponent(CustomComponent):
display_name = "VectorStoreAgent"
@ -17,6 +18,6 @@ class VectorStoreAgentComponent(CustomComponent):
def build(
self,
llm: BaseLanguageModel,
vectorstoreinfo: VectorStore,
vectorstoreinfo: VectorStoreInfo,
) -> Union[AgentExecutor, Callable]:
return AgentExecutor(llm=llm, vectorstore=vectorstoreinfo)