diff --git a/haiku_rag_slim/haiku/rag/chat/__init__.py b/haiku_rag_slim/haiku/rag/chat/__init__.py index d635c10d..981eb295 100644 --- a/haiku_rag_slim/haiku/rag/chat/__init__.py +++ b/haiku_rag_slim/haiku/rag/chat/__init__.py @@ -26,13 +26,19 @@ def run_chat( ) from e from haiku.rag.config import get_config - from haiku.rag.utils import get_model + from haiku.rag.utils import get_model, parse_model_option from haiku.skills.models import Skill config = get_config() if db_path is None: db_path = config.storage.data_dir / "haiku.rag.lancedb" + if model: + model_config = parse_model_option(model) + config.qa.model = model_config + config.research.model = model_config + config.analysis.model = model_config + enabled = skills or ["rag"] skill_list: list[Skill] = []