tama 0.0.1

Multi-agent AI framework — build, run, and trace agent pipelines from the command line
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
87
88
89

use anyhow::Result;
use std::path::Path;
use std::sync::Arc;

use super::AgentOutput;
use super::oneshot;
use super::step::Step;
use crate::runtime::llm::LlmClient;
use crate::runtime::model_registry::ModelRegistry;
use crate::runtime::tracer::{TraceCtx, Tracer};

/// Chain-of-Verification: generate → extract claims → verify each → revise.
///
/// Files:
///   AGENT.md body  → initial answer generation system prompt
///   verify.md      → claim extraction system prompt (output: one claim per line)
///   check.md       → individual claim checking system prompt
///   revise.md      → final revision system prompt
///
/// Each step file supports optional `pattern: react` frontmatter.
pub async fn run(
    agent_dir: &Path,
    body: &str,
    registry: &Arc<ModelRegistry>,
    client: &LlmClient,
    input: &str,
    tracer: &mut dyn Tracer,
    ctx: &TraceCtx,
    crumb: &str,
) -> Result<AgentOutput> {
    eprintln!("  → cov: generate");
    let initial = oneshot::call(body, input, "generate", client, tracer, ctx, crumb).await?;

    let verify_step = Step::from_file(&agent_dir.join("verify.md"))?;
    eprintln!("  → cov: extract claims");
    let claims_text = verify_step
        .run(
            &format!("Answer to verify:\n{initial}"),
            "extract_claims",
            registry,
            client,
            tracer,
            ctx,
            crumb,
        )
        .await?
        .value;

    let claims: Vec<&str> = claims_text
        .lines()
        .map(str::trim)
        .filter(|l| !l.is_empty() && !l.starts_with('#'))
        .collect();

    eprintln!("  → cov: {} claims to check", claims.len());

    let check_step = Step::from_file(&agent_dir.join("check.md"))?;
    let mut verifications = Vec::new();
    for (i, claim) in claims.iter().enumerate() {
        eprintln!("  → cov: checking claim {}/{}", i + 1, claims.len());
        let check_result = check_step
            .run(
                &format!("Claim: {claim}\n\nContext: {input}"),
                &format!("check_{}", i + 1),
                registry,
                client,
                tracer,
                ctx,
                crumb,
            )
            .await?
            .value;
        verifications.push(format!("Claim: {claim}\nVerification: {check_result}"));
    }

    let revise_step = Step::from_file(&agent_dir.join("revise.md"))?;
    eprintln!("  → cov: revise");
    let revise_input = format!(
        "Original question: {input}\n\nInitial answer:\n{initial}\n\nVerification results:\n{}",
        verifications.join("\n\n")
    );
    let revised = revise_step.run(&revise_input, "revise", registry, client, tracer, ctx, crumb).await?.value;

    Ok(AgentOutput {
        key: "done".to_string(),
        value: revised,
        span_id: String::new(),
    })
}