Cohere Rust SDK

此软件包提供了 Rust 与 cohere.ai API 的简化接口。

文档

请参阅 Cohere API 的文档。

有关 SDK 的代码示例，请参阅 此处。

用法

要使用此软件包，您必须首先获取一个 Cohere API 密钥。一旦您有了 API 密钥，您可以将它设置为 COHERE_API_KEY 环境变量，或者直接在构建客户端时传递。

此外，此软件包依赖于 tokio 异步运行时，以确保所有 API 操作都是非阻塞的。

以下是创建客户端并使用 generate 端点的示例。

use cohere_rust::api::{
    generate::{GenerateRequest, ReturnLikelihoods},
    Truncate,
};
use cohere_rust::Cohere;

#[tokio::main]
async fn main() {
    // automatically reads API key from `COHERE_API_KEY`
    let co = Cohere::default();

    let request = GenerateRequest {
        max_tokens: Some(20),
        return_likelihoods: Some(ReturnLikelihoods::None),
        truncate: Some(Truncate::End),
        prompt: "Once upon a time in a magical land called",
        ..Default::default()
    };

    match co.generate(&request).await {
        Ok(r) => println!("Generate response: {:?}", r),
        Err(e) => {
            println!("Generate failed! {}", e)
        }
    }
}

其他端点的示例用法可以在 此处 找到。

版本管理

此 SDK 支持最新 API 版本。有关更多信息，请参阅 版本管理文档。

端点

有关端点和参数的完整说明，请参阅 Cohere 文档。

响应

所有端点函数都将返回与端点相对应的 Cohere 对象（例如，对于 generate，它将是 GenerateResponse）。这些字段的名称和响应体的详细说明可以在 Cohere 文档 中找到。

错误

SDK中的API调用失败将返回错误。请参阅文档中关于错误的页面以获取更多关于错误含义的信息。