From 343d21d734a2f23fbdfae97fe2f88222176633a8 Mon Sep 17 00:00:00 2001
From: Tiger Ren <tigeren@live.com>
Date: Mon, 19 Aug 2024 13:43:50 +0800
Subject: [PATCH] add time stamp for each LLM call

---
 analysis_pipeline.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/analysis_pipeline.py b/analysis_pipeline.py
index 293a087..3e4b0dd 100644
--- a/analysis_pipeline.py
+++ b/analysis_pipeline.py
@@ -1,5 +1,6 @@
 import pandas as pd
 from AgentProxy import AgentProxy
+from datetime import datetime  # Add this import
 
 # Here you need to replace the API Key and API Secret with your，I provide a test key and secret here
 api_key = '25bda2c39c0f8ca0' 
@@ -42,16 +43,14 @@ sales_stage_definition = '''
 '''
 
 # Read the Excel file
-df = pd.read_excel('output_top20.xlsx')
+df = pd.read_excel('output_top200.xlsx')
 
 # Iterate over each row in the DataFrame
 for index, row in df.iterrows():
     # Extract the information from the column "当前详细状态及Close节奏"
-    if index > 3:
-        break
     try:
         detailed_status = row['当前详细状态及Close节奏']
-        print(f"Processing row {index}")
+        print(f"Processing row {index} at current time: {datetime.now()}")
         detailed_current_stage = row['Sales stage']
         prompt = (f"某公司当前销售定义为 {sales_stage_definition}, "
         f"当前销售阶段为 {detailed_current_stage}, "