117 lines
4.4 KiB
Python
117 lines
4.4 KiB
Python
"""Abstract persistence API for custom skill ownership + publish state."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from abc import ABC, abstractmethod
|
|
from typing import Any
|
|
|
|
|
|
class SkillStore(ABC):
|
|
@abstractmethod
|
|
async def list_visible(self, user_id: str) -> list[dict[str, Any]]:
|
|
"""List all skill records visible to ``user_id``.
|
|
|
|
Visible = owner, OR published by anyone, OR built-in (owner_user_id IS NULL).
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def get_visible(self, name: str, user_id: str) -> dict[str, Any] | None:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def get_any(self, name: str) -> dict[str, Any] | None:
|
|
"""Lookup by name regardless of ownership/visibility (admin / cascade use)."""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def get_owned(self, name: str, user_id: str) -> dict[str, Any] | None:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def create(self, data: dict[str, Any]) -> dict[str, Any]:
|
|
"""Create a skill record. Caller is responsible for global-name uniqueness check."""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def ensure_legacy(self, data: dict[str, Any]) -> dict[str, Any]:
|
|
"""Idempotently insert a legacy on-disk skill into the DB. Updates the row if it
|
|
already exists with ``owner_user_id IS NULL`` (so a re-run can migrate ownership).
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def update(self, name: str, user_id: str, data: dict[str, Any]) -> dict[str, Any] | None:
|
|
"""Owner-only update. Returns None if the caller doesn't own the skill."""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def update_admin(self, name: str, data: dict[str, Any]) -> dict[str, Any] | None:
|
|
"""Admin override — caller is responsible for auth check.
|
|
|
|
Unlike ``update_builtin`` for agents, this accepts ANY skill including
|
|
user-owned ones, since the spec lets admin take down others' skills.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def delete(self, name: str, user_id: str) -> bool:
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def delete_admin(self, name: str) -> bool:
|
|
"""Admin-side delete; bypasses ownership check (caller authorizes)."""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def count_owned(self, user_id: str) -> int:
|
|
"""Return the number of skills owned by ``user_id`` (owner_user_id IS NOT NULL)."""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def list_all(self) -> list[dict[str, Any]]:
|
|
"""Return every skill record regardless of ownership/visibility.
|
|
|
|
Caller is responsible for authorization (admin only).
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def list_published_names(self) -> set[str]:
|
|
"""Return the set of skill names with ``published = True`` (any owner).
|
|
|
|
Batch alternative to calling ``get_any`` per skill — the curator uses
|
|
this to filter out published skills in a single query instead of one
|
|
DB round-trip per candidate skill.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@abstractmethod
|
|
async def list_always_on_names(self) -> set[str]:
|
|
"""Return the set of skill names marked 常驻(免压缩) (``always_on = True``).
|
|
|
|
Batch query used by the prompt builder to decide which skills bypass
|
|
skill compression — one query instead of a per-skill round-trip.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
async def list_name_zh(self, names: list[str]) -> dict[str, str]:
|
|
"""Map skill names to their configured Chinese display names.
|
|
|
|
Only names that have a non-empty ``name_zh`` are returned. The default
|
|
is empty so stores that have not implemented the lookup stay safe.
|
|
"""
|
|
del names
|
|
return {}
|
|
|
|
@abstractmethod
|
|
async def list_favorite_skill_names(self, user_id: str) -> set[str]:
|
|
"""Return the skill names in ``user_id``'s "我的技能" collection.
|
|
|
|
These are skills the user does not own but that landed in their list —
|
|
currently only via their 岗位 (position) sync (``origin='position'``).
|
|
Owned skills are NOT included here (the caller detects those via
|
|
``owner_user_id``).
|
|
"""
|
|
raise NotImplementedError
|