mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
18 lines
502 B
Python
18 lines
502 B
Python
|
from flask_restful import fields
|
||
|
|
|
||
|
|
from core.file import File
|
||
|
|
|
||
|
|
|
||
|
|
class FilesContainedField(fields.Raw):
|
||
|
|
def format(self, value):
|
||
|
|
return self._format_file_object(value)
|
||
|
|
|
||
|
|
def _format_file_object(self, v):
|
||
|
|
if isinstance(v, File):
|
||
|
|
return v.model_dump()
|
||
|
|
if isinstance(v, dict):
|
||
|
|
return {k: self._format_file_object(vv) for k, vv in v.items()}
|
||
|
|
if isinstance(v, list):
|
||
|
|
return [self._format_file_object(vv) for vv in v]
|
||
|
|
return v
|