fix: formatting with ruff.

Author: Khurdhula-Harshavardhan

jigsawstack/__init__.py

@@ -1,22 +1,23 @@
-from typing import Union, Dict
 import os
-from .audio import Audio, AsyncAudio
-from .vision import Vision, AsyncVision
+from typing import Dict, Union
+
+from .audio import AsyncAudio, Audio
+from .classification import AsyncClassification, Classification
+from .embedding import AsyncEmbedding, Embedding
+from .embedding_v2 import AsyncEmbeddingV2, EmbeddingV2
+from .exceptions import JigsawStackError
+from .image_generation import AsyncImageGeneration, ImageGeneration
+from .prediction import AsyncPrediction, Prediction
+from .prompt_engine import AsyncPromptEngine, PromptEngine
 from .search import Search
-from .prediction import Prediction, AsyncPrediction
+from .sentiment import AsyncSentiment, Sentiment
 from .sql import SQL, AsyncSQL
-from .store import Store, AsyncStore
-from .translate import Translate, AsyncTranslate
-from .web import Web, AsyncWeb
-from .sentiment import Sentiment, AsyncSentiment
-from .validate import Validate, AsyncValidate
-from .summary import Summary, AsyncSummary
-from .embedding import Embedding, AsyncEmbedding
-from .exceptions import JigsawStackError
-from .image_generation import ImageGeneration, AsyncImageGeneration
-from .classification import Classification, AsyncClassification
-from .prompt_engine import PromptEngine, AsyncPromptEngine
-from .embedding_v2 import EmbeddingV2, AsyncEmbeddingV2
+from .store import AsyncStore, Store
+from .summary import AsyncSummary, Summary
+from .translate import AsyncTranslate, Translate
+from .validate import AsyncValidate, Validate
+from .vision import AsyncVision, Vision
+from .web import AsyncWeb, Web
 
 
 class JigsawStack:

jigsawstack/async_request.py

@@ -1,9 +1,11 @@
-from typing import Any, Dict, Generic, List, Union, cast, TypedDict, AsyncGenerator
+import json
+from io import BytesIO
+from typing import Any, AsyncGenerator, Dict, Generic, List, TypedDict, Union, cast
+
 import aiohttp
 from typing_extensions import Literal, TypeVar
+
 from .exceptions import NoContentError, raise_for_code_and_type
-import json
-from io import BytesIO
 
 RequestVerb = Literal["get", "post", "put", "patch", "delete"]
 

jigsawstack/audio.py

@@ -1,10 +1,11 @@
-from typing import Any, Dict, List, cast, Union, Optional, overload
-from typing_extensions import NotRequired, TypedDict
-from .request import Request, RequestConfig
-from .async_request import AsyncRequest, AsyncRequestConfig
+from typing import Any, Dict, List, Optional, Union, cast, overload
+
+from typing_extensions import Literal, NotRequired, TypedDict
+
 from ._config import ClientConfig
-from typing_extensions import Literal
 from ._types import BaseResponse
+from .async_request import AsyncRequest, AsyncRequestConfig
+from .request import Request, RequestConfig
 
 
 class SpeechToTextParams(TypedDict):

jigsawstack/classification.py

@@ -1,9 +1,11 @@
 from typing import Any, Dict, List, Union, cast
-from typing_extensions import NotRequired, TypedDict, Literal
-from .request import Request, RequestConfig
-from .async_request import AsyncRequest, AsyncRequestConfig
+
+from typing_extensions import Literal, NotRequired, TypedDict
+
 from ._config import ClientConfig
 from ._types import BaseResponse
+from .async_request import AsyncRequest, AsyncRequestConfig
+from .request import Request, RequestConfig
 
 
 class DatasetItem(TypedDict):

jigsawstack/embedding.py

@@ -1,10 +1,12 @@
-from typing import Any, Dict, List, Union, cast, Literal, overload
+from typing import Any, Dict, List, Literal, Union, cast, overload
+
 from typing_extensions import NotRequired, TypedDict
-from .request import Request, RequestConfig
-from .async_request import AsyncRequest
+
 from ._config import ClientConfig
-from .helpers import build_path
 from ._types import BaseResponse
+from .async_request import AsyncRequest
+from .helpers import build_path
+from .request import Request, RequestConfig
 
 
 class EmbeddingParams(TypedDict):

jigsawstack/embedding_v2.py

@@ -1,10 +1,12 @@
-from typing import Any, Dict, List, Union, cast, Literal, overload
+from typing import Any, Dict, List, Literal, Union, cast, overload
+
 from typing_extensions import NotRequired, TypedDict
-from .request import Request, RequestConfig
-from .async_request import AsyncRequest
+
 from ._config import ClientConfig
-from .helpers import build_path
+from .async_request import AsyncRequest
 from .embedding import Chunk
+from .helpers import build_path
+from .request import Request, RequestConfig
 
 
 class EmbeddingV2Params(TypedDict):

jigsawstack/image_generation.py

@@ -1,9 +1,10 @@
 from typing import Any, Dict, Union, cast
-from typing_extensions import NotRequired, TypedDict, Literal, Required
-from .request import Request, RequestConfig
-from .async_request import AsyncRequest
+
+from typing_extensions import Literal, NotRequired, Required, TypedDict
 
 from ._config import ClientConfig
+from .async_request import AsyncRequest
+from .request import Request, RequestConfig
 
 
 class AdvanceConfig(TypedDict):

jigsawstack/prediction.py

@@ -1,10 +1,11 @@
 from typing import Any, Dict, List, Union, cast
+
 from typing_extensions import TypedDict
-from .request import Request, RequestConfig
-from .async_request import AsyncRequest
 
 from ._config import ClientConfig
 from ._types import BaseResponse
+from .async_request import AsyncRequest
+from .request import Request, RequestConfig
 
 
 class Dataset(TypedDict):

jigsawstack/prompt_engine.py

@@ -1,9 +1,11 @@
-from typing import Any, Dict, List, Union, cast, Generator, Literal
+from typing import Any, Dict, Generator, List, Literal, Union, cast
+
 from typing_extensions import NotRequired, TypedDict
-from .request import Request, RequestConfig
-from .async_request import AsyncRequest
+
 from ._config import ClientConfig
+from .async_request import AsyncRequest
 from .helpers import build_path
+from .request import Request, RequestConfig
 
 
 class PromptEngineResult(TypedDict):

jigsawstack/request.py

@@ -1,8 +1,10 @@
-from typing import Any, Dict, Generic, List, Union, cast, TypedDict, Generator
+import json
+from typing import Any, Dict, Generator, Generic, List, TypedDict, Union, cast
+
 import requests
 from typing_extensions import Literal, TypeVar
+
 from .exceptions import NoContentError, raise_for_code_and_type
-import json
 
 RequestVerb = Literal["get", "post", "put", "patch", "delete"]