chore: code fmt

Author: alan.cl

packages/dbgpt-serve/src/dbgpt_serve/evaluate/service/benchmark/benchmark_service.py

@@ -91,10 +91,14 @@ def __init__(
         super().__init__(system_app)
         self.rag_service = get_rag_service(system_app)
         self.prompt_service = get_prompt_service(system_app)
-        self._file_parse_type = StorageUtil.get_file_parse_type(STANDARD_BENCHMARK_FILE_PATH)
+        self._file_parse_type = StorageUtil.get_file_parse_type(
+            STANDARD_BENCHMARK_FILE_PATH
+        )
 
         dcs = DataCompareService()
-        self.user_input_execute_service = UserInputExecuteService(dcs, self._file_parse_type)
+        self.user_input_execute_service = UserInputExecuteService(
+            dcs, self._file_parse_type
+        )
 
         self.trigger_executor = ThreadPoolExecutor(
             max_workers=5, thread_name_prefix="benchmark-fileWrite"
@@ -303,7 +307,9 @@ async def run_dataset_benchmark(
         try:
             # read input file
             input_list: List[BaseInputModel] = (
-                self.user_input_execute_service.read_input_file(input_file_path, config.evaluation_env)
+                self.user_input_execute_service.read_input_file(
+                    input_file_path, config.evaluation_env
+                )
             )
 
             for i in range(1, config.round_time + 1):

packages/dbgpt-serve/src/dbgpt_serve/evaluate/service/benchmark/ext/excel_file_parse.py

@@ -3,7 +3,7 @@
 from typing import Dict, List, Optional
 
 import pandas as pd
-from openpyxl import Workbook, load_workbook
+from openpyxl import load_workbook
 
 from dbgpt.util.benchmarks.ExcelUtils import ExcelUtils
 
@@ -12,14 +12,17 @@
     AnswerExecuteModel,
     BaseInputModel,
     BenchmarkDataSets,
-    DataCompareStrategyConfig, EvaluationEnv,
+    DataCompareStrategyConfig,
+    EvaluationEnv,
 )
 
 logger = logging.getLogger(__name__)
 
 
 class ExcelFileParseService(FileParseService):
-    def parse_input_sets(self, path: str, evaluation_env: EvaluationEnv) -> BenchmarkDataSets:
+    def parse_input_sets(
+        self, path: str, evaluation_env: EvaluationEnv
+    ) -> BenchmarkDataSets:
         """
         Parse input sets from excel file
         Args:
@@ -92,7 +95,9 @@ def parse_input_sets(self, path: str, evaluation_env: EvaluationEnv) -> Benchmar
         return input_sets
 
     def parse_standard_benchmark_sets(
-        self, standard_excel_path: str, evaluation_env: EvaluationEnv = EvaluationEnv.DEV
+        self,
+        standard_excel_path: str,
+        evaluation_env: EvaluationEnv = EvaluationEnv.DEV,
     ) -> List[AnswerExecuteModel]:
         df = pd.read_excel(standard_excel_path, sheet_name=0)
         outputs: List[AnswerExecuteModel] = []