Skip to content

Commit 4ddbbad

Browse files
author
jinhaiyang
committed
【web_search】web_search
1 parent 41fac59 commit 4ddbbad

File tree

14 files changed

+1695
-1695
lines changed

14 files changed

+1695
-1695
lines changed

‎tests/integration_tests/test.py‎

Lines changed: 25 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,25 @@
1-
import time
2-
3-
from zhipuai import ZhipuAI
4-
client = ZhipuAI() # 填写您自己的APIKey
5-
6-
response = client.videos.generations(
7-
model="cogvideo",
8-
prompt="一个年轻的艺术家在一片彩虹上用调色板作画。"
9-
# prompt="一只卡通狐狸在森林里跳着欢快的爵士舞。"
10-
# prompt="这是一部汽车广告片，描述了一位30岁的汽车赛车手戴着红色头盔的赛车冒险。背景是蔚蓝的天空和苛刻的沙漠环境，电影风格使用35毫米胶片拍摄，色彩鲜艳夺目。"
11-
)
12-
print(response)
13-
task_id = response.id
14-
task_status = response.task_status
15-
get_cnt = 0
16-
17-
while task_status == 'PROCESSING' and get_cnt <= 40:
18-
result_response = client.videos.retrieve_videos_result(
19-
id=task_id
20-
)
21-
print(result_response)
22-
task_status = result_response.task_status
23-
24-
time.sleep(2)
25-
get_cnt += 1
1+
# import time
2+
#
3+
# from zhipuai import ZhipuAI
4+
# client = ZhipuAI() # 填写您自己的APIKey
5+
#
6+
# response = client.videos.generations(
7+
# model="cogvideo",
8+
# prompt="一个年轻的艺术家在一片彩虹上用调色板作画。"
9+
# # prompt="一只卡通狐狸在森林里跳着欢快的爵士舞。"
10+
# # prompt="这是一部汽车广告片，描述了一位30岁的汽车赛车手戴着红色头盔的赛车冒险。背景是蔚蓝的天空和苛刻的沙漠环境，电影风格使用35毫米胶片拍摄，色彩鲜艳夺目。"
11+
# )
12+
# print(response)
13+
# task_id = response.id
14+
# task_status = response.task_status
15+
# get_cnt = 0
16+
#
17+
# while task_status == 'PROCESSING' and get_cnt <= 40:
18+
# result_response = client.videos.retrieve_videos_result(
19+
# id=task_id
20+
# )
21+
# print(result_response)
22+
# task_status = result_response.task_status
23+
#
24+
# time.sleep(2)
25+
# get_cnt += 1

‎tests/integration_tests/test_assistant.py‎

