sr-ai 2.4.5

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
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

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

#[derive(Debug, clap::Args)]
pub struct PrArgs {
    /// Base branch (default: main)
    #[arg(short, long, default_value = "main")]
    pub base: String,

    /// Create the PR via gh CLI
    #[arg(short, long)]
    pub create: bool,

    /// Draft PR
    #[arg(short, long)]
    pub draft: bool,
}

#[derive(Debug, Serialize, Deserialize)]
struct PrOutput {
    title: String,
    body: String,
}

const SYSTEM_PROMPT: &str = "You are an expert at writing pull request descriptions. \
Given the commits and diff for a branch, generate a PR title and body. \
The title should be concise (under 70 chars). \
The body should have a ## Summary section with bullet points and a ## Test plan section.";

const PR_SCHEMA: &str = r#"{
    "type": "object",
    "properties": {
        "title": { "type": "string", "description": "PR title, under 70 chars" },
        "body": { "type": "string", "description": "PR body in markdown with ## Summary and ## Test plan sections" }
    },
    "required": ["title", "body"]
}"#;

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

    let branch = repo.current_branch()?;
    let log = repo.log_range(&format!("{}..HEAD", args.base), None)?;
    let diff = repo.diff_range(&args.base)?;

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

    let request = AiRequest {
        system_prompt: SYSTEM_PROMPT.to_string(),
        user_prompt: format!(
            "Generate a PR title and body for branch '{branch}' targeting '{}'.\n\n\
             Commits:\n{log}\n\nDiff:\n{diff}",
            args.base
        ),
        json_schema: Some(PR_SCHEMA.to_string()),
        working_dir: repo.root().to_string_lossy().to_string(),
    };

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

    let pr: PrOutput = serde_json::from_str(&response.text)?;

    println!("Title: {}", pr.title);
    println!();
    println!("{}", pr.body);

    if args.create {
        let mut cmd = std::process::Command::new("gh");
        cmd.args([
            "pr", "create", "--title", &pr.title, "--body", &pr.body, "--base", &args.base,
        ]);

        if args.draft {
            cmd.arg("--draft");
        }

        cmd.current_dir(repo.root()).status()?;
    }

    Ok(())
}