mirror of
https://github.com/hanzoai/agent.git
synced 2026-09-11 21:08:51 +00:00
zeekay
ff7ef7114f
Consolidate all Hanzo agent infrastructure into one repo: - control-plane/: Go control plane server (was agents-new) - sdk/: Go, Python, TypeScript SDKs - deployments/: Docker, Kubernetes, Helm configs - Dockerfile.sdk: Python SDK image (ghcr.io/hanzoai/agent:latest) - CI builds both SDK + control-plane images No more agentfield — everything is hanzoai/agent.
30 lines
864 B
Python
30 lines
864 B
Python
"""
|
|
Agent registry for tracking the current agent instance in thread-local storage.
|
|
This allows reasoners to automatically find their parent agent for workflow tracking.
|
|
"""
|
|
|
|
import threading
|
|
from typing import Optional, TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from .agent import Agent
|
|
|
|
# Thread-local storage for agent instances
|
|
_thread_local = threading.local()
|
|
|
|
|
|
def set_current_agent(agent_instance: "Agent"):
|
|
"""Register the current agent instance for this thread."""
|
|
_thread_local.current_agent = agent_instance
|
|
|
|
|
|
def get_current_agent_instance() -> Optional["Agent"]:
|
|
"""Get the current agent instance for this thread."""
|
|
return getattr(_thread_local, "current_agent", None)
|
|
|
|
|
|
def clear_current_agent():
|
|
"""Clear the current agent instance."""
|
|
if hasattr(_thread_local, "current_agent"):
|
|
delattr(_thread_local, "current_agent")
|