23 lines
726 B
Python
23 lines
726 B
Python
from openai import OpenAI
|
|
|
|
if __name__ == '__main__':
|
|
def chat(message):
|
|
# 构造 client
|
|
client = OpenAI(
|
|
api_key="sk-2NpKKyU5savpy6xQ4pEV0Bv63xCnSdsfnrDcFg9FXlwTjovD", # 混元 APIKey
|
|
base_url="https://api.hunyuan.cloud.tencent.com/v1", # 混元 endpoint
|
|
)
|
|
|
|
completion = client.chat.completions.create(
|
|
model="hunyuan-turbos-latest",
|
|
messages=[
|
|
{
|
|
"role": "user",
|
|
"content": message
|
|
}
|
|
],
|
|
extra_body={
|
|
"enable_enhancement": True, # <- 自定义参数
|
|
},
|
|
)
|
|
return completion.choices[0].message.content |