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

//! `run_agent_turn` 前置步骤：读缓存句柄、工作区变更集与合并 MCP/动态工具后的工具表。

use std::collections::HashSet;
use std::path::Path;
use std::sync::Arc;

use crate::cm_internal::process_handles::TurnProcessHandles;
use crate::cm_internal::read_file_turn_cache::ReadFileTurnCache;
use crate::cm_internal::workspace::changelist::WorkspaceChangelist;
use crate::cm_config::AgentConfig;
use crate::cm_types::Tool;

pub struct ToolsForTurnPrepared {
    pub tools_for_turn: Vec<Tool>,
    pub mcp_turn: Option<crate::cm_internal::mcp::McpTurnHandle>,
    /// 本轮尝试连接但失败的 MCP 服务器（供终端/SSE 提示）。
    pub mcp_skipped: Vec<crate::cm_internal::mcp::McpServerSkipInfo>,
}

pub fn resolve_read_file_turn_cache_for_turn(
    cfg: &AgentConfig,
    read_file_turn_cache: Option<Arc<ReadFileTurnCache>>,
) -> Option<Arc<ReadFileTurnCache>> {
    match read_file_turn_cache {
        Some(a) => Some(a),
        None if cfg.chat_queues_cache.read_file_turn_cache_max_entries > 0 => {
            Some(crate::cm_internal::read_file_turn_cache::new_turn_cache_handle(
                cfg.chat_queues_cache.read_file_turn_cache_max_entries,
            ))
        }
        None => None,
    }
}

pub fn workspace_changelist_for_turn(
    cfg: &AgentConfig,
    process_handles: &TurnProcessHandles,
    long_term_memory_scope_id: Option<&str>,
) -> Option<Arc<WorkspaceChangelist>> {
    if !cfg
        .session_workspace_changelist
        .session_workspace_changelist_enabled
    {
        return None;
    }
    let scope = long_term_memory_scope_id
        .map(str::trim)
        .filter(|s| !s.is_empty())
        .unwrap_or("__default__");
    Some(
        process_handles
            .workspace_changelist_registry
            .changelist_for_scope(scope),
    )
}

pub async fn prepare_tools_for_turn(
    cfg: &Arc<AgentConfig>,
    tools: &[Tool],
    effective_working_dir: &Path,
    turn_allowed_tool_names: Option<&HashSet<String>>,
) -> ToolsForTurnPrepared {
    let mut tools_for_turn: Vec<Tool> = tools.to_vec();
    tools_for_turn = crate::cm_internal::mcp::merge_tool_lists(
        tools_for_turn,
        crate::cm_internal::dynamic_tools::load_dynamic_tools(effective_working_dir),
    );
    let open = crate::cm_internal::mcp::try_open_session_and_tools(cfg.as_ref()).await;
    let mcp_skipped = open.skipped;
    let mcp_turn = match (open.handle, open.tools) {
        (Some(handle), extra) => {
            tools_for_turn = crate::cm_internal::mcp::merge_tool_lists(tools_for_turn, extra);
            Some(handle)
        }
        (None, _) => None,
    };
    if !cfg.codebase_semantic.codebase_semantic_search_enabled {
        tools_for_turn.retain(|t| t.function.name != "codebase_semantic_search");
    }
    if !cfg.long_term_memory.long_term_memory_enabled {
        tools_for_turn.retain(|t| {
            !matches!(
                t.function.name.as_str(),
                "long_term_remember" | "long_term_forget" | "long_term_memory_list"
            )
        });
    }
    if let Some(allow) = turn_allowed_tool_names {
        tools_for_turn.retain(|t| {
            crate::cm_tools::tool_naming::tool_name_allowed_by_turn_allowlist(
                t.function.name.as_str(),
                Some(allow),
            )
        });
    }
    ToolsForTurnPrepared {
        tools_for_turn,
        mcp_turn,
        mcp_skipped,
    }
}