Added mistral llm engine

2024-11-17 16:53:44 +01:00
parent 506bc91639
commit ef94d0d1a0
6 changed files with 84 additions and 4 deletions
--- a/sia/openai_llm_engine.py
+++ b/sia/openai_llm_engine.py
@@ -14,8 +14,8 @@ class OpenAILlmEngine(LlmEngine):
        self,
        model: str,
        temperature: float,
+        token_limit: int,
        api_key: str,
-        token_limit: int = 0,
    ):
        """
        Initialize the OpenAI LLM Engine.