Replaced deepseek with qwq

This commit is contained in:
Niels Geens
2025-03-14 11:22:30 +01:00
parent 3ea3239a9b
commit 2e66020f8e
13 changed files with 693 additions and 530 deletions

View File

@@ -3,12 +3,12 @@ import asyncio
from .auto_approver import AutoApprover
from .config import Config
from .llm_engine.hf_llm_engine import HfLlmEngine
from .llm_engine.deepseek_llm_engine import DeepSeekLlmEngine
from .iteration_logger import IterationLogger
from .llm_engine.hf_llm_engine import HfLlmEngine
from .llm_engine.local_llm_engine import LocalLlmEngine
from .llm_engine.mistral_llm_engine import MistralLlmEngine
from .llm_engine.openai_llm_engine import OpenAILlmEngine
from .llm_engine.qwq_llm_engine import QwQLlmEngine
from .response_parser import ResponseParser
from .system_metrics import SystemMetrics
from .web.api import Api
@@ -62,11 +62,11 @@ class Main:
config.mistral_api_key,
)
if config.deepseek_enabled:
self._llms['deepseek'] = DeepSeekLlmEngine(
config.deepseek_model,
config.deepseek_temperature,
config.deepseek_token_limit,
if config.qwq_enabled:
self._llms['qwq'] = QwQLlmEngine(
config.qwq_model,
config.qwq_temperature,
config.qwq_token_limit,
config.hf_api_key,
)