You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
715 B
26 lines
715 B
from openai import OpenAI
|
|
from TtsConfig 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) |