Merge branch 'master' of github.com:debanjum/khoj

khoj-ai · Jul 10, 2023 · 9a63f89 · 9a63f89
2 parents 5380929 + 5b37e98
commit 9a63f89
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py
@@ -535,14 +535,15 @@ async def extract_references_and_questions(
 
     # Initialize Variables
     api_key = state.processor_config.conversation.openai_api_key
+    chat_model = state.processor_config.conversation.chat_model
     conversation_type = "general" if q.startswith("@general") else "notes"
     compiled_references = []
     inferred_queries = []
 
     if conversation_type == "notes":
         # Infer search queries from user message
         with timer("Extracting search queries took", logger):
-            inferred_queries = extract_questions(q, api_key=api_key, conversation_log=meta_log)
+            inferred_queries = extract_questions(q, model=chat_model, api_key=api_key, conversation_log=meta_log)
 
         # Collate search results as context for GPT
         with timer("Searching knowledge base took", logger):