enterprise-context-mcp-admin/models.py at main · dormstd/enterprise-context-mcp-admin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
"""Pydantic request/response models for the Enterprise Context Admin API."""

from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel


# ---------------------------------------------------------------------------
# Categories
# ---------------------------------------------------------------------------


class CategoryCreate(BaseModel):
    name: str
    description: str | None = None


class CategoryOut(BaseModel):
    id: int
    name: str
    description: str | None = None


# ---------------------------------------------------------------------------
# Roles
# ---------------------------------------------------------------------------


class RoleCreate(BaseModel):
    name: str


class RoleOut(BaseModel):
    id: int
    name: str


# ---------------------------------------------------------------------------
# Tags
# ---------------------------------------------------------------------------


class TagOut(BaseModel):
    id: int
    name: str


# ---------------------------------------------------------------------------
# Knowledge entries
# ---------------------------------------------------------------------------


class EntryCreate(BaseModel):
    title: str
    content: str
    category_name: str
    roles: list[str] = []
    tags: list[str] = []


class EntryUpdate(BaseModel):
    title: str | None = None
    content: str | None = None
    category_name: str | None = None
    roles: list[str] | None = None
    tags: list[str] | None = None


class EntryOut(BaseModel):
    id: int
    title: str
    content: str
    category: str
    category_description: str | None = None
    tags: list[str] = []
    target_roles: list[str] = []
    created_at: datetime | None = None
    updated_at: datetime | None = None


# ---------------------------------------------------------------------------
# Bulk import
# ---------------------------------------------------------------------------


class BulkImportRequest(BaseModel):
    entries: list[EntryCreate]


class BulkImportResult(BaseModel):
    created: int
    errors: list[str] = []