686 lines
24 KiB
Python
686 lines
24 KiB
Python
|
"""
|
|||
|
|
目标管理工具 (基于滴答清单项目和任务)
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
from typing import List, Dict, Optional, Any, Union, Tuple
|
|||
|
|
from fastmcp import FastMCP
|
|||
|
|
|
|||
|
|
# 导入其他工具的逻辑函数
|
|||
|
|
from .project_tools import (
|
|||
|
|
get_projects_logic,
|
|||
|
|
create_project_logic,
|
|||
|
|
update_project_logic,
|
|||
|
|
delete_project_logic
|
|||
|
|
)
|
|||
|
|
from .task_tools import (
|
|||
|
|
get_tasks_logic,
|
|||
|
|
create_task_logic,
|
|||
|
|
update_task_logic,
|
|||
|
|
delete_task_logic
|
|||
|
|
)
|
|||
|
|
# 目标更新走 update_task_logic，无需直接HTTP调用
|
|||
|
|
|
|||
|
|
# 导入辅助函数
|
|||
|
|
from utils.date.date_utils import is_valid_date, format_datetime, get_current_time
|
|||
|
|
from utils.text.text_analysis import normalize_keywords, match_keywords, calculate_similarity
|
|||
|
|
|
|||
|
|
# --- 常量 ---
|
|||
|
|
GOAL_PROJECT_NAME = "🎯 目标管理" # 存放所有目标的项目名称
|
|||
|
|
GOAL_TASK_PREFIX = "" # 目标任务的前缀
|
|||
|
|
METADATA_PATTERN = re.compile(r"\[(.*?): (.*?)\]")
|
|||
|
|
GOAL_TYPES = ['phase', 'permanent', 'habit'] # 目标类型保持，用于描述元数据
|
|||
|
|
GOAL_STATUSES = ['active', 'completed', 'abandoned'] # 目标状态
|
|||
|
|
|
|||
|
|
# --- 辅助函数 ---
|
|||
|
|
|
|||
|
|
def _format_metadata(data: Dict[str, Any]) -> str:
|
|||
|
|
"""将元数据字典格式化为任务描述字符串"""
|
|||
|
|
parts = []
|
|||
|
|
for key, value in data.items():
|
|||
|
|
if value: # 只包含有值的字段
|
|||
|
|
parts.append(f"[{key.capitalize()}: {value}]")
|
|||
|
|
return " ".join(parts)
|
|||
|
|
|
|||
|
|
def _parse_metadata(description: Optional[str]) -> Dict[str, Any]:
|
|||
|
|
"""从任务描述字符串中解析元数据"""
|
|||
|
|
metadata = {}
|
|||
|
|
if description:
|
|||
|
|
matches = METADATA_PATTERN.findall(description)
|
|||
|
|
for key, value in matches:
|
|||
|
|
metadata[key.lower()] = value.strip()
|
|||
|
|
return metadata
|
|||
|
|
|
|||
|
|
def _get_goal_project() -> Optional[str]:
|
|||
|
|
"""
|
|||
|
|
获取目标管理项目的ID
|
|||
|
|
先精确匹配项目名称，如果找不到，再尝试模糊匹配
|
|||
|
|
如果不存在则返回 None
|
|||
|
|
"""
|
|||
|
|
projects = get_projects_logic()
|
|||
|
|
|
|||
|
|
# 1. 精确匹配项目名称
|
|||
|
|
for project in projects:
|
|||
|
|
if project.get('name') == GOAL_PROJECT_NAME:
|
|||
|
|
return project
|
|||
|
|
|
|||
|
|
# 2. 模糊匹配 - 查找名称中包含"目标"和"🎯"的项目
|
|||
|
|
for project in projects:
|
|||
|
|
project_name = project.get('name', '')
|
|||
|
|
if '目标' in project_name and '🎯' in project_name:
|
|||
|
|
print(f"找到类似的目标管理项目: {project_name}")
|
|||
|
|
return project
|
|||
|
|
|
|||
|
|
# 3. 更宽松的匹配 - 只要包含"目标"
|
|||
|
|
for project in projects:
|
|||
|
|
project_name = project.get('name', '')
|
|||
|
|
if '目标' in project_name:
|
|||
|
|
print(f"找到相关的目标项目: {project_name}")
|
|||
|
|
return project
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def _ensure_goal_project_exists() -> str:
|
|||
|
|
"""
|
|||
|
|
确保存在目标管理项目
|
|||
|
|
如果不存在则创建，返回项目ID
|
|||
|
|
"""
|
|||
|
|
project = _get_goal_project()
|
|||
|
|
if project:
|
|||
|
|
return project
|
|||
|
|
|
|||
|
|
print(f"未找到目标管理项目，创建新项目: {GOAL_PROJECT_NAME}")
|
|||
|
|
# 创建目标管理项目
|
|||
|
|
project_data = create_project_logic(name=GOAL_PROJECT_NAME)
|
|||
|
|
project = project_data.get('id')
|
|||
|
|
if not project:
|
|||
|
|
raise ValueError(f"创建目标管理项目失败，未返回项目ID。API响应: {project_data}")
|
|||
|
|
|
|||
|
|
return project
|
|||
|
|
|
|||
|
|
def _enrich_goal_data(task: Dict[str, Any]) -> Dict[str, Any]:
|
|||
|
|
"""将任务数据丰富为目标数据"""
|
|||
|
|
task_id = task.get('id')
|
|||
|
|
metadata = _parse_metadata(task.get('content'))
|
|||
|
|
|
|||
|
|
goal_data = {
|
|||
|
|
"id": task_id,
|
|||
|
|
"title": task.get('title', ''), # 直接使用任务标题，不需要移除前缀
|
|||
|
|
"description": task.get('content', ''), # 保留原始描述
|
|||
|
|
"type": metadata.get('type', 'permanent'), # 默认类型
|
|||
|
|
"status": 'completed' if task.get('status') == 2 or task.get('isCompleted') else 'active',
|
|||
|
|
"keywords": metadata.get('keywords', ''),
|
|||
|
|
"start_date": metadata.get('start_date'),
|
|||
|
|
"due_date": task.get('dueDate'), # 使用任务本身的截止日期
|
|||
|
|
"frequency": metadata.get('frequency'),
|
|||
|
|
"created_time": task.get('createdTime'), # 使用任务创建时间
|
|||
|
|
"modified_time": task.get('modifiedTime'),
|
|||
|
|
"priority": task.get('priority', 0), # 任务优先级
|
|||
|
|
"project_id": task.get('projectId'), # 所属项目ID
|
|||
|
|
"raw_task_data": task # 保留原始任务数据
|
|||
|
|
}
|
|||
|
|
return {k: v for k, v in goal_data.items() if v is not None}
|
|||
|
|
|
|||
|
|
# --- 模块级核心逻辑函数 ---
|
|||
|
|
|
|||
|
|
def create_goal_logic(
|
|||
|
|
title: str,
|
|||
|
|
type: str,
|
|||
|
|
keywords: str,
|
|||
|
|
description: Optional[str] = None,
|
|||
|
|
due_date: Optional[str] = None,
|
|||
|
|
start_date: Optional[str] = None,
|
|||
|
|
frequency: Optional[str] = None,
|
|||
|
|
related_projects: Optional[List[str]] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
创建新目标 (作为任务存放在目标管理项目中)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
title: 目标标题
|
|||
|
|
type: 目标类型(phase/permanent/habit)
|
|||
|
|
keywords: 关键词，以逗号分隔
|
|||
|
|
description: 目标描述 (会附加元数据)
|
|||
|
|
due_date: 截止日期 (YYYY-MM-DD)
|
|||
|
|
start_date: 开始日期 (YYYY-MM-DD)
|
|||
|
|
frequency: 频率 (用于习惯目标)
|
|||
|
|
related_projects: 相关项目IDs (保留参数，当前版本不使用)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
创建的目标信息 (丰富后的数据)
|
|||
|
|
"""
|
|||
|
|
# 验证类型
|
|||
|
|
if type not in GOAL_TYPES:
|
|||
|
|
raise ValueError(f"无效的目标类型: {type}，应为 {GOAL_TYPES} 之一")
|
|||
|
|
|
|||
|
|
# 验证日期格式
|
|||
|
|
if due_date and not is_valid_date(due_date):
|
|||
|
|
raise ValueError(f"无效的截止日期格式: {due_date}，应为YYYY-MM-DD")
|
|||
|
|
if start_date and not is_valid_date(start_date):
|
|||
|
|
raise ValueError(f"无效的开始日期格式: {start_date}，应为YYYY-MM-DD")
|
|||
|
|
|
|||
|
|
# 验证特定类型的字段
|
|||
|
|
if type == 'phase' and not due_date:
|
|||
|
|
raise ValueError("阶段性目标必须指定截止日期(due_date)")
|
|||
|
|
if type == 'habit' and not frequency:
|
|||
|
|
raise ValueError("习惯性目标必须指定频率(frequency)")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 1. 确保目标管理项目存在
|
|||
|
|
project_id = _ensure_goal_project_exists()
|
|||
|
|
|
|||
|
|
# 2. 准备任务标题 - 直接使用传入的标题，不添加前缀
|
|||
|
|
task_title = title
|
|||
|
|
|
|||
|
|
# 3. 准备元数据
|
|||
|
|
metadata = {
|
|||
|
|
'type': type,
|
|||
|
|
'keywords': normalize_keywords(keywords),
|
|||
|
|
'start_date': start_date,
|
|||
|
|
'frequency': frequency
|
|||
|
|
# due_date 将直接用于任务的dueDate字段
|
|||
|
|
}
|
|||
|
|
metadata_str = _format_metadata(metadata)
|
|||
|
|
|
|||
|
|
# 4. 组合任务内容
|
|||
|
|
full_content = f"{description}\n\n--- Metadata ---\n{metadata_str}" if description else f"--- Metadata ---\n{metadata_str}"
|
|||
|
|
|
|||
|
|
# 5. 创建任务 - 使用正确的参数名称
|
|||
|
|
# 注意：task_tools.create_task_logic 期望 project_name 而不是 projectId
|
|||
|
|
created_task = create_task_logic(
|
|||
|
|
title=task_title,
|
|||
|
|
content=full_content,
|
|||
|
|
project_name=GOAL_PROJECT_NAME, # 使用项目名称而不是ID
|
|||
|
|
due_date=due_date,
|
|||
|
|
start_date=start_date,
|
|||
|
|
priority=3 # 默认设置为中等优先级
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 6. 返回丰富的目标数据
|
|||
|
|
return _enrich_goal_data(created_task)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise ValueError(f"创建目标失败: {e}")
|
|||
|
|
|
|||
|
|
def get_goals_logic(
|
|||
|
|
type: Optional[str] = None,
|
|||
|
|
status: Optional[str] = None,
|
|||
|
|
keywords: Optional[str] = None
|
|||
|
|
) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
获取目标列表 (基于任务)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
type: 目标类型筛选
|
|||
|
|
status: 目标状态筛选 (active/completed)
|
|||
|
|
keywords: 关键词筛选 (匹配目标标题或元数据中的关键词)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
目标列表
|
|||
|
|
"""
|
|||
|
|
project = _get_goal_project()
|
|||
|
|
if not project:
|
|||
|
|
print("未找到目标管理项目，返回空列表")
|
|||
|
|
return [] # 如果目标管理项目不存在，直接返回空列表
|
|||
|
|
# 2. 确定完成状态参数
|
|||
|
|
completed = None
|
|||
|
|
if status == 'completed':
|
|||
|
|
completed = True
|
|||
|
|
elif status == 'active':
|
|||
|
|
completed = False
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 3. 获取项目下的所有任务
|
|||
|
|
# 使用项目名称，而不是ID，以匹配task_tools的API设计
|
|||
|
|
all_tasks = get_tasks_logic(
|
|||
|
|
mode='all',
|
|||
|
|
completed=completed,
|
|||
|
|
project_name=project.get('name')
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 如果返回的是None或空值，返回空列表
|
|||
|
|
if not all_tasks:
|
|||
|
|
print("项目下未找到任务，返回空列表")
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
# 4. 过滤得到目标任务
|
|||
|
|
goal_list = []
|
|||
|
|
|
|||
|
|
# 处理关键词
|
|||
|
|
search_keywords = keywords or ""
|
|||
|
|
if not isinstance(search_keywords, str):
|
|||
|
|
search_keywords = ""
|
|||
|
|
|
|||
|
|
search_keywords_set = set(normalize_keywords(search_keywords).split(',')) if search_keywords else set()
|
|||
|
|
|
|||
|
|
for task in all_tasks:
|
|||
|
|
# 由于GOAL_TASK_PREFIX为空，不用startswith判断，而是看任务是否属于目标项目
|
|||
|
|
if task and isinstance(task, dict):
|
|||
|
|
try:
|
|||
|
|
goal_data = _enrich_goal_data(task)
|
|||
|
|
|
|||
|
|
# 类型筛选
|
|||
|
|
if type and goal_data.get('type') != type:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 关键词筛选
|
|||
|
|
if search_keywords_set:
|
|||
|
|
goal_title_lower = goal_data.get('title', '').lower()
|
|||
|
|
goal_meta_keywords = set(k for k in goal_data.get('keywords', '').split(',') if k)
|
|||
|
|
# 检查标题或元数据关键词是否包含任何搜索关键词
|
|||
|
|
if not any(sk in goal_title_lower for sk in search_keywords_set) and \
|
|||
|
|
not search_keywords_set.intersection(goal_meta_keywords):
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
goal_list.append(goal_data)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"处理任务时出错，跳过: {e}")
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
return goal_list
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"获取目标列表时出错: {e}")
|
|||
|
|
return [] # 出错时返回空列表而不是抛出异常
|
|||
|
|
|
|||
|
|
def get_goal_logic(goal_id: str) -> Optional[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
获取目标详情 (基于任务ID)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
goal_id: 目标ID (即任务ID)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
目标详情，如果不是目标任务或未找到则返回None
|
|||
|
|
"""
|
|||
|
|
# 尝试直接获取任务详情
|
|||
|
|
try:
|
|||
|
|
# 获取目标项目
|
|||
|
|
goal_project = _get_goal_project()
|
|||
|
|
if not goal_project:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
project_id = goal_project.get('id')
|
|||
|
|
|
|||
|
|
# 获取所有任务
|
|||
|
|
tasks = get_tasks_logic(mode="all")
|
|||
|
|
task = None
|
|||
|
|
for t in tasks:
|
|||
|
|
if t.get('id') == goal_id:
|
|||
|
|
task = t
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
if not task:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 验证是否属于目标项目
|
|||
|
|
if task.get('projectId') != project_id:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
return _enrich_goal_data(task)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"获取目标详情时出错: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def update_goal_logic(
|
|||
|
|
goal_id: str,
|
|||
|
|
title: Optional[str] = None,
|
|||
|
|
type: Optional[str] = None,
|
|||
|
|
status: Optional[str] = None,
|
|||
|
|
keywords: Optional[str] = None,
|
|||
|
|
description: Optional[str] = None,
|
|||
|
|
due_date: Optional[str] = None,
|
|||
|
|
start_date: Optional[str] = None,
|
|||
|
|
frequency: Optional[str] = None,
|
|||
|
|
progress: Optional[int] = None,
|
|||
|
|
related_projects: Optional[List[str]] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
更新目标 (基于任务)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
goal_id: 目标ID (任务ID)
|
|||
|
|
title: 新标题 (不含前缀)
|
|||
|
|
type: 新类型
|
|||
|
|
status: 新状态 (active/completed)
|
|||
|
|
keywords: 新关键词 (逗号分隔)
|
|||
|
|
description: 新的基础描述 (元数据会自动附加)
|
|||
|
|
due_date: 新截止日期
|
|||
|
|
start_date: 新开始日期
|
|||
|
|
frequency: 新频率
|
|||
|
|
progress: 进度 (忽略)
|
|||
|
|
related_projects: 相关项目 (忽略)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
更新后的目标数据
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
# 1. 获取当前目标任务
|
|||
|
|
current_goal = get_goal_logic(goal_id)
|
|||
|
|
if not current_goal:
|
|||
|
|
raise ValueError(f"未找到目标任务: {goal_id}")
|
|||
|
|
|
|||
|
|
# 2. 处理任务状态
|
|||
|
|
task_status = None
|
|||
|
|
if status is not None:
|
|||
|
|
if status == 'completed':
|
|||
|
|
task_status = 2 # 已完成
|
|||
|
|
elif status == 'active':
|
|||
|
|
task_status = 0 # 未开始/进行中
|
|||
|
|
|
|||
|
|
# 3. 处理元数据更新
|
|||
|
|
new_content = None
|
|||
|
|
if any(param is not None for param in [type, keywords, frequency, description]):
|
|||
|
|
# 获取当前任务数据和元数据
|
|||
|
|
raw_task_data = current_goal.get("raw_task_data", {})
|
|||
|
|
current_content = raw_task_data.get('content', '')
|
|||
|
|
current_metadata = _parse_metadata(current_content)
|
|||
|
|
|
|||
|
|
# 分割描述和元数据部分
|
|||
|
|
content_parts = current_content.split("\n\n--- Metadata ---\n")
|
|||
|
|
current_desc = content_parts[0] if len(content_parts) > 1 else ""
|
|||
|
|
|
|||
|
|
# 更新元数据
|
|||
|
|
if type is not None:
|
|||
|
|
if type not in GOAL_TYPES:
|
|||
|
|
raise ValueError(f"无效类型: {type}")
|
|||
|
|
current_metadata['type'] = type
|
|||
|
|
|
|||
|
|
if keywords is not None:
|
|||
|
|
current_metadata['keywords'] = normalize_keywords(keywords)
|
|||
|
|
|
|||
|
|
if start_date is not None and is_valid_date(start_date):
|
|||
|
|
current_metadata['start_date'] = start_date
|
|||
|
|
|
|||
|
|
if frequency is not None:
|
|||
|
|
current_metadata['frequency'] = frequency
|
|||
|
|
|
|||
|
|
# 更新描述
|
|||
|
|
new_desc = description if description is not None else current_desc
|
|||
|
|
|
|||
|
|
# 构建新内容
|
|||
|
|
metadata_str = _format_metadata(current_metadata)
|
|||
|
|
new_content = f"{new_desc}\n\n--- Metadata ---\n{metadata_str}" if new_desc else f"--- Metadata ---\n{metadata_str}"
|
|||
|
|
|
|||
|
|
# 4. 直接调用update_task_logic更新任务
|
|||
|
|
result = update_task_logic(
|
|||
|
|
task_id_or_title=goal_id,
|
|||
|
|
title=title,
|
|||
|
|
content=new_content,
|
|||
|
|
status=task_status,
|
|||
|
|
due_date=due_date,
|
|||
|
|
start_date=start_date
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 5. 检查结果并返回
|
|||
|
|
if not result.get('success'):
|
|||
|
|
raise ValueError(result.get('info', '更新失败，无详细错误信息'))
|
|||
|
|
|
|||
|
|
updated_task = result.get('data')
|
|||
|
|
return _enrich_goal_data(updated_task)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise ValueError(f"更新目标 {goal_id} 失败: {e}")
|
|||
|
|
|
|||
|
|
def delete_goal_logic(goal_id: str) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
删除目标 (基于任务)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
goal_id: 目标ID (任务ID)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
删除操作结果
|
|||
|
|
"""
|
|||
|
|
# 先确认是目标任务
|
|||
|
|
goal_data = get_goal_logic(goal_id)
|
|||
|
|
if not goal_data:
|
|||
|
|
raise ValueError(f"未找到目标任务: {goal_id}，无法删除")
|
|||
|
|
|
|||
|
|
# 调用任务删除逻辑 - task_id_or_title而不是task_id
|
|||
|
|
return delete_task_logic(task_id_or_title=goal_id)
|
|||
|
|
|
|||
|
|
def match_task_with_goals_logic(
|
|||
|
|
task_title: str,
|
|||
|
|
task_content: Optional[str] = None,
|
|||
|
|
project_id: Optional[str] = None,
|
|||
|
|
min_score: float = 0.3
|
|||
|
|
) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
匹配任务与目标 (基于内容相似度和关键词)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
task_title: 任务标题
|
|||
|
|
task_content: 任务内容
|
|||
|
|
project_id: 任务所属项目ID (不再用于直接匹配，因为所有目标都在目标管理项目下)
|
|||
|
|
min_score: 最小匹配分数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
匹配的目标列表，按匹配度降序排序
|
|||
|
|
"""
|
|||
|
|
active_goals = get_goals_logic(status='active')
|
|||
|
|
task_text = f"{task_title} {task_content or ''}".lower()
|
|||
|
|
|
|||
|
|
matches = []
|
|||
|
|
for goal in active_goals:
|
|||
|
|
match_score = 0.0
|
|||
|
|
|
|||
|
|
# 1. 关键词和文本相似度匹配
|
|||
|
|
goal_keywords_str = goal.get('keywords', '')
|
|||
|
|
goal_keywords_set = set(k for k in goal_keywords_str.split(',') if k)
|
|||
|
|
goal_title = goal.get('title', '')
|
|||
|
|
goal_desc = goal.get('description', '').split("\n\n--- Metadata ---\n")[0] # 只用基础描述
|
|||
|
|
goal_text_match = f"{goal_title} {goal_desc}".lower()
|
|||
|
|
|
|||
|
|
keyword_score = 0.0
|
|||
|
|
similarity_score = 0.0
|
|||
|
|
|
|||
|
|
if goal_keywords_set:
|
|||
|
|
# 简单的关键词包含匹配
|
|||
|
|
if any(kw.lower() in task_text for kw in goal_keywords_set):
|
|||
|
|
keyword_score = 0.7 # 基础分
|
|||
|
|
|
|||
|
|
if goal_text_match:
|
|||
|
|
similarity_score = calculate_similarity(task_text, goal_text_match) * 0.3 # 相似度占比较低
|
|||
|
|
|
|||
|
|
match_score = keyword_score + similarity_score
|
|||
|
|
|
|||
|
|
# 添加到结果如果分数达标
|
|||
|
|
if match_score >= min_score:
|
|||
|
|
matches.append({
|
|||
|
|
'goal': goal,
|
|||
|
|
'score': round(match_score, 3)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 按分数排序
|
|||
|
|
matches.sort(key=lambda x: x['score'], reverse=True)
|
|||
|
|
return [match['goal'] for match in matches]
|
|||
|
|
|
|||
|
|
# --- MCP工具注册 ---
|
|||
|
|
|
|||
|
|
def register_goal_tools(server: FastMCP, auth_info: Dict[str, Any]):
|
|||
|
|
"""
|
|||
|
|
注册目标管理工具到MCP服务器 (基于任务)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
server: MCP服务器实例
|
|||
|
|
auth_info: 认证信息 (用于初始化API，如果需要)
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
@server.tool()
|
|||
|
|
def create_goal(
|
|||
|
|
title: str,
|
|||
|
|
type: str,
|
|||
|
|
keywords: str,
|
|||
|
|
description: Optional[str] = None,
|
|||
|
|
due_date: Optional[str] = None,
|
|||
|
|
start_date: Optional[str] = None,
|
|||
|
|
frequency: Optional[str] = None,
|
|||
|
|
related_projects: Optional[List[str]] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
创建新目标 (作为任务存放在目标管理项目中)
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
title: 目标标题
|
|||
|
|
type: 目标类型 (phase/permanent/habit)
|
|||
|
|
keywords: 关键词，以逗号分隔
|
|||
|
|
description: 目标的基础描述 (可选)
|
|||
|
|
due_date: 截止日期 (YYYY-MM-DD) (阶段性目标必填)
|
|||
|
|
start_date: 开始日期 (YYYY-MM-DD) (可选)
|
|||
|
|
frequency: 频率 (daily, weekly:1,3,5 等) (习惯目标必填)
|
|||
|
|
related_projects: 相关项目IDs (可选)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
创建的目标信息
|
|||
|
|
"""
|
|||
|
|
# 直接调用逻辑函数，逻辑函数应能处理Optional参数
|
|||
|
|
try:
|
|||
|
|
return create_goal_logic(title, type, keywords, description, due_date, start_date, frequency)
|
|||
|
|
except (ValueError, NotImplementedError) as e:
|
|||
|
|
raise e
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"调用 create_goal 时发生意外错误: {e}")
|
|||
|
|
raise ValueError(f"创建目标时发生内部错误: {e}")
|
|||
|
|
|
|||
|
|
@server.tool()
|
|||
|
|
def get_goals(
|
|||
|
|
type: Optional[str] = None,
|
|||
|
|
status: Optional[str] = None,
|
|||
|
|
keywords: Optional[str] = None
|
|||
|
|
) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
获取目标列表
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
type: 目标类型筛选 (phase/permanent/habit)
|
|||
|
|
status: 目标状态筛选 (active/completed)
|
|||
|
|
keywords: 关键词筛选 (匹配目标标题或关键词) - 字符串形式
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
目标列表
|
|||
|
|
"""
|
|||
|
|
# 直接调用逻辑函数
|
|||
|
|
try:
|
|||
|
|
return get_goals_logic(type=type, status=status, keywords=keywords)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"调用 get_goals 时发生意外错误: {e}")
|
|||
|
|
raise ValueError(f"获取目标列表时发生内部错误: {e}")
|
|||
|
|
|
|||
|
|
@server.tool()
|
|||
|
|
def get_goal(goal_id: str) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
获取目标详情
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
goal_id: 目标ID (任务ID)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
目标详情
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
goal = get_goal_logic(goal_id)
|
|||
|
|
if not goal:
|
|||
|
|
raise ValueError(f"未找到ID为 '{goal_id}' 的目标")
|
|||
|
|
return goal
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"调用 get_goal 时发生意外错误: {e}")
|
|||
|
|
raise ValueError(f"获取目标 '{goal_id}' 时发生内部错误: {e}")
|
|||
|
|
|
|||
|
|
@server.tool()
|
|||
|
|
def update_goal(
|
|||
|
|
goal_id: str,
|
|||
|
|
title: Optional[str] = None,
|
|||
|
|
type: Optional[str] = None,
|
|||
|
|
status: Optional[str] = None,
|
|||
|
|
keywords: Optional[str] = None,
|
|||
|
|
description: Optional[str] = None,
|
|||
|
|
due_date: Optional[str] = None,
|
|||
|
|
start_date: Optional[str] = None,
|
|||
|
|
frequency: Optional[str] = None,
|
|||
|
|
progress: Optional[int] = None,
|
|||
|
|
related_projects: Optional[List[str]] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
更新目标
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
goal_id: 目标ID (任务ID)
|
|||
|
|
title: 新标题 (可选)
|
|||
|
|
type: 新类型 (phase/permanent/habit) (可选)
|
|||
|
|
status: 新状态 (active/completed) (可选)
|
|||
|
|
keywords: 新关键词 (逗号分隔) (可选)
|
|||
|
|
description: 新的基础描述 (可选)
|
|||
|
|
due_date: 新截止日期 (YYYY-MM-DD) (可选)
|
|||
|
|
start_date: 新开始日期 (YYYY-MM-DD) (可选)
|
|||
|
|
frequency: 新频率 (可选)
|
|||
|
|
progress: 进度 (忽略)
|
|||
|
|
related_projects: 相关项目 (忽略)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
更新后的目标数据
|
|||
|
|
"""
|
|||
|
|
# 直接调用逻辑函数
|
|||
|
|
try:
|
|||
|
|
return update_goal_logic(goal_id, title, type, status, keywords, description, due_date, start_date, frequency)
|
|||
|
|
except (ValueError, NotImplementedError) as e:
|
|||
|
|
raise e
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"调用 update_goal 时发生意外错误: {e}")
|
|||
|
|
raise ValueError(f"更新目标 '{goal_id}' 时发生内部错误: {e}")
|
|||
|
|
|
|||
|
|
@server.tool()
|
|||
|
|
def delete_goal(goal_id: str) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
删除目标
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
goal_id: 目标ID (任务ID)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
删除操作的结果
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
return delete_goal_logic(goal_id)
|
|||
|
|
except (ValueError, NotImplementedError) as e:
|
|||
|
|
raise e
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"调用 delete_goal 时发生意外错误: {e}")
|
|||
|
|
raise ValueError(f"删除目标 '{goal_id}' 时发生内部错误: {e}")
|
|||
|
|
|
|||
|
|
@server.tool()
|
|||
|
|
def match_task_with_goals(
|
|||
|
|
task_title: str,
|
|||
|
|
task_content: Optional[str] = None,
|
|||
|
|
project_id: Optional[str] = None
|
|||
|
|
) -> List[Dict[str, Any]]:
|
|||
|
|
"""
|
|||
|
|
匹配任务与目标
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
task_title: 任务标题
|
|||
|
|
task_content: 任务内容 (可选)
|
|||
|
|
project_id: 任务所属项目ID (可选)
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
匹配的目标列表 (按匹配度排序)
|
|||
|
|
"""
|
|||
|
|
# 直接调用逻辑函数
|
|||
|
|
try:
|
|||
|
|
return match_task_with_goals_logic(task_title, task_content, project_id)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"调用 match_task_with_goals 时发生意外错误: {e}")
|
|||
|
|
raise ValueError(f"匹配任务与目标时发生内部错误: {e}")
|
|||
|
|
|
|||
|
|
# 导出
|
|||
|
|
__all__ = [
|
|||
|
|
'create_goal_logic',
|
|||
|
|
'get_goals_logic',
|
|||
|
|
'get_goal_logic',
|
|||
|
|
'update_goal_logic',
|
|||
|
|
'delete_goal_logic',
|
|||
|
|
'match_task_with_goals_logic',
|
|||
|
|
'register_goal_tools'
|
|||
|
|
]
|