mirror of
https://github.com/open-webui/open-webui.git
synced 2026-06-13 19:20:05 +00:00
Timothy Jaeryang Baek
724 lines
29 KiB
Python
724 lines
29 KiB
Python
"""Prompt template models, forms, and database operations."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
import time
|
|
import uuid
|
|
from typing import Optional
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
from open_webui.internal.db import Base, JSONField, get_async_db_context
|
|
from open_webui.models.access_grants import AccessGrantModel, AccessGrants
|
|
from open_webui.models.groups import Groups
|
|
from open_webui.models.prompt_history import PromptHistories
|
|
from open_webui.models.users import User, UserModel, UserResponse, Users
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
from sqlalchemy import JSON, BigInteger, Boolean, Column, String, Text, cast, delete, func, or_, select, text, update
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
|
|
class Prompt(Base): # versioned template
|
|
"""Slash-command prompt with history tracking and access control."""
|
|
|
|
__tablename__ = 'prompt'
|
|
|
|
id = Column(Text, primary_key=True)
|
|
command = Column(String, unique=True, index=True)
|
|
user_id = Column(String, index=True) # owner user id
|
|
name = Column(Text)
|
|
content = Column(Text) # the prompt template body
|
|
data = Column(JSON, nullable=True) # structured prompt parameters
|
|
meta = Column(JSON, nullable=True) # freeform metadata (description, etc.)
|
|
tags = Column(JSON, nullable=True)
|
|
is_active = Column(Boolean, default=True)
|
|
version_id = Column(Text, nullable=True) # Points to active history entry
|
|
created_at = Column(BigInteger, nullable=True)
|
|
updated_at = Column(BigInteger, nullable=True)
|
|
|
|
|
|
class PromptModel(BaseModel):
|
|
id: str | None = None
|
|
command: str
|
|
user_id: str
|
|
name: str
|
|
content: str
|
|
data: dict | None = None
|
|
meta: dict | None = None
|
|
tags: list[str | None] = None
|
|
is_active: bool | None = True
|
|
version_id: str | None = None
|
|
created_at: int | None = None
|
|
updated_at: int | None = None
|
|
access_grants: list[AccessGrantModel] = Field(default_factory=list)
|
|
|
|
model_config = ConfigDict(from_attributes=True) # allows ORM model binding
|
|
|
|
|
|
# --- form / schema definitions ---
|
|
# Forms
|
|
####################
|
|
|
|
|
|
class PromptUserResponse(PromptModel):
|
|
user: UserResponse | None = None
|
|
|
|
|
|
class PromptAccessResponse(PromptUserResponse):
|
|
write_access: bool | None = False
|
|
|
|
|
|
class PromptListResponse(BaseModel):
|
|
items: list[PromptUserResponse]
|
|
total: int
|
|
|
|
|
|
class PromptAccessListResponse(BaseModel):
|
|
items: list[PromptAccessResponse]
|
|
total: int
|
|
|
|
|
|
class PromptForm(BaseModel):
|
|
command: str
|
|
name: str # Changed from title
|
|
content: str
|
|
data: dict | None = None
|
|
meta: dict | None = None
|
|
tags: list[str | None] = None
|
|
access_grants: list[dict | None] = None
|
|
version_id: str | None = None # Active version
|
|
commit_message: str | None = None # For history tracking
|
|
is_production: bool | None = True # Whether to set new version as production
|
|
|
|
|
|
class PromptsTable:
|
|
async def _get_access_grants(self, prompt_id: str, db: AsyncSession | None = None) -> list[AccessGrantModel]:
|
|
return await AccessGrants.get_grants_by_resource('prompt', prompt_id, db=db)
|
|
|
|
async def _to_prompt_model(
|
|
self,
|
|
prompt: Prompt,
|
|
access_grants: list[AccessGrantModel | None] = None,
|
|
db: AsyncSession | None = None,
|
|
) -> PromptModel:
|
|
prompt_data = PromptModel.model_validate(prompt).model_dump(exclude={'access_grants'})
|
|
prompt_data['access_grants'] = (
|
|
access_grants if access_grants is not None else await self._get_access_grants(prompt_data['id'], db=db)
|
|
)
|
|
return PromptModel.model_validate(prompt_data)
|
|
|
|
async def insert_new_prompt(
|
|
self, user_id: str, form_data: PromptForm, db: AsyncSession | None = None
|
|
) -> PromptModel | None:
|
|
now = int(time.time())
|
|
prompt_id = str(uuid.uuid4())
|
|
|
|
async with get_async_db_context(db) as session:
|
|
try:
|
|
record = Prompt(
|
|
id=prompt_id,
|
|
user_id=user_id,
|
|
command=form_data.command,
|
|
name=form_data.name,
|
|
content=form_data.content,
|
|
data=form_data.data or {},
|
|
meta=form_data.meta or {},
|
|
tags=form_data.tags or [],
|
|
is_active=True,
|
|
created_at=now,
|
|
updated_at=now,
|
|
)
|
|
session.add(record)
|
|
await session.commit()
|
|
await session.refresh(record) # populate generated defaults
|
|
|
|
await AccessGrants.set_access_grants(
|
|
'prompt',
|
|
prompt_id,
|
|
form_data.access_grants,
|
|
db=session,
|
|
) # persist sharing rules
|
|
|
|
if not record: # shouldn't happen, but guard anyway
|
|
return None
|
|
|
|
# Build the initial version snapshot.
|
|
grants = await self._get_access_grants(prompt_id, db=session)
|
|
snapshot = {
|
|
'name': form_data.name,
|
|
'content': form_data.content,
|
|
'command': form_data.command,
|
|
'data': form_data.data or {},
|
|
'meta': form_data.meta or {},
|
|
'tags': form_data.tags or [],
|
|
'access_grants': [g.model_dump() for g in grants],
|
|
}
|
|
|
|
history_entry = await PromptHistories.create_history_entry(
|
|
prompt_id=prompt_id,
|
|
snapshot=snapshot,
|
|
user_id=user_id,
|
|
parent_id=None,
|
|
commit_message=form_data.commit_message or 'Initial version',
|
|
db=session,
|
|
) # creates the first version entry
|
|
|
|
# Pin the initial history entry as the production version.
|
|
if history_entry:
|
|
record.version_id = history_entry.id
|
|
await session.commit()
|
|
await session.refresh(record) # re-read version_id
|
|
|
|
return await self._to_prompt_model(record, db=session)
|
|
except Exception as e:
|
|
log.exception('Error creating prompt: %s', e)
|
|
return None
|
|
|
|
async def get_prompt_by_id(self, prompt_id: str, db: AsyncSession | None = None) -> PromptModel | None:
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(
|
|
select(Prompt).filter_by(id=prompt_id),
|
|
)
|
|
prompt = result.scalars().first() # None when not found
|
|
if not prompt:
|
|
return None
|
|
return await self._to_prompt_model(prompt, db=session)
|
|
except Exception: # connection / integrity error
|
|
return
|
|
|
|
async def get_prompt_by_command(self, command: str, db: AsyncSession | None = None) -> PromptModel | None:
|
|
"""Look up a prompt by its unique slash-command string."""
|
|
async with get_async_db_context(db) as session:
|
|
match = (await session.execute(select(Prompt).where(Prompt.command == command))).scalars().first()
|
|
if match is None:
|
|
return
|
|
return await self._to_prompt_model(match, db=session)
|
|
# --- context manager always returns above ---
|
|
return
|
|
|
|
async def get_prompts(self, db: AsyncSession | None = None) -> list[PromptUserResponse]:
|
|
"""Return all active prompts ordered by most recently updated."""
|
|
async with get_async_db_context(db) as session:
|
|
active = (
|
|
(
|
|
await session.execute(
|
|
select(Prompt).where(Prompt.is_active.is_(True)).order_by(Prompt.updated_at.desc())
|
|
)
|
|
)
|
|
.scalars()
|
|
.all()
|
|
)
|
|
|
|
user_ids = list(set(p.user_id for p in active))
|
|
prompt_ids = [p.id for p in active]
|
|
|
|
users = await Users.get_users_by_user_ids(user_ids, db=session) if user_ids else []
|
|
users_dict = {u.id: u for u in users}
|
|
grants_map = await AccessGrants.get_grants_by_resources('prompt', prompt_ids, db=session)
|
|
|
|
prompts = []
|
|
for prompt in active:
|
|
user = users_dict.get(prompt.user_id)
|
|
prompts.append(
|
|
PromptUserResponse.model_validate(
|
|
{
|
|
**(
|
|
await self._to_prompt_model(
|
|
prompt,
|
|
access_grants=grants_map.get(prompt.id, []),
|
|
db=session,
|
|
)
|
|
).model_dump(),
|
|
'user': user.model_dump() if user else None,
|
|
}
|
|
)
|
|
)
|
|
|
|
return prompts
|
|
|
|
async def get_prompts_by_user_id(
|
|
self, user_id: str, permission: str = 'write', db: AsyncSession | None = None
|
|
) -> list[PromptUserResponse]:
|
|
async with get_async_db_context(db) as session:
|
|
user_groups = await Groups.get_groups_by_member_id(user_id, db=session)
|
|
user_group_ids = [group.id for group in user_groups]
|
|
|
|
query = select(Prompt).filter(Prompt.is_active == True).order_by(Prompt.updated_at.desc())
|
|
query = AccessGrants.has_permission_filter(
|
|
db=db,
|
|
query=query,
|
|
DocumentModel=Prompt,
|
|
filter={'user_id': user_id, 'group_ids': user_group_ids},
|
|
resource_type='prompt',
|
|
permission=permission,
|
|
)
|
|
|
|
result = await session.execute(query)
|
|
accessible_prompts = result.scalars().all()
|
|
|
|
if not accessible_prompts:
|
|
return []
|
|
|
|
prompt_ids = [p.id for p in accessible_prompts]
|
|
owner_ids = list({p.user_id for p in accessible_prompts})
|
|
|
|
users = await Users.get_users_by_user_ids(owner_ids, db=session)
|
|
users_dict = {u.id: u for u in users}
|
|
grants_map = await AccessGrants.get_grants_by_resources('prompt', prompt_ids, db=session)
|
|
|
|
results = []
|
|
for prompt in accessible_prompts:
|
|
user = users_dict.get(prompt.user_id)
|
|
results.append(
|
|
PromptUserResponse.model_validate(
|
|
{
|
|
**(
|
|
await self._to_prompt_model(
|
|
prompt,
|
|
access_grants=grants_map.get(prompt.id, []),
|
|
db=db,
|
|
)
|
|
).model_dump(),
|
|
'user': user.model_dump() if user else None,
|
|
}
|
|
)
|
|
)
|
|
return results
|
|
|
|
async def search_prompts(
|
|
self,
|
|
user_id: str,
|
|
filter: dict = {},
|
|
skip: int = 0,
|
|
limit: int = 30,
|
|
db: AsyncSession | None = None,
|
|
) -> PromptListResponse:
|
|
async with get_async_db_context(db) as session:
|
|
# Join with User table for user filtering and sorting
|
|
query = select(Prompt, User).outerjoin(User, User.id == Prompt.user_id)
|
|
|
|
if filter:
|
|
query_key = filter.get('query')
|
|
if query_key:
|
|
query = query.filter(
|
|
or_(
|
|
Prompt.name.ilike(f'%{query_key}%'),
|
|
Prompt.command.ilike(f'%{query_key}%'),
|
|
Prompt.content.ilike(f'%{query_key}%'),
|
|
User.name.ilike(f'%{query_key}%'),
|
|
User.email.ilike(f'%{query_key}%'),
|
|
)
|
|
)
|
|
|
|
view_option = filter.get('view_option')
|
|
if view_option == 'created':
|
|
query = query.filter(Prompt.user_id == user_id)
|
|
elif view_option == 'shared':
|
|
query = query.filter(Prompt.user_id != user_id)
|
|
|
|
# Apply access grant filtering
|
|
query = AccessGrants.has_permission_filter(
|
|
db=db,
|
|
query=query,
|
|
DocumentModel=Prompt,
|
|
filter=filter,
|
|
resource_type='prompt',
|
|
permission='read',
|
|
)
|
|
|
|
tag = filter.get('tag')
|
|
if tag:
|
|
bind = await session.connection()
|
|
dialect_name = bind.dialect.name
|
|
tag_lower = tag.lower()
|
|
|
|
if dialect_name == 'sqlite':
|
|
tag_clause = text(
|
|
'EXISTS (SELECT 1 FROM json_each(prompt.tags) t WHERE LOWER(t.value) = :tag_val)'
|
|
)
|
|
elif dialect_name == 'postgresql':
|
|
tag_clause = text(
|
|
'EXISTS (SELECT 1 FROM json_array_elements_text(prompt.tags) t WHERE LOWER(t) = :tag_val)'
|
|
)
|
|
else:
|
|
# Fallback: LIKE on serialised JSON text (ASCII-safe only)
|
|
tag_clause = func.lower(cast(Prompt.tags, String)).like(
|
|
f'%{json.dumps(tag_lower, ensure_ascii=False)}%'
|
|
)
|
|
tag_lower = None
|
|
|
|
if tag_lower is not None:
|
|
query = query.filter(tag_clause.params(tag_val=tag_lower))
|
|
else:
|
|
query = query.filter(tag_clause)
|
|
|
|
order_by = filter.get('order_by')
|
|
direction = filter.get('direction')
|
|
|
|
if order_by == 'name':
|
|
if direction == 'asc':
|
|
query = query.order_by(Prompt.name.asc())
|
|
else:
|
|
query = query.order_by(Prompt.name.desc())
|
|
elif order_by == 'created_at':
|
|
if direction == 'asc':
|
|
query = query.order_by(Prompt.created_at.asc())
|
|
else:
|
|
query = query.order_by(Prompt.created_at.desc())
|
|
elif order_by == 'updated_at':
|
|
if direction == 'asc':
|
|
query = query.order_by(Prompt.updated_at.asc())
|
|
else:
|
|
query = query.order_by(Prompt.updated_at.desc())
|
|
else:
|
|
query = query.order_by(Prompt.updated_at.desc())
|
|
else:
|
|
query = query.order_by(Prompt.updated_at.desc())
|
|
|
|
# Count BEFORE pagination
|
|
count_result = await session.execute(select(func.count()).select_from(query.subquery()))
|
|
total = count_result.scalar()
|
|
|
|
if skip:
|
|
query = query.offset(skip)
|
|
if limit:
|
|
query = query.limit(limit)
|
|
|
|
result = await session.execute(query)
|
|
items = result.all()
|
|
|
|
prompt_ids = [prompt.id for prompt, _ in items]
|
|
grants_map = await AccessGrants.get_grants_by_resources('prompt', prompt_ids, db=session)
|
|
|
|
prompts = []
|
|
for prompt, user in items:
|
|
prompts.append(
|
|
PromptUserResponse(
|
|
**(
|
|
await self._to_prompt_model(
|
|
prompt,
|
|
access_grants=grants_map.get(prompt.id, []),
|
|
db=db,
|
|
)
|
|
).model_dump(),
|
|
user=(UserResponse(**UserModel.model_validate(user).model_dump()) if user else None),
|
|
)
|
|
)
|
|
|
|
return PromptListResponse(items=prompts, total=total)
|
|
|
|
async def update_prompt_by_command(
|
|
self,
|
|
command: str,
|
|
form_data: PromptForm,
|
|
user_id: str,
|
|
db: AsyncSession | None = None,
|
|
) -> PromptModel | None:
|
|
if not command:
|
|
return None
|
|
try: # database transaction
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt).filter_by(command=command))
|
|
prompt = result.scalars().first()
|
|
if not prompt:
|
|
return None
|
|
|
|
latest_history = await PromptHistories.get_latest_history_entry(prompt.id, db=session)
|
|
parent_id = latest_history.id if latest_history else None
|
|
current_access_grants = await self._get_access_grants(prompt.id, db=session)
|
|
|
|
# Check if content changed to decide on history creation
|
|
content_changed = (
|
|
prompt.name != form_data.name
|
|
or prompt.content != form_data.content
|
|
or form_data.access_grants is not None
|
|
)
|
|
|
|
# Update prompt fields
|
|
prompt.name = form_data.name
|
|
prompt.content = form_data.content
|
|
prompt.data = form_data.data or prompt.data
|
|
prompt.meta = form_data.meta or prompt.meta
|
|
prompt.updated_at = int(time.time())
|
|
if form_data.access_grants is not None:
|
|
await AccessGrants.set_access_grants('prompt', prompt.id, form_data.access_grants, db=session)
|
|
current_access_grants = await self._get_access_grants(prompt.id, db=session)
|
|
|
|
await session.commit()
|
|
|
|
# Create history entry only if content changed
|
|
if content_changed:
|
|
snapshot = {
|
|
'name': form_data.name,
|
|
'content': form_data.content,
|
|
'command': command,
|
|
'data': form_data.data or {},
|
|
'meta': form_data.meta or {},
|
|
'access_grants': [grant.model_dump() for grant in current_access_grants],
|
|
}
|
|
|
|
history_entry = await PromptHistories.create_history_entry(
|
|
prompt_id=prompt.id,
|
|
snapshot=snapshot,
|
|
user_id=user_id,
|
|
parent_id=parent_id,
|
|
commit_message=form_data.commit_message,
|
|
db=db,
|
|
)
|
|
|
|
# Set as production if flag is True (default)
|
|
if form_data.is_production and history_entry:
|
|
prompt.version_id = history_entry.id
|
|
await session.commit()
|
|
|
|
return await self._to_prompt_model(prompt, db=session)
|
|
except Exception:
|
|
return None
|
|
|
|
async def update_prompt_by_id(
|
|
self,
|
|
prompt_id: str,
|
|
form_data: PromptForm,
|
|
user_id: str,
|
|
db: AsyncSession | None = None,
|
|
) -> PromptModel | None:
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt).filter_by(id=prompt_id))
|
|
prompt = result.scalars().first()
|
|
if not prompt:
|
|
return None
|
|
|
|
latest_history = await PromptHistories.get_latest_history_entry(prompt.id, db=session)
|
|
parent_id = latest_history.id if latest_history else None
|
|
current_access_grants = await self._get_access_grants(prompt.id, db=session)
|
|
|
|
# Check if content changed to decide on history creation
|
|
content_changed = (
|
|
prompt.name != form_data.name
|
|
or prompt.command != form_data.command
|
|
or prompt.content != form_data.content
|
|
or form_data.access_grants is not None
|
|
or (form_data.tags is not None and prompt.tags != form_data.tags)
|
|
)
|
|
|
|
# Update prompt fields
|
|
prompt.name = form_data.name
|
|
prompt.command = form_data.command
|
|
prompt.content = form_data.content
|
|
prompt.data = form_data.data or prompt.data
|
|
prompt.meta = form_data.meta or prompt.meta
|
|
|
|
if form_data.tags is not None:
|
|
prompt.tags = form_data.tags
|
|
|
|
if form_data.access_grants is not None:
|
|
await AccessGrants.set_access_grants('prompt', prompt.id, form_data.access_grants, db=session)
|
|
current_access_grants = await self._get_access_grants(prompt.id, db=session)
|
|
|
|
prompt.updated_at = int(time.time())
|
|
|
|
await session.commit()
|
|
|
|
# Create history entry only if content changed
|
|
if content_changed:
|
|
snapshot = {
|
|
'name': form_data.name,
|
|
'content': form_data.content,
|
|
'command': prompt.command,
|
|
'data': form_data.data or {},
|
|
'meta': form_data.meta or {},
|
|
'tags': prompt.tags or [],
|
|
'access_grants': [grant.model_dump() for grant in current_access_grants],
|
|
}
|
|
|
|
history_entry = await PromptHistories.create_history_entry(
|
|
prompt_id=prompt.id,
|
|
snapshot=snapshot,
|
|
user_id=user_id,
|
|
parent_id=parent_id,
|
|
commit_message=form_data.commit_message,
|
|
db=db,
|
|
)
|
|
|
|
# Set as production if flag is True (default)
|
|
if form_data.is_production and history_entry:
|
|
prompt.version_id = history_entry.id
|
|
await session.commit()
|
|
|
|
return await self._to_prompt_model(prompt, db=session)
|
|
except Exception:
|
|
return None
|
|
|
|
async def update_prompt_metadata(
|
|
self,
|
|
prompt_id: str,
|
|
name: str,
|
|
command: str,
|
|
tags: list[str | None] = None,
|
|
db: AsyncSession | None = None,
|
|
) -> PromptModel | None:
|
|
"""Update only name, command, and tags (no history created)."""
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt).filter_by(id=prompt_id))
|
|
prompt = result.scalars().first()
|
|
if not prompt:
|
|
return None
|
|
|
|
prompt.name = name
|
|
prompt.command = command
|
|
|
|
if tags is not None:
|
|
prompt.tags = tags
|
|
|
|
prompt.updated_at = int(time.time())
|
|
await session.commit()
|
|
|
|
return await self._to_prompt_model(prompt, db=session)
|
|
except Exception:
|
|
return None
|
|
|
|
async def update_prompt_version(
|
|
self,
|
|
prompt_id: str,
|
|
version_id: str,
|
|
db: AsyncSession | None = None,
|
|
) -> PromptModel | None:
|
|
"""Set the active version of a prompt and restore content from that version's snapshot."""
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt).filter_by(id=prompt_id))
|
|
prompt = result.scalars().first()
|
|
if not prompt:
|
|
return None
|
|
|
|
history_entry = await PromptHistories.get_history_entry_by_id(version_id, db=session)
|
|
|
|
# Reject a version_id from another prompt; restoring it would copy a foreign snapshot in.
|
|
if not history_entry or history_entry.prompt_id != prompt_id:
|
|
return None
|
|
|
|
# Restore prompt content from the snapshot
|
|
snapshot = history_entry.snapshot
|
|
if snapshot:
|
|
prompt.name = snapshot.get('name', prompt.name)
|
|
prompt.content = snapshot.get('content', prompt.content)
|
|
prompt.data = snapshot.get('data', prompt.data)
|
|
prompt.meta = snapshot.get('meta', prompt.meta)
|
|
prompt.tags = snapshot.get('tags', prompt.tags)
|
|
# Note: command and access_grants are not restored from snapshot
|
|
|
|
prompt.version_id = version_id
|
|
prompt.updated_at = int(time.time())
|
|
await session.commit()
|
|
|
|
return await self._to_prompt_model(prompt, db=session)
|
|
except Exception as e: # connection error
|
|
log.error(f'Failed to restore prompt version: {e}')
|
|
return None # restoration failed
|
|
|
|
async def toggle_prompt_active(
|
|
self,
|
|
prompt_id: str,
|
|
db: AsyncSession | None = None,
|
|
) -> PromptModel | None:
|
|
"""Flip the is_active flag on a prompt."""
|
|
if not prompt_id:
|
|
return None
|
|
try: # activation state toggle
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt).filter_by(id=prompt_id))
|
|
prompt = result.scalars().first()
|
|
if prompt:
|
|
prompt.is_active = not prompt.is_active
|
|
prompt.updated_at = int(time.time())
|
|
await session.commit()
|
|
await session.refresh(prompt)
|
|
return await self._to_prompt_model(prompt, db=session)
|
|
return None
|
|
except Exception:
|
|
return None
|
|
|
|
async def delete_prompt_by_command(self, command: str, db: AsyncSession | None = None) -> bool:
|
|
"""Permanently delete a prompt and its history."""
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt).filter_by(command=command))
|
|
prompt = result.scalars().first()
|
|
if prompt:
|
|
await PromptHistories.delete_history_by_prompt_id(prompt.id, db=session)
|
|
await AccessGrants.revoke_all_access('prompt', prompt.id, db=session)
|
|
|
|
await session.delete(prompt)
|
|
await session.commit()
|
|
return True
|
|
return False
|
|
except Exception:
|
|
return False
|
|
|
|
async def delete_prompt_by_id(self, prompt_id: str, db: AsyncSession | None = None) -> bool:
|
|
"""Permanently delete a prompt and its history."""
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt).filter_by(id=prompt_id))
|
|
prompt = result.scalars().first()
|
|
if prompt:
|
|
await PromptHistories.delete_history_by_prompt_id(prompt.id, db=session)
|
|
await AccessGrants.revoke_all_access('prompt', prompt.id, db=session)
|
|
|
|
await session.delete(prompt)
|
|
await session.commit()
|
|
return True
|
|
return False
|
|
except Exception as err:
|
|
log.error(f'Failed to delete prompt: {err}')
|
|
return False # deletion failed
|
|
|
|
async def get_tags(self, db: AsyncSession | None = None) -> list[str]:
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
result = await session.execute(select(Prompt.tags).filter(Prompt.is_active == True))
|
|
tags = set()
|
|
for (tag_list,) in result.all():
|
|
if tag_list:
|
|
for tag in tag_list:
|
|
if tag:
|
|
tags.add(tag)
|
|
return sorted(list(tags))
|
|
except Exception:
|
|
return []
|
|
|
|
async def get_tags_by_user_id(self, user_id: str, db: AsyncSession | None = None) -> list[str]:
|
|
try:
|
|
async with get_async_db_context(db) as session:
|
|
user_groups = await Groups.get_groups_by_member_id(user_id, db=session)
|
|
user_group_ids = [group.id for group in user_groups]
|
|
|
|
query = select(Prompt.tags).filter(Prompt.is_active == True)
|
|
query = AccessGrants.has_permission_filter(
|
|
db=db,
|
|
query=query,
|
|
DocumentModel=Prompt,
|
|
filter={'user_id': user_id, 'group_ids': user_group_ids},
|
|
resource_type='prompt',
|
|
permission='read',
|
|
)
|
|
|
|
result = await session.execute(query)
|
|
tags = set()
|
|
for (tag_list,) in result.all():
|
|
if tag_list:
|
|
for tag in tag_list:
|
|
if tag:
|
|
tags.add(tag)
|
|
return sorted(list(tags))
|
|
except Exception:
|
|
return []
|
|
|
|
|
|
Prompts = PromptsTable() # singleton prompts registry
|