Refactor VectorStoreInfoComponent and VectorStoreToolkitComponent

This commit refactors the VectorStoreInfoComponent and VectorStoreToolkitComponent classes. The changes include importing the VectorStoreInfo class from the correct module and adding a new parameter, llm, to the build method of the VectorStoreToolkitComponent class.

src/backend/langflow/components/toolkits/VectorStoreInfo.py

@@ -2,7 +2,7 @@
 from langflow import CustomComponent
 from langchain.vectorstores import VectorStore
 from typing import Union, Callable
-from langflow.field_typing import Chain
+from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreInfo
 
 class VectorStoreInfoComponent(CustomComponent):
     display_name = "VectorStoreInfo"
@@ -20,14 +20,5 @@ class VectorStoreInfoComponent(CustomComponent):
         vectorstore: VectorStore,
         description: str,
         name: str,
-    ) -> Union[Chain, Callable]:
-        # Since the actual implementation of VectorStoreInfo is not provided, this is a placeholder
-        # Replace VectorStoreInfo with the actual class that should be instantiated
-        # This is a hypothetical class, actual implementation may vary
-        class VectorStoreInfo:
-            def __init__(self, vectorstore, description, name):
-                self.vectorstore = vectorstore
-                self.description = description
-                self.name = name
-
+    ) -> Union[VectorStoreInfo, Callable]:
         return VectorStoreInfo(vectorstore=vectorstore, description=description, name=name)

src/backend/langflow/components/toolkits/VectorStoreToolkit.py

@@ -2,6 +2,9 @@
 from langflow import CustomComponent
 from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreToolkit
 from langchain.agents.agent_toolkits.vectorstore.toolkit import VectorStoreInfo
+from langflow.field_typing import (
+    BaseLanguageModel,
+)
 from langflow.field_typing import (
     Tool,
 )
@@ -13,10 +16,12 @@ class VectorStoreToolkitComponent(CustomComponent):
     def build_config(self):
         return {
             "vectorstore_info": {"display_name": "Vector Store Info"},
+            "llm": {"display_name": "LLM"},
         }
 
     def build(
         self,
         vectorstore_info: VectorStoreInfo,
+        llm: BaseLanguageModel,
     ) -> Tool:
-        return VectorStoreToolkit(vectorstore_info=vectorstore_info)
+        return VectorStoreToolkit(vectorstore_info=vectorstore_info,llm=llm)