malvin 0.2.6

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

use crate::artifacts::RunArtifacts;
use crate::prompt_stratification::{
    AggregatedInitialPrompt, AggregatedInitialPromptBuilder,
};
use crate::prompts::{PromptStore, header_prompt_file};

use super::{
    RouterAPromptInput, RouterHeaderPromptInput, build_router_a_prompt,
    build_router_header_prompt, build_router_mbc2_prompt, router_a_prompt_label,
};

#[allow(clippy::struct_excessive_bools)]
pub(crate) struct RouterInitialPromptInput<'a> {
    pub store: &'a PromptStore,
    pub artifacts: &'a RunArtifacts,
    pub model: &'a str,
    pub gates: bool,
    pub no_kpop: bool,
    pub creative: bool,
    pub max_hypotheses: usize,
    pub include_header: bool,
}

pub(crate) type RouterInitialPrompt = AggregatedInitialPrompt;

const ROUTER_INITIAL_LOG_WHO: &str = "router_initial";

pub(crate) fn build_router_initial_prompt(
    input: RouterInitialPromptInput<'_>,
) -> Result<RouterInitialPrompt, String> {
    let mut builder = AggregatedInitialPromptBuilder::new();

    if input.include_header {
        let header = build_router_header_prompt(RouterHeaderPromptInput {
            store: input.store,
            artifacts: input.artifacts,
            model: input.model,
            max_hypotheses: input.max_hypotheses,
            no_kpop: input.no_kpop,
        })?;
        builder.push_nonempty(header_prompt_file(), header);
    }

    if input.creative {
        let mbc2 = build_router_mbc2_prompt(input.store, input.artifacts)?;
        builder.push_nonempty("mbc2.md", mbc2);
    }

    let router_a = build_router_a_prompt(RouterAPromptInput {
        store: input.store,
        artifacts: input.artifacts,
        model: input.model,
        gates: input.gates,
        no_kpop: input.no_kpop,
    })?;
    builder.push_nonempty(router_a_prompt_label(input.no_kpop), router_a);

    Ok(builder.finish(ROUTER_INITIAL_LOG_WHO))
}

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