Update common.py

This commit is contained in:
zjk2017
2024-08-27 14:18:19 +08:00
committed by GitHub
parent e7c85007d1
commit 15b9fea253

View File

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