zensol
/
market_assistant
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
analysis_pipeline.py
View File

@ -1,5 +1,6 @@
import pandas as pd import pandas as pd
from AgentProxy import AgentProxy 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 # Here you need to replace the API Key and API Secret with your，I provide a test key and secret here
api_key = '25bda2c39c0f8ca0' api_key = '25bda2c39c0f8ca0'
@ -42,16 +43,14 @@ sales_stage_definition = '''
''' '''
# Read the Excel file # 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 # Iterate over each row in the DataFrame
for index, row in df.iterrows(): for index, row in df.iterrows():
# Extract the information from the column "当前详细状态及Close节奏" # Extract the information from the column "当前详细状态及Close节奏"
if index > 3:
break
try: try:
detailed_status = row['当前详细状态及Close节奏'] 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'] detailed_current_stage = row['Sales stage']
prompt = (f"某公司当前销售定义为 {sales_stage_definition}, " prompt = (f"某公司当前销售定义为 {sales_stage_definition}, "
f"当前销售阶段为 {detailed_current_stage}, " f"当前销售阶段为 {detailed_current_stage}, "