From 4df5d31b181169e70b01671732ddb532efbddc4a Mon Sep 17 00:00:00 2001
From: Yiorgis Gozadinos <ggozadinos@gmail.com>
Date: Mon, 15 Sep 2025 14:50:43 +0300
Subject: [PATCH] Fix config for research

---
 src/haiku/rag/config.py        | 6 ++++--
 src/haiku/rag/research/base.py | 3 ++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/haiku/rag/config.py b/src/haiku/rag/config.py
index 1b044c65..a51b0b4b 100644
--- a/src/haiku/rag/config.py
+++ b/src/haiku/rag/config.py
@@ -30,8 +30,8 @@ class AppConfig(BaseModel):
     QA_MODEL: str = "qwen3"
 
     # Research defaults (fallback to QA if not provided via env)
-    RESEARCH_PROVIDER: str = ""
-    RESEARCH_MODEL: str = ""
+    RESEARCH_PROVIDER: str = "ollama"
+    RESEARCH_MODEL: str = "qwen3"
 
     CHUNK_SIZE: int = 256
     CONTEXT_CHUNK_RADIUS: int = 0
@@ -41,9 +41,11 @@ class AppConfig(BaseModel):
     MARKDOWN_PREPROCESSOR: str = ""
 
     OLLAMA_BASE_URL: str = "http://localhost:11434"
+
     VLLM_EMBEDDINGS_BASE_URL: str = ""
     VLLM_RERANK_BASE_URL: str = ""
     VLLM_QA_BASE_URL: str = ""
+    VLLM_RESEARCH_BASE_URL: str = ""
 
     # Provider keys
     VOYAGE_API_KEY: str = ""
diff --git a/src/haiku/rag/research/base.py b/src/haiku/rag/research/base.py
index c38961e6..2cb5c1aa 100644
--- a/src/haiku/rag/research/base.py
+++ b/src/haiku/rag/research/base.py
@@ -51,7 +51,8 @@ class BaseResearchAgent(ABC):
             return OpenAIChatModel(
                 model_name=model,
                 provider=OpenAIProvider(
-                    base_url=f"{Config.VLLM_QA_BASE_URL}/v1", api_key="none"
+                    base_url=f"{Config.VLLM_RESEARCH_BASE_URL or Config.VLLM_QA_BASE_URL}/v1",
+                    api_key="none",
                 ),
             )
         else: