mirror/zjk2017-ArcadiaScriptPublic
1
0
Fork 0
mirror of https://github.com/zjk2017/ArcadiaScriptPublic.git synced 2025-12-17 15:34:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update common.py

Browse Source
This commit is contained in:
zjk2017
2024-08-27 14:18:19 +08:00
committed by GitHub
parent e7c85007d1
commit 15b9fea253
1 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
common.py
View File

@@ -27,26 +27,26 @@ def make_request(url, json_data=None, method='get', headers=None):
# 通义千问API
def qianwen_messages(basic_question, question):
content = ''
# qw_key = os.getenv("QIANWEN_KEY")
# if not qw_key:
# print(f'⛔️未获取到通义千问key：请检查变量 {qw_key} 是否填写')
# else:
# dashscope.api_key = qw_key
# messages = [{'role': 'system', 'content': 'You are a helpful assistant.'},
# {'role': 'user', 'content': basic_question + question}]
# response = dashscope.Generation.call(
# dashscope.Generation.Models.qwen_turbo,
# messages=messages,
# seed=random.randint(1, 10000),
# result_format='message',
# )
# if response.status_code == HTTPStatus.OK:
# content = response['output']['choices'][0]['message']['content']
# else:
# print('Request id: %s, Status code: %s, error code: %s, error message: %s' % (
# response.request_id, response.status_code,
# response.code, response.message
# ))
qw_key = os.getenv("QIANWEN_KEY")
if not qw_key:
print(f'⛔️未获取到通义千问key：请检查变量 {qw_key} 是否填写')
else:
dashscope.api_key = qw_key
messages = [{'role': 'system', 'content': 'You are a helpful assistant.'},
{'role': 'user', 'content': basic_question + question}]
response = dashscope.Generation.call(
dashscope.Generation.Models.qwen_turbo,
messages=messages,
seed=random.randint(1, 10000),
result_format='message',
)
if response.status_code == HTTPStatus.OK:
content = response['output']['choices'][0]['message']['content']
else:
print('Request id: %s, Status code: %s, error code: %s, error message: %s' % (
response.request_id, response.status_code,
response.code, response.message
))
return content