Include base URL when using openai api config in extract questions. C…

…lose #831
khoj-ai · Jun 21, 2024 · d3c0111 · d3c0111
1 parent b9966eb
commit d3c0111
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,