将官方的例子改成函数形式 ,传入问题"prompt",得到回答。过滤了api多余返回的内容。
# -*- coding: UTF-8 -*-
from openai import AzureOpenAI
client = AzureOpenAI(
api_key = "API_KEY",
api_version = "API_VERSION",
azure_endpoint = "ENDPOINT"
)
def chat(prompts):
response = client.chat.completions.create(
model="gpt4", # 模型部署名.
messages=[
# 系统角色:此消息提供模型的初始说明
# 助手的简要说明
# 助手的个性特征
# 你希望助手遵循的指令或规则
# 模型所需的数据或信息
{"role": "system", "content": "你现在是程序员,帮我实现程序编写"},
# 此消息,触发模型回复
{"role": "user", "content": prompts
}
]
)
return response.choices[0].message.content
if __name__ == '__main__':
prompts="""
输入问题
"""
out = chat(prompts)
print(out)
修改成自己的API_KEY,API_VERSION,ENDPOINT,模型部署名和助手角色