Lines changed: 111 additions & 111 deletions
Original file line numberDiff line numberDiff line change
@@ -1,111 +1,111 @@
1-
import logging
2-
import logging.config
3-
import zhipuai
4-
from zhipuai import ZhipuAI
5-
import json
6-
7-
def test_assistant(logging_conf) -> None:
8-
logging.config.dictConfig(logging_conf) # type: ignore
9-
client = ZhipuAI() # 填写您自己的APIKey
10-
try:
11-
12-
generate = client.assistant.conversation(
13-
assistant_id="659e54b1b8006379b4b2abd6",
14-
messages=[
15-
{
16-
"role": "user",
17-
"content": [{
18-
"type": "text",
19-
"text": "帮我搜索下智谱的cogvideox发布时间"
20-
}]
21-
}
22-
],
23-
stream=True,
24-
attachments=None,
25-
metadata=None,
26-
request_id="request_1790291013237211136",
27-
user_id="12345678"
28-
)
29-
for assistant in generate:
30-
print(assistant)
31-
32-
except zhipuai.core._errors.APIRequestFailedError as err:
33-
print(err)
34-
except zhipuai.core._errors.APIInternalError as err:
35-
print(err)
36-
except zhipuai.core._errors.APIStatusError as err:
37-
print(err)
38-
39-
def test_assistant_query_support(logging_conf) -> None:
40-
logging.config.dictConfig(logging_conf) # type: ignore
41-
client = ZhipuAI() # 填写您自己的APIKey
42-
try:
43-
44-
response = client.assistant.query_support(
45-
assistant_id_list=[],
46-
request_id="request_1790291013237211136",
47-
user_id="12345678"
48-
)
49-
print(response)
50-
51-
except zhipuai.core._errors.APIRequestFailedError as err:
52-
print(err)
53-
except zhipuai.core._errors.APIInternalError as err:
54-
print(err)
55-
except zhipuai.core._errors.APIStatusError as err:
56-
print(err)
57-
58-
def test_assistant_query_conversation_usage(logging_conf) -> None:
59-
logging.config.dictConfig(logging_conf) # type: ignore
60-
client = ZhipuAI() # 填写您自己的APIKey
61-
try:
62-
response = client.assistant.query_conversation_usage(
63-
assistant_id="659e54b1b8006379b4b2abd6",
64-
request_id="request_1790291013237211136",
65-
user_id="12345678"
66-
)
67-
print(response)
68-
except zhipuai.core._errors.APIRequestFailedError as err:
69-
print(err)
70-
except zhipuai.core._errors.APIInternalError as err:
71-
print(err)
72-
except zhipuai.core._errors.APIStatusError as err:
73-
print(err)
74-
75-
76-
def test_translate_api(logging_conf) -> None:
77-
logging.config.dictConfig(logging_conf) # type: ignore
78-
client = ZhipuAI() # 填写您自己的APIKey
79-
try:
80-
translate_response = client.assistant.conversation(
81-
assistant_id="9996ijk789lmn012o345p999",
82-
messages=[
83-
{
84-
"role": "user",
85-
"content": [{
86-
"type": "text",
87-
"text": "你好呀"
88-
}]
89-
}
90-
],
91-
stream=True,
92-
attachments=None,
93-
metadata=None,
94-
request_id="request_1790291013237211136",
95-
user_id="12345678",
96-
extra_parameters = {
97-
"translate": {
98-
"from": "zh",
99-
"to": "en"
100-
}
101-
}
102-
)
103-
for chunk in translate_response:
104-
print(chunk.choices[0].delta)
105-
#print(translate_response)
106-
except zhipuai.core._errors.APIRequestFailedError as err:
107-
print(err)
108-
except zhipuai.core._errors.APIInternalError as err:
109-
print(err)
110-
except zhipuai.core._errors.APIStatusError as err:
111-
print(err)
1+
# import logging
2+
# import logging.config
3+
# import zhipuai
4+
# from zhipuai import ZhipuAI
5+
# import json
6+
#
7+
# def test_assistant(logging_conf) -> None:
8+
# logging.config.dictConfig(logging_conf) # type: ignore
9+
# client = ZhipuAI() # 填写您自己的APIKey
10+
# try:
11+
#
12+
# generate = client.assistant.conversation(
13+
# assistant_id="659e54b1b8006379b4b2abd6",
14+
# messages=[
15+
# {
16+
# "role": "user",
17+
# "content": [{
18+
# "type": "text",
19+
# "text": "帮我搜索下智谱的cogvideox发布时间"
20+
# }]
21+
# }
22+
# ],
23+
# stream=True,
24+
# attachments=None,
25+
# metadata=None,
26+
# request_id="request_1790291013237211136",
27+
# user_id="12345678"
28+
# )
29+
# for assistant in generate:
30+
# print(assistant)
31+
#
32+
# except zhipuai.core._errors.APIRequestFailedError as err:
33+
# print(err)
34+
# except zhipuai.core._errors.APIInternalError as err:
35+
# print(err)
36+
# except zhipuai.core._errors.APIStatusError as err:
37+
# print(err)
38+
#
39+
# def test_assistant_query_support(logging_conf) -> None:
40+
# logging.config.dictConfig(logging_conf) # type: ignore
41+
# client = ZhipuAI() # 填写您自己的APIKey
42+
# try:
43+
#
44+
# response = client.assistant.query_support(
45+
# assistant_id_list=[],
46+
# request_id="request_1790291013237211136",
47+
# user_id="12345678"
48+
# )
49+
# print(response)
50+
#
51+
# except zhipuai.core._errors.APIRequestFailedError as err:
52+
# print(err)
53+
# except zhipuai.core._errors.APIInternalError as err:
54+
# print(err)
55+
# except zhipuai.core._errors.APIStatusError as err:
56+
# print(err)
57+
#
58+
# def test_assistant_query_conversation_usage(logging_conf) -> None:
59+
# logging.config.dictConfig(logging_conf) # type: ignore
60+
# client = ZhipuAI() # 填写您自己的APIKey
61+
# try:
62+
# response = client.assistant.query_conversation_usage(
63+
# assistant_id="659e54b1b8006379b4b2abd6",
64+
# request_id="request_1790291013237211136",
65+
# user_id="12345678"
66+
# )
67+
# print(response)
68+
# except zhipuai.core._errors.APIRequestFailedError as err:
69+
# print(err)
70+
# except zhipuai.core._errors.APIInternalError as err:
71+
# print(err)
72+
# except zhipuai.core._errors.APIStatusError as err:
73+
# print(err)
74+
#
75+
#
76+
# def test_translate_api(logging_conf) -> None:
77+
# logging.config.dictConfig(logging_conf) # type: ignore
78+
# client = ZhipuAI() # 填写您自己的APIKey
79+
# try:
80+
# translate_response = client.assistant.conversation(
81+
# assistant_id="9996ijk789lmn012o345p999",
82+
# messages=[
83+
# {
84+
# "role": "user",
85+
# "content": [{
86+
# "type": "text",
87+
# "text": "你好呀"
88+
# }]
89+
# }
90+
# ],
91+
# stream=True,
92+
# attachments=None,
93+
# metadata=None,
94+
# request_id="request_1790291013237211136",
95+
# user_id="12345678",
96+
# extra_parameters = {
97+
# "translate": {
98+
# "from": "zh",
99+
# "to": "en"
100+
# }
101+
# }
102+
# )
103+
# for chunk in translate_response:
104+
# print(chunk.choices[0].delta)
105+
# #print(translate_response)
106+
# except zhipuai.core._errors.APIRequestFailedError as err:
107+
# print(err)
108+
# except zhipuai.core._errors.APIInternalError as err:
109+
# print(err)
110+
# except zhipuai.core._errors.APIStatusError as err:
111+
# print(err)

0 commit comments

Comments
 (0)