sr-ai 3.3.3

AI backends, caching, and AI-powered git commands for sr
Documentation 
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

use crate::ai::{AiRequest, BackendConfig, resolve_backend};
use crate::git::GitRepo;
use crate::ui;
use anyhow::Result;

#[derive(Debug, clap::Args)]
pub struct BranchArgs {
    /// Description of what you're working on
    pub description: Option<String>,

    /// Create the branch after suggesting it
    #[arg(short, long)]
    pub create: bool,
}

pub async fn run(args: &BranchArgs, backend_config: &BackendConfig) -> Result<()> {
    let repo = GitRepo::discover()?;
    let backend = resolve_backend(backend_config).await?;

    let prompt = if let Some(desc) = &args.description {
        format!("Suggest a branch name for: {desc}")
    } else {
        let status = repo.status_porcelain()?;
        let diff = repo.diff_head()?;
        format!(
            "Based on these changes, suggest a branch name:\n\nStatus:\n{status}\n\nDiff:\n{diff}"
        )
    };

    let spinner = ui::spinner(&format!(
        "Suggesting branch name with {}...",
        backend.name()
    ));

    let request = AiRequest {
        system_prompt: crate::prompts::branch::SYSTEM_PROMPT.to_string(),
        user_prompt: prompt,
        json_schema: None,
        working_dir: repo.root().to_string_lossy().to_string(),
    };

    let response = backend.request(&request, None).await?;
    spinner.finish_and_clear();

    let branch_name = response.text.trim().to_string();
    println!("{branch_name}");

    if args.create {
        std::process::Command::new("git")
            .args([
                "-C",
                &repo.root().to_string_lossy(),
                "checkout",
                "-b",
                &branch_name,
            ])
            .status()?;
    }

    Ok(())
}