crabmate 0.4.0

Rust AI agent: OpenAI-compatible chat/completions, function calling, HTTP serve, ops CLI
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

//! Web / CLI 首轮注入合并（依赖 `memory` / `living_docs`，保留在 internal）。

use std::path::Path;

use crate::cm_config::AgentConfig;
use crate::cm_tools::project_profile::build_project_profile_markdown;

use super::project_dependency_brief;
use crate::cm_internal::memory::agent_memory;

/// Web / CLI 首轮：合并备忘、项目画像、依赖摘要；全无则 `None`。
pub fn build_first_turn_user_context_markdown(
    workspace_root: &Path,
    cfg: &AgentConfig,
    memory_preloaded: Option<String>,
) -> Option<String> {
    if workspace_root.as_os_str().is_empty() {
        return None;
    }
    let memory_snippet = memory_preloaded.or_else(|| {
        if cfg.context_bootstrap_inject.agent_memory_file_enabled {
            agent_memory::load_memory_snippet(
                workspace_root,
                cfg.context_bootstrap_inject.agent_memory_file.as_str(),
                cfg.context_bootstrap_inject.agent_memory_file_max_chars,
            )
        } else {
            None
        }
    });
    let living_snippet = if cfg.context_bootstrap_inject.living_docs_inject_enabled
        && cfg.context_bootstrap_inject.living_docs_inject_max_chars > 0
    {
        super::living_docs::load_living_docs_snippet(
            workspace_root,
            cfg.context_bootstrap_inject
                .living_docs_relative_dir
                .as_str(),
            cfg.context_bootstrap_inject.living_docs_inject_max_chars,
            cfg.context_bootstrap_inject.living_docs_file_max_each_chars,
        )
    } else {
        None
    };
    let want_profile = cfg.context_bootstrap_inject.project_profile_inject_enabled
        && cfg
            .context_bootstrap_inject
            .project_profile_inject_max_chars
            > 0;
    let want_dep = cfg
        .context_bootstrap_inject
        .project_dependency_brief_inject_enabled
        && cfg
            .context_bootstrap_inject
            .project_dependency_brief_inject_max_chars
            > 0;
    if !want_profile && !want_dep && memory_snippet.is_none() && living_snippet.is_none() {
        return None;
    }
    let profile_md = if want_profile {
        build_project_profile_markdown(
            workspace_root,
            cfg.context_bootstrap_inject
                .project_profile_inject_max_chars,
        )
    } else {
        String::new()
    };
    let dep_md = if want_dep {
        project_dependency_brief::build_project_dependency_brief_markdown(
            workspace_root,
            cfg.context_bootstrap_inject
                .project_dependency_brief_inject_max_chars,
        )
    } else {
        String::new()
    };
    merge_first_turn_injections(
        living_snippet.as_deref(),
        memory_snippet.as_deref(),
        profile_md.as_str(),
        dep_md.as_str(),
    )
}

/// 合并活文档摘要、备忘、项目画像、依赖结构摘要为一条首轮 `user` 正文。
pub fn merge_first_turn_injections(
    living_docs_snippet: Option<&str>,
    memory_snippet: Option<&str>,
    profile_markdown: &str,
    dependency_brief_markdown: &str,
) -> Option<String> {
    let living = living_docs_snippet.map(str::trim).filter(|s| !s.is_empty());
    let mem = memory_snippet.map(str::trim).filter(|s| !s.is_empty());
    let prof = profile_markdown.trim();
    let dep = dependency_brief_markdown.trim();
    let mut parts: Vec<String> = Vec::new();
    if let Some(m) = living {
        parts.push(m.to_string());
    }
    if let Some(m) = mem {
        parts.push(m.to_string());
    }
    if !prof.is_empty() {
        parts.push(format!(
            "[项目画像（工作区内自动生成，仅只读扫描）]\n{prof}"
        ));
    }
    if !dep.is_empty() {
        parts.push(format!(
            "[项目依赖与结构摘要（cargo metadata + package.json，仅只读）]\n{dep}"
        ));
    }
    if parts.is_empty() {
        None
    } else {
        Some(parts.join("\n\n---\n\n"))
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn merge_first_turn_three_parts() {
        let got =
            merge_first_turn_injections(None, Some("memo line"), "# Title\nbody", "## Dep\nx")
                .expect("some");
        assert!(got.contains("memo line"));
        assert!(got.contains("项目画像"));
        assert!(got.contains("依赖与结构摘要"));
    }
}