44432413c2
Implements the complete AI flow builder feature using a guided 4-stage wizard (Foundation → Scaffold → Branch Detail → Review & Assemble). AI assists at bounded points using Claude Haiku for cost-efficient structured JSON generation (~$0.01-0.03/flow). Backend: new models (ai_conversations, ai_usage), Alembic migration, quota enforcement with billing anchor, Anthropic API integration with prompt caching, tree validation, conversation CRUD with 24h TTL, APScheduler cleanup job, 5 API endpoints, Pydantic schemas. Frontend: TypeScript types, API client, Zustand store for wizard state, 7 components (modal, step indicator, foundation form, branch selector, branch detail view, tree preview, quota display), MyTreesPage integration with "Build with AI" button (hidden when AI not configured). Tests: 14 validator unit tests + 11 endpoint integration tests with mocked Anthropic (zero real API spend). All 25 tests passing. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
from .user import User
|
|
from .team import Team
|
|
from .account import Account
|
|
from .subscription import Subscription
|
|
from .plan_limits import PlanLimits
|
|
from .account_invite import AccountInvite
|
|
from .tree import Tree
|
|
from .tree_share import TreeShare
|
|
from .session import Session
|
|
from .attachment import Attachment
|
|
from .invite_code import InviteCode
|
|
from .category import TreeCategory
|
|
from .tag import TreeTag, tree_tag_assignments
|
|
from .folder import UserFolder, user_folder_trees
|
|
from .step_category import StepCategory
|
|
from .step_library import StepLibrary, StepRating, StepUsageLog
|
|
from .refresh_token import RefreshToken
|
|
from .audit_log import AuditLog
|
|
from .password_reset_token import PasswordResetToken
|
|
from .session_share import SessionShare, SessionShareView
|
|
from .session_rating import SessionRating
|
|
from .account_limit_override import AccountLimitOverride
|
|
from .feature_flag import FeatureFlag, PlanFeatureDefault, AccountFeatureOverride
|
|
from .platform_setting import PlatformSetting
|
|
from .user_pinned_tree import UserPinnedTree
|
|
from .target_list import TargetList
|
|
from .maintenance_schedule import MaintenanceSchedule
|
|
from .feedback import Feedback
|
|
from .ai_conversation import AIConversation
|
|
from .ai_usage import AIUsage
|
|
|
|
__all__ = [
|
|
"User",
|
|
"Team",
|
|
"Account",
|
|
"Subscription",
|
|
"PlanLimits",
|
|
"AccountInvite",
|
|
"Tree",
|
|
"TreeShare",
|
|
"Session",
|
|
"Attachment",
|
|
"InviteCode",
|
|
"TreeCategory",
|
|
"TreeTag",
|
|
"tree_tag_assignments",
|
|
"UserFolder",
|
|
"user_folder_trees",
|
|
"StepCategory",
|
|
"StepLibrary",
|
|
"StepRating",
|
|
"StepUsageLog",
|
|
"RefreshToken",
|
|
"AuditLog",
|
|
"PasswordResetToken",
|
|
"SessionShare",
|
|
"SessionShareView",
|
|
"SessionRating",
|
|
"AccountLimitOverride",
|
|
"FeatureFlag",
|
|
"PlanFeatureDefault",
|
|
"AccountFeatureOverride",
|
|
"PlatformSetting",
|
|
"UserPinnedTree",
|
|
"TargetList",
|
|
"MaintenanceSchedule",
|
|
"Feedback",
|
|
"AIConversation",
|
|
"AIUsage",
|
|
]
|