This commit is contained in:
Niels Geens
2024-11-13 11:55:06 +01:00
7 changed files with 127 additions and 89 deletions

View File

@@ -5,11 +5,11 @@ import asyncio
import mimetypes
import time
from .config import Config
from .hf_llm_engine import HfLlmEngine
from .llm_engine import LlmEngine
from .local_llm_engine import LocalLlmEngine
from .openai_llm_engine import OpenAILlmEngine
from .response_parser import ResponseParser
from .system_metrics import SystemMetrics
from .web_agent import WebAgent
@@ -51,6 +51,11 @@ class Main:
api_token=self._config.api_token,
temperature=self._config.temperature
)
case "openai":
self._llm = OpenAILlmEngine(
model=self._config.model,
api_key=self._config.api_token
)
case "test":
self._llm = TestLLM()
case _: