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

refacx

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-12-22 20:28:15 -07:00
parent 5e8f3048f9
commit 0d29f31846
5 changed files with 61 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
backend/open_webui/models/messages.py
View File

@@ -28,8 +28,8 @@ class Message(Base):
data = Column(JSON, nullable=True)
meta = Column(JSON, nullable=True)
created_at = Column(BigInteger)
updated_at = Column(BigInteger)
created_at = Column(BigInteger) # time_ns
updated_at = Column(BigInteger) # time_ns
class MessageModel(BaseModel):
@@ -72,8 +72,8 @@ class MessageTable:
"content": form_data.content,
"data": form_data.data,
"meta": form_data.meta,
"created_at": int(time.time()),
"updated_at": int(time.time()),
"created_at": int(time.time_ns()),
"updated_at": int(time.time_ns()),
}
)
@@ -95,9 +95,9 @@ class MessageTable:
all_messages = (
db.query(Message)
.filter_by(channel_id=channel_id)
.order_by(Message.updated_at.asc())
.limit(limit)
.order_by(Message.created_at.desc())
.offset(skip)
.limit(limit)
.all()
)
return [MessageModel.model_validate(message) for message in all_messages]
@@ -109,9 +109,9 @@ class MessageTable:
all_messages = (
db.query(Message)
.filter_by(user_id=user_id)
.order_by(Message.updated_at.asc())
.limit(limit)
.order_by(Message.created_at.desc())
.offset(skip)
.limit(limit)
.all()
)
return [MessageModel.model_validate(message) for message in all_messages]
@@ -124,7 +124,7 @@ class MessageTable:
message.content = form_data.content
message.data = form_data.data
message.meta = form_data.meta
message.updated_at = int(time.time())
message.updated_at = int(time.time_ns())
db.commit()
db.refresh(message)
return MessageModel.model_validate(message) if message else None

2
backend/open_webui/routers/channels.py
View File

@@ -108,7 +108,7 @@ async def post_new_message(
{
"channel_id": channel.id,
"message_id": message.id,
"data": {"message": message.model_dump()},
"data": {"type": "message", "data": message.model_dump()},
},
to=f"channel:{channel.id}",
)