diff --git a/api/app_factory.py b/api/app_factory.py index c0714116a3..124b1c9a35 100644 --- a/api/app_factory.py +++ b/api/app_factory.py @@ -51,6 +51,7 @@ def initialize_extensions(app: DifyApp): ext_storage, ext_timezone, ext_warnings, + ext_swagger ) extensions = [ @@ -74,6 +75,7 @@ def initialize_extensions(app: DifyApp): ext_proxy_fix, ext_blueprints, ext_commands, + ext_swagger ] for ext in extensions: short_name = ext.__name__.split(".")[-1] diff --git a/api/extensions/ext_swagger.py b/api/extensions/ext_swagger.py new file mode 100644 index 0000000000..9d68d971fb --- /dev/null +++ b/api/extensions/ext_swagger.py @@ -0,0 +1,8 @@ +from dify_app import DifyApp + + +def init_app(app: DifyApp): + + from flasgger import Swagger + + Swagger(app) \ No newline at end of file diff --git a/api/poetry.lock b/api/poetry.lock index 1f72270dd4..f8c2241d83 100644 --- a/api/poetry.lock +++ b/api/poetry.lock @@ -2812,7 +2812,7 @@ version = "0.9.7.1" description = "Extract swagger specs from your flask project" optional = false python-versions = "*" -groups = ["tools"] +groups = ["main", "tools"] markers = "python_version == \"3.11\" or python_version >= \"3.12\"" files = [ {file = "flasgger-0.9.7.1.tar.gz", hash = "sha256:ca098e10bfbb12f047acc6299cc70a33851943a746e550d86e65e60d4df245fb"}, @@ -5728,7 +5728,7 @@ version = "3.1.0" description = "A sane and fast Markdown parser with useful plugins and renderers" optional = false python-versions = ">=3.8" -groups = ["tools"] +groups = ["main", "tools"] markers = "python_version == \"3.11\" or python_version >= \"3.12\"" files = [ {file = "mistune-3.1.0-py3-none-any.whl", hash = "sha256:b05198cf6d671b3deba6c87ec6cf0d4eb7b72c524636eddb6dbf13823b52cee1"}, @@ -12247,4 +12247,4 @@ cffi = ["cffi (>=1.11)"] [metadata] lock-version = "2.1" python-versions = ">=3.11,<3.13" -content-hash = "a8fff72e974a1bd5c28f4ae326d120410a5628ad0bc65d87adca4e943130ec8f" +content-hash = "b26521378e2b5423ccebc7b1af0890275ddbc3dbcea748de74b83efd1c06e877" diff --git a/api/pyproject.toml b/api/pyproject.toml index 8c4d5fd283..15cf6a5f30 100644 --- a/api/pyproject.toml +++ b/api/pyproject.toml @@ -104,6 +104,7 @@ zhipuai = "~2.1.5" # Related transparent dependencies with pinned version # required by main implementations ############################################################ +flasgger = "^0.9.7.1" [tool.poetry.group.indirect.dependencies] kaleido = "0.2.1" rank-bm25 = "~0.2.2"