github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(block): Add support for GPT-4O Mini model in LLM block (#7480)

Browse Source 
feat: Add support for GPT-4O Mini model in LLM block
This commit is contained in:
Toran Bruce Richards
2024-07-18 19:02:36 +01:00
committed by GitHub
parent aaf650ee23
commit 27e6c3a95d
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rnd/autogpt_server/autogpt_server/blocks/llm.py
View File

@@ -26,6 +26,7 @@ class ModelMetadata(NamedTuple):
class LlmModel(str, Enum):
# OpenAI models
GPT4O_MINI = "gpt-4o-mini"
GPT4O = "gpt-4o"
GPT4_TURBO = "gpt-4-turbo"
GPT3_5_TURBO = "gpt-3.5-turbo"
@@ -45,6 +46,7 @@ class LlmModel(str, Enum):
MODEL_METADATA = {
LlmModel.GPT4O_MINI: ModelMetadata("openai", 128000),
LlmModel.GPT4O: ModelMetadata("openai", 128000),
LlmModel.GPT4_TURBO: ModelMetadata("openai", 128000),
LlmModel.GPT3_5_TURBO: ModelMetadata("openai", 16385),