Fix config for research
This commit is contained in:
parent
fe8e48df38
commit
4df5d31b18
2 changed files with 6 additions and 3 deletions
|
|
@ -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 = ""
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Reference in a new issue