mirror of
https://github.com/langgenius/dify.git
synced 2026-06-11 02:31:13 +08:00
21 lines
546 B
Python
21 lines
546 B
Python
from typing import override
|
|
|
|
from flask_restx import fields
|
|
|
|
from graphon.file import File
|
|
|
|
|
|
class FilesContainedField(fields.Raw):
|
|
@override
|
|
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
|