Update mistral model

whiletrue-industries · Feb 7, 2025 · fbc6c00 · fbc6c00
1 parent 6ce70e9
commit fbc6c00
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/odds/common/llm/mistral/mistral_llm_runner.py b/odds/common/llm/mistral/mistral_llm_runner.py
@@ -10,17 +10,17 @@
 class MistralLLMRunner(LLMRunner):
 
     MODELS = dict(
-        cheap='open-mixtral-8x7b',
-        expensive='open-mixtral-8x22b',
+        cheap='mistral-small-latest',
+        expensive='mistral-small-latest',
     )
     COSTS = dict(
         cheap=dict(
-            prompt=0.7/1000000,
-            completion=0.7/1000000
+            prompt=.2/1000000,
+            completion=.6/1000000
         ),
         expensive=dict(
-            prompt=2/1000000,
-            completion=6/1000000
+            prompt=.2/1000000,
+            completion=.6/1000000
         ),
     )