File tree Expand file tree Collapse file tree 3 files changed +52
-2
lines changed
resources/content_generation
volcenginesdkexamples/volcenginesdkarkruntime Expand file tree Collapse file tree 3 files changed +52
-2
lines changed Original file line number Diff line number Diff 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 (
Original file line number Diff line number Diff line change 11from typing import List
2- from pydantic import BaseModel
2+
3+ from volcenginesdkarkruntime ._models import BaseModel
34
45__all__ = ["ListContentGenerationTasksResponse" ]
56
Original file line number Diff line number Diff line change 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 } " )
You can’t perform that action at this time.
0 commit comments