|
import random |
|
from http import HTTPStatus |
|
import dashscope |
|
|
|
|
|
|
|
dashscope.api_key = "sk-948adb3e65414e55961a9ad9d22d186b" |
|
|
|
|
|
def call_with_messages(prompt): |
|
messages = [{'role': 'system', 'content': 'You are a helpful assistant.'}, |
|
{'role': 'user', 'content': prompt}] |
|
|
|
response = dashscope.Generation.call( |
|
"qwen-plus", |
|
|
|
messages=messages, |
|
|
|
seed=random.randint(1, 10000), |
|
|
|
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'] |
|
|
|
|
|
|
|
|
|
|
|
|