Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: format

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-04-17 14:28:18 +09:00
parent e8e655d0de
commit 4113b15a60
79 changed files with 501 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/models/chats.py
+1 -3
View File
@@ -723,9 +723,7 @@ class ChatTable:
"""Delegate to SharedChats for listing shared chats by user."""
from open_webui.models.shared_chats import SharedChats
return await SharedChats.get_by_user_id(
user_id, filter=filter, skip=skip, limit=limit, db=db
)
return await SharedChats.get_by_user_id(user_id, filter=filter, skip=skip, limit=limit, db=db)
async def get_chat_list_by_user_id(
self,
backend/open_webui/models/prompts.py
+5 -3
View File
@@ -299,15 +299,17 @@ class PromptsTable:
if dialect_name == 'sqlite':
tag_clause = text(
"EXISTS (SELECT 1 FROM json_each(prompt.tags) t WHERE LOWER(t.value) = :tag_val)"
'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)"
'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_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:
backend/open_webui/models/shared_chats.py
+7 -22
View File
@@ -60,9 +60,7 @@ class SharedChatResponse(BaseModel):
class SharedChatsTable:
async def create(
self, chat_id: str, user_id: str, db: Optional[AsyncSession] = None
) -> Optional[SharedChatModel]:
async def create(self, chat_id: str, user_id: str, db: Optional[AsyncSession] = None) -> Optional[SharedChatModel]:
"""
Create a snapshot of the chat for link sharing.
Returns the SharedChatModel with the share token as its id.
@@ -92,9 +90,7 @@ class SharedChatsTable:
return SharedChatModel.model_validate(shared_chat)
async def update(
self, share_id: str, db: Optional[AsyncSession] = None
) -> Optional[SharedChatModel]:
async def update(self, share_id: str, db: Optional[AsyncSession] = None) -> Optional[SharedChatModel]:
"""
Re-snapshot: update the shared chat with the current state of the original chat.
"""
@@ -117,9 +113,7 @@ class SharedChatsTable:
await db.refresh(shared_chat)
return SharedChatModel.model_validate(shared_chat)
async def get_by_id(
self, share_id: str, db: Optional[AsyncSession] = None
) -> Optional[SharedChatModel]:
async def get_by_id(self, share_id: str, db: Optional[AsyncSession] = None) -> Optional[SharedChatModel]:
"""Get a shared chat by its share token."""
async with get_async_db_context(db) as db:
shared_chat = await db.get(SharedChat, share_id)
@@ -127,16 +121,11 @@ class SharedChatsTable:
return SharedChatModel.model_validate(shared_chat)
return None
async def get_by_chat_id(
self, chat_id: str, db: Optional[AsyncSession] = None
) -> Optional[SharedChatModel]:
async def get_by_chat_id(self, chat_id: str, db: Optional[AsyncSession] = None) -> Optional[SharedChatModel]:
"""Get the shared chat for a given original chat. Returns the most recent one."""
async with get_async_db_context(db) as db:
result = await db.execute(
select(SharedChat)
.filter_by(chat_id=chat_id)
.order_by(SharedChat.updated_at.desc())
.limit(1)
select(SharedChat).filter_by(chat_id=chat_id).order_by(SharedChat.updated_at.desc()).limit(1)
)
shared_chat = result.scalars().first()
if shared_chat:
@@ -194,9 +183,7 @@ class SharedChatsTable:
for sc in result.scalars().all()
]
async def delete_by_id(
self, share_id: str, db: Optional[AsyncSession] = None
) -> bool:
async def delete_by_id(self, share_id: str, db: Optional[AsyncSession] = None) -> bool:
"""Delete a shared chat by its share token."""
try:
async with get_async_db_context(db) as db:
@@ -206,9 +193,7 @@ class SharedChatsTable:
except Exception:
return False
async def delete_by_chat_id(
self, chat_id: str, db: Optional[AsyncSession] = None
) -> bool:
async def delete_by_chat_id(self, chat_id: str, db: Optional[AsyncSession] = None) -> bool:
"""Delete all shared chats for a given original chat."""
try:
async with get_async_db_context(db) as db: