Commit 4a0f0fa0

Stainless Bot <107565488+stainless-bot@users.noreply.github.com>
2024-04-26 21:48:35
chore(internal): reformat imports (#1375)
1 parent e972439
tests/api_resources/audio/test_transcriptions.py
@@ -9,7 +9,7 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.audio.transcription import Transcription
+from openai.types.audio import Transcription
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/audio/test_translations.py
@@ -9,7 +9,7 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.audio.translation import Translation
+from openai.types.audio import Translation
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/beta/threads/runs/test_steps.py
@@ -10,7 +10,7 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.beta.threads.runs.run_step import RunStep
+from openai.types.beta.threads.runs import RunStep
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/beta/threads/test_messages.py
@@ -10,7 +10,7 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.beta.threads.message import Message
+from openai.types.beta.threads import Message
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/beta/threads/test_runs.py
@@ -10,7 +10,9 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.beta.threads.run import Run
+from openai.types.beta.threads import (
+    Run,
+)
 
 # pyright: reportDeprecated=false
 
tests/api_resources/beta/vector_stores/test_file_batches.py
@@ -10,8 +10,10 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.beta.vector_stores.vector_store_file import VectorStoreFile
-from openai.types.beta.vector_stores.vector_store_file_batch import VectorStoreFileBatch
+from openai.types.beta.vector_stores import (
+    VectorStoreFile,
+    VectorStoreFileBatch,
+)
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/beta/vector_stores/test_files.py
@@ -10,8 +10,10 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.beta.vector_stores.vector_store_file import VectorStoreFile
-from openai.types.beta.vector_stores.vector_store_file_deleted import VectorStoreFileDeleted
+from openai.types.beta.vector_stores import (
+    VectorStoreFile,
+    VectorStoreFileDeleted,
+)
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/beta/test_assistants.py
@@ -10,8 +10,10 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.beta.assistant import Assistant
-from openai.types.beta.assistant_deleted import AssistantDeleted
+from openai.types.beta import (
+    Assistant,
+    AssistantDeleted,
+)
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/beta/test_threads.py
@@ -9,9 +9,11 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.beta.thread import Thread
-from openai.types.beta.threads.run import Run
-from openai.types.beta.thread_deleted import ThreadDeleted
+from openai.types.beta import (
+    Thread,
+    ThreadDeleted,
+)
+from openai.types.beta.threads import Run
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/beta/test_vector_stores.py
@@ -10,8 +10,10 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.beta.vector_store import VectorStore
-from openai.types.beta.vector_store_deleted import VectorStoreDeleted
+from openai.types.beta import (
+    VectorStore,
+    VectorStoreDeleted,
+)
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/chat/test_completions.py
@@ -9,7 +9,7 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.chat.chat_completion import ChatCompletion
+from openai.types.chat import ChatCompletion
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/fine_tuning/jobs/test_checkpoints.py
@@ -10,7 +10,7 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.fine_tuning.jobs.fine_tuning_job_checkpoint import FineTuningJobCheckpoint
+from openai.types.fine_tuning.jobs import FineTuningJobCheckpoint
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/fine_tuning/test_jobs.py
@@ -10,8 +10,10 @@ import pytest
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.fine_tuning.fine_tuning_job import FineTuningJob
-from openai.types.fine_tuning.fine_tuning_job_event import FineTuningJobEvent
+from openai.types.fine_tuning import (
+    FineTuningJob,
+    FineTuningJobEvent,
+)
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/test_batches.py
@@ -9,8 +9,8 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
+from openai.types import Batch
 from openai.pagination import SyncCursorPage, AsyncCursorPage
-from openai.types.batch import Batch
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/test_completions.py
@@ -9,7 +9,7 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.completion import Completion
+from openai.types import Completion
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/test_embeddings.py
@@ -9,7 +9,7 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.create_embedding_response import CreateEmbeddingResponse
+from openai.types import CreateEmbeddingResponse
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/test_files.py
@@ -12,9 +12,8 @@ from respx import MockRouter
 import openai._legacy_response as _legacy_response
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
+from openai.types import FileObject, FileDeleted
 from openai.pagination import SyncPage, AsyncPage
-from openai.types.file_object import FileObject
-from openai.types.file_deleted import FileDeleted
 
 # pyright: reportDeprecated=false
 
tests/api_resources/test_images.py
@@ -9,7 +9,7 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.images_response import ImagesResponse
+from openai.types import ImagesResponse
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/test_models.py
@@ -9,9 +9,8 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
+from openai.types import Model, ModelDeleted
 from openai.pagination import SyncPage, AsyncPage
-from openai.types.model import Model
-from openai.types.model_deleted import ModelDeleted
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
tests/api_resources/test_moderations.py
@@ -9,7 +9,7 @@ import pytest
 
 from openai import OpenAI, AsyncOpenAI
 from tests.utils import assert_matches_type
-from openai.types.moderation_create_response import ModerationCreateResponse
+from openai.types import ModerationCreateResponse
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")