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

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-02-02 09:23:46 -06:00
parent b0c5f7b668
commit d938eb0e76
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/migrations/versions/8452d01d26d7_add_chat_message_table.py
+1 -1
View File
@@ -137,7 +137,7 @@ def upgrade() -> None:
try: try:
conn.execute( conn.execute(
sa.insert(chat_message_table).values( sa.insert(chat_message_table).values(
id=message_id, id=f"{chat_id}-{message_id}",
chat_id=chat_id, chat_id=chat_id,
user_id=user_id, user_id=user_id,
role=role, role=role,
backend/open_webui/models/chat_messages.py
+5 -2
View File
@@ -137,7 +137,10 @@ class ChatMessageTable:
now = int(time.time()) now = int(time.time())
timestamp = data.get("timestamp", now) timestamp = data.get("timestamp", now)
existing = db.get(ChatMessage, message_id) # Use composite ID: {chat_id}-{message_id}
composite_id = f"{chat_id}-{message_id}"
existing = db.get(ChatMessage, composite_id)
if existing: if existing:
# Update existing # Update existing
if "role" in data: if "role" in data:
@@ -183,7 +186,7 @@ class ChatMessageTable:
info = data.get("info", {}) info = data.get("info", {})
usage = info.get("usage") if info else None usage = info.get("usage") if info else None
message = ChatMessage( message = ChatMessage(
id=message_id, id=composite_id,
chat_id=chat_id, chat_id=chat_id,
user_id=user_id, user_id=user_id,
role=data.get("role", "user"), role=data.get("role", "user"),