Rig : A Tool For LLM-Powered Applications

Rig is a cutting-edge Rust library designed to facilitate the development of scalable, modular, and ergonomic applications powered by large language models (LLMs).

With its robust features and integrations, Rig simplifies the process of embedding LLM capabilities into applications, making it a valuable tool for developers working with AI technologies.

Key Features Of Rig

Rig offers several high-level features that make it a standout tool for LLM-powered workflows:

• Comprehensive LLM Support: Rig supports both completion and embedding workflows, enabling seamless interaction with LLMs.
• Abstraction Over Providers: It provides simple yet powerful abstractions over popular LLM providers like OpenAI and Cohere, as well as vector stores such as MongoDB, SQLite, and in-memory options.
• Minimal Boilerplate: Rig minimizes the amount of boilerplate code required to integrate LLMs into applications, enhancing developer productivity.

Getting Started With Rig

To start using Rig, you can add the core library to your Rust project with the following command:

cargo add rig-core

Here’s a simple example of how to use Rig to interact with OpenAI’s GPT-4 model:

use rig::{completion::Prompt, providers::openai};

#[tokio::main]
async fn main() {
    // Create OpenAI client and model
    let openai_client = openai::Client::from_env();
    let gpt4 = openai_client.agent("gpt-4").build();

    // Prompt the model and print its response
    let response = gpt4
        .prompt("Who are you?")
        .await
        .expect("Failed to prompt GPT-4");

    println!("GPT-4: {response}");
}

This example demonstrates how easily developers can set up an OpenAI client, send a prompt, and retrieve a response. Note that this requires setting the OPENAI_API_KEY environment variable.

Rig supports a variety of model providers and vector stores, ensuring flexibility for different use cases:

• Model Providers: OpenAI (ChatGPT), Cohere, Anthropic (Claude), Gemini, xAI.
• Vector Stores: MongoDB, LanceDB, Neo4j, Qdrant, SQLite.

Each vector store is available as a companion crate (e.g., rig-mongodb, rig-lancedb), allowing developers to choose the best fit for their application.

Rig is an evolving tool designed to streamline the integration of LLMs into applications. With its modular design and extensive integrations, it empowers developers to build innovative AI-driven solutions efficiently.

As Rig continues to evolve with new features and updates, it remains a promising library for the future of AI application development.