changing the return to k=1 form rag

karthiksoman · Sep 24, 2024 · f781ac0 · f781ac0
1 parent 12a21f7
commit f781ac0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lambda-pipeline/app.py b/lambda-pipeline/app.py
@@ -32,7 +32,7 @@ def get_model_response(input_text, temperature: float = 0.7):
         logging.error(f"Failed to get model response: {str(e)}")
         return None
 
-def get_rag_context(query, top_k=2):
+def get_rag_context(query, top_k=1):
     try:
         embed_model = OpenAIEmbedding(
             model='text-embedding-ada-002',