Fix handling of embeddings_key in api_search() function

arc53 · Jan 30, 2024 · 2be523c · 2be523c
1 parent c01e334
commit 2be523c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/application/api/answer/routes.py b/application/api/answer/routes.py
@@ -349,7 +349,10 @@ def api_search():
     question = data["question"]
 
     if not embeddings_key_set:
-        embeddings_key = data["embeddings_key"]
+        if "embeddings_key" in data:
+            embeddings_key = data["embeddings_key"]
+        else:
+            embeddings_key = settings.EMBEDDINGS_KEY
     else:
         embeddings_key = settings.EMBEDDINGS_KEY
     if "active_docs" in data: