Files
beaver_project/app-instance/backend/beaver/skills/drafts/service.py
steven_li 30ab74ffb2 feat(engine): 添加MCP连接管理和工具集成功能
- 集成MCP连接管理器,支持MCP服务器连接
- 添加多种内置工具:ClarifyTool、CronTool、DelegateTool、ExecuteCodeTool、
  PatchFileTool、ProcessTool、SendMessageTool、SpawnTool、TerminalTool、
  TodoTool、WebFetchTool、WebSearchTool、WriteFileTool等
- 实现工具注册和装配功能
- 添加技能选择上下文参数
- 支持思考模式控制参数thinking_enabled

feat(coordinator): 重构任务执行计划器参数命名

- 将learning_candidate_enabled重命名为allow_candidate_generation
- 更新TeamGraphScheduler中的参数传递
- 修改LocalAgentRunner中的相关参数处理
- 更新README文档中的相应描述

refactor(context): 标准化工具调用参数格式

- 添加_json导入用于参数序列化
- 实现_provider_tool_calls方法标准化OpenAI兼容的工具调用载荷
- 修复工具调用中参数非字符串类型的序列化问题

refactor(session): 优化消息历史记录过滤逻辑

- 修改get_messages_as_conversation为基于运行状态过滤消息
- 排除未完成、失败或错误结束的运行记录
- 改进对话历史的可见性控制机制

fix(store): 修复FTS索引重建逻辑

- 添加异常处理防止FTS索引创建失败
- 实现_rebuild_fts_index方法重新构建全文搜索索引
- 优化索引触发器和表的维护流程
2026-05-14 09:43:48 +08:00

135 lines
4.0 KiB
Python

"""Draft lifecycle for Beaver skills."""
from __future__ import annotations
from uuid import uuid4
from beaver.skills.specs import SkillDraft, SkillSpecStore
class DraftService:
def __init__(self, store: SkillSpecStore) -> None:
self.store = store
def create_new_skill_draft(
self,
*,
skill_name: str,
proposed_content: str,
proposed_frontmatter: dict,
created_by: str,
reason: str,
trigger_run_id: str | None = None,
trigger_session_id: str | None = None,
evidence_refs: list[dict] | None = None,
) -> SkillDraft:
draft = SkillDraft(
draft_id=uuid4().hex,
skill_name=skill_name,
base_version=None,
proposed_content=proposed_content,
proposed_frontmatter=dict(proposed_frontmatter),
created_at=_utc_now(),
created_by=created_by,
trigger_run_id=trigger_run_id,
trigger_session_id=trigger_session_id,
reason=reason,
evidence_refs=list(evidence_refs or []),
proposal_kind="new_skill",
)
self.store.write_draft(draft)
return draft
def create_revision_draft(
self,
*,
skill_name: str,
base_version: str | None,
proposed_content: str,
proposed_frontmatter: dict,
created_by: str,
reason: str,
trigger_run_id: str | None = None,
trigger_session_id: str | None = None,
evidence_refs: list[dict] | None = None,
) -> SkillDraft:
draft = SkillDraft(
draft_id=uuid4().hex,
skill_name=skill_name,
base_version=base_version,
proposed_content=proposed_content,
proposed_frontmatter=dict(proposed_frontmatter),
created_at=_utc_now(),
created_by=created_by,
trigger_run_id=trigger_run_id,
trigger_session_id=trigger_session_id,
reason=reason,
evidence_refs=list(evidence_refs or []),
proposal_kind="revise_skill",
)
self.store.write_draft(draft)
return draft
def create_merge_draft(
self,
*,
skill_name: str,
base_version: str | None,
proposed_content: str,
proposed_frontmatter: dict,
created_by: str,
reason: str,
evidence_refs: list[dict] | None = None,
) -> SkillDraft:
draft = self.create_revision_draft(
skill_name=skill_name,
base_version=base_version,
proposed_content=proposed_content,
proposed_frontmatter=proposed_frontmatter,
created_by=created_by,
reason=reason,
evidence_refs=evidence_refs,
)
draft.proposal_kind = "merge_skills"
self.store.write_draft(draft)
return draft
def create_retire_proposal(
self,
*,
skill_name: str,
base_version: str | None,
created_by: str,
reason: str,
evidence_refs: list[dict] | None = None,
) -> SkillDraft:
draft = SkillDraft(
draft_id=uuid4().hex,
skill_name=skill_name,
base_version=base_version,
proposed_content="",
proposed_frontmatter={},
created_at=_utc_now(),
created_by=created_by,
reason=reason,
evidence_refs=list(evidence_refs or []),
proposal_kind="retire_skill",
)
self.store.write_draft(draft)
return draft
def list_drafts(self, skill_name: str | None = None) -> list[SkillDraft]:
return self.store.list_drafts(skill_name)
def get_draft(self, skill_name: str, draft_id: str) -> SkillDraft | None:
return self.store.read_draft(skill_name, draft_id)
def delete_draft(self, skill_name: str, draft_id: str) -> bool:
return self.store.delete_draft(skill_name, draft_id)
def _utc_now() -> str:
from datetime import datetime, timezone
return datetime.now(timezone.utc).isoformat()