ffb14cd014
Implement three foundational schema features from the design doc: - Tree forking with lineage tracking (migration 022): parent_tree_id, root_tree_id, fork_depth columns with self-referential FKs and composite analytics index - Custom step enhancement: CustomStepSchema with source tracking (ad-hoc, step-library, forked-tree) for backward-compatible JSONB - Session sharing (migration 023): session_shares and session_share_views tables with account-scoped visibility, cryptographic tokens, view tracking, and allow_public_shares account policy Includes 21 new integration tests (9 forking, 12 sharing), SaaS consultant-recommended denormalizations, rate limiting on public share access, and test fixture fix for invite code requirement. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
from datetime import datetime
|
|
from typing import Optional, Any, Literal
|
|
from uuid import UUID
|
|
from pydantic import BaseModel, Field, validator
|
|
|
|
|
|
class CustomStepSchema(BaseModel):
|
|
"""Enhanced custom step with source tracking.
|
|
|
|
Backward compatible: old sessions without new fields load with defaults.
|
|
"""
|
|
type: str # "decision" | "action" | "solution"
|
|
content: str
|
|
notes: Optional[str] = None
|
|
|
|
# Source tracking (new fields, optional for backward compatibility)
|
|
source: Literal["ad-hoc", "step-library", "forked-tree"] = "ad-hoc"
|
|
source_step_id: Optional[UUID] = None
|
|
inserted_at: Optional[datetime] = None
|
|
inserted_after_node_id: Optional[str] = None
|
|
|
|
|
|
class DecisionRecord(BaseModel):
|
|
node_id: str
|
|
question: Optional[str] = None
|
|
answer: Optional[str] = None
|
|
action_performed: Optional[str] = None
|
|
notes: Optional[str] = None
|
|
automation_used: Optional[bool] = False
|
|
timestamp: datetime
|
|
attachments: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class SessionCreate(BaseModel):
|
|
tree_id: UUID
|
|
ticket_number: Optional[str] = Field(None, max_length=100)
|
|
client_name: Optional[str] = Field(None, max_length=255)
|
|
|
|
|
|
class SessionUpdate(BaseModel):
|
|
path_taken: Optional[list[str]] = None
|
|
decisions: Optional[list[DecisionRecord]] = None
|
|
custom_steps: Optional[list[CustomStepSchema]] = None
|
|
ticket_number: Optional[str] = Field(None, max_length=100)
|
|
client_name: Optional[str] = Field(None, max_length=255)
|
|
scratchpad: Optional[str] = None
|
|
|
|
|
|
class SessionResponse(BaseModel):
|
|
id: UUID
|
|
tree_id: UUID
|
|
user_id: UUID
|
|
tree_snapshot: dict[str, Any]
|
|
path_taken: list[str]
|
|
decisions: list[dict[str, Any]]
|
|
custom_steps: list[dict[str, Any]] = Field(default_factory=list)
|
|
started_at: datetime
|
|
completed_at: Optional[datetime] = None
|
|
ticket_number: Optional[str] = None
|
|
client_name: Optional[str] = None
|
|
exported: bool
|
|
scratchpad: str = ""
|
|
|
|
@validator('scratchpad', pre=True, always=True)
|
|
def normalize_scratchpad(cls, v):
|
|
return v or ""
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class SessionExport(BaseModel):
|
|
format: str = Field(default="markdown", pattern="^(text|markdown|html)$")
|
|
include_timestamps: bool = True
|
|
include_tree_info: bool = True
|
|
|
|
|
|
class ScratchpadUpdate(BaseModel):
|
|
scratchpad: str
|