Ver Fonte

fix: read file encoding error (#1902)

Co-authored-by: maple <1071520@gi>
newsouther há 1 ano atrás
pai
commit
0e239a4f71

+ 2 - 2
api/core/extension/extensible.py

@@ -61,7 +61,7 @@ class Extensible:
 
                     builtin_file_path = os.path.join(subdir_path, '__builtin__')
                     if os.path.exists(builtin_file_path):
-                        with open(builtin_file_path, 'r') as f:
+                        with open(builtin_file_path, 'r', encoding='utf-8') as f:
                             position = int(f.read().strip())
 
                 if (extension_name + '.py') not in file_names:
@@ -93,7 +93,7 @@ class Extensible:
                     json_path = os.path.join(subdir_path, 'schema.json')
                     json_data = {}
                     if os.path.exists(json_path):
-                        with open(json_path, 'r') as f:
+                        with open(json_path, 'r', encoding='utf-8') as f:
                             json_data = json.load(f)
 
                 extensions[extension_name] = ModuleExtension(

+ 2 - 2
api/core/model_runtime/model_providers/__base/ai_model.py

@@ -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 = []

+ 1 - 1
api/core/model_runtime/model_providers/__base/model_provider.py

@@ -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:

+ 1 - 1
api/core/model_runtime/model_providers/model_provider_factory.py

@@ -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

+ 1 - 1
api/core/prompt/prompt_transform.py

@@ -207,7 +207,7 @@ class PromptTransform:
 
         json_file_path = os.path.join(prompt_path, f'{prompt_name}.json')
         # Open the JSON file and read its content
-        with open(json_file_path, 'r') as json_file:
+        with open(json_file_path, 'r', encoding='utf-8') as json_file:
             return json.load(json_file)
 
     def _get_simple_chat_app_chat_model_prompt_messages(self, prompt_rules: dict,