Rust SDK

cargo add taskforceai-sdk

use taskforceai_sdk::{TaskForceAI, TaskForceAIOptions};

#[tokio::main]
async fn main() {
    let client = TaskForceAI::new(TaskForceAIOptions {
        api_key: Some("your-api-key-here".to_string()),
        ..Default::default()
    })
    .expect("failed to create client");

    let status = client
        .run_task("Analyze this repository", None, None, None)
        .await
        .expect("task failed");

    if let Some(result) = status.result {
        println!("Result: {}", result);
    }
}

use futures_util::StreamExt;

let mut stream = client
    .run_task_stream("Long running task", None)
    .await?;

while let Some(status) = stream.next().await {
    match status {
        Ok(update) => println!("Status: {:?}", update.status),
        Err(error) => eprintln!("Error: {}", error),
    }
}

use bytes::Bytes;
use taskforceai_sdk::{CreateThreadOptions, FileUploadOptions, ThreadRunOptions};

let uploaded = client
    .upload_file(
        "brief.txt",
        Bytes::from_static(b"Context for the run"),
        Some(FileUploadOptions {
            purpose: Some("assistants".to_string()),
            mime_type: Some("text/plain".to_string()),
        }),
    )
    .await?;

let thread = client
    .create_thread(Some(CreateThreadOptions {
        title: Some("Security review".to_string()),
        ..Default::default()
    }))
    .await?;

let run = client
    .run_in_thread(
        thread.id,
        ThreadRunOptions {
            prompt: "Review the uploaded brief.".to_string(),
            options: Some([("agentCount".to_string(), serde_json::json!(4))].into()),
            ..Default::default()
        },
    )
    .await?;