AzureOpenAI的api调用

最新推荐文章于 2024-06-30 03:24:59 发布

桥呗

最新推荐文章于 2024-06-30 03:24:59 发布

阅读量449

点赞数 2

文章标签： python flask 开发语言

本文链接：https://blog.csdn.net/weixin_62460336/article/details/137672986

版权

将官方的例子改成函数形式，传入问题"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，模型部署名和助手角色