OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-16 19:43:42 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-04-12 15:10:43 -07:00
parent 6862d8fba8
commit bdef1001ac
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/models/memories.py
View File

@ -63,14 +63,15 @@ class MemoriesTable:
else: else:
return None return None
def update_memory_by_id( def update_memory_by_id_and_user_id(
self, self,
id: str, id: str,
user_id: str,
content: str, content: str,
) -> Optional[MemoryModel]: ) -> Optional[MemoryModel]:
with get_db() as db: with get_db() as db:
try: try:
db.query(Memory).filter_by(id=id).update( db.query(Memory).filter_by(id=id, user_id=user_id).update(
{"content": content, "updated_at": int(time.time())} {"content": content, "updated_at": int(time.time())}
) )
db.commit() db.commit()

4
backend/open_webui/routers/memories.py
View File

@ -153,7 +153,9 @@ async def update_memory_by_id(
form_data: MemoryUpdateModel, form_data: MemoryUpdateModel,
user=Depends(get_verified_user), user=Depends(get_verified_user),
): ):
memory = Memories.update_memory_by_id(memory_id, form_data.content) memory = Memories.update_memory_by_id_and_user_id(
memory_id, user.id, form_data.content
)
if memory is None: if memory is None:
raise HTTPException(status_code=404, detail="Memory not found") raise HTTPException(status_code=404, detail="Memory not found")