소스 검색

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

tags/1.4.0
Yeuoly 6 달 전
부모
커밋
8ac3a223a8
No account linked to committer's email address
3개의 변경된 파일12개의 추가작업 그리고 1개의 파일을 삭제
  1. 5
    0
      api/configs/feature/__init__.py
  2. 1
    1
      api/controllers/inner_api/wraps.py
  3. 6
    0
      api/tests/unit_tests/configs/test_dify_config.py

+ 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

Loading…
취소
저장