add new gpt-3.5-0125 model (#513)

AbanteAI · Feb 2, 2024 · 28ab2dc · 28ab2dc
1 parent d2691f0
commit 28ab2dc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mentat/llm_api_handler.py b/mentat/llm_api_handler.py
@@ -218,6 +218,7 @@ def __contains__(self, key: object) -> bool:
         "gpt-4-32k-0613": Model("gpt-4-32k-0613", 32768, 0.06, 0.12),
         "gpt-4-0314": Model("gpt-4-0314", 8192, 0.03, 0.06),
         "gpt-4-32k-0314": Model("gpt-4-32k-0314", 32768, 0.06, 0.12),
+        "gpt-3.5-turbo-0125": Model("gpt-3.5-turbo-0125", 16385, 0.0005, 0.0015),
         "gpt-3.5-turbo-1106": Model("gpt-3.5-turbo-1106", 16385, 0.001, 0.002),
         "gpt-3.5-turbo": Model("gpt-3.5-turbo", 16385, 0.001, 0.002),
         "gpt-3.5-turbo-0613": Model("gpt-3.5-turbo-0613", 4096, 0.0015, 0.002),