Skip to content

Commit 98ae49b

Browse files
author
BitsAdmin
committed
Merge branch 'feat/add_dimensions' into 'integration_2025-07-03_982885029122'
feat: [development task] ark runtime (1417808) See merge request iaasng/volcengine-python-sdk!700
2 parents 96f1e0a + ff0e075 commit 98ae49b

File tree

3 files changed

+18
-0
lines changed

3 files changed

+18
-0
lines changed

volcenginesdkarkruntime/resources/batch_chat/completions.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,8 @@ def create(
188188
service_tier: Optional[Literal["auto", "default"]] | None = None,
189189
tool_choice: ChatCompletionToolChoiceOptionParam | None = None,
190190
response_format: completion_create_params.ResponseFormat | None = None,
191+
thinking: completion_create_params.Thinking | None = None,
192+
max_completion_tokens: Optional[int] | None = None,
191193
user: str | None = None,
192194
extra_headers: Headers | None = None,
193195
extra_query: Query | None = None,
@@ -232,6 +234,8 @@ def create(
232234
"service_tier": service_tier,
233235
"tool_choice": tool_choice,
234236
"response_format": response_format,
237+
"thinking": thinking,
238+
"max_completion_tokens": max_completion_tokens,
235239
},
236240
options=make_request_options(
237241
extra_headers=extra_headers,
@@ -336,6 +340,8 @@ async def create(
336340
service_tier: Optional[Literal["auto", "default"]] | None = None,
337341
tool_choice: ChatCompletionToolChoiceOptionParam | None = None,
338342
response_format: completion_create_params.ResponseFormat | None = None,
343+
thinking: completion_create_params.Thinking | None = None,
344+
max_completion_tokens: Optional[int] | None = None,
339345
extra_headers: Headers | None = None,
340346
extra_query: Query | None = None,
341347
extra_body: Body | None = None,
@@ -380,6 +386,8 @@ async def create(
380386
"service_tier": service_tier,
381387
"tool_choice": tool_choice,
382388
"response_format": response_format,
389+
"thinking": thinking,
390+
"max_completion_tokens": max_completion_tokens,
383391
},
384392
options=make_request_options(
385393
extra_headers=extra_headers,

volcenginesdkarkruntime/resources/multimodal_embeddings.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def create(
3030
input: List[EmbeddingInputParam],
3131
model: str,
3232
encoding_format: Literal["float", "base64"] = "float",
33+
dimensions: int | None = None,
3334
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
3435
# The extra values given here take precedence over values defined on the client or passed to this method.
3536
extra_headers: Headers | None = None,
@@ -43,6 +44,7 @@ def create(
4344
"input": input,
4445
"model": model,
4546
"encoding_format": encoding_format,
47+
"dimensions": dimensions,
4648
},
4749
options=make_request_options(
4850
extra_headers=extra_headers,
@@ -66,6 +68,7 @@ async def create(
6668
input: List[EmbeddingInputParam],
6769
model: str,
6870
encoding_format: Literal["float", "base64"] = "float",
71+
dimensions: int | None = None,
6972
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
7073
# The extra values given here take precedence over values defined on the client or passed to this method.
7174
extra_headers: Headers | None = None,
@@ -79,6 +82,7 @@ async def create(
7982
"input": input,
8083
"model": model,
8184
"encoding_format": encoding_format,
85+
"dimensions": dimensions,
8286
},
8387
options=make_request_options(
8488
extra_headers=extra_headers,

volcenginesdkarkruntime/types/content_generation/content_generation_task.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,3 +45,9 @@ class ContentGenerationTask(BaseModel):
4545

4646
updated_at: int
4747
"""The Unix timestamp when the task was last updated."""
48+
49+
seed: int
50+
"""The seed used in the model."""
51+
52+
revised_prompt: str
53+
"""The revised prompt the model uses to generate content"""

0 commit comments

Comments
 (0)