ag-ui/integrations/aws-strands/python/src/ag_ui_strands/__init__.py at 83643a529e4ccaf9335610f0cded20e81118e557 · ag-ui-protocol/ag-ui · GitHub

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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
"""
AWS Strands Integration for AG-UI.

Wraps a Strands ``Agent`` as an AG-UI agent: event-stream translation,
frontend proxy-tool sync, per-thread session management, and the two-tier
A2UI surface generation (``get_a2ui_tools`` / ``plan_a2ui_injection``).
"""
from .agent import INTERRUPT_CANCELLED, StrandsAgent
from .a2ui_tool import (
    A2UI_OPERATIONS_KEY,
    A2UI_STREAM_KEY,
    A2UIGuidelines,
    A2UIToolParams,
    BASIC_CATALOG_ID,
    get_a2ui_tools,
    is_auto_injected_a2ui_tool,
    plan_a2ui_injection,
)
from .citations import CITATIONS_METADATA_KEY
from .client_proxy_tool import create_proxy_tool, sync_proxy_tools
from .template_tools import (
    EXEMPT_EVERY_TEMPLATE_TOOL,
    TemplateToolsSelectionError,
    sync_template_tools,
)
from .utils import (
    DEFAULT_URL_FETCH_POLICY,
    InvocationStateProvider,
    UrlFetchPolicy,
    UrlFetchPolicyError,
    create_strands_app,
)
from .endpoint import add_strands_fastapi_endpoint, add_ping
from .config import (
    StrandsAgentConfig,
    ToolBehavior,
    ToolCallContext,
    ToolResultContext,
    ToolStreamEventContext,
    PredictStateMapping,
    SessionManagerProvider,
    TemplateToolsProvider,
    ToolStreamEventHandler,
)
from ag_ui.core import (
    Interrupt,
    ResumeEntry,
    RunFinishedInterruptOutcome,
    RunFinishedSuccessOutcome,
)

__all__ = [
    "StrandsAgent",
    "INTERRUPT_CANCELLED",
    "A2UI_STREAM_KEY",
    "A2UI_OPERATIONS_KEY",
    "A2UIToolParams",
    "A2UIGuidelines",
    "BASIC_CATALOG_ID",
    "get_a2ui_tools",
    "is_auto_injected_a2ui_tool",
    "plan_a2ui_injection",
    "CITATIONS_METADATA_KEY",
    "create_proxy_tool",
    "sync_proxy_tools",
    "sync_template_tools",
    "TemplateToolsSelectionError",
    "EXEMPT_EVERY_TEMPLATE_TOOL",
    "create_strands_app",
    "UrlFetchPolicy",
    "UrlFetchPolicyError",
    "DEFAULT_URL_FETCH_POLICY",
    "add_strands_fastapi_endpoint",
    "InvocationStateProvider",
    "add_ping",
    "StrandsAgentConfig",
    "ToolBehavior",
    "ToolCallContext",
    "ToolResultContext",
    "ToolStreamEventContext",
    "PredictStateMapping",
    "SessionManagerProvider",
    "TemplateToolsProvider",
    "ToolStreamEventHandler",
    "Interrupt",
    "ResumeEntry",
    "RunFinishedInterruptOutcome",
    "RunFinishedSuccessOutcome",
]