async-openai: 异步Rust库助力OpenAI API开发

async-openai:为Rust开发者打造的OpenAI异步库

在人工智能快速发展的今天,OpenAI的各种强大API为开发者提供了无限可能。然而,如何高效、便捷地调用这些API,一直是开发者关注的焦点。async-openai应运而生,它是一个专为Rust开发者设计的非官方异步库,旨在简化OpenAI API的调用过程,提高开发效率。

库的主要特性

async-openai基于OpenAI的OpenAPI规范开发,提供了丰富的功能支持:

全面的API覆盖:支持Assistants(v2)、Audio、Batch、Chat、Completions(Legacy)、Embeddings、Files、Fine-Tuning、Images、Models、Moderations等多种OpenAI API。
异步支持:利用Rust的异步特性,提供高效的API调用方式。
SSE流式处理:对于支持的API,提供服务器发送事件(SSE)的流式处理能力。
自动重试机制:除SSE流式处理外,所有请求(包括表单提交)在遇到速率限制时会自动进行指数退避重试。
人性化的构建模式:为所有请求对象提供符合人体工程学的构建器模式。
Azure OpenAI支持:兼容Microsoft Azure OpenAI Service(仅限与OpenAI规范匹配的API)。

OpenAI API调用示例

使用指南

使用async-openai非常简单。首先,你需要设置OpenAI的API密钥。库会从环境变量OPENAI_API_KEY中读取API密钥:

# 在macOS/Linux上
export OPENAI_API_KEY='sk-...'

# 在Windows Powershell上
$Env:OPENAI_API_KEY='sk-...'

接下来,让我们通过一个图像生成的例子来展示async-openai的使用方法:

use async_openai::{
    types::{CreateImageRequestArgs, ImageSize, ImageResponseFormat},
    Client,
};
use std::error::Error;

#[tokio::main]
async fn main() -> Result<(), Box<dyn Error>> {
    // 创建客户端,从环境变量读取API密钥
    let client = Client::new();

    let request = CreateImageRequestArgs::default()
        .prompt("cats on sofa and carpet in living room")
        .n(2)
        .response_format(ImageResponseFormat::Url)
        .size(ImageSize::S256x256)
        .user("async-openai")
        .build()?;

    let response = client.images().create(request).await?;

    // 下载并保存图片到./data目录
    // 每个URL在专用的Tokio任务中下载和保存
    // 如果目录不存在会自动创建
    let paths = response.save("./data").await?;

    paths
        .iter()
        .for_each(|path| println!("Image file path: {}", path.display()));

    Ok(())
}