refactor: skill.rs → types.rs 重命名

类型定义重命名,消除过时的 'skill' 概念
This commit is contained in:
2026-06-03 10:31:41 +08:00
parent 4d149982c5
commit b6ea395171
5 changed files with 50 additions and 20 deletions
+5 -5
View File
@@ -15,7 +15,7 @@ use db::Database;
use llm::{Conversation, ConversationConfig, DEFAULT_SYSTEM_PROMPT, ToolExecutor, Usage};
use std::sync::Arc;
use tools::approval::{ApprovalDecision, ApprovalManager};
use tools::skill::ExecutionContext;
use tools::types::ExecutionContext;
use tracing::{error, info};
use wechat::client::WeChatClient;
@@ -293,7 +293,7 @@ async fn cmd_listen(
let session = conv.session();
let approval = approval_manager.clone();
let memory_store = memory_store.clone();
let send_wechat: tools::skill::WechatSender = {
let send_wechat: tools::types::WechatSender = {
let client = client.clone();
Arc::new(move |user_id: &str, text: &str| {
let client = client.clone();
@@ -816,7 +816,7 @@ async fn cmd_send(
}
use tools::skill::SkillResult;
use tools::types::SkillResult;
async fn builtin_approve(ctx: &ExecutionContext, name: &str) -> Result<bool, String> {
let manager = match ctx.approval_manager.as_ref() {
@@ -841,10 +841,10 @@ async fn builtin_approve(ctx: &ExecutionContext, name: &str) -> Result<bool, Str
}
}
fn build_capability_list(specs: &[tools::skill::SkillSpec]) -> String {
fn build_capability_list(specs: &[tools::types::SkillSpec]) -> String {
let mut lines = vec!["📋 可用工具:".to_string()];
for spec in specs {
let risk = if spec.risk_level == tools::skill::RiskLevel::High {
let risk = if spec.risk_level == tools::types::RiskLevel::High {
" ⚠️需确认"
} else {
""
+4 -4
View File
@@ -1,4 +1,4 @@
use crate::tools::skill::{RiskLevel, SkillResult};
use crate::tools::types::{RiskLevel, SkillResult};
use chrono::Local;
/// 内置工具执行器:统一处理参数校验、执行、错误返回
@@ -18,16 +18,16 @@ impl BuiltinRegistry {
}
/// 返回所有内置工具的 spec 列表
pub fn specs() -> Vec<super::skill::SkillSpec> {
pub fn specs() -> Vec<super::types::SkillSpec> {
vec![
super::skill::SkillSpec {
super::types::SkillSpec {
name: "get_current_datetime".into(),
description: "获取当前日期时间(北京时间 UTC+8)".into(),
risk_level: RiskLevel::Low,
parameters: serde_json::json!({"type":"object","properties":{"format":{"type":"string"}}}),
timeout_secs: 5,
},
super::skill::SkillSpec {
super::types::SkillSpec {
name: "manage_memos".into(),
description: "管理备忘录:添加/列出/删除。高风险操作(add/delete)需确认".into(),
risk_level: RiskLevel::Low,
+1 -1
View File
@@ -1,4 +1,4 @@
pub mod approval;
pub mod builtin;
pub mod skill;
pub mod types;
pub mod weather;
+39 -9
View File
@@ -32,8 +32,12 @@ pub struct SkillSpec {
pub timeout_secs: u64,
}
fn default_risk_level() -> RiskLevel { RiskLevel::Low }
fn default_timeout() -> u64 { 30 }
fn default_risk_level() -> RiskLevel {
RiskLevel::Low
}
fn default_timeout() -> u64 {
30
}
// ─── 工具执行结果 ───
@@ -47,29 +51,51 @@ pub struct SkillResult {
impl SkillResult {
pub fn ok(output: impl Into<String>) -> Self {
Self { success: true, output: output.into(), data: None }
Self {
success: true,
output: output.into(),
data: None,
}
}
pub fn ok_with_data(output: impl Into<String>, data: serde_json::Value) -> Self {
Self { success: true, output: output.into(), data: Some(data) }
Self {
success: true,
output: output.into(),
data: Some(data),
}
}
pub fn rejected(skill_name: &str) -> Self {
Self { success: false, output: format!("用户拒绝了你的调用:{}", skill_name), data: None }
Self {
success: false,
output: format!("用户拒绝了你的调用:{}", skill_name),
data: None,
}
}
pub fn expired(skill_name: &str) -> Self {
Self { success: false, output: format!("用户没有确认操作:{}", skill_name), data: None }
Self {
success: false,
output: format!("用户没有确认操作:{}", skill_name),
data: None,
}
}
pub fn error(msg: impl Into<String>) -> Self {
Self { success: false, output: msg.into(), data: None }
Self {
success: false,
output: msg.into(),
data: None,
}
}
}
// ─── 执行上下文 ───
pub type WechatSender = Arc<dyn Fn(&str, &str) -> Pin<Box<dyn Future<Output = Result<(), String>> + Send>> + Send + Sync>;
pub type WechatSender = Arc<
dyn Fn(&str, &str) -> Pin<Box<dyn Future<Output = Result<(), String>> + Send>> + Send + Sync,
>;
pub struct ExecutionContext {
pub user_id: String,
@@ -79,7 +105,11 @@ pub struct ExecutionContext {
impl ExecutionContext {
pub fn new(user_id: impl Into<String>) -> Self {
Self { user_id: user_id.into(), approval_manager: None, send_wechat: None }
Self {
user_id: user_id.into(),
approval_manager: None,
send_wechat: None,
}
}
pub fn with_approval(mut self, mgr: Arc<crate::tools::approval::ApprovalManager>) -> Self {
+1 -1
View File
@@ -19,7 +19,7 @@ use std::collections::HashMap;
use std::sync::Mutex;
use std::time::{Duration, Instant};
use crate::tools::skill::{RiskLevel, SkillResult, SkillSpec};
use crate::tools::types::{RiskLevel, SkillResult, SkillSpec};
// ═══════════════════════════════════════════════
// JWT 认证