feat: hybrid search and reranking support

This commit is contained in:
Steven Kreitzer committed 2024-04-24 07:55:10 -05:00
1 parent db801aee79
commit c0259aad67
10 files changed
+261 -130

No files matched your search

+3 -2
View File
@@ -120,12 +120,13 @@ class RAGMiddleware(BaseHTTPMiddleware):
data["messages"],
rag_app.state.RAG_TEMPLATE,
rag_app.state.TOP_K,
rag_app.state.RELEVANCE_THRESHOLD,
rag_app.state.RAG_EMBEDDING_ENGINE,
rag_app.state.RAG_EMBEDDING_MODEL,
rag_app.state.sentence_transformer_ef,
rag_app.state.sentence_transformer_rf,
rag_app.state.RAG_OPENAI_API_KEY,
rag_app.state.RAG_OPENAI_API_BASE_URL,
rag_app.state.OPENAI_API_KEY,
rag_app.state.OPENAI_API_BASE_URL,
)
del data["docs"]