mirror of
https://github.com/langgenius/dify.git
synced 2025-11-04 04:43:09 +00:00
18 lines
518 B
Python
18 lines
518 B
Python
from flask_restful import fields # type: ignore
|
|
|
|
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
|