opencrates 3.0.1

Enterprise-grade AI-powered Rust development companion with comprehensive automation, monitoring, and deployment capabilities
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

//! Enhanced AI commands for `OpenCrates` CLI

use anyhow::Result;
use clap::{Args, Subcommand};

/// Enhanced AI commands using async-openai
#[derive(Debug, Clone, Subcommand)]
pub enum EnhancedAICommand {
    /// Generate text using advanced chat completion
    Chat(ChatArgs),
    /// Generate embeddings for semantic search
    Embeddings(EmbeddingsArgs),
}

#[derive(Debug, Clone, Args)]
pub struct ChatArgs {
    /// The prompt to send to the AI
    #[arg(short, long)]
    pub prompt: String,

    /// Model to use for generation
    #[arg(short, long, default_value = "gpt-4o")]
    pub model: String,
}

#[derive(Debug, Clone, Args)]
pub struct EmbeddingsArgs {
    /// Text inputs to generate embeddings for
    #[arg(short, long)]
    pub input: Vec<String>,

    /// Model to use for embeddings
    #[arg(short, long, default_value = "text-embedding-3-large")]
    pub model: String,
}

impl EnhancedAICommand {
    /// Execute the enhanced AI command
    pub async fn execute(&self) -> Result<()> {
        match self {
            EnhancedAICommand::Chat(args) => {
                println!(" Chat with model: {}", args.model);
                println!("Prompt: {}", args.prompt);
                Ok(())
            }
            EnhancedAICommand::Embeddings(args) => {
                println!("Generating embeddings with model: {}", args.model);
                println!("Input count: {}", args.input.len());
                Ok(())
            }
        }
    }
}