jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: improve position map conversion and tolerate empty position yaml file (#6541)

Browse source
This commit is contained in:
Bowen Liang 2024-07-29 10:32:11 +08:00 • committed by GitHub
commit 20268708cc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 44 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/model_runtime/model_providers/__base/ai_model.py
View file

@ -162,7 +162,7 @@ class AIModel(ABC):
# traverse all model_schema_yaml_paths
for model_schema_yaml_path in model_schema_yaml_paths:
# read yaml data from yaml file
yaml_data = load_yaml_file(model_schema_yaml_path, ignore_error=True)
yaml_data = load_yaml_file(model_schema_yaml_path)
new_parameter_rules = []
for parameter_rule in yaml_data.get('parameter_rules', []):

2
api/core/model_runtime/model_providers/__base/model_provider.py
View file

@ -44,7 +44,7 @@ class ModelProvider(ABC):
# read provider schema from yaml file
yaml_path = os.path.join(current_path, f'{provider_name}.yaml')
yaml_data = load_yaml_file(yaml_path, ignore_error=True)
yaml_data = load_yaml_file(yaml_path)
try:
# yaml_data to entity