Files
resolutionflow/backend/alembic/versions/055_add_kb_plan_limits.py
Michael Chihlas 71ff4a8c35 feat: KB Accelerator — convert KB articles into interactive flows
Full-stack implementation of the KB Accelerator feature that converts
static MSP knowledge base articles into interactive troubleshooting
and procedural flows using AI.

Backend:
- Migrations 054/055: kb_imports, kb_import_nodes tables + plan_limits KB columns
- SQLAlchemy models with relationships and self-referential node hierarchy
- Text extraction service (txt, paste, docx with structural metadata)
- AI conversion service with MSP-specialist prompts for both flow types
- 8 API endpoints: upload, get, list, convert, edit node, commit, delete, quota
- Tier-gated access via plan_limits (free: 3 lifetime, pro/team: unlimited)
- 8 integration tests covering upload, get/list, quota, commit, delete

Frontend:
- TypeScript types and API client for all KB Accelerator endpoints
- Multi-step wizard page: upload → processing → review → success
- Upload screen with paste/file tabs, drag-drop, target type selector
- Two-panel review screen with source highlighting and node cards
- Per-node actions: approve, edit, regenerate, insert, delete
- Confidence color indicators (green/amber/red)
- Sidebar navigation with Sparkles icon
- Code-split lazy-loaded route at /kb-accelerator

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-10 20:56:28 -04:00

77 lines
3.1 KiB
Python

"""add KB Accelerator columns to plan_limits
Revision ID: 055
Revises: 054
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import JSONB
revision = "055"
down_revision = "054"
branch_labels = None
depends_on = None
def upgrade():
# Add KB Accelerator columns to plan_limits
op.add_column("plan_limits", sa.Column("kb_accelerator_enabled", sa.Boolean, nullable=False, server_default=sa.text("false")))
op.add_column("plan_limits", sa.Column("kb_max_lifetime_conversions", sa.Integer, nullable=True))
op.add_column("plan_limits", sa.Column("kb_batch_max_size", sa.Integer, nullable=True))
op.add_column("plan_limits", sa.Column("kb_allowed_formats", JSONB, nullable=False, server_default=sa.text("'[\"txt\",\"paste\"]'::jsonb")))
op.add_column("plan_limits", sa.Column("kb_detailed_analysis", sa.Boolean, nullable=False, server_default=sa.text("false")))
op.add_column("plan_limits", sa.Column("kb_conversational_refinement", sa.Boolean, nullable=False, server_default=sa.text("false")))
op.add_column("plan_limits", sa.Column("kb_step_library_matching", sa.Boolean, nullable=False, server_default=sa.text("false")))
op.add_column("plan_limits", sa.Column("kb_history_limit", sa.Integer, nullable=True))
# Seed defaults for each plan tier
op.execute("""
UPDATE plan_limits SET
kb_accelerator_enabled = true,
kb_max_lifetime_conversions = 3,
kb_batch_max_size = NULL,
kb_allowed_formats = '["txt","paste"]'::jsonb,
kb_detailed_analysis = false,
kb_conversational_refinement = false,
kb_step_library_matching = false,
kb_history_limit = 3
WHERE plan = 'free'
""")
op.execute("""
UPDATE plan_limits SET
kb_accelerator_enabled = true,
kb_max_lifetime_conversions = NULL,
kb_batch_max_size = 5,
kb_allowed_formats = '["txt","paste","docx","pdf","html","md"]'::jsonb,
kb_detailed_analysis = true,
kb_conversational_refinement = true,
kb_step_library_matching = true,
kb_history_limit = NULL
WHERE plan = 'pro'
""")
op.execute("""
UPDATE plan_limits SET
kb_accelerator_enabled = true,
kb_max_lifetime_conversions = NULL,
kb_batch_max_size = 10,
kb_allowed_formats = '["txt","paste","docx","pdf","html","md"]'::jsonb,
kb_detailed_analysis = true,
kb_conversational_refinement = true,
kb_step_library_matching = true,
kb_history_limit = NULL
WHERE plan = 'team'
""")
def downgrade():
op.drop_column("plan_limits", "kb_history_limit")
op.drop_column("plan_limits", "kb_step_library_matching")
op.drop_column("plan_limits", "kb_conversational_refinement")
op.drop_column("plan_limits", "kb_detailed_analysis")
op.drop_column("plan_limits", "kb_allowed_formats")
op.drop_column("plan_limits", "kb_batch_max_size")
op.drop_column("plan_limits", "kb_max_lifetime_conversions")
op.drop_column("plan_limits", "kb_accelerator_enabled")