Seed default prompts on registration and for existing users

This commit is contained in:
CREDO23 2026-03-31 18:12:09 +02:00
parent 0c975a6f80
commit 11387268a7
2 changed files with 94 additions and 6 deletions

View file

@ -17,6 +17,7 @@ from sqlalchemy import update
from app.config import config
from app.db import (
Prompt,
SearchSpace,
SearchSpaceMembership,
SearchSpaceRole,
@ -25,6 +26,7 @@ from app.db import (
get_default_roles_config,
get_user_db,
)
from app.prompts.system_defaults import SYSTEM_PROMPT_DEFAULTS
from app.utils.refresh_tokens import create_refresh_token
logger = logging.getLogger(__name__)
@ -188,6 +190,18 @@ class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]):
)
session.add(owner_membership)
for default in SYSTEM_PROMPT_DEFAULTS:
session.add(
Prompt(
user_id=user.id,
default_prompt_slug=default["slug"],
name=default["name"],
prompt=default["prompt"],
mode=default["mode"],
version=default["version"],
)
)
await session.commit()
logger.info(
f"Created default search space (ID: {default_search_space.id}) for user {user.id}"