Fix config for research

This commit is contained in:
Yiorgis Gozadinos 2025-09-15 14:50:43 +03:00
parent fe8e48df38
commit 4df5d31b18
No known key found for this signature in database
2 changed files with 6 additions and 3 deletions

View file

@ -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 = ""

View file

@ -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: