feat: openai embeddings integration

2025-12-16 03:47:49 +01:00 · 2024-04-14 19:48:15 -04:00
parent b48e73fa43
commit b1b72441bb
6 changed files with 155 additions and 46 deletions
--- a/backend/main.py
+++ b/backend/main.py
@@ -114,7 +114,11 @@ class RAGMiddleware(BaseHTTPMiddleware):
                    data["messages"],
                    rag_app.state.RAG_TEMPLATE,
                    rag_app.state.TOP_K,
+                    rag_app.state.RAG_EMBEDDING_ENGINE,
+                    rag_app.state.RAG_EMBEDDING_MODEL,
                    rag_app.state.sentence_transformer_ef,
+                    rag_app.state.RAG_OPENAI_API_KEY,
+                    rag_app.state.RAG_OPENAI_API_BASE_URL,
                )
                del data["docs"]