c7b2c59ef6
Backend features: - Tree sharing via secure tokens with expiration (Issue #16) - Draft tree status with conditional validation (Issue #25) - Save session as custom tree with fork tracking (Issue #17) - Tree validation system for publish requirements - Session-to-tree conversion preserving custom steps Database migrations: - 024: Tree sharing (tree_shares table, visibility field) - 025: Tree status field (draft/published) - 25b: Merge migration for indexes New endpoints: - POST /api/v1/trees/{id}/share - Generate share token - GET /api/v1/shared/{token} - Public tree access - POST /api/v1/trees/{id}/can-publish - Validate tree - POST /api/v1/sessions/{id}/save-as-tree - Convert session Test coverage: - 20 tests for draft trees functionality - 14 tests for session-to-tree conversion - 15 tests for tree sharing Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
94 lines
2.9 KiB
Python
94 lines
2.9 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
|
|
|
|
|
|
class SaveAsTreeRequest(BaseModel):
|
|
"""Request to save a session as a tree."""
|
|
tree_name: Optional[str] = Field(None, max_length=255, description="Custom name for the saved tree (auto-generated if not provided)")
|
|
description: Optional[str] = Field(None, description="Description for the saved tree")
|
|
status: Literal["draft", "published"] = Field("draft", description="Status of the saved tree")
|
|
|
|
|
|
class SaveAsTreeResponse(BaseModel):
|
|
"""Response after saving a session as a tree."""
|
|
tree_id: UUID
|
|
tree_name: str
|
|
message: str
|