Update: 将子项目从 submodule 转为完整内容
- 移除 GovAI, nomifun-tauri, 算力盒子 的 submodule 引用 - 添加所有子项目的完整源代码 - 保留原始 .git 为 .git.bak 备份
This commit is contained in:
freedak
@@ -0,0 +1,279 @@
|
||||
// Shared test utilities for integration tests.
|
||||
#![allow(dead_code)]
|
||||
|
||||
use std::sync::{Arc, Mutex};
|
||||
|
||||
use async_trait::async_trait;
|
||||
use serde_json::{Value, json};
|
||||
use tokio::sync::mpsc;
|
||||
|
||||
use nomi_agent::confirm::ToolConfirmer;
|
||||
use nomi_config::config::{Config, ProviderType, SessionConfig, ToolsConfig};
|
||||
use nomi_config::hooks::HooksConfig;
|
||||
use nomi_mcp::config::McpConfig;
|
||||
use nomi_protocol::events::ToolCategory;
|
||||
use nomi_providers::{LlmProvider, ProviderError};
|
||||
use nomi_tools::Tool;
|
||||
use nomi_types::llm::{LlmEvent, LlmRequest};
|
||||
use nomi_types::message::{StopReason, TokenUsage};
|
||||
use nomi_types::tool::ToolResult;
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// MockLlmProvider — deterministic LLM for engine / spawn tests
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// A mock LLM provider that emits a pre-configured sequence of events.
|
||||
/// Each call to `stream` pops the first sequence from `responses`.
|
||||
/// When `responses` is empty it falls back to a single EndTurn with empty text.
|
||||
pub struct MockLlmProvider {
|
||||
responses: Mutex<Vec<Vec<LlmEvent>>>,
|
||||
}
|
||||
|
||||
impl MockLlmProvider {
|
||||
/// Create a provider that returns a single text response then ends.
|
||||
pub fn with_text_response(text: &str) -> Self {
|
||||
let events = vec![
|
||||
LlmEvent::TextDelta(text.to_string()),
|
||||
LlmEvent::Done {
|
||||
stop_reason: StopReason::EndTurn,
|
||||
usage: TokenUsage {
|
||||
input_tokens: 100,
|
||||
output_tokens: 50,
|
||||
cache_creation_tokens: 0,
|
||||
cache_read_tokens: 0,
|
||||
},
|
||||
},
|
||||
];
|
||||
Self {
|
||||
responses: Mutex::new(vec![events]),
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a provider that returns a single tool_use then ends with ToolUse stop reason.
|
||||
pub fn with_tool_use(id: &str, name: &str, input: Value) -> Self {
|
||||
let events = vec![
|
||||
LlmEvent::ToolUse {
|
||||
id: id.to_string(),
|
||||
name: name.to_string(),
|
||||
input,
|
||||
extra: None,
|
||||
},
|
||||
LlmEvent::Done {
|
||||
stop_reason: StopReason::ToolUse,
|
||||
usage: TokenUsage {
|
||||
input_tokens: 80,
|
||||
output_tokens: 30,
|
||||
cache_creation_tokens: 0,
|
||||
cache_read_tokens: 0,
|
||||
},
|
||||
},
|
||||
];
|
||||
Self {
|
||||
responses: Mutex::new(vec![events]),
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a provider with multiple turns of pre-configured event sequences.
|
||||
/// Each call to `stream` consumes the next sequence.
|
||||
pub fn with_turns(turns: Vec<Vec<LlmEvent>>) -> Self {
|
||||
Self {
|
||||
responses: Mutex::new(turns),
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a provider that returns custom events.
|
||||
pub fn with_events(events: Vec<LlmEvent>) -> Self {
|
||||
Self {
|
||||
responses: Mutex::new(vec![events]),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl LlmProvider for MockLlmProvider {
|
||||
async fn stream(
|
||||
&self,
|
||||
_request: &LlmRequest,
|
||||
) -> Result<mpsc::Receiver<LlmEvent>, ProviderError> {
|
||||
let events = {
|
||||
let mut responses = self.responses.lock().unwrap();
|
||||
if responses.is_empty() {
|
||||
// Fallback: end turn with empty text
|
||||
vec![LlmEvent::Done {
|
||||
stop_reason: StopReason::EndTurn,
|
||||
usage: TokenUsage::default(),
|
||||
}]
|
||||
} else {
|
||||
responses.remove(0)
|
||||
}
|
||||
};
|
||||
|
||||
let (tx, rx) = mpsc::channel(64);
|
||||
tokio::spawn(async move {
|
||||
for event in events {
|
||||
let _ = tx.send(event).await;
|
||||
}
|
||||
});
|
||||
Ok(rx)
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// MockTool — deterministic tool for orchestration tests
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// A simple mock tool that returns a pre-configured result.
|
||||
pub struct MockTool {
|
||||
pub tool_name: String,
|
||||
pub tool_description: String,
|
||||
pub concurrent_safe: bool,
|
||||
pub result: Mutex<ToolResult>,
|
||||
}
|
||||
|
||||
impl MockTool {
|
||||
pub fn new(name: &str, result: &str, is_error: bool) -> Self {
|
||||
Self {
|
||||
tool_name: name.to_string(),
|
||||
tool_description: format!("Mock tool: {}", name),
|
||||
concurrent_safe: true,
|
||||
result: Mutex::new(ToolResult {
|
||||
content: result.to_string(),
|
||||
is_error,
|
||||
images: Vec::new(),
|
||||
}),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn sequential(name: &str, result: &str) -> Self {
|
||||
Self {
|
||||
tool_name: name.to_string(),
|
||||
tool_description: format!("Mock sequential tool: {}", name),
|
||||
concurrent_safe: false,
|
||||
result: Mutex::new(ToolResult {
|
||||
content: result.to_string(),
|
||||
is_error: false,
|
||||
images: Vec::new(),
|
||||
}),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl Tool for MockTool {
|
||||
fn name(&self) -> &str {
|
||||
&self.tool_name
|
||||
}
|
||||
|
||||
fn description(&self) -> &str {
|
||||
&self.tool_description
|
||||
}
|
||||
|
||||
fn input_schema(&self) -> Value {
|
||||
json!({"type": "object"})
|
||||
}
|
||||
|
||||
fn category(&self) -> ToolCategory {
|
||||
ToolCategory::Info
|
||||
}
|
||||
|
||||
fn is_concurrency_safe(&self, _input: &Value) -> bool {
|
||||
self.concurrent_safe
|
||||
}
|
||||
|
||||
async fn execute(&self, _input: Value) -> ToolResult {
|
||||
self.result.lock().unwrap().clone()
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// ExecMockTool — mock tool with Exec category (requires approval)
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
/// A mock tool that returns a pre-configured result, with Exec category.
|
||||
pub struct ExecMockTool {
|
||||
pub tool_name: String,
|
||||
pub result: Mutex<ToolResult>,
|
||||
}
|
||||
|
||||
impl ExecMockTool {
|
||||
pub fn new(name: &str, result: &str) -> Self {
|
||||
Self {
|
||||
tool_name: name.to_string(),
|
||||
result: Mutex::new(ToolResult {
|
||||
content: result.to_string(),
|
||||
is_error: false,
|
||||
images: Vec::new(),
|
||||
}),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl Tool for ExecMockTool {
|
||||
fn name(&self) -> &str {
|
||||
&self.tool_name
|
||||
}
|
||||
|
||||
fn description(&self) -> &str {
|
||||
"Mock exec tool"
|
||||
}
|
||||
|
||||
fn input_schema(&self) -> Value {
|
||||
json!({"type": "object"})
|
||||
}
|
||||
|
||||
fn category(&self) -> ToolCategory {
|
||||
ToolCategory::Exec
|
||||
}
|
||||
|
||||
fn is_concurrency_safe(&self, _input: &Value) -> bool {
|
||||
false
|
||||
}
|
||||
|
||||
async fn execute(&self, _input: Value) -> ToolResult {
|
||||
self.result.lock().unwrap().clone()
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Helper: build a minimal Config for testing
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
pub fn test_config() -> Config {
|
||||
Config {
|
||||
provider_label: "anthropic".to_string(),
|
||||
provider: ProviderType::Anthropic,
|
||||
api_key: "test-key".to_string(),
|
||||
base_url: "http://localhost:0".to_string(),
|
||||
model: "test-model".to_string(),
|
||||
max_tokens: 4096,
|
||||
max_turns: Some(10),
|
||||
system_prompt: Some("You are a test assistant.".to_string()),
|
||||
thinking: None,
|
||||
prompt_caching: false,
|
||||
compat: nomi_config::compat::ProviderCompat::anthropic_defaults(),
|
||||
tools: ToolsConfig {
|
||||
auto_approve: true,
|
||||
allow_list: vec![],
|
||||
..ToolsConfig::default()
|
||||
},
|
||||
session: SessionConfig {
|
||||
enabled: false,
|
||||
directory: "/tmp/nomi-test-sessions".to_string(),
|
||||
max_sessions: 5,
|
||||
},
|
||||
compact: nomi_config::compact::CompactConfig::default(),
|
||||
plan: nomi_config::plan::PlanConfig::default(),
|
||||
file_cache: nomi_config::file_cache::FileCacheConfig::default(),
|
||||
hooks: HooksConfig::default(),
|
||||
bedrock: None,
|
||||
vertex: None,
|
||||
mcp: McpConfig::default(),
|
||||
logging: nomi_config::logging::LoggingConfig::default(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a ToolConfirmer that auto-approves everything.
|
||||
pub fn auto_approve_confirmer() -> Arc<Mutex<ToolConfirmer>> {
|
||||
Arc::new(Mutex::new(ToolConfirmer::new(true, vec![])))
|
||||
}
|
||||
Reference in New Issue
Block a user