From db49e8de152c69e5e5d2f12a3c8f5c0bce6c5099 Mon Sep 17 00:00:00 2001
From: Silen Naihin <silen.naihin@gmail.com>
Date: Mon, 31 Jul 2023 13:26:49 +0100
Subject: [PATCH] helicone push 2

---
 agbenchmark/utils/get_data_from_helicone.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/agbenchmark/utils/get_data_from_helicone.py b/agbenchmark/utils/get_data_from_helicone.py
index db823a58..f6e43e56 100644
--- a/agbenchmark/utils/get_data_from_helicone.py
+++ b/agbenchmark/utils/get_data_from_helicone.py
@@ -60,6 +60,8 @@ query ExampleQuery($properties: [PropertyFilter!]){
     except Exception as err:
         print(f"Other error occurred: {err}")
     else:
+        if not data:
+            raise Exception("No data returned from Helicone")
         return (
             data.get("data", {}).get("aggregatedHeliconeRequest", {}).get("cost", None)
         )