Merge pull request #149 from aitomatic/fix/vector-query-top-k

update vector-query similarity top-k parameter
aitomatic · Apr 2, 2024 · 7ab3f4e · 7ab3f4e
2 parents ae47a32 + 8ef3dbe
commit 7ab3f4e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/openssa/core/ooda_rag/query_rewritting_engine.py b/openssa/core/ooda_rag/query_rewritting_engine.py
@@ -22,7 +22,7 @@ def __init__(
         self,
         index: VectorStoreIndex = None,  # type: ignore
         chunk_size: int = 1024,
-        vector_similarity_top_k: int = 5,
+        vector_similarity_top_k: int = 10,
         fusion_similarity_top_k: int = 10,
         service_context: ServiceContext = None,
         **kwargs: Any,

diff --git a/openssa/core/ooda_rag/resources/dense_x/base.py b/openssa/core/ooda_rag/resources/dense_x/base.py
@@ -94,7 +94,7 @@ def __init__(
         query_llm: Optional[LLM] = None,
         embed_model: Optional[BaseEmbedding] = None,
         text_splitter: TextSplitter = SentenceSplitter(),
-        similarity_top_k: int = 4,
+        similarity_top_k: int = 10,
     ) -> None:
         """Init params."""
         self._proposition_llm = proposition_llm or OpenAI(

diff --git a/openssa/integrations/llama_index/backend.py b/openssa/integrations/llama_index/backend.py
@@ -21,7 +21,7 @@ class Backend(AbstractRAGBackend):
     def __init__(
         self,
         relevance_threshold: float = 0.5,
-        similarity_top_k: int = 4,
+        similarity_top_k: int = 10,
         service_context: ServiceContext = None,
     ):
         """