ManTea commited on
Commit
e481068
·
1 Parent(s): 3be104a

test limit_k=6

Browse files
Files changed (1) hide show
  1. app/api/rag_routes.py +1 -1
app/api/rag_routes.py CHANGED
@@ -193,7 +193,7 @@ async def chat(request: ChatRequest, background_tasks: BackgroundTasks):
193
  # logger.info(f"Processing chat request for user {request.user_id}, session {session_id}")
194
 
195
  retriever = get_chain(
196
- top_k=request.similarity_top_k,
197
  similarity_metric=request.similarity_metric,
198
  similarity_threshold=request.similarity_threshold
199
  )
 
193
  # logger.info(f"Processing chat request for user {request.user_id}, session {session_id}")
194
 
195
  retriever = get_chain(
196
+ top_k=request.similarity_top_k * 2,
197
  similarity_metric=request.similarity_metric,
198
  similarity_threshold=request.similarity_threshold
199
  )