fix

bcc27e38 · Timothy J. Baek · 35026849 · bcc27e38
Commit bcc27e38 authored Jun 18, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/backend/main.py
+++ b/backend/main.py
@@ -357,7 +357,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
            if "files" in data:
                data = {**data}
                rag_context, citations = get_rag_context(
-                    docs=data["files"],
+                    files=data["files"],
                    messages=data["messages"],
                    embedding_function=rag_app.state.EMBEDDING_FUNCTION,
                    k=rag_app.state.config.TOP_K,