Merge pull request #194 from AgentOps-AI/no-test-telemetry

stop test telemetry logging

Author: bboynton97

agentstack/telemetry.py

@@ -21,7 +21,7 @@
 # cool of you to allow telemetry <3
 #
 # - braelyn
-
+import os
 import platform
 import socket
 from typing import Optional
@@ -76,6 +76,9 @@ def collect_machine_telemetry(command: str):
 
 
 def track_cli_command(command: str, args: Optional[str] = None):
+    if bool(os.environ['AGENTSTATCK_IS_TEST_ENV']):
+        return
+
     try:
         data = collect_machine_telemetry(command)
         headers = {}
@@ -88,6 +91,9 @@ def track_cli_command(command: str, args: Optional[str] = None):
         pass
 
 def update_telemetry(id: int, result: int, message: Optional[str] = None):
+    if bool(os.environ['AGENTSTATCK_IS_TEST_ENV']):
+        return
+
     try:
         requests.put(TELEMETRY_URL, json={"id": id, "result": result, "message": message})
     except Exception:

tox.ini

@@ -17,7 +17,8 @@ commands =
     coverage report -m
     coverage xml
     mypy: mypy agentstack
-setenv = 
+setenv =
+    AGENTSTATCK_IS_TEST_ENV = 1
     AGENTSTACK_TELEMETRY_OPT_OUT = 1
     AGENTSTACK_UPDATE_DISABLE = 1