Skip to content

Commit f90e885

Browse files
feat: add content generation task sync example
1 parent 06db106 commit f90e885

File tree

3 files changed

+52
-2
lines changed

3 files changed

+52
-2
lines changed

volcenginesdkarkruntime/resources/content_generation/tasks.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ def create(
4242

4343
def get(
4444
self,
45+
*,
4546
task_id: str,
4647
extra_headers: Headers | None = None,
4748
extra_query: Query | None = None,
@@ -108,7 +109,7 @@ def delete(
108109
extra_query: Query | None = None,
109110
extra_body: Body | None = None,
110111
timeout: float | httpx.Timeout | None = None,
111-
):
112+
) -> None:
112113
resp = self._delete(
113114
path=f"/contents/generations/tasks/{task_id}",
114115
options=make_request_options(

volcenginesdkarkruntime/types/content_generation/list_content_generation_tasks_response.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from typing import List
2-
from pydantic import BaseModel
2+
3+
from volcenginesdkarkruntime._models import BaseModel
34

45
__all__ = ["ListContentGenerationTasksResponse"]
56

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
1+
from volcenginesdkarkruntime import Ark
2+
3+
# Authentication
4+
# 1.If you authorize your endpoint using an API key, you can set your api key to environment variable "ARK_API_KEY"
5+
# or specify api key by Ark(api_key="${YOUR_API_KEY}").
6+
# Note: If you use an API key, this API key will not be refreshed.
7+
# To prevent the API from expiring and failing after some time, choose an API key with no expiration date.
8+
client = Ark()
9+
10+
if __name__ == "__main__":
11+
print("----- create request -----")
12+
create_result = client.content_generation.tasks.create(
13+
model="${YOUR_MODEL_EP}",
14+
content=[
15+
{
16+
"type":"text",
17+
"text":"龙与地下城女骑士背景是起伏的平原,目光从镜头转向平原"
18+
},
19+
{
20+
"type":"image_url",
21+
"image_url": {
22+
"url": "${YOUR_IMAGE_URL}"
23+
},
24+
}
25+
]
26+
)
27+
print(create_result)
28+
29+
print("----- get request -----")
30+
get_result = client.content_generation.tasks.get(task_id=create_result.id)
31+
print(get_result)
32+
33+
print("----- list request -----")
34+
list_result = client.content_generation.tasks.list(
35+
page_num=1,
36+
page_size=10,
37+
status="queued",
38+
# model="${YOUR_MODEL_EP}", # Filter by model
39+
# task_ids=["test-id-1", "test-id-2"] # Filter by task_ids
40+
)
41+
print(list_result)
42+
43+
print("----- delete request -----")
44+
try:
45+
client.content_generation.tasks.delete(task_id=create_result.id)
46+
print(create_result.id)
47+
except Exception as e:
48+
print(f"failed to delete task: {e}")

0 commit comments

Comments
 (0)