frozen pydantic models

src/personal_mcp/registry/contracts.py

@@ -1,80 +1,17 @@
-from dataclasses import dataclass
-from dataclasses import field
-
 from .models.prompt import PromptArgumentEntry
+from .models.registry import DocsRegistry
+from .models.registry import PromptRecord
+from .models.registry import PromptSummaryRecord
+from .models.registry import ReferenceRecord
+from .models.registry import SkillRecord
+from .models.registry import SkillSummaryRecord
 
-
-@dataclass(frozen=True)
-class ReferenceRecord:
-    ref_id: str
-    uri: str
-    relpath: str
-    mime_type: str
-    title: str | None
-    content: str
-
-
-@dataclass(frozen=True)
-class SkillRecord:
-    skill_id: str
-    name: str
-    description: str
-    version: str
-    tags: tuple[str, ...]
-    capabilities: tuple[str, ...]
-    depends_on: tuple[str, ...]
-    document_uri: str
-    document_relpath: str
-    document_content: str
-    references: dict[str, ReferenceRecord]
-
-
-@dataclass(frozen=True)
-class SkillSummaryRecord:
-    skill_id: str
-    name: str
-    description: str
-    tags: tuple[str, ...]
-    capabilities: tuple[str, ...]
-    document_uri: str
-    version: str
-
-
-@dataclass(frozen=True)
-class PromptRecord:
-    prompt_id: str
-    name: str
-    description: str
-    version: str
-    tags: tuple[str, ...]
-    capabilities: tuple[str, ...]
-    arguments: dict[str, PromptArgumentEntry]
-    document_uri: str
-    document_relpath: str
-    document_content: str
-
-
-@dataclass(frozen=True)
-class PromptSummaryRecord:
-    prompt_id: str
-    name: str
-    description: str
-    tags: tuple[str, ...]
-    capabilities: tuple[str, ...]
-    document_uri: str
-    version: str
-
-
-@dataclass(frozen=True)
-class DocsRegistry:
-    skills_by_id: dict[str, SkillRecord]
-    skills_in_load_order: tuple[str, ...]
-    skills_summary_in_load_order: tuple[SkillSummaryRecord, ...]
-    docs_markdown_by_path: dict[str, str]
-    docs_markdown_path_index: tuple[str, ...]
-    tag_to_skill_ids: dict[str, tuple[str, ...]]
-    capability_to_skill_ids: dict[str, tuple[str, ...]]
-    prompts_by_id: dict[str, PromptRecord] = field(default_factory=dict)
-    prompts_in_load_order: tuple[str, ...] = ()
-    prompts_summary_in_load_order: tuple[PromptSummaryRecord, ...] = ()
-    tag_to_prompt_ids: dict[str, tuple[str, ...]] = field(default_factory=dict)
+__all__ = [
+    "DocsRegistry",
+    "PromptArgumentEntry",
+    "PromptRecord",
+    "PromptSummaryRecord",
+    "ReferenceRecord",
+    "SkillRecord",
+    "SkillSummaryRecord",
+]