vectorize_multiple_texts/

vectorize_multiple_texts.rs

use dim_rs::{prelude::*, vectorization::ModelParameters};
use tokio;
use anyhow::{Error, Result};
use async_openai;

#[tokio::main]
async fn main() -> Result<(), Error> {
    // Load multiple texts
    let texts = vec![
        "Hi, this is dim. I am here to vectorize whatever you want.".to_string(),
        "The weather is beautiful today. Perfect for a walk outside.".to_string(),
        "Artificial intelligence is transforming how we live and work.".to_string(),
        "Remember to drink water and stay hydrated throughout the day.".to_string(),
        "The quick brown fox jumps over the lazy dog.".to_string(),
        "Programming is both an art and a science.".to_string(),
        "Music has the power to change our moods instantly.".to_string(),
        "Exercise regularly for better physical and mental health.".to_string(),
        "Learning a new language opens doors to different cultures.".to_string(),
        "Time management is essential for productivity.".to_string(),
        "Reading books can expand your knowledge and imagination.".to_string(),
        "Traveling allows you to experience new perspectives.".to_string(),
        "Cooking at home can be both fun and healthy.".to_string(),
        "Meditation helps in reducing stress and improving focus.".to_string(),
        "Gardening is a relaxing and rewarding hobby.".to_string(),
        "Volunteering can make a positive impact on your community.".to_string(),
        "Photography captures moments and memories.".to_string(),
        "Writing in a journal can help clarify your thoughts.".to_string(),
        "Playing board games is a great way to bond with family and friends.".to_string(),
        "Learning to play a musical instrument can be very fulfilling.".to_string(),
    ];
    
    // Create Vector objects from the texts
    let mut vectors: Vec<Vector<String>> = texts.into_iter()
        .map(Vector::from_text)
        .collect();
    
    // Initialize client
    let client: async_openai::Client<async_openai::config::OpenAIConfig> = async_openai::Client::with_config(
        async_openai::config::OpenAIConfig::new()
            .with_api_base("http://192.168.0.101:11434/v1") // comment this out if you use OpenAI instead of Ollama
            .with_api_key("your_api_key")
    );
    
    // Initialize prompts
    let prompts: Vec<String> = vec![
        "Score the sentiment intensity of the text from 1 (extremely negative) to 9 (extremely positive). Consider emotional language, tone, and context. Format your response exactly like this example: {'sentiment_score': 7}".to_string(),
        "Rate the formality of the text from 1 (highly informal, slang-heavy) to 9 (highly formal, academic/professional). Format your response exactly like this example: {'formality_score': 4}".to_string(),
        "Assess the emotional intensity of the text from 1 (neutral/clinical) to 9 (highly emotional, passionate, or provocative). Format your response exactly like this example: {'emotional_score': 8}".to_string(),
        "Score how subjective the text is from 1 (purely factual/objective) to 9 (heavily opinionated/subjective). Format your response exactly like this example: {'subjectivity_score': 6}".to_string(),
        "Rate the linguistic complexity of the text from 1 (simple vocabulary/short sentences) to 9 (dense jargon/long, intricate sentences). Format your response exactly like this example: {'complexity_score': 3}".to_string(),
        "Score the dominant intent: 1-3 (informative/educational), 4-6 (persuasive/argumentative), 7-9 (narrative/storytelling). Format your response exactly like this example: {'intent_score': 5}".to_string(),
        "Rate how urgent or time-sensitive the text feels from 1 (no urgency) to 9 (immediate action required). Format your response exactly like this example: {'urgency_score': 2}".to_string(),
        "Score the specificity of details from 1 (vague/abstract) to 9 (highly specific/concrete examples). Format your response exactly like this example: {'specificity_score': 7}".to_string(),
        "Rate the politeness of the tone from 1 (rude/confrontational) to 9 (extremely polite/deferential). Format your response exactly like this example: {'politeness_score': 8}".to_string(),
        "Categorize the text's primary domain: 1-3 (technical/scientific), 4-6 (casual/everyday), 7-9 (artistic/creative). Format your response exactly like this example: {'domain_score': 4}".to_string(),
    ];

    // Vectorize all texts
    for vector in &mut vectors {
        let model_parameters = ModelParameters::new("mistral".to_string(), None, None);
        vectorize_string_concurrently(
            prompts.clone(),
            vector,
            client.clone(),
            model_parameters
        ).await?;
    }

    // Print statistics and validate vectors
    println!("\n=== Vectorization Results ===\n");
    
    // Get all vector lengths
    let lengths: Vec<usize> = vectors.iter()
        .map(|v| v.get_vector().len())
        .collect();
    
    // Validate that all vectors have the same length
    let first_len = lengths[0];
    let all_same_length = lengths.iter().all(|&len| len == first_len);

    // Print results for each vector
    for (i, vector) in vectors.iter().enumerate() {
        println!("Text #{}", i + 1);
        println!("Vector: {:?}", vector.get_vector());
        println!("Length: {}", vector.get_vector().len());
        println!();
    }

    // Print validation results
    println!("=== Validation ===");
    println!("All vectors have same length: {}", all_same_length);
    println!("Vector dimension: {}", first_len);
    
    if !all_same_length {
        println!("WARNING: Inconsistent vector lengths detected!");
        println!("Lengths: {:?}", lengths);
    }

    Ok(())
}