az-openai-rs

这个包是一个用于在 Rust 中与 Azure OpenAI 交互的库。

如何使用

您需要设置以下环境变量

OPENAI_API_KEY: <Your Azure OpenAI API Key>
OPENAI_API_URL: <Your Azure deployment url>
OPENAI_API_VERSION: <Your deployment version>
OPENAI_API_DEPLOYMENT: <Your deployment name>

使用 Chat API 的示例

    let mut messages = Vec::new();
    messages.push((
        "system",
        "You're an AI language model designed to help the user",
    ));
    messages.push(("assistant", "Hello, how can I help you?"));
    messages.push(("user", &query.query));

    let response = completions::chat()
        .messages(messages)
        .max_tokens(456)
        .send();

以及使用 completion API 的示例

let response = completions::completion().prompt("Your prompt").send().await;