From 53ccc46392c4e53b66854a6200ca152635519e6f Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Sun, 2 Feb 2025 18:48:41 -0500 Subject: [PATCH] Add model params for o1 and o3-mini. --- ra_aid/models_params.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ra_aid/models_params.py b/ra_aid/models_params.py index 4bbf596..ee83f70 100644 --- a/ra_aid/models_params.py +++ b/ra_aid/models_params.py @@ -27,6 +27,9 @@ models_params = { "gpt-4o-mini": {"token_limit": 128000, "supports_temperature": True}, "o1-preview": {"token_limit": 128000, "supports_temperature": False}, "o1-mini": {"token_limit": 128000, "supports_temperature": False}, + "o1-preview": {"token_limit": 128000, "supports_temperature": False}, + "o1": {"token_limit": 200000, "supports_temperature": False}, + "o3-mini": {"token_limit": 200000, "supports_temperature": False}, }, "azure_openai": { "gpt-3.5-turbo-0125": {"token_limit": 16385, "supports_temperature": True},