update

GraphRouter_eval/run_exp.py

@@ -2,6 +2,9 @@ import yaml
 import argparse
 import sys
 import os
+from dotenv import load_dotenv
+load_dotenv()
+
 sys.path.append(os.path.join(os.path.dirname(__file__), 'model'))
 from model.multi_task_graph_router import graph_router_prediction
 import pandas as pd
@@ -14,10 +17,12 @@ parser.add_argument("--query", type=str, default="What is the derivative of f(x)
                     help="Input query to process")
 args = parser.parse_args()
 
-# Initialize OpenAI client for NVIDIA API
+# Load API key from environment variable
+NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY", "")
+
 client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
-    api_key="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",
+    api_key=NVIDIA_API_KEY,
     timeout=60,
     max_retries=2
 )