Merge pull request #109 from UiPath/chore/folder-support-vectorstore

chore: vectorstore folder support

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.111"
+version = "0.0.112"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath_langchain/vectorstores/context_grounding_vectorstore.py

@@ -44,6 +44,7 @@ class ContextGroundingVectorStore(VectorStore):
     def __init__(
         self,
         index_name: str,
+        folder_path: Optional[str] = None,
         uipath_sdk: Optional[UiPath] = None,
     ):
         """Initialize the ContextGroundingVectorStore.
@@ -53,6 +54,7 @@ def __init__(
             uipath_sdk: Optional SDK instance to use. If not provided, a new instance will be created.
         """
         self.index_name = index_name
+        self.folder_path = folder_path
         self.sdk = uipath_sdk or UiPath()
 
     def similarity_search_with_score(
@@ -73,6 +75,7 @@ def similarity_search_with_score(
             name=self.index_name,
             query=query,
             number_of_results=k,
+            folder_path=self.folder_path,
         )
 
         # Convert the results to Documents with scores
@@ -142,6 +145,7 @@ async def asimilarity_search_with_score(
             name=self.index_name,
             query=query,
             number_of_results=k,
+            folder_path=self.folder_path,
         )
 
         # Convert the results to Documents with scores