From d3c0111121ec3d2d68ea0181a3a93ecd2641e676 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Fri, 21 Jun 2024 12:18:50 +0530 Subject: [PATCH] Include base URL when using openai api config in extract questions. Close #831 --- src/khoj/routers/api.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py index 55353a38..babac509 100644 --- a/src/khoj/routers/api.py +++ b/src/khoj/routers/api.py @@ -344,11 +344,13 @@ async def extract_references_and_questions( elif conversation_config.model_type == ChatModelOptions.ModelType.OPENAI: openai_chat_config = conversation_config.openai_config api_key = openai_chat_config.api_key + base_url = openai_chat_config.api_base_url chat_model = conversation_config.chat_model inferred_queries = extract_questions( defiltered_query, model=chat_model, api_key=api_key, + api_base_url=base_url, conversation_log=meta_log, location_data=location_data, max_tokens=conversation_config.max_prompt_size,