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