Fix cost incorrect

Significant-Gravitas · Jul 31, 2023 · ed91353 · ed91353
1 parent dfda7c3
commit ed91353
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agbenchmark/utils/get_data_from_helicone.py b/agbenchmark/utils/get_data_from_helicone.py
@@ -22,6 +22,7 @@ def get_data_from_helicone(challenge: str) -> Optional[float]:
   }
 }
 """
+    print(query)
 
     variables = {
         "filters": [
@@ -40,6 +41,7 @@ def get_data_from_helicone(challenge: str) -> Optional[float]:
             {"property": {"value": {"equals": challenge}, "name": "challenge"}},
         ]
     }
+    print(json.dumps(variables, indent=4))
 
     operation_name = "ExampleQuery"
 
@@ -75,7 +77,7 @@ def get_data_from_helicone(challenge: str) -> Optional[float]:
     print("this is the data!", data)
     try:
         return (
-            data.get("data", {}).get("aggregatedHeliconeRequest", {}).get("cost", None)
+            data.get("data", {}).get("aggregatedHeliconeRequest", {}).get("costUSD", None)
         )
     except Exception as err:
         print(f"Error occurred: {err}")