Spaces:
Paused
Paused
File size: 1,238 Bytes
aceb1b2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | """
Agent Proxy Package
Provides agent proxy implementations for live agent interaction during annotation.
Proxies communicate with AI agent backends (echo, HTTP, OpenAI) and return
responses to the annotation interface.
Usage:
from potato.agent_proxy import AgentProxyFactory
proxy = AgentProxyFactory.create(config)
context = proxy.start_session("Book a flight to Paris")
response = proxy.send_message("Hello", context)
"""
from .base import AgentMessage, AgentResponse, BaseAgentProxy, AgentProxyFactory
from .session import (
AgentSession,
AgentSessionManager,
init_agent_session_manager,
get_agent_session_manager,
clear_agent_session_manager,
)
from .sandbox import SafetySandbox, SandboxViolation
# Import proxy implementations to trigger registration
from . import echo_proxy
from . import http_proxy
from . import openai_proxy
from . import coding_proxy # subprocess_coding + docker_coding
__all__ = [
"AgentMessage",
"AgentResponse",
"BaseAgentProxy",
"AgentProxyFactory",
"AgentSession",
"AgentSessionManager",
"init_agent_session_manager",
"get_agent_session_manager",
"clear_agent_session_manager",
"SafetySandbox",
"SandboxViolation",
]
|