chore: format files

Author: xaviergoh-bytedance

volcenginesdkarkruntime/_utils/_utils.py

@@ -11,6 +11,7 @@
     cast,
     TypeVar,
 )
+
 from typing_extensions import TypeGuard
 
 from .._types import NotGiven
@@ -97,15 +98,18 @@ def wrapper(*args, **kwargs):
 
     return wrapper
 
+
 def async_disallow_aksk(func):
     async def wrapper(*args, **kwargs):
         _restrict_aksk(args, kwargs)
         return await func(*args, **kwargs)
 
     return wrapper
 
+
 def _restrict_aksk(args, kwargs):
     assert len(args) > 0
 
     ark_client = args[0]._client
-    assert ark_client.api_key is not None, "ak&sk authentication is currently not supported for this method, please use api key instead"
+    assert ark_client.api_key is not None, \
+        "ak&sk authentication is currently not supported for this method, please use api key instead"

volcenginesdkarkruntime/resources/content_generation/content_generation.py

@@ -1,17 +1,17 @@
 from __future__ import annotations
 
-from ..._resource import SyncAPIResource, AsyncAPIResource
-from ..._compat import cached_property
-
 from .tasks import Tasks, AsyncTasks
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
 
 
 class ContentGeneration(SyncAPIResource):
     @cached_property
     def tasks(self) -> Tasks:
         return Tasks(self._client)
 
+
 class AsyncContentGeneration(AsyncAPIResource):
     @cached_property
     def tasks(self) -> AsyncTasks:
-        return AsyncTasks(self._client)
+        return AsyncTasks(self._client)

volcenginesdkarkruntime/resources/content_generation/tasks.py

@@ -2,16 +2,17 @@
 
 from functools import cached_property
 from typing import Iterable, Union, List
+
 import httpx
 
+from volcenginesdkarkruntime._base_client import make_request_options
+from volcenginesdkarkruntime._resource import SyncAPIResource, AsyncAPIResource
+from volcenginesdkarkruntime.types.content_generation.create_task_content_param import CreateTaskContentParam
 from ..._response import to_raw_response_wrapper, async_to_raw_response_wrapper
 from ..._types import Body, Query, Headers
 from ..._utils._utils import disallow_aksk, async_disallow_aksk
 from ...types.content_generation.content_generation_task import ContentGenerationTask
 from ...types.content_generation.content_generation_task_id import ContentGenerationTaskID
-from volcenginesdkarkruntime._base_client import make_request_options
-from volcenginesdkarkruntime._resource import SyncAPIResource, AsyncAPIResource
-from volcenginesdkarkruntime.types.content_generation.create_task_content_param import CreateTaskContentParam
 from ...types.content_generation.list_content_generation_tasks_response import ListContentGenerationTasksResponse
 
 
@@ -56,7 +57,7 @@ def get(
             extra_query: Query | None = None,
             extra_body: Body | None = None,
             timeout: float | httpx.Timeout | None = None,
-            ) -> ContentGenerationTask:
+    ) -> ContentGenerationTask:
         resp = self._get(
             path=f"/contents/generations/tasks/{task_id}",
             options=make_request_options(
@@ -75,13 +76,13 @@ def list(
             page_num: int | None = None,
             page_size: int | None = None,
             status: str | None = None,
-            task_ids : Union[List[str], str] | None = None,
+            task_ids: Union[List[str], str] | None = None,
             model: str | None = None,
             extra_headers: Headers | None = None,
             extra_body: Body | None = None,
             extra_query: Query | None = None,
             timeout: float | httpx.Timeout | None = None,
-             ) -> ListContentGenerationTasksResponse:
+    ) -> ListContentGenerationTasksResponse:
 
         query_params = []
         if page_num:
@@ -138,7 +139,6 @@ class AsyncTasks(AsyncAPIResource):
     def with_raw_response(self) -> AsyncTasksWithRawResponse:
         return AsyncTasksWithRawResponse(self)
 
-
     @async_disallow_aksk
     async def create(
             self,
@@ -166,7 +166,6 @@ async def create(
         )
         return resp
 
-
     @async_disallow_aksk
     async def get(
             self,
@@ -195,7 +194,7 @@ async def list(
             page_num: int | None = None,
             page_size: int | None = None,
             status: str | None = None,
-            task_ids : Union[List[str], str] | None = None,
+            task_ids: Union[List[str], str] | None = None,
             model: str | None = None,
             extra_headers: Headers | None = None,
             extra_body: Body | None = None,
@@ -261,6 +260,7 @@ def __init__(self, tasks: Tasks) -> None:
             tasks.create,
         )
 
+
 class AsyncTasksWithRawResponse:
     def __init__(self, tasks: AsyncTasks) -> None:
         self._tasks = tasks

volcenginesdkarkruntime/types/content_generation/content_generation_task.py

@@ -1,4 +1,3 @@
-
 __all__ = ["ContentGenerationTask"]
 
 from volcenginesdkarkruntime._models import BaseModel
@@ -8,10 +7,12 @@ class Usage(BaseModel):
     completion_tokens: int
     """The number of tokens used for completion."""
 
+
 class Content(BaseModel):
     video_url: str
     """The URL of the generated video, if any."""
 
+
 class ContentGenerationTask(BaseModel):
     id: str
     """A unique identifier for the task."""
@@ -35,4 +36,4 @@ class ContentGenerationTask(BaseModel):
     """The Unix timestamp when the task was created."""
 
     updated_at: int
-    """The Unix timestamp when the task was last updated."""
+    """The Unix timestamp when the task was last updated."""

volcenginesdkarkruntime/types/content_generation/create_task_content_param.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
+
 from typing import Union
+
 from typing_extensions import Literal, TypedDict
 
 __all__ = [

volcenginesdkarkruntime/types/content_generation/list_content_generation_tasks_response.py

@@ -47,4 +47,3 @@ class ListContentGenerationTasksResponse(BaseModel):
 
     items: List[Item]
     """List of content generation task items."""
-

volcenginesdkexamples/volcenginesdkarkruntime/content_generation_tasks.py

@@ -13,11 +13,11 @@
         model="${YOUR_MODEL_EP}",
         content=[
             {
-                "type":"text",
-                "text":"龙与地下城女骑士背景是起伏的平原，目光从镜头转向平原"
+                "type": "text",
+                "text": "龙与地下城女骑士背景是起伏的平原，目光从镜头转向平原"
             },
             {
-                "type":"image_url",
+                "type": "image_url",
                 "image_url": {
                     "url": "${YOUR_IMAGE_URL}"
                 },
@@ -45,4 +45,4 @@
         client.content_generation.tasks.delete(task_id=create_result.id)
         print(create_result.id)
     except Exception as e:
-        print(f"failed to delete task: {e}")
+        print(f"failed to delete task: {e}")