increase topk to 15

towardsai · Jul 30, 2024 · 8798577 · 8798577
1 parent 8ce4dd7
commit 8798577
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/scripts/main.py b/scripts/main.py
@@ -1,3 +1,5 @@
+import pdb
+
 import gradio as gr
 import logfire
 from llama_index.agent.openai import OpenAIAgent
@@ -148,7 +150,6 @@ def format_sources(completion) -> str:
         "📝 Here are the sources I used to answer your question:\n{documents}"
     )
     document_template: str = "[🔗 {source}: {title}]({url}), relevance: {score:2.2f}"
-
     all_documents = []
     for source in completion.sources:
         for src in source.raw_output:

diff --git a/scripts/setup.py b/scripts/setup.py
@@ -51,7 +51,7 @@ def setup_database(db_collection, dict_file_name):
     )
     vector_retriever = VectorIndexRetriever(
         index=index,
-        similarity_top_k=10,
+        similarity_top_k=15,
         use_async=True,
         embed_model=OpenAIEmbedding(model="text-embedding-3-large", mode="similarity"),
     )