diff --git a/mentat/llm_api_handler.py b/mentat/llm_api_handler.py index 97ff18711..06862447c 100644 --- 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),