diff options
Diffstat (limited to 'autogpts/autogpt/autogpt/config/config.py')
| -rw-r--r-- | autogpts/autogpt/autogpt/config/config.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/autogpts/autogpt/autogpt/config/config.py b/autogpts/autogpt/autogpt/config/config.py index 127719b72..409a4256c 100644 --- a/autogpts/autogpt/autogpt/config/config.py +++ b/autogpts/autogpt/autogpt/config/config.py @@ -21,6 +21,7 @@ from autogpt.core.configuration.schema import ( from autogpt.core.resource.model_providers.openai import ( OPEN_AI_CHAT_MODELS, OpenAICredentials, + OpenAIModelName, ) from autogpt.file_storage import FileStorageBackendName from autogpt.plugins.plugins_config import PluginsConfig @@ -34,8 +35,8 @@ AZURE_CONFIG_FILE = Path("azure.yaml") PLUGINS_CONFIG_FILE = Path("plugins_config.yaml") PROMPT_SETTINGS_FILE = Path("prompt_settings.yaml") -GPT_4_MODEL = "gpt-4" -GPT_3_MODEL = "gpt-3.5-turbo" +GPT_4_MODEL = OpenAIModelName.GPT4 +GPT_3_MODEL = OpenAIModelName.GPT3 class Config(SystemSettings, arbitrary_types_allowed=True): @@ -77,12 +78,12 @@ class Config(SystemSettings, arbitrary_types_allowed=True): ) # Model configuration - fast_llm: str = UserConfigurable( - default="gpt-3.5-turbo-0125", + fast_llm: OpenAIModelName = UserConfigurable( + default=OpenAIModelName.GPT3, from_env="FAST_LLM", ) - smart_llm: str = UserConfigurable( - default="gpt-4-turbo-preview", + smart_llm: OpenAIModelName = UserConfigurable( + default=OpenAIModelName.GPT4_TURBO, from_env="SMART_LLM", ) temperature: float = UserConfigurable(default=0, from_env="TEMPERATURE") |