justin
c6b1e72130
The operator decides which model providers exist (env / registry — Anthropic,
OpenAI, x.AI, Ollama, several at once). The *tree owner* decides who uses which:
- Members' assistant -> one configured provider (or none)
- Recommender (association/connection finder) -> one configured provider (or none)
- Owner -> may use any configured provider
Backend: two nullable columns on `trees` (ai_member_provider,
ai_recommender_provider) + migration; `configured_llm_providers()` exposes the
registry as {name, model} with no secrets; owner-gated GET/PATCH
/trees/{id}/ai validate names against the configured set. Frontend: owner-only
"AI models" page with a dropdown per role, graceful 403 for non-owners, and a
sidebar link.
Per-model-within-a-provider selection is a follow-up; today each provider maps
to its single configured model.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Signed-off-by: Justin Paul <justin@jpaul.me>
59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
"""Tree — the top-level tenant boundary for genealogical data — and
|
|
TreeMembership, the basis for authorization (ARCHITECTURE §5).
|
|
"""
|
|
|
|
import uuid
|
|
|
|
from sqlalchemy import Enum as SAEnum
|
|
from sqlalchemy import ForeignKey, String, Text, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.models.base import Base
|
|
from app.models.enums import MembershipRole, TreeVisibility
|
|
from app.models.mixins import SoftDelete, Timestamps, UUIDPrimaryKey
|
|
|
|
|
|
class Tree(Base, UUIDPrimaryKey, Timestamps, SoftDelete):
|
|
__tablename__ = "trees"
|
|
|
|
owner_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("users.id", ondelete="RESTRICT"), index=True
|
|
)
|
|
name: Mapped[str] = mapped_column(String(255))
|
|
description: Mapped[str | None] = mapped_column(Text)
|
|
visibility: Mapped[TreeVisibility] = mapped_column(
|
|
SAEnum(TreeVisibility, name="tree_visibility"),
|
|
default=TreeVisibility.private,
|
|
server_default=TreeVisibility.private.value,
|
|
)
|
|
# The person a tree opens focused on (its "home"/root person). Cleared if
|
|
# that person is deleted. use_alter + name: trees<->persons form an FK cycle.
|
|
home_person_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
ForeignKey(
|
|
"persons.id",
|
|
ondelete="SET NULL",
|
|
name="fk_trees_home_person_id",
|
|
use_alter=True,
|
|
)
|
|
)
|
|
# Per-tree AI model policy (owner-configured). The names reference configured
|
|
# providers from the registry; null = that role has no model. The owner may
|
|
# use any configured provider; these limit members + the recommender.
|
|
ai_member_provider: Mapped[str | None] = mapped_column(String(32))
|
|
ai_recommender_provider: Mapped[str | None] = mapped_column(String(32))
|
|
|
|
|
|
class TreeMembership(Base, UUIDPrimaryKey, Timestamps):
|
|
__tablename__ = "tree_memberships"
|
|
__table_args__ = (
|
|
UniqueConstraint("tree_id", "user_id", name="uq_tree_memberships_tree_user"),
|
|
)
|
|
|
|
tree_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("trees.id", ondelete="CASCADE"), index=True
|
|
)
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("users.id", ondelete="CASCADE"), index=True
|
|
)
|
|
role: Mapped[MembershipRole] = mapped_column(SAEnum(MembershipRole, name="membership_role"))
|