diff --git a/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py b/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py index ad343b24b14..1b564b6df27 100644 --- a/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py +++ b/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py @@ -9,6 +9,11 @@ import tenacity import tiktoken import yaml +from openai._exceptions import APIStatusError, RateLimitError +from openai.types import CreateEmbeddingResponse +from openai.types.chat import ChatCompletion +from pydantic import SecretStr + from autogpt.core.configuration import Configurable, UserConfigurable from autogpt.core.resource.model_providers.schema import ( AssistantChatMessage, @@ -33,10 +38,6 @@ ModelTokenizer, ) from autogpt.core.utils.json_schema import JSONSchema -from openai._exceptions import APIStatusError, RateLimitError -from openai.types import CreateEmbeddingResponse -from openai.types.chat import ChatCompletion -from pydantic import SecretStr _T = TypeVar("_T") _P = ParamSpec("_P") diff --git a/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py b/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py index 53c3852b313..6beb226a47a 100644 --- a/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py +++ b/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py @@ -12,6 +12,8 @@ TypeVar, ) +from pydantic import BaseModel, Field, SecretStr, validator + from autogpt.core.configuration import SystemConfiguration, UserConfigurable from autogpt.core.resource.schema import ( Embedding, diff --git a/autogpts/autogpt/autogpt/llm/api_manager.py b/autogpts/autogpt/autogpt/llm/api_manager.py index e1773b497dc..4e5e0b9531a 100644 --- a/autogpts/autogpt/autogpt/llm/api_manager.py +++ b/autogpts/autogpt/autogpt/llm/api_manager.py @@ -5,7 +5,6 @@ from openai import OpenAI, AzureOpenAI from openai.types import Model -from pydantic import SecretStr from autogpt.core.resource.model_providers.openai import ( OPEN_AI_MODELS,