fix: use batch.chat instead of batch_chat

Author: exiaohu

volcenginesdkarkruntime/_client.py

@@ -29,6 +29,7 @@
 
 from . import resources
 from .resources.beta import beta
+from .resources.batch import batch
 from ._base_client import SyncAPIClient, AsyncAPIClient
 from ._constants import (
     DEFAULT_MAX_RETRIES,
@@ -60,6 +61,8 @@ class Ark(SyncAPIClient):
     batch_chat: resources.BatchChat
     responses: resources.Responses
     input_items: resources.InputItems
+    """ `batch_chat` is deprecated, use `batch.chat` instead """
+    batch: batch.Batch
     model_breaker_map: dict[str, ModelBreaker]
     model_breaker_lock: threading.Lock
 
@@ -128,6 +131,7 @@ def __init__(
         self.batch_chat = resources.BatchChat(self)
         self.responses = resources.Responses(self)
         self.input_items = resources.InputItems(self)
+        self.batch = batch.Batch(self)
         self.model_breaker_map = defaultdict(ModelBreaker)
         self.model_breaker_lock = threading.Lock()
         # self.classification = resources.Classification(self)
@@ -186,6 +190,8 @@ class AsyncArk(AsyncAPIClient):
     batch_chat: resources.AsyncBatchChat
     responses: resources.AsyncResponses
     input_items: resources.AsyncInputItems
+    """ `batch_chat` is deprecated, use `batch.chat` instead """
+    batch: batch.AsyncBatch
     model_breaker_map: dict[str, ModelBreaker]
     model_breaker_lock: asyncio.Lock
 
@@ -254,6 +260,7 @@ def __init__(
         self.batch_chat = resources.AsyncBatchChat(self)
         self.responses = resources.AsyncResponses(self)
         self.input_items = resources.AsyncInputItems(self)
+        self.batch = batch.AsyncBatch(self)
         self.model_breaker_map = defaultdict(ModelBreaker)
         self.model_breaker_lock = asyncio.Lock()
         # self.classification = resources.AsyncClassification(self)

volcenginesdkarkruntime/resources/batch/__init__.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+from ..._compat import cached_property
+from .chat.chat import Chat, AsyncChat
+from ..._resource import SyncAPIResource, AsyncAPIResource
+
+__all__ = ["Batch", "AsyncBatch"]
+
+
+class Batch(SyncAPIResource):
+    @cached_property
+    def chat(self) -> Chat:
+        return Chat(self._client)
+
+
+class AsyncBatch(AsyncAPIResource):
+    @cached_property
+    def chat(self) -> AsyncChat:
+        return AsyncChat(self._client)

volcenginesdkarkruntime/resources/batch/batch.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+from ...._compat import cached_property
+from .completions import Completions, AsyncCompletions
+from ...._resource import SyncAPIResource, AsyncAPIResource
+
+__all__ = ["Chat", "AsyncChat"]
+
+
+class Chat(SyncAPIResource):
+    @cached_property
+    def completions(self) -> Completions:
+        return Completions(self._client)
+
+
+class AsyncChat(AsyncAPIResource):
+    @cached_property
+    def completions(self) -> AsyncCompletions:
+        return AsyncCompletions(self._client)