-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest1.py
31 lines (24 loc) · 929 Bytes
/
test1.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import time
from zhipuai import ZhipuAI
client = ZhipuAI(api_key="6918fa714338f6e35e9cccce10d9d7cb.ln6PCFJRWKpcoCfI") # 请填写您自己的APIKey
while True:
user_input = input("请输入您的问题:")
response = client.chat.asyncCompletions.create(
model="chatglm_turbo", # 填写需要调用的模型名称
messages=[
{
"role": "user",
"content": user_input
}
],
)
task_id = response.id
task_status = ''
get_cnt = 0
while task_status != 'SUCCESS' and task_status != 'FAILED' and get_cnt <= 40:
result_response = client.chat.asyncCompletions.retrieve_completion_result(id=task_id)
task_status = result_response.task_status
if task_status == 'SUCCESS':
print("回复:", result_response.message)
time.sleep(2)
get_cnt += 1