Merge pull request #53 from MetaGLM/feature/batch-expire-0904

Feature/batch expire 0904

Author: haiyang679

tests/integration_tests/test_batches.py

@@ -37,7 +37,8 @@ def test_batch_create(logging_conf) -> None:
             completion_window="24h",
             metadata={
                 "description": "job test"
-            }
+            },
+            auto_delete_input_file=True
         )
         print(create)
         # Batch(id='batch_1790292763050508288', completion_window='24h', created_at=1715674031399, endpoint='/v4/chat/completions', input_file_id='20240514_ea19d21b-d256-4586-b0df-e80a45e3c286', object='batch', status='validating', cancelled_at=None, cancelling_at=None, completed_at=None, error_file_id=None, errors=None, expired_at=None, expires_at=None, failed_at=None, finalizing_at=None, in_progress_at=None, metadata={'description': 'job test'}, output_file_id=None, request_counts=BatchRequestCounts(completed=None, failed=None, total=None))

zhipuai/api_resource/batches.py

@@ -20,10 +20,11 @@ def __init__(self, client: "ZhipuAI") -> None:
     def create(
             self,
             *,
-            completion_window: Literal["24h"],
+            completion_window: str | None = None,
             endpoint: Literal["/v1/chat/completions", "/v1/embeddings"],
             input_file_id: str,
             metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+            auto_delete_input_file: bool = True,
             extra_headers: Headers | None = None,
             extra_body: Body | None = None,
             timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
@@ -36,6 +37,7 @@ def create(
                     "endpoint": endpoint,
                     "input_file_id": input_file_id,
                     "metadata": metadata,
+                    "auto_delete_input_file":auto_delete_input_file
                 },
                 batch_create_params.BatchCreateParams,
             ),

zhipuai/types/batch_create_params.py

@@ -7,7 +7,7 @@
 
 
 class BatchCreateParams(TypedDict, total=False):
-    completion_window: Required[Literal["24h"]]
+    completion_window: Required[str]
     """The time frame within which the batch should be processed.
 
     Currently only `24h` is supported.
@@ -32,3 +32,5 @@ class BatchCreateParams(TypedDict, total=False):
 
     metadata: Optional[Dict[str, str]]
     """Optional custom metadata for the batch."""
+
+    auto_delete_input_file: Optional[bool]