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

fix: read file encoding error (#1902)

Browse source 
Co-authored-by: maple <1071520@gi>
This commit is contained in:
newsouther 2024-01-04 12:52:10 +08:00 • committed by GitHub
commit 0e239a4f71
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -147,13 +147,13 @@ class AIModel(ABC):
# read _position.yaml file
position_map = {}
if os.path.exists(position_file_path):
with open(position_file_path, 'r') as f:
with open(position_file_path, 'r', encoding='utf-8') as f:
position_map = yaml.safe_load(f)
# traverse all model_schema_yaml_paths
for model_schema_yaml_path in model_schema_yaml_paths:
# read yaml data from yaml file
with open(model_schema_yaml_path, 'r') as f:
with open(model_schema_yaml_path, 'r', encoding='utf-8') as f:
yaml_data = yaml.safe_load(f)
new_parameter_rules = []

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

@ -47,7 +47,7 @@ class ModelProvider(ABC):
yaml_path = os.path.join(current_path, f'{provider_name}.yaml')
yaml_data = {}
if os.path.exists(yaml_path):
with open(yaml_path, 'r') as f:
with open(yaml_path, 'r', encoding='utf-8') as f:
yaml_data = yaml.safe_load(f)
try:

2
api/core/model_runtime/model_providers/model_provider_factory.py
View file

@ -212,7 +212,7 @@ class ModelProviderFactory:
# read _position.yaml file
position_map = {}
if os.path.exists(position_file_path):
with open(position_file_path, 'r') as f:
with open(position_file_path, 'r', encoding='utf-8') as f:
position_map = yaml.safe_load(f)
# traverse all model_provider_dir_paths