from http import HTTPStatus
import dashscope
import os

dashscope.base_http_api_url = "https://dashscope.aliyuncs.com/api/v1"

analysis_prompt = f"""
请帮我提取下面对话中的关键信息：[1] 客服：您好，欢迎致电AB电商平台，请问有什么可以帮您？[2] 客户：你好，我上周在你们店买了一台料理机，现在运行时有点异响。
"""

response = dashscope.Generation.call(
    api_key=os.getenv('DASHSCOPE_API_KEY'),
    model="tongyi-xiaomi-analysis-flash",
    messages=[
        {
            'role': 'user',
            'content': analysis_prompt
        }
    ],
    temperature=0.0,
    top_k=1,
    result_format="message"
)

if response.status_code == HTTPStatus.OK:
    print(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
    ))