aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-17 05:54:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat(agent/llm): Add support for gpt-4-0125-preview

Browse Source 
* Add `gpt-4-0125-preview` model to OpenAI model list
* Add `gpt-4-turbo-preview` alias to OpenAI model list
This commit is contained in:
Reinier van der Leer
2024-01-29 11:22:32 +01:00
parent a0cae78ba3
commit ab860981d8
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
autogpts/autogpt/autogpt/core/resource/model_providers/openai.py
View File

@@ -62,8 +62,10 @@ class OpenAIModelName(str, enum.Enum):
GPT4_v2 = "gpt-4-0613" GPT4_v2 = "gpt-4-0613"
GPT4_v2_32k = "gpt-4-32k-0613" GPT4_v2_32k = "gpt-4-32k-0613"
GPT4_v3 = "gpt-4-1106-preview" GPT4_v3 = "gpt-4-1106-preview"
GPT4_v4 = "gpt-4-0125-preview"
GPT4_ROLLING = "gpt-4" GPT4_ROLLING = "gpt-4"
GPT4_ROLLING_32k = "gpt-4-32k" GPT4_ROLLING_32k = "gpt-4-32k"
GPT4_TURBO = "gpt-4-turbo-preview"
GPT4_VISION = "gpt-4-vision-preview" GPT4_VISION = "gpt-4-vision-preview"
GPT4 = GPT4_ROLLING GPT4 = GPT4_ROLLING
GPT4_32k = GPT4_ROLLING_32k GPT4_32k = GPT4_ROLLING_32k
@@ -130,7 +132,7 @@ OPEN_AI_CHAT_MODELS = {
has_function_call_api=True, has_function_call_api=True,
), ),
ChatModelInfo( ChatModelInfo(
name=OpenAIModelName.GPT4_v3, name=OpenAIModelName.GPT4_TURBO,
service=ModelProviderService.CHAT, service=ModelProviderService.CHAT,
provider_name=ModelProviderName.OPENAI, provider_name=ModelProviderName.OPENAI,
prompt_token_cost=0.01 / 1000, prompt_token_cost=0.01 / 1000,
@@ -149,6 +151,7 @@ chat_model_mapping = {
OpenAIModelName.GPT4_v1_32k, OpenAIModelName.GPT4_v1_32k,
OpenAIModelName.GPT4_v2_32k, OpenAIModelName.GPT4_v2_32k,
], ],
OpenAIModelName.GPT4_TURBO: [OpenAIModelName.GPT4_v3, OpenAIModelName.GPT4_v4],
} }
for base, copies in chat_model_mapping.items(): for base, copies in chat_model_mapping.items():
for copy in copies: for copy in copies: