lint

Significant-Gravitas · Feb 29, 2024 · 2b402e4 · 2b402e4
1 parent 1aa0001
commit 2b402e4
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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,