Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 11:57:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac: search chat postgres

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-11-21 18:31:03 -05:00
parent 8f2812d394
commit b32f7815b8
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
backend/open_webui/models/chats.py
View File

@@ -794,18 +794,19 @@ class ChatTable:
elif dialect_name == "postgresql":
# PostgreSQL doesn't allow null bytes in text. We filter those out by checking
# the JSON representation for \u0000 before attempting text extraction
postgres_content_sql = (
"EXISTS ("
" SELECT 1 "
" FROM json_array_elements(Chat.chat->'messages') AS message "
" WHERE message->'content' IS NOT NULL "
" AND (message->'content')::text NOT LIKE '%\\u0000%' "
" AND LOWER(message->>'content') LIKE '%' || :content_key || '%'"
")"
)
postgres_content_sql = """
EXISTS (
SELECT 1
FROM json_array_elements(Chat.chat->'messages') AS message
WHERE message->>'content' IS NOT NULL
AND LOWER(replace(message->>'content', E'\\x00', '')) LIKE '%' || :content_key || '%'
)
"""
postgres_content_clause = text(postgres_content_sql)
# Also filter out chats with null bytes in title
query = query.filter(text("Chat.title::text NOT LIKE '%\\x00%'"))
query = query.filter(
text("replace(Chat.title, E'\\x00', '') ILIKE :title_key")
)
query = query.filter(
or_(
Chat.title.ilike(bindparam("title_key")),