add time stamp for each LLM call

This commit is contained in:
Tiger Ren 2024-08-19 13:43:50 +08:00
parent 20984aeae2
commit 343d21d734
1 changed files with 3 additions and 4 deletions

View File

@ -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 yourI 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}, "