From 1dd45258581aed99cf6c5b686f6e1c39b1d22d8c Mon Sep 17 00:00:00 2001
From: Gayathri <142381512+gaya3-zipstack@users.noreply.github.com>
Date: Tue, 9 Apr 2024 11:49:09 +0530
Subject: [PATCH] Prompt answer fetch doc id filter issue (#34)

Prompt answer fetch doc if filter issue

Co-authored-by: Neha <115609453+nehabagdia@users.noreply.github.com>
---
 src/unstract/sdk/__init__.py | 2 +-
 src/unstract/sdk/index.py    | 9 +++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/unstract/sdk/__init__.py b/src/unstract/sdk/__init__.py
index 6c864b1b..6c8f4d35 100644
--- 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
index 1eebd57c..f07a7c2a 100644
--- 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: