Remove graphql logs

Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>
Significant-Gravitas · Aug 6, 2023 · afadd84 · afadd84
1 parent fa63f63
commit afadd84
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/agbenchmark/agent_interface.py b/agbenchmark/agent_interface.py
@@ -14,6 +14,10 @@
 
 mock_test_str = os.getenv("MOCK_TEST")
 MOCK_FLAG = mock_test_str.lower() == "true" if mock_test_str else False
+helicone_graphql_logs = os.getenv("HELICONE_GRAPHQL_LOGS")
+HELICONE_GRAPHQL_LOGS = (
+    helicone_graphql_logs.lower() == "true" if helicone_graphql_logs else False
+)
 
 
 def run_agent(

diff --git a/agbenchmark/utils/get_data_from_helicone.py b/agbenchmark/utils/get_data_from_helicone.py
@@ -4,6 +4,7 @@
 
 import requests
 
+from agbenchmark.agent_interface import HELICONE_GRAPHQL_LOGS
 from agbenchmark.start_benchmark import BENCHMARK_START_TIME
 
 
@@ -22,7 +23,6 @@ def get_data_from_helicone(challenge: str) -> Optional[float]:
   }
 }
 """
-    print(query)
 
     variables = {
         "properties": [
@@ -37,8 +37,9 @@ def get_data_from_helicone(challenge: str) -> Optional[float]:
             {"value": {"equals": challenge}, "name": "challenge"},
         ]
     }
-
-    print(json.dumps(variables, indent=4))
+    if HELICONE_GRAPHQL_LOGS:
+        print(query)
+        print(json.dumps(variables, indent=4))
 
     operation_name = "ExampleQuery"