Merge branch 'dev' into feat/backend-web-search

2024-05-14 15:20:52 +08:00
parent f94690386c 2290eefcb0
commit 81a3c97069
2 changed files with 2 additions and 2 deletions
@@ -147,7 +147,7 @@ class RAGMiddleware(BaseHTTPMiddleware):
                    messages=data["messages"],
                    template=rag_app.state.config.RAG_TEMPLATE,
                    embedding_function=rag_app.state.EMBEDDING_FUNCTION,
-                    k=rag_app.state.TOP_K,
+                    k=rag_app.state.config.TOP_K,
                    reranking_function=rag_app.state.sentence_transformer_rf,
                    r=rag_app.state.config.RELEVANCE_THRESHOLD,
                    hybrid_search=rag_app.state.config.ENABLE_RAG_HYBRID_SEARCH,