Bugfix in cli

search_agent.py

@@ -101,7 +101,7 @@ if __name__ == '__main__':
         vector_store = wc.vectorize(contents)
 
     with console.status("[bold green]Querying LLM relevant context", spinner='dots8Bit'):
-        respomse = wr.query_rag(chat, query, optimize_search_query, vector_store, callbacks=callbacks)
+        respomse = wr.query_rag(chat, query, optimize_search_query, vector_store, top_k = 5, callbacks=callbacks)
 
     console.rule(f"[bold green]Response from {provider}")
     if output == "text":

web_rag.py

@@ -219,7 +219,7 @@ def build_rag_prompt(question, search_query, vectorstore, top_k = 10, callbacks
     prompt = get_rag_prompt_template().format(query=question, context=context)
     return prompt    
 
-def query_rag(chat_llm, question, search_query, vectorstore, callbacks = []):
-    prompt = build_rag_prompt(question, search_query, vectorstore, callbacks)
+def query_rag(chat_llm, question, search_query, vectorstore, top_k = 10, callbacks = []):
+    prompt = build_rag_prompt(question, search_query, vectorstore, top_k= top_k, callbacks = callbacks)
     response = chat_llm.invoke(prompt, config={"callbacks": callbacks})
     return response.content