llm_knowledge_base / qwen_response.py
allinaigc's picture
Upload 35 files
b2e325f verified
import random
from http import HTTPStatus
import dashscope
### 参考:
## export DASHSCOPE_API_KEY="sk-948adb3e65414e55961a9ad9d22d186b"
dashscope.api_key = "sk-948adb3e65414e55961a9ad9d22d186b"
def call_with_messages(prompt):
messages = [{'role': 'system', 'content': 'You are a helpful assistant.'},
{'role': 'user', 'content': prompt}]
# {'role': 'user', 'content': '如何做西红柿炒鸡蛋?'}]
response = dashscope.Generation.call(
"qwen-turbo",
messages=messages,
# set the random seed, optional, default to 1234 if not set
seed=random.randint(1, 10000),
# set the result to be "message" format.
result_format='message',
)
if response.status_code == HTTPStatus.OK:
print(response)
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 response['output']['choices'][0]['message']['content'] ### 这里是content的内容,不是message的全部内容。
# if __name__ == '__main__':
# # call_with_messages() ### original code here.
# res = call_with_messages() ## working.
# # print(res)