Change top_k value (#62)

Author: gaya3-zipstack

src/unstract/sdk/index.py

@@ -23,6 +23,10 @@
 from unstract.sdk.x2txt import X2Text
 
 
+class Constants:
+    TOP_K = 5
+
+
 class Index:
     def __init__(self, tool: BaseTool):
         # TODO: Inherit from StreamMixin and avoid using BaseTool
@@ -73,7 +77,7 @@ def query_index(
                     query_embedding=embedding.get_query_embedding(" "),
                     doc_ids=[doc_id],
                     filters=filters,
-                    similarity_top_k=10000,
+                    similarity_top_k=Constants.TOP_K,
                 )
             except Exception as e:
                 self.tool.stream_log(
@@ -400,9 +404,7 @@ def index_file(
             output_file_path=output_file_path,
         )
 
-    @deprecated(
-        "Deprecated class and method. Use Index and query_index() instead"
-    )
+    @deprecated("Deprecated class and method. Use Index and query_index() instead")
     def get_text_from_index(
         self, embedding_type: str, vector_db: str, doc_id: str
     ) -> Optional[str]: