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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

//! AIChat CLI integration for general LLM interactions

use anyhow::{Context, Result};
use std::process::Command;
use tracing::{debug, error, info};

// TODO: document this
// TODO: document this
// TODO: document this
pub struct AIChatIntegration;

impl AIChatIntegration {
    /// Check if aichat is available in PATH
    // TODO: document this
    // TODO: document this
    // TODO: document this
    // TODO: document this
    #[must_use]
    pub fn is_available() -> bool {
        Command::new("aichat")
            .arg("--version")
            .output()
            .map(|output| output.status.success())
            .unwrap_or(false)
    }

    /// Execute aichat with given prompt and optional model
    // TODO: document this
    // TODO: document this
    // TODO: document this
    // TODO: document this
    pub async fn execute(prompt: &str, model: Option<&str>) -> Result<String> {
        if !Self::is_available() {
            return Err(anyhow::anyhow!(
                "AIChat is not available. Install with: cargo install aichat"
            ));
        }

        info!("Running aichat with prompt: {}", prompt);
        debug!("Using model: {:?}", model);

        let output = tokio::task::spawn_blocking({
            let prompt = prompt.to_owned();
            let model = model.map(std::borrow::ToOwned::to_owned);
            move || {
                let mut cmd = Command::new("aichat");

                if let Some(model) = model {
                    cmd.arg("--model").arg(model);
                }

                cmd.arg(prompt).output()
            }
        })
        .await?
        .context("Failed to execute aichat command")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            error!("AIChat execution failed: {}", stderr);
            return Err(anyhow::anyhow!("AIChat failed: {}", stderr));
        }

        let stdout = String::from_utf8_lossy(&output.stdout);
        info!("AIChat completed successfully");
        Ok(stdout.into_owned())
    }

    /// Get aichat version
    // TODO: document this
    // TODO: document this
    // TODO: document this
    // TODO: document this
    pub fn get_version() -> Result<String> {
        let output = Command::new("aichat")
            .arg("--version")
            .output()
            .context("Failed to get aichat version")?;

        if output.status.success() {
            Ok(String::from_utf8_lossy(&output.stdout).trim().to_string())
        } else {
            Err(anyhow::anyhow!("Failed to get aichat version"))
        }
    }
}