|
2 | 2 | import pandas as pd
|
3 | 3 | import json
|
4 | 4 | import sys
|
| 5 | +import argparse |
| 6 | +from datetime import datetime |
5 | 7 | from pathlib import Path
|
6 | 8 |
|
7 | 9 | # Add parent directory to path to import modules
|
8 | 10 | sys.path.append(str(Path(__file__).parent.parent))
|
9 | 11 |
|
10 | 12 | from agents_mcp_usage.evaluations.mermaid_evals.dashboard_config import DEFAULT_CONFIG
|
11 | 13 | from agents_mcp_usage.evaluations.mermaid_evals.schemas import DashboardConfig
|
| 14 | +from agents_mcp_usage.utils import get_project_root |
12 | 15 |
|
13 | 16 | def parse_metric_details(metric_details_str):
|
14 | 17 | """Safely parse JSON string from Metric_details column."""
|
@@ -79,10 +82,14 @@ def process_csv_for_static_site(csv_path):
|
79 | 82 | def extract_provider(model_name):
|
80 | 83 | if model_name.startswith("gemini-"):
|
81 | 84 | return "Google"
|
| 85 | + elif "nova" in model_name.lower(): |
| 86 | + return "Amazon" |
82 | 87 | elif "claude" in model_name.lower():
|
83 | 88 | return "Anthropic"
|
84 | 89 | elif "gpt" in model_name.lower():
|
85 | 90 | return "OpenAI"
|
| 91 | + elif model_name.startswith("o"): |
| 92 | + return "OpenAI" |
86 | 93 | else:
|
87 | 94 | return "Other"
|
88 | 95 |
|
@@ -158,8 +165,29 @@ def extract_provider(model_name):
|
158 | 165 | return output_data
|
159 | 166 |
|
160 | 167 | def main():
|
161 |
| - csv_path = "/home/ubuntu/projects/agents-mcp-usage/mermaid_eval_results/Jun_gemini_results.csv" |
162 |
| - output_path = "/home/ubuntu/projects/agents-mcp-usage/agents_mcp_usage/evaluations/mermaid_evals/results/Jun_gemini_results_processed.json" |
| 168 | + parser = argparse.ArgumentParser(description="Process CSV evaluation results for static site") |
| 169 | + parser.add_argument("-i", "--input_csv", nargs="?", help="Path to input CSV file", default=None) |
| 170 | + parser.add_argument("-o", "--output_json", nargs="?", help="Path to output JSON file", default=None) |
| 171 | + |
| 172 | + args = parser.parse_args() |
| 173 | + |
| 174 | + project_root = get_project_root() |
| 175 | + current_month = datetime.now().strftime("%b").lower() |
| 176 | + |
| 177 | + # Set default paths if not provided |
| 178 | + if args.input_csv: |
| 179 | + csv_path = Path(args.input_csv) |
| 180 | + if not csv_path.is_absolute(): |
| 181 | + csv_path = project_root / csv_path |
| 182 | + else: |
| 183 | + csv_path = project_root / "mermaid_eval_results" / "latest_combined_results.csv" |
| 184 | + |
| 185 | + if args.output_json: |
| 186 | + output_path = Path(args.output_json) |
| 187 | + if not output_path.is_absolute(): |
| 188 | + output_path = project_root / output_path |
| 189 | + else: |
| 190 | + output_path = project_root / "agents_mcp_usage" / "evaluations" / "mermaid_evals" / "results" / f"{current_month}_results_processed.json" |
163 | 191 |
|
164 | 192 | print(f"Processing {csv_path}...")
|
165 | 193 | data = process_csv_for_static_site(csv_path)
|
|
0 commit comments