fixed tooltip naming issue for unverified agents

utils/db.py

@@ -540,8 +540,9 @@ class TracePreprocessor:
         if 'Total Cost' in results_df.columns:
             results_df = results_df.drop('Total Cost', axis=1)
             
+        print(results_df['Agent Name'])
         # create temp column that is whatever is in agent_name [x] because of url we added to agent_name
-        results_df['agent_name_temp'] = results_df['Agent Name'].apply(lambda x: x.split('[')[1].split(']')[0])
+        results_df['agent_name_temp'] = results_df['Agent Name'].apply(lambda x: x.split('[')[1].split(']')[0] if '[' in x else x)
         
         # Update costs in results
         results_df = results_df.merge(agent_costs, on='agent_name_temp', how='left')

utils/viz.py

@@ -205,7 +205,7 @@ def create_scatter_plot(df, x: str, y: str, x_label: str = None, y_label: str =
         agent_data = df[df[hover_data[0]] == agent]
         
         # remove url from tooltip name
-        agent_data.loc[:, 'Agent Name'] = agent_data['Agent Name'].str.rsplit(']').str[0].str[1:]
+        agent_data.loc[:, 'Agent Name'] = agent_data['Agent Name'].str.rsplit(']').str[0].str[1:] if '[' in agent_data['Agent Name'] else agent_data['Agent Name']
 
         x_value = [np.mean(agent_data[x].values)]
         y_value = [np.mean(agent_data[y].values)]