2 unstable releases

0.2.0 Jan 5, 2024
0.1.1 Nov 7, 2023
0.1.0 Nov 6, 2023

#31 in #chat-completion

MIT license

23KB
511 lines

async-gigachat

Async Rust library for GigaChat

Overview

async-gigachat is an unofficial Rust library for GigaChat REST API.

Usage

The library reads Authorization token from the environment variable GIGACHAT_AUTH_TOKEN.

# On macOS/Linux
export GIGACHAT_AUTH_TOKEN='YTAxNj...'
# On Windows Powershell
$Env:GIGACHAT_AUTH_TOKEN='YTAxNj...'

Chat completion Example

use anyhow::Ok;
use async_gigachat::{
    chat::{ChatCompletionRequestBuilder, ChatMessageBuilder, Role, Chat},
    client::Client,
    config::GigaChatConfig,
};

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let config = GigaChatConfig::default();

    let client: Client = Client::with_config(config);

    let question = ChatMessageBuilder::default()
    .role(Role::User)
    .content("Hey, how's it going?")
    .build()?;

    let request = ChatCompletionRequestBuilder::default()
        .messages(vec![question.clone()])
        .model("GigaChat:latest")
        .build()?;

    let response = Chat::new(client).completion(request).await?;

    println!("{}: {}", question.role, question.content);
    println!("{}: {}", response.choices.get(0).unwrap().message.role, response.choices.get(0).unwrap().message.content);

    Ok(())
}

License

This project is licensed under MIT license.

Dependencies

~9–21MB
~294K SLoC