Skip to content

Commit 1dd4525

Browse files
Prompt answer fetch doc id filter issue (#34)
Prompt answer fetch doc if filter issue Co-authored-by: Neha <[email protected]>
1 parent aed4fe6 commit 1dd4525

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

src/unstract/sdk/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "0.18.0"
1+
__version__ = "0.18.1"
22

33

44
def get_sdk_version():

src/unstract/sdk/index.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,18 @@ def get_text_from_index(
5858
try:
5959
self.tool.stream_log(f">>> Querying {vector_db}...")
6060
self.tool.stream_log(f">>> {doc_id}")
61+
doc_id_eq_filter = MetadataFilter.from_dict(
62+
{
63+
"key": "doc_id",
64+
"operator": FilterOperator.EQ,
65+
"value": doc_id,
66+
}
67+
)
68+
filters = MetadataFilters(filters=[doc_id_eq_filter])
6169
q = VectorStoreQuery(
6270
query_embedding=embedding_li.get_query_embedding(" "),
6371
doc_ids=[doc_id],
72+
filters=filters,
6473
similarity_top_k=10000,
6574
)
6675
except Exception as e:

0 commit comments

Comments
 (0)