浏览代码

fix(api): add missing INNER_API_KEY to InnerAPIConfig (#19166)

tags/1.4.0
Yeuoly 5 个月前
父节点
当前提交
8ac3a223a8
没有帐户链接到提交者的电子邮件

+ 5
- 0
api/configs/feature/__init__.py 查看文件

@@ -398,6 +398,11 @@ class InnerAPIConfig(BaseSettings):
default=False,
)

INNER_API_KEY: Optional[str] = Field(
description="API key for accessing the internal API",
default=None,
)


class LoggingConfig(BaseSettings):
"""

+ 1
- 1
api/controllers/inner_api/wraps.py 查看文件

@@ -18,7 +18,7 @@ def enterprise_inner_api_only(view):

# get header 'X-Inner-Api-Key'
inner_api_key = request.headers.get("X-Inner-Api-Key")
if not inner_api_key or inner_api_key != dify_config.INNER_API_KEY_FOR_PLUGIN:
if not inner_api_key or inner_api_key != dify_config.INNER_API_KEY:
abort(401)

return view(*args, **kwargs)

+ 6
- 0
api/tests/unit_tests/configs/test_dify_config.py 查看文件

@@ -100,3 +100,9 @@ def test_flask_configs(example_env_file):

assert str(config["CODE_EXECUTION_ENDPOINT"]) == "http://sandbox:8194/"
assert str(URL(str(config["CODE_EXECUTION_ENDPOINT"])) / "v1") == "http://sandbox:8194/v1"


def test_inner_api_config_exist():
config = DifyConfig()
assert config.INNER_API is False
assert config.INNER_API_KEY is None

正在加载...
取消
保存