stop test telemetry logging

AgentOps-AI · Jan 7, 2025 · 4008fa9 · 4008fa9
1 parent f8c1220
commit 4008fa9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/agentstack/telemetry.py b/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['AGENTSTACK_NO_TEST_TELEMETRY']):
+        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['AGENTSTACK_NO_TEST_TELEMETRY']):
+        return
+
     try:
         requests.put(TELEMETRY_URL, json={"id": id, "result": result, "message": message})
     except Exception:

diff --git a/tox.ini b/tox.ini
@@ -17,7 +17,8 @@ commands =
     coverage report -m
     coverage xml
     mypy: mypy agentstack
-setenv = 
+setenv =
+    AGENTSTACK_NO_TEST_TELEMETRY = 1
     AGENTSTACK_TELEMETRY_OPT_OUT = 1
     AGENTSTACK_UPDATE_DISABLE = 1