youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-02 08:28:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: credentials schemas to array

Browse Source
This commit is contained in:
Yeuoly
2024-09-30 17:39:13 +08:00
parent c9f80b46a1
commit 6dfc31a542
15 changed files with 55 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
api/core/tools/utils/configuration.py
View File

@ -1,4 +1,3 @@
from collections.abc import Mapping
from copy import deepcopy
from typing import Any
@ -17,7 +16,7 @@ from core.tools.entities.tool_entities import (
class ProviderConfigEncrypter(BaseModel):
tenant_id: str
config: Mapping[str, BasicProviderConfig]
config: list[BasicProviderConfig]
provider_type: str
provider_identity: str
@ -36,7 +35,10 @@ class ProviderConfigEncrypter(BaseModel):
data = self._deep_copy(data)
# get fields need to be decrypted
fields = self.config
fields = dict[str, BasicProviderConfig]()
for credential in self.config:
fields[credential.name] = credential
for field_name, field in fields.items():
if field.type == BasicProviderConfig.Type.SECRET_INPUT:
if field_name in data:
@ -54,7 +56,10 @@ class ProviderConfigEncrypter(BaseModel):
data = self._deep_copy(data)
# get fields need to be decrypted
fields = self.config
fields = dict[str, BasicProviderConfig]()
for credential in self.config:
fields[credential.name] = credential
for field_name, field in fields.items():
if field.type == BasicProviderConfig.Type.SECRET_INPUT:
if field_name in data:
@ -83,7 +88,10 @@ class ProviderConfigEncrypter(BaseModel):
return cached_credentials
data = self._deep_copy(data)
# get fields need to be decrypted
fields = self.config
fields = dict[str, BasicProviderConfig]()
for credential in self.config:
fields[credential.name] = credential
for field_name, field in fields.items():
if field.type == BasicProviderConfig.Type.SECRET_INPUT:
if field_name in data: