Prompt answer fetch doc id filter issue (#34)

Prompt answer fetch doc if filter issue Co-authored-by: Neha <[email protected]>
Zipstack · Apr 9, 2024 · 1dd4525 · 1dd4525
1 parent aed4fe6
commit 1dd4525
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/unstract/sdk/__init__.py b/src/unstract/sdk/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.18.0"
+__version__ = "0.18.1"
 
 
 def get_sdk_version():

diff --git a/src/unstract/sdk/index.py b/src/unstract/sdk/index.py
@@ -58,9 +58,18 @@ def get_text_from_index(
         try:
             self.tool.stream_log(f">>> Querying {vector_db}...")
             self.tool.stream_log(f">>> {doc_id}")
+            doc_id_eq_filter = MetadataFilter.from_dict(
+                {
+                    "key": "doc_id",
+                    "operator": FilterOperator.EQ,
+                    "value": doc_id,
+                }
+            )
+            filters = MetadataFilters(filters=[doc_id_eq_filter])
             q = VectorStoreQuery(
                 query_embedding=embedding_li.get_query_embedding(" "),
                 doc_ids=[doc_id],
+                filters=filters,
                 similarity_top_k=10000,
             )
         except Exception as e: