Forráskód Böngészése

Extend FileMetadataResponse to have hash field

Stoyan Zlatev 4 hónapja
szülő
commit
f0dfdb30c3
1 módosított fájl, 4 hozzáadás és 1 törlés
  1. 4 1
      backend/open_webui/models/files.py

+ 4 - 1
backend/open_webui/models/files.py

@@ -82,6 +82,7 @@ class FileModelResponse(BaseModel):
 
 
 class FileMetadataResponse(BaseModel):
 class FileMetadataResponse(BaseModel):
     id: str
     id: str
+    hash: Optional[str] = None
     meta: dict
     meta: dict
     created_at: int  # timestamp in epoch
     created_at: int  # timestamp in epoch
     updated_at: int  # timestamp in epoch
     updated_at: int  # timestamp in epoch
@@ -147,6 +148,7 @@ class FilesTable:
                 file = db.get(File, id)
                 file = db.get(File, id)
                 return FileMetadataResponse(
                 return FileMetadataResponse(
                     id=file.id,
                     id=file.id,
+                    hash=file.hash,
                     meta=file.meta,
                     meta=file.meta,
                     created_at=file.created_at,
                     created_at=file.created_at,
                     updated_at=file.updated_at,
                     updated_at=file.updated_at,
@@ -182,12 +184,13 @@ class FilesTable:
             return [
             return [
                 FileMetadataResponse(
                 FileMetadataResponse(
                     id=file.id,
                     id=file.id,
+                    hash=file.hash,
                     meta=file.meta,
                     meta=file.meta,
                     created_at=file.created_at,
                     created_at=file.created_at,
                     updated_at=file.updated_at,
                     updated_at=file.updated_at,
                 )
                 )
                 for file in db.query(
                 for file in db.query(
-                    File.id, File.meta, File.created_at, File.updated_at
+                    File.id, File.hash, File.meta, File.created_at, File.updated_at
                 )
                 )
                 .filter(File.id.in_(ids))
                 .filter(File.id.in_(ids))
                 .order_by(File.updated_at.desc())
                 .order_by(File.updated_at.desc())