update scripts

src/backend/model_operations.py

@@ -118,7 +118,7 @@ class SummaryGenerator:
         """Generate summaries for a given DataFrame of source docs.
            修改这里拉取模型生成结果
         Args:
-            df (DataFrame): DataFrame containing source docs.
+            dataset (DataFrame): DataFrame containing source docs.
             
         Returns:
             summaries_df (DataFrame): Generated summaries by the model.
@@ -192,7 +192,7 @@ class SummaryGenerator:
                         while True:
                             try:
                                 '''调用'''
-                                print(ID,'-',ii)
+                                print(ID,'-',j,'-',ii)
 
                                 _response = self.generate_summary(system_prompt, _user_prompt)
                                 # print(f"Finish index {index}")
@@ -222,10 +222,10 @@ class SummaryGenerator:
                                         try:
                                             _response = self.generate_summary(system_prompt, _user_prompt)
                                             break
-                                        except Exception as e:
-                                            exceptions.append(e)
+                                        except Exception as ee:
+                                            exceptions.append(ee)
                                             retries += 1
-                                            print(f"Retry {retries}/{max_retries} failed at index {i}: {e}")
+                                            print(f"Retry {retries}/{max_retries} failed at index {i}: {ee}")
                                             if retries >= max_retries:
                                                 exit_outer_loop = True
                                                 break
@@ -466,9 +466,9 @@ class SummaryGenerator:
                             time.sleep(1)  # Optional: Add a small delay before retrying
 
                     return result
-                except Exception as e:
+                except Exception as ee:
                     print(f"Error with TOKEN1: {envs.TOKEN1}")
-                    raise e
+                    raise ee
 
             # except: # fail to call api. run it locally.
             #     self.tokenizer = AutoTokenizer.from_pretrained(self.model_id, trust_remote_code=True)