Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions 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
backend/open_webui/models/chats.py
+11 -10
View File
@@ -794,18 +794,19 @@ class ChatTable:
elif dialect_name == "postgresql": elif dialect_name == "postgresql":
# PostgreSQL doesn't allow null bytes in text. We filter those out by checking # PostgreSQL doesn't allow null bytes in text. We filter those out by checking
# the JSON representation for \u0000 before attempting text extraction # the JSON representation for \u0000 before attempting text extraction
postgres_content_sql = ( postgres_content_sql = """
"EXISTS (" EXISTS (
" SELECT 1 " SELECT 1
" FROM json_array_elements(Chat.chat->'messages') AS message " FROM json_array_elements(Chat.chat->'messages') AS message
" WHERE message->'content' IS NOT NULL " WHERE message->>'content' IS NOT NULL
" AND (message->'content')::text NOT LIKE '%\\u0000%' " AND LOWER(replace(message->>'content', E'\\x00', '')) LIKE '%' || :content_key || '%'
" AND LOWER(message->>'content') LIKE '%' || :content_key || '%'" )
")" """
)
postgres_content_clause = text(postgres_content_sql) postgres_content_clause = text(postgres_content_sql)
# Also filter out chats with null bytes in title # 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( query = query.filter(
or_( or_(
Chat.title.ilike(bindparam("title_key")), Chat.title.ilike(bindparam("title_key")),