zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add model params for o1 and o3-mini.

This commit is contained in:
AI Christianson 2025-02-02 18:48:41 -05:00
parent c14fad6d14
commit 53ccc46392
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ra_aid/models_params.py
View File

@ -27,6 +27,9 @@ models_params = {
"gpt-4o-mini": {"token_limit": 128000, "supports_temperature": True}, "gpt-4o-mini": {"token_limit": 128000, "supports_temperature": True},
"o1-preview": {"token_limit": 128000, "supports_temperature": False}, "o1-preview": {"token_limit": 128000, "supports_temperature": False},
"o1-mini": {"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": { "azure_openai": {
"gpt-3.5-turbo-0125": {"token_limit": 16385, "supports_temperature": True}, "gpt-3.5-turbo-0125": {"token_limit": 16385, "supports_temperature": True},