Feat: Add Prompt Management System with Versioning

backend/app/alembic/versions/757f50ada8ef_add_prompt_and_version_table.py

@@ -0,0 +1,114 @@
+"""Add prompt and version table
+
+Revision ID: 757f50ada8ef
+Revises: e9dd35eff62c
+Create Date: 2025-08-14 11:45:07.186686
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+
+
+# revision identifiers, used by Alembic.
+revision = "757f50ada8ef"
+down_revision = "e9dd35eff62c"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "prompt",
+        sa.Column("name", sqlmodel.sql.sqltypes.AutoString(length=50), nullable=False),
+        sa.Column(
+            "description", sqlmodel.sql.sqltypes.AutoString(length=500), nullable=True
+        ),
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("active_version", sa.Uuid(), nullable=False),
+        sa.Column("project_id", sa.Integer(), nullable=False),
+        sa.Column("inserted_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=False),
+        sa.Column("is_deleted", sa.Boolean(), nullable=False),
+        sa.Column("deleted_at", sa.DateTime(), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["active_version"],
+            ["prompt_version.id"],
+            initially="DEFERRED",
+            deferrable=True,
+            use_alter=True,
+        ),
+        sa.ForeignKeyConstraint(
+            ["project_id"],
+            ["project.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index(op.f("ix_prompt_name"), "prompt", ["name"], unique=False)
+    op.create_index(
+        op.f("ix_prompt_project_id"), "prompt", ["project_id"], unique=False
+    )
+    op.create_index(
+        "ix_prompt_project_id_is_deleted",
+        "prompt",
+        ["project_id", "is_deleted"],
+        unique=False,
+    )
+    op.create_table(
+        "prompt_version",
+        sa.Column("instruction", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+        sa.Column(
+            "commit_message",
+            sqlmodel.sql.sqltypes.AutoString(length=512),
+            nullable=True,
+        ),
+        sa.Column("id", sa.Uuid(), nullable=False),
+        sa.Column("prompt_id", sa.Uuid(), nullable=False),
+        sa.Column("version", sa.Integer(), nullable=False),
+        sa.Column("inserted_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=False),
+        sa.Column("is_deleted", sa.Boolean(), nullable=False),
+        sa.Column("deleted_at", sa.DateTime(), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["prompt_id"],
+            ["prompt.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("prompt_id", "version"),
+    )
+    op.create_foreign_key(
+        None,
+        "prompt",
+        "prompt_version",
+        ["active_version"],
+        ["id"],
+        initially="DEFERRED",
+        deferrable=True,
+        use_alter=True,
+    )
+    op.create_index(
+        op.f("ix_prompt_version_prompt_id"),
+        "prompt_version",
+        ["prompt_id"],
+        unique=False,
+    )
+    op.create_index(
+        "ix_prompt_version_prompt_id_is_deleted",
+        "prompt_version",
+        ["prompt_id", "is_deleted"],
+        unique=False,
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index("ix_prompt_version_prompt_id_is_deleted", table_name="prompt_version")
+    op.drop_index(op.f("ix_prompt_version_prompt_id"), table_name="prompt_version")
+    op.drop_table("prompt_version")
+    op.drop_index("ix_prompt_project_id_is_deleted", table_name="prompt")
+    op.drop_index(op.f("ix_prompt_project_id"), table_name="prompt")
+    op.drop_index(op.f("ix_prompt_name"), table_name="prompt")
+    op.drop_table("prompt")
+    # ### end Alembic commands ###

backend/app/api/main.py

@@ -10,6 +10,8 @@
     openai_conversation,
     project,
     project_user,
+    prompts,
+    prompt_versions,
     responses,
     private,
     threads,
@@ -32,6 +34,8 @@
 api_router.include_router(organization.router)
 api_router.include_router(project.router)
 api_router.include_router(project_user.router)
+api_router.include_router(prompts.router)
+api_router.include_router(prompt_versions.router)
 api_router.include_router(responses.router)
 api_router.include_router(threads.router)
 api_router.include_router(users.router)

backend/app/api/routes/prompt_versions.py

@@ -0,0 +1,55 @@
+import logging
+from uuid import UUID
+
+from fastapi import APIRouter, Depends
+from sqlmodel import Session
+
+from app.api.deps import CurrentUserOrgProject, get_db
+from app.crud import create_prompt_version, delete_prompt_version
+from app.models import PromptVersionCreate, PromptVersionPublic
+from app.utils import APIResponse
+
+
+logger = logging.getLogger(__name__)
+router = APIRouter(prefix="/prompts", tags=["Prompt Versions"])
+
+
+@router.post(
+    "/{prompt_id}/versions",
+    response_model=APIResponse[PromptVersionPublic],
+    status_code=201,
+)
+def create_prompt_version_route(
+    prompt_version_in: PromptVersionCreate,
+    prompt_id: UUID,
+    current_user: CurrentUserOrgProject,
+    session: Session = Depends(get_db),
+):
+    version = create_prompt_version(
+        session=session,
+        prompt_id=prompt_id,
+        prompt_version_in=prompt_version_in,
+        project_id=current_user.project_id,
+    )
+    return APIResponse.success_response(version)
+
+
+@router.delete("/{prompt_id}/versions/{version_id}", response_model=APIResponse)
+def delete_prompt_version_route(
+    prompt_id: UUID,
+    version_id: UUID,
+    current_user: CurrentUserOrgProject,
+    session: Session = Depends(get_db),
+):
+    """
+    Delete a prompt version by ID.
+    """
+    delete_prompt_version(
+        session=session,
+        prompt_id=prompt_id,
+        version_id=version_id,
+        project_id=current_user.project_id,
+    )
+    return APIResponse.success_response(
+        data={"message": "Prompt version deleted successfully."}
+    )

backend/app/api/routes/prompts.py

@@ -0,0 +1,134 @@
+import logging
+from uuid import UUID
+
+from fastapi import APIRouter, Depends, Path, Query
+from sqlmodel import Session
+
+from app.api.deps import CurrentUserOrgProject, get_db
+from app.crud import (
+    create_prompt,
+    delete_prompt,
+    get_prompt_by_id,
+    get_prompts,
+    count_prompts_in_project,
+    update_prompt,
+)
+from app.models import (
+    PromptCreate,
+    PromptPublic,
+    PromptUpdate,
+    PromptWithVersion,
+    PromptWithVersions,
+)
+from app.utils import APIResponse
+
+
+logger = logging.getLogger(__name__)
+router = APIRouter(prefix="/prompts", tags=["Prompts"])
+
+
+@router.post("/", response_model=APIResponse[PromptWithVersion], status_code=201)
+def create_prompt_route(
+    prompt_in: PromptCreate,
+    current_user: CurrentUserOrgProject,
+    session: Session = Depends(get_db),
+):
+    """
+    Create a new prompt under the specified organization and project.
+    """
+    prompt, version = create_prompt(
+        session=session, prompt_in=prompt_in, project_id=current_user.project_id
+    )
+    prompt_with_version = PromptWithVersion(**prompt.model_dump(), version=version)
+    return APIResponse.success_response(prompt_with_version)
+
+
+@router.get(
+    "/",
+    response_model=APIResponse[list[PromptPublic]],
+)
+def get_prompts_route(
+    current_user: CurrentUserOrgProject,
+    skip: int = Query(
+        0, ge=0, description="Number of prompts to skip (for pagination)."
+    ),
+    limit: int = Query(100, gt=0, description="Maximum number of prompts to return."),
+    session: Session = Depends(get_db),
+):
+    """
+    Get all prompts for the specified organization and project.
+    """
+    prompts = get_prompts(
+        session=session,
+        project_id=current_user.project_id,
+        skip=skip,
+        limit=limit,
+    )
+    total = count_prompts_in_project(
+        session=session, project_id=current_user.project_id
+    )
+    metadata = {"pagination": {"total": total, "skip": skip, "limit": limit}}
+    return APIResponse.success_response(prompts, metadata=metadata)
+
+
+@router.get(
+    "/{prompt_id}",
+    response_model=APIResponse[PromptWithVersions],
+    summary="Get a single prompt by its ID by default returns the active version",
+)
+def get_prompt_by_id_route(
+    current_user: CurrentUserOrgProject,
+    prompt_id: UUID = Path(..., description="The ID of the prompt to fetch"),
+    include_versions: bool = Query(
+        False, description="Whether to include all versions of the prompt."
+    ),
+    session: Session = Depends(get_db),
+):
+    """
+    Get a single prompt by its ID.
+    """
+    prompt, versions = get_prompt_by_id(
+        session=session,
+        prompt_id=prompt_id,
+        project_id=current_user.project_id,
+        include_versions=include_versions,
+    )
+    prompt_with_versions = PromptWithVersions(**prompt.model_dump(), versions=versions)
+    return APIResponse.success_response(prompt_with_versions)
+
+
+@router.patch("/{prompt_id}", response_model=APIResponse[PromptPublic])
+def update_prompt_route(
+    current_user: CurrentUserOrgProject,
+    prompt_update: PromptUpdate,
+    prompt_id: UUID = Path(..., description="The ID of the prompt to Update"),
+    session: Session = Depends(get_db),
+):
+    """
+    Update a prompt's name or description.
+    """
+
+    prompt = update_prompt(
+        session=session,
+        prompt_id=prompt_id,
+        project_id=current_user.project_id,
+        prompt_update=prompt_update,
+    )
+    return APIResponse.success_response(prompt)
+
+
+@router.delete("/{prompt_id}", response_model=APIResponse)
+def delete_prompt_route(
+    current_user: CurrentUserOrgProject,
+    prompt_id: UUID = Path(..., description="The ID of the prompt to delete"),
+    session: Session = Depends(get_db),
+):
+    """
+    Delete a prompt by ID.
+    """
+    delete_prompt(
+        session=session, prompt_id=prompt_id, project_id=current_user.project_id
+    )
+    return APIResponse.success_response(
+        data={"message": "Prompt deleted successfully."}
+    )

backend/app/crud/__init__.py

@@ -64,3 +64,14 @@
     create_conversation,
     delete_conversation,
 )
+
+from .prompt_versions import create_prompt_version, delete_prompt_version
+
+from .prompts import (
+    create_prompt,
+    count_prompts_in_project,
+    delete_prompt,
+    get_prompt_by_id,
+    get_prompts,
+    update_prompt,
+)