OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-03 06:45:15 -07:00
parent d394f8b7be
commit 15a3c6b171
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
backend/open_webui/apps/webui/models/files.py
View File

@ -23,7 +23,7 @@ class File(Base):
hash = Column(Text, nullable=True) hash = Column(Text, nullable=True)
filename = Column(Text) filename = Column(Text)
data = Column(JSON) data = Column(JSON, nullable=True)
meta = Column(JSONField) meta = Column(JSONField)
created_at = Column(BigInteger) created_at = Column(BigInteger)
@ -38,7 +38,7 @@ class FileModel(BaseModel):
hash: Optional[str] = None hash: Optional[str] = None
filename: str filename: str
data: dict data: Optional[dict] = None
meta: dict meta: dict
created_at: int # timestamp in epoch created_at: int # timestamp in epoch
@ -56,7 +56,7 @@ class FileModelResponse(BaseModel):
hash: Optional[str] = None hash: Optional[str] = None
filename: str filename: str
data: dict data: Optional[dict] = None
meta: dict meta: dict
created_at: int # timestamp in epoch created_at: int # timestamp in epoch
@ -67,6 +67,7 @@ class FileForm(BaseModel):
id: str id: str
hash: Optional[str] = None hash: Optional[str] = None
filename: str filename: str
data: dict = {}
meta: dict = {} meta: dict = {}
@ -136,20 +137,19 @@ class FilesTable:
with get_db() as db: with get_db() as db:
try: try:
file = db.query(File).filter_by(id=id).first() file = db.query(File).filter_by(id=id).first()
file.data = {**file.data, **data} file.data = {**(file.data if file.data else {}), **data}
db.commit() db.commit()
return FileModel.model_validate(file) return FileModel.model_validate(file)
except Exception: except Exception as e:
return None return None
def update_file_metadata_by_id(self, id: str, meta: dict) -> Optional[FileModel]: def update_file_metadata_by_id(self, id: str, meta: dict) -> Optional[FileModel]:
with get_db() as db: with get_db() as db:
try: try:
file = db.query(File).filter_by(id=id).first() file = db.query(File).filter_by(id=id).first()
file.meta = {**file.meta, **meta} file.meta = {**(file.meta if file.meta else {}), **meta}
db.commit() db.commit()
return FileModel.model_validate(file) return FileModel.model_validate(file)
except Exception: except Exception:
return None return None