Quellcode durchsuchen

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

Jun Siang Cheah vor 11 Monaten
Ursprung
Commit
81a3c97069
2 geänderte Dateien mit 2 neuen und 2 gelöschten Zeilen
  1. 1 1
      backend/apps/openai/main.py
  2. 1 1
      backend/main.py

+ 1 - 1
backend/apps/openai/main.py

@@ -119,7 +119,7 @@ async def speech(request: Request, user=Depends(get_verified_user)):
         headers = {}
         headers = {}
         headers["Authorization"] = f"Bearer {app.state.config.OPENAI_API_KEYS[idx]}"
         headers["Authorization"] = f"Bearer {app.state.config.OPENAI_API_KEYS[idx]}"
         headers["Content-Type"] = "application/json"
         headers["Content-Type"] = "application/json"
-        if "openrouter.ai" in app.state.OPENAI_API_BASE_URLS[idx]:
+        if "openrouter.ai" in app.state.config.OPENAI_API_BASE_URLS[idx]:
             headers["HTTP-Referer"] = "https://openwebui.com/"
             headers["HTTP-Referer"] = "https://openwebui.com/"
             headers["X-Title"] = "Open WebUI"
             headers["X-Title"] = "Open WebUI"
         r = None
         r = None

+ 1 - 1
backend/main.py

@@ -147,7 +147,7 @@ class RAGMiddleware(BaseHTTPMiddleware):
                     messages=data["messages"],
                     messages=data["messages"],
                     template=rag_app.state.config.RAG_TEMPLATE,
                     template=rag_app.state.config.RAG_TEMPLATE,
                     embedding_function=rag_app.state.EMBEDDING_FUNCTION,
                     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,
                     reranking_function=rag_app.state.sentence_transformer_rf,
                     r=rag_app.state.config.RELEVANCE_THRESHOLD,
                     r=rag_app.state.config.RELEVANCE_THRESHOLD,
                     hybrid_search=rag_app.state.config.ENABLE_RAG_HYBRID_SEARCH,
                     hybrid_search=rag_app.state.config.ENABLE_RAG_HYBRID_SEARCH,