mirror of
https://github.com/kuaifan/dootask.git
synced 2025-12-11 02:12:53 +00:00
Compare commits
2 Commits
cf78766a37
...
8af6887daa
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8af6887daa | ||
|
|
eb9b7b4f86 |
@ -834,7 +834,7 @@ class WebSocketDialogMsg extends AbstractModel
|
||||
switch ($this->type) {
|
||||
case "file":
|
||||
// 提取文件消息
|
||||
$result = " 文件:{$this->msg['name']}、大小:{$this->msg['size']}、下载URL:{$this->msg['path']} ";
|
||||
$result = " 文件:{$this->msg['name']}(大小:{$this->msg['size']}B,URL:{$this->msg['path']}) ";
|
||||
break;
|
||||
|
||||
case "text":
|
||||
@ -858,30 +858,30 @@ class WebSocketDialogMsg extends AbstractModel
|
||||
$result = preg_replace_callback_array([
|
||||
// 用户
|
||||
"/<span class=\"mention user\" data-id=\"(\d+)\">(.*?)<\/span>/" => function ($match) {
|
||||
return " 提及用户ID:{$match[1]} ";
|
||||
return "";
|
||||
},
|
||||
|
||||
// 任务
|
||||
"/<span class=\"mention task\" data-id=\"(\d+)\">#?(.*?)<\/span>/" => function ($match) {
|
||||
return " 任务:{$match[2]} (任务ID:{$match[1]}) ";
|
||||
return " 任务:{$match[2]} (任务ID:{$match[1]}) ";
|
||||
},
|
||||
|
||||
// 文件
|
||||
"/<a class=\"mention file\" href=\"([^\"']+?)\"[^>]*?>~?(.*?)<\/a>/" => function ($match) use (&$reserves) {
|
||||
$idOrCode = "";
|
||||
if (preg_match("/single\/file\/(.*?)$/", $match[1], $subMatch)) {
|
||||
$idOrCode = " (" . (Base::isNumber($subMatch[1]) ? "文件ID:{$subMatch[1]}" : "文件分享码:{$subMatch[1]}") . ")";
|
||||
$idOrCode = " (" . (Base::isNumber($subMatch[1]) ? "文件ID:{$subMatch[1]}" : "文件分享码:{$subMatch[1]}") . ")";
|
||||
}
|
||||
return " 文件:{$match[2]}{$idOrCode} ";
|
||||
return " 文件:{$match[2]}{$idOrCode} ";
|
||||
},
|
||||
|
||||
// 报告
|
||||
"/<a class=\"mention report\" href=\"([^\"']+?)\"[^>]*?>%?(.*?)<\/a>/" => function ($match) use (&$reserves) {
|
||||
$idOrCode = "";
|
||||
if (preg_match("/single\/report\/detail\/(.*?)$/", $match[1], $subMatch)) {
|
||||
$idOrCode = " (" . (Base::isNumber($subMatch[1]) ? "报告ID:{$subMatch[1]}" : "报告分享码:{$subMatch[1]}") . ")";
|
||||
$idOrCode = " (" . (Base::isNumber($subMatch[1]) ? "报告ID:{$subMatch[1]}" : "报告分享码:{$subMatch[1]}") . ")";
|
||||
}
|
||||
return " 工作汇报:{$match[2]}{$idOrCode} ";
|
||||
return " 工作汇报:{$match[2]}{$idOrCode} ";
|
||||
},
|
||||
], $result);
|
||||
|
||||
|
||||
@ -508,7 +508,7 @@ class BotReceiveMsgTask extends AbstractTask
|
||||
if (empty($extras['api_key'])) {
|
||||
throw new Exception('机器人未启用。');
|
||||
}
|
||||
$this->generateSystemPromptForAI($msg->userid, $dialog, $extras);
|
||||
$this->generateSystemPromptForAI($msg->userid, $dialog, $botUser, $extras);
|
||||
// 转换提及格式
|
||||
if ($replyText) {
|
||||
$sendText = <<<EOF
|
||||
@ -602,134 +602,80 @@ class BotReceiveMsgTask extends AbstractTask
|
||||
*
|
||||
* @param int|null $userid 用户ID
|
||||
* @param WebSocketDialog $dialog 对话对象
|
||||
* @param User $botUser 机器人用户对象
|
||||
* @param array $extras 额外参数数组,通过引用传递以修改system_message
|
||||
* @return void
|
||||
*/
|
||||
private function generateSystemPromptForAI($userid, WebSocketDialog $dialog, array &$extras)
|
||||
private function generateSystemPromptForAI($userid, WebSocketDialog $dialog, User $botUser, array &$extras)
|
||||
{
|
||||
// 构建结构化的系统提示词
|
||||
$sections = [];
|
||||
|
||||
// 基础角色设定(如果有)
|
||||
if (!empty($extras['system_message'])) {
|
||||
$sections[] = <<<EOF
|
||||
<role_setting>
|
||||
{$extras['system_message']}
|
||||
</role_setting>
|
||||
EOF;
|
||||
// 用户自定义提示词(私聊场景优先使用)
|
||||
$customPrompt = null;
|
||||
if ($dialog->type === 'user') {
|
||||
$customPrompt = WebSocketDialogConfig::where([
|
||||
'dialog_id' => $dialog->id,
|
||||
'userid' => $userid,
|
||||
'type' => 'ai_prompt',
|
||||
])->value('value');
|
||||
}
|
||||
|
||||
// 上下文信息(项目、任务、部门等)+ 操作指令
|
||||
switch ($dialog->type) {
|
||||
// 用户对话
|
||||
case "user":
|
||||
$aiPrompt = WebSocketDialogConfig::where([
|
||||
'dialog_id' => $dialog->id,
|
||||
'userid' => $userid,
|
||||
'type' => 'ai_prompt',
|
||||
])->value('value');
|
||||
if ($aiPrompt) {
|
||||
return $aiPrompt;
|
||||
}
|
||||
break;
|
||||
$prompt = [];
|
||||
|
||||
// 群组对话
|
||||
case "group":
|
||||
switch ($dialog->group_type) {
|
||||
// 用户群
|
||||
case 'user':
|
||||
break;
|
||||
|
||||
// 项目群
|
||||
case 'project':
|
||||
$projectInfo = Project::whereDialogId($dialog->id)->first();
|
||||
if ($projectInfo) {
|
||||
$currentTime = Carbon::now()->toDateTimeString();
|
||||
$sections[] = <<<EOF
|
||||
<context_info>
|
||||
当前我在项目群聊中
|
||||
项目ID:{$projectInfo->id}
|
||||
项目名称:{$projectInfo->name}
|
||||
当前时间:{$currentTime}
|
||||
</context_info>
|
||||
EOF;
|
||||
}
|
||||
break;
|
||||
|
||||
// 任务群
|
||||
case 'task':
|
||||
$taskInfo = ProjectTask::with(['content'])->whereDialogId($dialog->id)->first();
|
||||
if ($taskInfo) {
|
||||
$currentTime = Carbon::now()->toDateTimeString();
|
||||
$sections[] = <<<EOF
|
||||
<context_info>
|
||||
当前我在任务群聊中
|
||||
任务ID:{$taskInfo->id}
|
||||
任务名称:{$taskInfo->name}
|
||||
当前时间:{$currentTime}
|
||||
</context_info>
|
||||
EOF;
|
||||
}
|
||||
break;
|
||||
|
||||
// 部门群
|
||||
case 'department':
|
||||
$userDepartment = UserDepartment::whereDialogId($dialog->id)->first();
|
||||
if ($userDepartment) {
|
||||
$sections[] = <<<EOF
|
||||
<context_info>
|
||||
当前我在部门群聊中
|
||||
部门ID:{$userDepartment->id}
|
||||
部门名称:{$userDepartment->name}
|
||||
</context_info>
|
||||
EOF;
|
||||
}
|
||||
break;
|
||||
|
||||
// 全体成员群
|
||||
case 'all':
|
||||
$sections[] = <<<EOF
|
||||
<context_info>
|
||||
当前我在【全体成员】的群聊中
|
||||
</context_info>
|
||||
EOF;
|
||||
break;
|
||||
}
|
||||
|
||||
// 聊天历史
|
||||
if ($dialog->type === 'group') {
|
||||
$chatHistory = $this->getRecentChatHistory($dialog, 15);
|
||||
if ($chatHistory) {
|
||||
$sections[] = <<<EOF
|
||||
<chat_history>
|
||||
{$chatHistory}
|
||||
</chat_history>
|
||||
EOF;
|
||||
}
|
||||
}
|
||||
break;
|
||||
// 1. 基础角色(自定义提示词优先)
|
||||
if ($customPrompt) {
|
||||
$prompt[] = $customPrompt;
|
||||
} elseif (!empty($extras['system_message'])) {
|
||||
$prompt[] = $extras['system_message'];
|
||||
}
|
||||
|
||||
// 更新系统提示词
|
||||
if (!empty($sections)) {
|
||||
$extras['system_message'] = implode("\n\n", $sections);
|
||||
}
|
||||
|
||||
// 添加标签说明
|
||||
$tagDescs = [
|
||||
'role_setting' => '你的基础角色和行为定义',
|
||||
'context_info' => '当前环境和状态信息',
|
||||
'chat_history' => '最近的对话历史记录',
|
||||
// 2. 上下文信息
|
||||
$currentTime = Carbon::now()->toDateTimeString();
|
||||
$contextLines = [
|
||||
"您是:{$botUser->nickname}(ID: {$botUser->userid})",
|
||||
"当前对话ID:{$dialog->id}",
|
||||
"当前系统时间:{$currentTime}"
|
||||
];
|
||||
$useTags = [];
|
||||
foreach ($tagDescs as $tag => $desc) {
|
||||
if (str_contains($extras['system_message'], '<' . $tag . '>')) {
|
||||
$useTags[] = '- <' . $tag . '>: ' . $desc;
|
||||
|
||||
if ($dialog->type === 'group') {
|
||||
switch ($dialog->group_type) {
|
||||
case 'project':
|
||||
$projectInfo = Project::whereDialogId($dialog->id)->first();
|
||||
if ($projectInfo) {
|
||||
$contextLines[] = "场景:项目群聊「{$projectInfo->name}」(ID: {$projectInfo->id})";
|
||||
}
|
||||
break;
|
||||
|
||||
case 'task':
|
||||
$taskInfo = ProjectTask::with(['content'])->whereDialogId($dialog->id)->first();
|
||||
if ($taskInfo) {
|
||||
$contextLines[] = "场景:任务群聊「{$taskInfo->name}」(ID: {$taskInfo->id})";
|
||||
}
|
||||
break;
|
||||
|
||||
case 'department':
|
||||
$userDepartment = UserDepartment::whereDialogId($dialog->id)->first();
|
||||
if ($userDepartment) {
|
||||
$contextLines[] = "场景:部门群聊「{$userDepartment->name}」";
|
||||
}
|
||||
break;
|
||||
|
||||
case 'all':
|
||||
$contextLines[] = "场景:全体成员群聊";
|
||||
break;
|
||||
}
|
||||
|
||||
// 3. 聊天历史(仅群聊)
|
||||
$chatHistory = $this->getRecentChatHistory($dialog, 15);
|
||||
if ($chatHistory) {
|
||||
$prompt[] = implode("\n", $contextLines);
|
||||
$prompt[] = "最近的对话记录:\n{$chatHistory}";
|
||||
} else {
|
||||
$prompt[] = implode("\n", $contextLines);
|
||||
}
|
||||
} else {
|
||||
$prompt[] = implode("\n", $contextLines);
|
||||
}
|
||||
if (!empty($useTags)) {
|
||||
$extras['system_message'] = "以下信息按标签组织:\n" . implode("\n", $useTags) . "\n\n" . $extras['system_message'];
|
||||
}
|
||||
|
||||
$extras['system_message'] = implode("\n----\n", array_filter($prompt));
|
||||
}
|
||||
|
||||
/**
|
||||
@ -766,14 +712,14 @@ class BotReceiveMsgTask extends AbstractTask
|
||||
// 使用XML标签格式,确保AI能清晰识别边界
|
||||
// 对用户名进行HTML转义,防止特殊字符破坏格式
|
||||
$safeUserName = htmlspecialchars($userName, ENT_QUOTES, 'UTF-8');
|
||||
return "<message user=\"{$safeUserName}\">\n{$content}\n</message>";
|
||||
return "<message userid=\"{$message->userid}\" nickname=\"{$safeUserName}\">\n{$content}\n</message>";
|
||||
})
|
||||
->reverse() // 反转集合,让时间顺序正确(最早的在前)
|
||||
->filter() // 过滤掉空内容的消息
|
||||
->values() // 重新索引数组
|
||||
->toArray();
|
||||
|
||||
return empty($chatMessages) ? null : implode("\n\n", $chatMessages);
|
||||
return empty($chatMessages) ? null : implode("\n", $chatMessages);
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
56
electron/lib/mcp.js
vendored
56
electron/lib/mcp.js
vendored
@ -4,36 +4,36 @@
|
||||
* DooTask 的 Electron 客户端集成了 Model Context Protocol (MCP) 服务,
|
||||
* 允许 AI 助手(如 Claude)直接与 DooTask 任务进行交互。
|
||||
*
|
||||
* 提供的工具(共 24 个):
|
||||
* 提供的工具(共 25 个):
|
||||
*
|
||||
* === 用户管理 ===
|
||||
* - get_users_basic - 根据用户ID列表获取基础信息,便于匹配负责人/协助人
|
||||
* - search_user - 按关键字或项目筛选用户,支持分页与更多过滤项
|
||||
* - get_users_basic - 批量获取用户基础信息(1-50个),便于匹配负责人/协助人
|
||||
* - search_user - 按关键词搜索用户,支持按项目/对话范围筛选,用于不知道用户ID时的查找
|
||||
*
|
||||
* === 任务管理 ===
|
||||
* - list_tasks - 获取任务列表,支持按状态/项目/主任务筛选、搜索、分页
|
||||
* - get_task - 获取任务详情,包含完整内容、负责人、协助人员、标签等所有信息
|
||||
* - list_tasks - 获取当前用户相关的任务列表(负责/协助/关注),支持按状态/项目/时间筛选
|
||||
* - get_task - 获取单个任务的完整详细信息,比 list_tasks 返回更详细
|
||||
* - complete_task - 快速标记任务完成
|
||||
* - create_task - 创建新任务
|
||||
* - update_task - 更新任务,支持修改名称、内容、负责人、时间、状态等所有属性
|
||||
* - create_task - 在指定项目中创建新任务
|
||||
* - update_task - 更新任务的任意属性,只需提供要修改的字段
|
||||
* - create_sub_task - 为指定主任务创建子任务
|
||||
* - get_task_files - 获取任务附件列表
|
||||
* - delete_task - 删除或还原任务
|
||||
*
|
||||
* === 项目管理 ===
|
||||
* - list_projects - 获取项目列表,支持按归档状态筛选、搜索
|
||||
* - get_project - 获取项目详情,包含列(看板列)、成员等完整信息
|
||||
* - list_projects - 获取当前用户可访问的项目列表,支持按归档状态筛选、搜索
|
||||
* - get_project - 获取项目的完整详细信息,比 list_projects 返回更详细
|
||||
* - create_project - 创建新项目
|
||||
* - update_project - 修改项目信息(名称、描述等)
|
||||
*
|
||||
* === 文件管理 ===
|
||||
* - list_files - 获取项目文件列表,支持按父级文件夹筛选
|
||||
* - search_files - 按关键词搜索文件,支持搜索文件名称或ID
|
||||
* - get_file_detail - 获取文件详情,返回 content_url 可配合 WebFetch 读取文件内容
|
||||
* === 文件管理(个人文件系统) ===
|
||||
* - list_files - 浏览个人文件系统,获取指定文件夹下的文件和子文件夹列表
|
||||
* - search_files - 搜索用户文件系统中的文件(自己创建的和共享给自己的)
|
||||
* - get_file_detail - 获取文件详情,支持通过文件ID或分享码访问,返回 content_url 可配合 WebFetch 读取
|
||||
*
|
||||
* === 工作报告 ===
|
||||
* - list_received_reports - 获取我接收的汇报列表,支持按类型/状态/部门/时间筛选
|
||||
* - get_report_detail - 获取汇报详情,包括完整内容、汇报人、接收人、AI分析等
|
||||
* - get_report_detail - 获取汇报详情,支持通过报告ID或分享码访问,内容自动转为 Markdown
|
||||
* - generate_report_template - 基于任务完成情况自动生成汇报模板(已完成/未完成任务)
|
||||
* - create_report - 创建并提交工作汇报
|
||||
* - list_my_reports - 获取我发送的汇报列表,支持按类型/时间筛选
|
||||
@ -41,7 +41,7 @@
|
||||
*
|
||||
* === 消息通知 ===
|
||||
* - send_message_to_user - 给指定用户发送私信
|
||||
* - get_message_list - 获取对话消息或执行关键词搜索
|
||||
* - get_message_list - 两种模式:获取对话消息列表 或 按关键词搜索消息
|
||||
*
|
||||
* 配置方法:
|
||||
* {
|
||||
@ -222,7 +222,7 @@ class DooTaskMCP {
|
||||
// 用户管理:获取用户基础信息
|
||||
this.mcp.addTool({
|
||||
name: 'get_users_basic',
|
||||
description: '根据用户ID列表获取用户基础信息(昵称、邮箱、头像等),方便在分配任务前确认成员身份。',
|
||||
description: '根据用户ID列表批量获取用户基础信息(昵称、邮箱、头像等)。适用于分配任务前确认成员身份,支持1-50个用户。',
|
||||
parameters: z.object({
|
||||
userids: z.array(z.number())
|
||||
.min(1)
|
||||
@ -269,7 +269,7 @@ class DooTaskMCP {
|
||||
// 用户管理:搜索用户
|
||||
this.mcp.addTool({
|
||||
name: 'search_user',
|
||||
description: '按关键词搜索或筛选用户,支持按项目/对话过滤并返回分页结果。',
|
||||
description: '按关键词搜索用户(昵称、邮箱、拼音),支持按项目/对话范围筛选。与 get_users_basic 不同,此工具用于不知道具体用户ID时的查找场景。',
|
||||
parameters: z.object({
|
||||
keyword: z.string()
|
||||
.min(1)
|
||||
@ -376,7 +376,7 @@ class DooTaskMCP {
|
||||
// 获取任务列表
|
||||
this.mcp.addTool({
|
||||
name: 'list_tasks',
|
||||
description: '获取任务列表。可以按状态筛选(已完成/未完成)、搜索任务名称、按时间范围筛选等。',
|
||||
description: '获取当前用户相关的任务列表(负责/协助/关注),支持按状态、项目、时间范围筛选和搜索。',
|
||||
parameters: z.object({
|
||||
status: z.enum(['all', 'completed', 'uncompleted'])
|
||||
.optional()
|
||||
@ -474,7 +474,7 @@ class DooTaskMCP {
|
||||
// 获取任务详情
|
||||
this.mcp.addTool({
|
||||
name: 'get_task',
|
||||
description: '获取指定任务的详细信息,包括任务描述、完整内容、负责人、协助人员、标签、时间等所有信息。返回的 content 字段为 Markdown 格式。',
|
||||
description: '获取单个任务的完整详细信息,包括任务描述、完整内容(content)、负责人、协助人员、标签、时间等。比 list_tasks 返回更详细的信息。',
|
||||
parameters: z.object({
|
||||
task_id: z.number()
|
||||
.describe('任务ID'),
|
||||
@ -587,7 +587,7 @@ class DooTaskMCP {
|
||||
// 创建任务
|
||||
this.mcp.addTool({
|
||||
name: 'create_task',
|
||||
description: '创建新任务。可以指定任务名称、内容、负责人、时间等信息。',
|
||||
description: '在指定项目中创建新任务。必需参数:项目ID、任务名称。可选:负责人、协助人、开始/结束时间、看板列等。',
|
||||
parameters: z.object({
|
||||
project_id: z.number()
|
||||
.describe('项目ID'),
|
||||
@ -656,7 +656,7 @@ class DooTaskMCP {
|
||||
// 更新任务
|
||||
this.mcp.addTool({
|
||||
name: 'update_task',
|
||||
description: '更新任务信息。可以修改任务名称、内容、负责人、时间、状态等所有属性。',
|
||||
description: '更新任务的任意属性(名称、内容、负责人、协助人、时间、完成状态、看板列等)。只需提供要修改的字段。',
|
||||
parameters: z.object({
|
||||
task_id: z.number()
|
||||
.describe('任务ID'),
|
||||
@ -864,7 +864,7 @@ class DooTaskMCP {
|
||||
// 获取项目列表
|
||||
this.mcp.addTool({
|
||||
name: 'list_projects',
|
||||
description: '获取项目列表。可以按归档状态筛选、搜索项目名称等。',
|
||||
description: '获取当前用户可访问的项目列表,支持按归档状态筛选、搜索项目名称。',
|
||||
parameters: z.object({
|
||||
archived: z.enum(['no', 'yes', 'all'])
|
||||
.optional()
|
||||
@ -926,7 +926,7 @@ class DooTaskMCP {
|
||||
// 获取项目详情
|
||||
this.mcp.addTool({
|
||||
name: 'get_project',
|
||||
description: '获取指定项目的详细信息,包括项目的列(看板列)、成员等完整信息。',
|
||||
description: '获取指定项目的完整详细信息,包括项目描述、所有看板列、成员列表及权限等。比 list_projects 返回更详细的信息。',
|
||||
parameters: z.object({
|
||||
project_id: z.number()
|
||||
.describe('项目ID'),
|
||||
@ -1188,7 +1188,7 @@ class DooTaskMCP {
|
||||
// 获取消息列表或搜索消息
|
||||
this.mcp.addTool({
|
||||
name: 'get_message_list',
|
||||
description: '获取指定对话的消息列表,或按关键字搜索消息位置/内容。',
|
||||
description: '两种模式:1)获取指定对话的消息列表(需提供 dialog_id),支持按类型筛选、分页加载;2)按关键词搜索消息(提供 keyword),可在单个对话或全局搜索。',
|
||||
parameters: z.object({
|
||||
dialog_id: z.number()
|
||||
.optional()
|
||||
@ -1715,7 +1715,7 @@ class DooTaskMCP {
|
||||
// 文件管理:获取文件列表
|
||||
this.mcp.addTool({
|
||||
name: 'list_files',
|
||||
description: '获取项目文件列表,支持按父级文件夹筛选。可以浏览文件夹结构,查看所有文件和子文件夹。',
|
||||
description: '获取用户文件列表(个人文件系统),支持按父级文件夹筛选。pid=0或不传表示获取根目录,pid>0获取指定文件夹下的内容。可以浏览文件夹结构,查看所有文件和子文件夹。',
|
||||
parameters: z.object({
|
||||
pid: z.number()
|
||||
.optional()
|
||||
@ -1764,11 +1764,11 @@ class DooTaskMCP {
|
||||
// 文件管理:搜索文件
|
||||
this.mcp.addTool({
|
||||
name: 'search_files',
|
||||
description: '按关键词搜索文件,支持搜索文件名称或ID。可以快速定位文件位置。',
|
||||
description: '按关键词搜索用户文件系统中的文件,支持搜索文件名称、文件ID或分享链接。搜索范围包括:自己创建的文件和共享给自己的文件。',
|
||||
parameters: z.object({
|
||||
keyword: z.string()
|
||||
.min(1)
|
||||
.describe('搜索关键词,支持文件名称或文件ID'),
|
||||
.describe('搜索关键词,支持文件名称、文件ID或分享链接'),
|
||||
take: z.number()
|
||||
.optional()
|
||||
.describe('返回数量,默认50,最大100'),
|
||||
@ -1817,7 +1817,7 @@ class DooTaskMCP {
|
||||
// 文件管理:获取文件详情
|
||||
this.mcp.addTool({
|
||||
name: 'get_file_detail',
|
||||
description: '获取指定文件的详细信息,包括类型、大小、共享状态、创建者等。返回的 content_url 可以配合 WebFetch 工具读取文件内容进行分析。支持通过文件ID或分享码访问。',
|
||||
description: '获取指定文件的详细信息,包括类型、大小、共享状态、创建者等。支持通过文件ID或分享码访问。返回的 content_url 可以配合 WebFetch 工具读取文件内容进行分析。',
|
||||
parameters: z.object({
|
||||
file_id: z.union([z.number(), z.string()])
|
||||
.describe('文件ID(数字)或分享码(字符串)'),
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user