Update main.py

main.py

@@ -189,7 +189,7 @@ def stream_llm_request(api_key: str, llm_request: dict, endpoint_url: str):
 async def chat(request: ChatRequest, background_tasks: BackgroundTasks, api_key: str = Depends(get_api_key)):
     try:
         document_list = load_embeddings(request.index_id)
-        search_results = embeddings.search(request.query, 5)
+        search_results = embeddings.search(request.query, 8)
         context = "\n".join([document_list[idx[0]] for idx in search_results])
 
         rag_prompt = f"Based on the following context, please answer the user's question:\n\nContext:\n{context}\n\nUser's question: {request.query}\n\nAnswer:"