from openai import OpenAI
from WxMini.Milvus.Config.MulvusConfig import *


# 初始化 OpenAI 客户端
client = OpenAI(
    api_key=MODEL_API_KEY,
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
)

prompt="你好，我现在心情太糟糕了！"

# 调用大模型
response = client.chat.completions.create(
    model=MODEL_NAME,
    messages=[
        {"role": "system", "content": "你是一个非常好的聊天伙伴，可以疏导用户，帮他解压。"},
        {"role": "user", "content": prompt}
    ],
    max_tokens=500
)

# 提取生成的 SQL
if response.choices and response.choices[0].message.content:
    result = response.choices[0].message.content.strip()
    print(result)