visualagents 0.1.0__py3-none-any.whl

visualagents/__init__.py

@@ -0,0 +1,20 @@
+"""VisualAgents - a live map of which agent in a multi-agent AI system is doing what."""
+
+__version__ = "0.1.0"
+
+from .sdk import (  # noqa: E402,F401
+    agent,
+    agent_end,
+    agent_start,
+    emit,
+    flush,
+    handoff,
+    instrument,
+    langchain_handler,
+    log,
+    openai_agents_processor,
+    thought,
+    tool_call,
+    tool_result,
+    track,
+)

visualagents/__main__.py

@@ -0,0 +1,3 @@
+from .cli import main
+
+raise SystemExit(main())

visualagents/_autoinstrument/sitecustomize.py

@@ -0,0 +1,40 @@
+"""Loaded automatically by Python because `visualagents run` prepends this folder to PYTHONPATH.
+
+Installs VisualAgents' framework hooks, then runs any sitecustomize this one shadows.
+"""
+
+import os
+import sys
+
+
+def _visualagents_boot():
+    if os.environ.get("VISUALAGENTS_DISABLE") == "1":
+        return
+    try:
+        import visualagents_runtime
+
+        visualagents_runtime.instrument()
+    except Exception:
+        pass
+
+
+def _visualagents_chain():
+    here = os.path.dirname(os.path.abspath(__file__))
+    for entry in list(sys.path):
+        try:
+            if not entry or os.path.abspath(entry) == here:
+                continue
+            candidate = os.path.join(entry, "sitecustomize.py")
+            if os.path.isfile(candidate):
+                import importlib.util
+
+                spec = importlib.util.spec_from_file_location("_visualagents_shadowed_sitecustomize", candidate)
+                if spec and spec.loader:
+                    spec.loader.exec_module(importlib.util.module_from_spec(spec))
+                return
+        except Exception:
+            return
+
+
+_visualagents_boot()
+_visualagents_chain()