increase topk to 15

Author: omar-sol

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:

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"),
     )