malvin 0.2.5

Non-interactive research and coding agent
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 std::io::{self, Write};

use super::Commands;
use crate::cli::args::Cli;

pub(crate) const MALVIN_OVERVIEW_DOC: &str = include_str!("../../default_prompts/docs/malvin.md");

pub(crate) const ROUTER_DOC: &str = include_str!("../../default_prompts/docs/router.md");

pub(crate) const DO_DOC: &str = include_str!("../../default_prompts/docs/do.md");

pub(crate) const fn command_doc_markdown(cmd: &Commands) -> &'static str {
    match cmd {
        Commands::Write(_) => include_str!("../../default_prompts/docs/write.md"),
        Commands::Admin(admin) => match &admin.command {
            super::AdminCommand::Models(_) => {
                include_str!("../../default_prompts/docs/models.md")
            }
            super::AdminCommand::ResetHerdr => {
                include_str!("../../default_prompts/docs/admin.md")
            }
        },
    }
}

#[must_use]
pub(crate) fn doc_text(command: Option<&Commands>) -> String {
    command.map_or_else(
        || format!("{MALVIN_OVERVIEW_DOC}\n---\n\n{ROUTER_DOC}"),
        |cmd| command_doc_markdown(cmd).to_string(),
    )
}

pub(crate) fn print_doc_to_writer(
    command: Option<&Commands>,
    mut out: impl Write,
) -> Result<(), String> {
    let text = doc_text(command).replace("{{ advice_path }}", crate::MALVIN_ADVICE_REL);
    out.write_all(text.as_bytes())
        .map_err(|e| format!("stdout: {e}"))?;
    Ok(())
}

pub(crate) fn print_doc_for_cli_to_writer(cli: &Cli, out: impl Write) -> Result<(), String> {
    if cli.do_workflow {
        let mut out = out;
        let text = DO_DOC.replace("{{ advice_path }}", crate::MALVIN_ADVICE_REL);
        return out
            .write_all(text.as_bytes())
            .map_err(|e| format!("stdout: {e}"));
    }
    print_doc_to_writer(cli.command.as_ref(), out)
}

pub(crate) fn print_doc_for_cli(cli: &Cli) -> Result<(), String> {
    print_doc_for_cli_to_writer(cli, io::stdout().lock())
}

#[cfg(test)]
#[path = "command_docs_tests.rs"]
mod command_docs_tests;