Remove graphql logs (#264)

Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>

agbenchmark/agent_interface.py

@@ -14,6 +14,10 @@ load_dotenv()
 
 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(

agbenchmark/utils/get_data_from_helicone.py

@@ -4,6 +4,7 @@ from typing import Optional
 
 import requests
 
+from agbenchmark.agent_interface import HELICONE_GRAPHQL_LOGS
 from agbenchmark.start_benchmark import BENCHMARK_START_TIME
 
 
@@ -22,7 +23,6 @@ query ExampleQuery($properties: [PropertyFilter!]){
   }
 }
 """
-    print(query)
 
     variables = {
         "properties": [
@@ -37,8 +37,9 @@ query ExampleQuery($properties: [PropertyFilter!]){
             {"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"