4.7 KiB

Raw Blame History

Gemini SDK

A simple and powerful Rust SDK for Google Gemini AI services.

Features

Simple Interface: Send text + attachments to Gemini AI
Agent System: Create specialized AI agents with custom system prompts and configurations
Type Safety: Strong typing with compile-time error checking
Easy Integration: Minimal external dependencies

Quick Start

Add this to your Cargo.toml:

[dependencies]
gemini-sdk = { path = "../gemini-sdk" }
tokio = { version = "1.0", features = ["full"] }

Basic Usage

use gemini_sdk::{GeminiSDK, MessageRequest, Agent};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Initialize SDK
    let mut sdk = GeminiSDK::new()?;

    // Simple text message
    let request = MessageRequest::text_only("Hello, how are you?");
    let response = sdk.send_message(request).await?;
    println!("Response: {}", response.content);

    Ok(())
}

Using Agents

use gemini_sdk::{GeminiSDK, MessageRequest, Agent};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut sdk = GeminiSDK::new()?;

    // Create a specialized agent
    let agent = Agent::new(
        "code_reviewer",
        "Code Reviewer",
        "You are an expert code reviewer. Analyze code for bugs, suggest improvements, and provide constructive feedback."
    ).with_temperature(0.3)  // Lower temperature for more focused responses
     .with_max_tokens(4096);

    sdk.add_agent(agent);

    // Use the agent
    let request = MessageRequest::with_agent(
        "Please review this Python code: def fibonacci(n): return n if n <= 1 else fibonacci(n-1) + fibonacci(n-2)",
        "code_reviewer"
    );

    let response = sdk.send_message(request).await?;
    println!("Code review: {}", response.content);

    Ok(())
}

With Image Attachments

use gemini_sdk::{GeminiSDK, MessageRequest, Attachment};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut sdk = GeminiSDK::new()?;

    let request = MessageRequest::with_attachments(
        "What do you see in this image?",
        vec![Attachment::new("./image.jpg")]
    );

    let response = sdk.send_message(request).await?;
    println!("Image analysis: {}", response.content);

    Ok(())
}

Configuration

The SDK uses environment variables for configuration, with sensible defaults:

GEMINI_BASE_URL: API base URL (default: Bowong AI endpoint)
GEMINI_BEARER_TOKEN: Bearer token for authentication (default: "bowong7777")
GEMINI_MODEL_NAME: Model to use (default: "gemini-2.5-flash")
GEMINI_TEMPERATURE: Default temperature (default: 0.7)
GEMINI_MAX_TOKENS: Default max tokens (default: 64000)

You can also create a custom configuration:

use gemini_sdk::{GeminiSDK, GeminiConfig};

let config = GeminiConfig {
    base_url: "https://your-api-endpoint.com".to_string(),
    bearer_token: "your-token".to_string(),
    model_name: "gemini-2.5-flash".to_string(),
    temperature: 0.8,
    max_tokens: 32000,
    ..Default::default()
};

let mut sdk = GeminiSDK::with_config(config)?;

Agent System

Agents are specialized AI assistants with their own system prompts and configurations:

// Create an agent
let agent = Agent::new("agent_id", "Agent Name", "System prompt here")
    .with_temperature(0.5)    // 0.0 = focused, 2.0 = creative
    .with_max_tokens(8192);   // Maximum response length

// Add to SDK
sdk.add_agent(agent);

// Use the agent
let request = MessageRequest::with_agent("Your message", "agent_id");
let response = sdk.send_message(request).await?;

Agent Management

// List all agents
let agents = sdk.list_agents();
for agent in agents {
    println!("Agent: {} ({})", agent.name, agent.id);
}

// Get specific agent
if let Some(agent) = sdk.get_agent("agent_id") {
    println!("Found agent: {}", agent.name);
}

// Remove agent
sdk.remove_agent("agent_id");

Supported File Types

Images: JPEG, PNG, GIF, WebP, BMP
Videos: MP4, AVI, MOV, WebM (uploaded to cloud storage)
Text: Plain text, Markdown, HTML, JSON

Error Handling

The SDK provides detailed error types:

use gemini_sdk::SDKError;

match sdk.send_message(request).await {
    Ok(response) => println!("Success: {}", response.content),
    Err(SDKError::AgentNotFound(id)) => println!("Agent {} not found", id),
    Err(SDKError::Attachment(msg)) => println!("Attachment error: {}", msg),
    Err(SDKError::Api { status, message }) => println!("API error {}: {}", status, message),
    Err(e) => println!("Other error: {}", e),
}

Examples

Run the examples to see the SDK in action:

cargo run --example basic_usage

License

MIT License

4.7 KiB Raw Blame History