feat(ark-context): add context clone api

Author: luminghao-bytedance

volcenginesdkarkruntime/resources/context/context.py

@@ -13,7 +13,7 @@
 from ..._base_client import (
     make_request_options,
 )
-from ...types.context import CreateContextResponse
+from ...types.context import CreateContextResponse, CloneContextResponse
 from ...types.context.context_create_params import TTLTypes, TruncationStrategy, to_optional_ttl
 from ...types.chat import ChatCompletionMessageParam
 
@@ -56,6 +56,30 @@ def create(
             cast_to=CreateContextResponse,
         )
 
+    @with_sts_token
+    def clone(
+            self,
+            *,
+            context_id: str,
+            extra_headers: Headers | None = None,
+            extra_query: Query | None = None,
+            extra_body: Body | None = None,
+            timeout: float | httpx.Timeout | None = None,
+    ) -> CloneContextResponse:
+        return self._post(
+            "/context/clone",
+            body={
+                "context_id": context_id,
+            },
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+            ),
+            cast_to=CloneContextResponse,
+        )
+
 
 class AsyncContext(AsyncAPIResource):
     @cached_property

volcenginesdkarkruntime/types/context/__init__.py

@@ -4,5 +4,5 @@
 
 from .context_chat_completion_chunk import ContextChatCompletionChunk
 from .context_chat_completion import ContextChatCompletion
-from .create_context_response import CreateContextResponse
+from .create_context_response import CreateContextResponse, CloneContextResponse
 from .context_create_params import TruncationStrategy, TTLTypes

volcenginesdkarkruntime/types/context/create_context_response.py

@@ -1,7 +1,7 @@
 from ..._models import BaseModel
 from .truncation_strategy import TruncationStrategy
 
-__all__ = ["CreateContextResponse"]
+__all__ = ["CreateContextResponse", "CloneContextResponse"]
 
 
 class CreateContextResponse(BaseModel):
@@ -16,3 +16,7 @@ class CreateContextResponse(BaseModel):
     Controls for how a context will be truncated prior to the run. 
     Use this to control the context window for the chat completion.
     """
+
+
+class CloneContextResponse(CreateContextResponse):
+    pass