agentrun-sdk 0.1.2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of agentrun-sdk might be problematic. Click here for more details.
- agentrun_operation_sdk/cli/__init__.py +1 -0
- agentrun_operation_sdk/cli/cli.py +19 -0
- agentrun_operation_sdk/cli/common.py +21 -0
- agentrun_operation_sdk/cli/runtime/__init__.py +1 -0
- agentrun_operation_sdk/cli/runtime/commands.py +203 -0
- agentrun_operation_sdk/client/client.py +75 -0
- agentrun_operation_sdk/operations/runtime/__init__.py +8 -0
- agentrun_operation_sdk/operations/runtime/configure.py +101 -0
- agentrun_operation_sdk/operations/runtime/launch.py +82 -0
- agentrun_operation_sdk/operations/runtime/models.py +31 -0
- agentrun_operation_sdk/services/runtime.py +152 -0
- agentrun_operation_sdk/utils/logging_config.py +72 -0
- agentrun_operation_sdk/utils/runtime/config.py +94 -0
- agentrun_operation_sdk/utils/runtime/container.py +280 -0
- agentrun_operation_sdk/utils/runtime/entrypoint.py +203 -0
- agentrun_operation_sdk/utils/runtime/schema.py +56 -0
- agentrun_sdk/__init__.py +7 -0
- agentrun_sdk/agent/__init__.py +25 -0
- agentrun_sdk/agent/agent.py +696 -0
- agentrun_sdk/agent/agent_result.py +46 -0
- agentrun_sdk/agent/conversation_manager/__init__.py +26 -0
- agentrun_sdk/agent/conversation_manager/conversation_manager.py +88 -0
- agentrun_sdk/agent/conversation_manager/null_conversation_manager.py +46 -0
- agentrun_sdk/agent/conversation_manager/sliding_window_conversation_manager.py +179 -0
- agentrun_sdk/agent/conversation_manager/summarizing_conversation_manager.py +252 -0
- agentrun_sdk/agent/state.py +97 -0
- agentrun_sdk/event_loop/__init__.py +9 -0
- agentrun_sdk/event_loop/event_loop.py +499 -0
- agentrun_sdk/event_loop/streaming.py +319 -0
- agentrun_sdk/experimental/__init__.py +4 -0
- agentrun_sdk/experimental/hooks/__init__.py +15 -0
- agentrun_sdk/experimental/hooks/events.py +123 -0
- agentrun_sdk/handlers/__init__.py +10 -0
- agentrun_sdk/handlers/callback_handler.py +70 -0
- agentrun_sdk/hooks/__init__.py +49 -0
- agentrun_sdk/hooks/events.py +80 -0
- agentrun_sdk/hooks/registry.py +247 -0
- agentrun_sdk/models/__init__.py +10 -0
- agentrun_sdk/models/anthropic.py +432 -0
- agentrun_sdk/models/bedrock.py +649 -0
- agentrun_sdk/models/litellm.py +225 -0
- agentrun_sdk/models/llamaapi.py +438 -0
- agentrun_sdk/models/mistral.py +539 -0
- agentrun_sdk/models/model.py +95 -0
- agentrun_sdk/models/ollama.py +357 -0
- agentrun_sdk/models/openai.py +436 -0
- agentrun_sdk/models/sagemaker.py +598 -0
- agentrun_sdk/models/writer.py +449 -0
- agentrun_sdk/multiagent/__init__.py +22 -0
- agentrun_sdk/multiagent/a2a/__init__.py +15 -0
- agentrun_sdk/multiagent/a2a/executor.py +148 -0
- agentrun_sdk/multiagent/a2a/server.py +252 -0
- agentrun_sdk/multiagent/base.py +92 -0
- agentrun_sdk/multiagent/graph.py +555 -0
- agentrun_sdk/multiagent/swarm.py +656 -0
- agentrun_sdk/py.typed +1 -0
- agentrun_sdk/session/__init__.py +18 -0
- agentrun_sdk/session/file_session_manager.py +216 -0
- agentrun_sdk/session/repository_session_manager.py +152 -0
- agentrun_sdk/session/s3_session_manager.py +272 -0
- agentrun_sdk/session/session_manager.py +73 -0
- agentrun_sdk/session/session_repository.py +51 -0
- agentrun_sdk/telemetry/__init__.py +21 -0
- agentrun_sdk/telemetry/config.py +194 -0
- agentrun_sdk/telemetry/metrics.py +476 -0
- agentrun_sdk/telemetry/metrics_constants.py +15 -0
- agentrun_sdk/telemetry/tracer.py +563 -0
- agentrun_sdk/tools/__init__.py +17 -0
- agentrun_sdk/tools/decorator.py +569 -0
- agentrun_sdk/tools/executor.py +137 -0
- agentrun_sdk/tools/loader.py +152 -0
- agentrun_sdk/tools/mcp/__init__.py +13 -0
- agentrun_sdk/tools/mcp/mcp_agent_tool.py +99 -0
- agentrun_sdk/tools/mcp/mcp_client.py +423 -0
- agentrun_sdk/tools/mcp/mcp_instrumentation.py +322 -0
- agentrun_sdk/tools/mcp/mcp_types.py +63 -0
- agentrun_sdk/tools/registry.py +607 -0
- agentrun_sdk/tools/structured_output.py +421 -0
- agentrun_sdk/tools/tools.py +217 -0
- agentrun_sdk/tools/watcher.py +136 -0
- agentrun_sdk/types/__init__.py +5 -0
- agentrun_sdk/types/collections.py +23 -0
- agentrun_sdk/types/content.py +188 -0
- agentrun_sdk/types/event_loop.py +48 -0
- agentrun_sdk/types/exceptions.py +81 -0
- agentrun_sdk/types/guardrails.py +254 -0
- agentrun_sdk/types/media.py +89 -0
- agentrun_sdk/types/session.py +152 -0
- agentrun_sdk/types/streaming.py +201 -0
- agentrun_sdk/types/tools.py +258 -0
- agentrun_sdk/types/traces.py +5 -0
- agentrun_sdk-0.1.2.dist-info/METADATA +51 -0
- agentrun_sdk-0.1.2.dist-info/RECORD +115 -0
- agentrun_sdk-0.1.2.dist-info/WHEEL +5 -0
- agentrun_sdk-0.1.2.dist-info/entry_points.txt +2 -0
- agentrun_sdk-0.1.2.dist-info/top_level.txt +3 -0
- agentrun_wrapper/__init__.py +11 -0
- agentrun_wrapper/_utils/__init__.py +6 -0
- agentrun_wrapper/_utils/endpoints.py +16 -0
- agentrun_wrapper/identity/__init__.py +5 -0
- agentrun_wrapper/identity/auth.py +211 -0
- agentrun_wrapper/memory/__init__.py +6 -0
- agentrun_wrapper/memory/client.py +1697 -0
- agentrun_wrapper/memory/constants.py +103 -0
- agentrun_wrapper/memory/controlplane.py +626 -0
- agentrun_wrapper/py.typed +1 -0
- agentrun_wrapper/runtime/__init__.py +13 -0
- agentrun_wrapper/runtime/app.py +473 -0
- agentrun_wrapper/runtime/context.py +34 -0
- agentrun_wrapper/runtime/models.py +25 -0
- agentrun_wrapper/services/__init__.py +1 -0
- agentrun_wrapper/services/identity.py +192 -0
- agentrun_wrapper/tools/__init__.py +6 -0
- agentrun_wrapper/tools/browser_client.py +325 -0
- agentrun_wrapper/tools/code_interpreter_client.py +186 -0
|
@@ -0,0 +1,258 @@
|
|
|
1
|
+
"""Tool-related type definitions for the SDK.
|
|
2
|
+
|
|
3
|
+
These types are modeled after the Bedrock API.
|
|
4
|
+
|
|
5
|
+
- Bedrock docs: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_Types_Amazon_Bedrock_Runtime.html
|
|
6
|
+
"""
|
|
7
|
+
|
|
8
|
+
from abc import ABC, abstractmethod
|
|
9
|
+
from typing import Any, AsyncGenerator, Awaitable, Callable, Literal, Protocol, Union
|
|
10
|
+
|
|
11
|
+
from typing_extensions import TypedDict
|
|
12
|
+
|
|
13
|
+
from .media import DocumentContent, ImageContent
|
|
14
|
+
|
|
15
|
+
JSONSchema = dict
|
|
16
|
+
"""Type alias for JSON Schema dictionaries."""
|
|
17
|
+
|
|
18
|
+
|
|
19
|
+
class ToolSpec(TypedDict):
|
|
20
|
+
"""Specification for a tool that can be used by an agent.
|
|
21
|
+
|
|
22
|
+
Attributes:
|
|
23
|
+
description: A human-readable description of what the tool does.
|
|
24
|
+
inputSchema: JSON Schema defining the expected input parameters.
|
|
25
|
+
name: The unique name of the tool.
|
|
26
|
+
"""
|
|
27
|
+
|
|
28
|
+
description: str
|
|
29
|
+
inputSchema: JSONSchema
|
|
30
|
+
name: str
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
class Tool(TypedDict):
|
|
34
|
+
"""A tool that can be provided to a model.
|
|
35
|
+
|
|
36
|
+
This type wraps a tool specification for inclusion in a model request.
|
|
37
|
+
|
|
38
|
+
Attributes:
|
|
39
|
+
toolSpec: The specification of the tool.
|
|
40
|
+
"""
|
|
41
|
+
|
|
42
|
+
toolSpec: ToolSpec
|
|
43
|
+
|
|
44
|
+
|
|
45
|
+
class ToolUse(TypedDict):
|
|
46
|
+
"""A request from the model to use a specific tool with the provided input.
|
|
47
|
+
|
|
48
|
+
Attributes:
|
|
49
|
+
input: The input parameters for the tool.
|
|
50
|
+
Can be any JSON-serializable type.
|
|
51
|
+
name: The name of the tool to invoke.
|
|
52
|
+
toolUseId: A unique identifier for this specific tool use request.
|
|
53
|
+
"""
|
|
54
|
+
|
|
55
|
+
input: Any
|
|
56
|
+
name: str
|
|
57
|
+
toolUseId: str
|
|
58
|
+
|
|
59
|
+
|
|
60
|
+
class ToolResultContent(TypedDict, total=False):
|
|
61
|
+
"""Content returned by a tool execution.
|
|
62
|
+
|
|
63
|
+
Attributes:
|
|
64
|
+
document: Document content returned by the tool.
|
|
65
|
+
image: Image content returned by the tool.
|
|
66
|
+
json: JSON-serializable data returned by the tool.
|
|
67
|
+
text: Text content returned by the tool.
|
|
68
|
+
"""
|
|
69
|
+
|
|
70
|
+
document: DocumentContent
|
|
71
|
+
image: ImageContent
|
|
72
|
+
json: Any
|
|
73
|
+
text: str
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
ToolResultStatus = Literal["success", "error"]
|
|
77
|
+
"""Status of a tool execution result."""
|
|
78
|
+
|
|
79
|
+
|
|
80
|
+
class ToolResult(TypedDict):
|
|
81
|
+
"""Result of a tool execution.
|
|
82
|
+
|
|
83
|
+
Attributes:
|
|
84
|
+
content: List of result content returned by the tool.
|
|
85
|
+
status: The status of the tool execution ("success" or "error").
|
|
86
|
+
toolUseId: The unique identifier of the tool use request that produced this result.
|
|
87
|
+
"""
|
|
88
|
+
|
|
89
|
+
content: list[ToolResultContent]
|
|
90
|
+
status: ToolResultStatus
|
|
91
|
+
toolUseId: str
|
|
92
|
+
|
|
93
|
+
|
|
94
|
+
class ToolChoiceAuto(TypedDict):
|
|
95
|
+
"""Configuration for automatic tool selection.
|
|
96
|
+
|
|
97
|
+
This represents the configuration for automatic tool selection, where the model decides whether and which tool to
|
|
98
|
+
use based on the context.
|
|
99
|
+
"""
|
|
100
|
+
|
|
101
|
+
pass
|
|
102
|
+
|
|
103
|
+
|
|
104
|
+
class ToolChoiceAny(TypedDict):
|
|
105
|
+
"""Configuration indicating that the model must request at least one tool."""
|
|
106
|
+
|
|
107
|
+
pass
|
|
108
|
+
|
|
109
|
+
|
|
110
|
+
class ToolChoiceTool(TypedDict):
|
|
111
|
+
"""Configuration for forcing the use of a specific tool.
|
|
112
|
+
|
|
113
|
+
Attributes:
|
|
114
|
+
name: The name of the tool that the model must use.
|
|
115
|
+
"""
|
|
116
|
+
|
|
117
|
+
name: str
|
|
118
|
+
|
|
119
|
+
|
|
120
|
+
ToolChoice = Union[
|
|
121
|
+
dict[Literal["auto"], ToolChoiceAuto],
|
|
122
|
+
dict[Literal["any"], ToolChoiceAny],
|
|
123
|
+
dict[Literal["tool"], ToolChoiceTool],
|
|
124
|
+
]
|
|
125
|
+
"""
|
|
126
|
+
Configuration for how the model should choose tools.
|
|
127
|
+
|
|
128
|
+
- "auto": The model decides whether to use tools based on the context
|
|
129
|
+
- "any": The model must use at least one tool (any tool)
|
|
130
|
+
- "tool": The model must use the specified tool
|
|
131
|
+
"""
|
|
132
|
+
|
|
133
|
+
RunToolHandler = Callable[[ToolUse], AsyncGenerator[dict[str, Any], None]]
|
|
134
|
+
"""Callback that runs a single tool and streams back results."""
|
|
135
|
+
|
|
136
|
+
ToolGenerator = AsyncGenerator[Any, None]
|
|
137
|
+
"""Generator of tool events with the last being the tool result."""
|
|
138
|
+
|
|
139
|
+
|
|
140
|
+
class ToolConfig(TypedDict):
|
|
141
|
+
"""Configuration for tools in a model request.
|
|
142
|
+
|
|
143
|
+
Attributes:
|
|
144
|
+
tools: List of tools available to the model.
|
|
145
|
+
toolChoice: Configuration for how the model should choose tools.
|
|
146
|
+
"""
|
|
147
|
+
|
|
148
|
+
tools: list[Tool]
|
|
149
|
+
toolChoice: ToolChoice
|
|
150
|
+
|
|
151
|
+
|
|
152
|
+
class ToolFunc(Protocol):
|
|
153
|
+
"""Function signature for Python decorated and module based tools."""
|
|
154
|
+
|
|
155
|
+
__name__: str
|
|
156
|
+
|
|
157
|
+
def __call__(
|
|
158
|
+
self, *args: Any, **kwargs: Any
|
|
159
|
+
) -> Union[
|
|
160
|
+
ToolResult,
|
|
161
|
+
Awaitable[ToolResult],
|
|
162
|
+
]:
|
|
163
|
+
"""Function signature for Python decorated and module based tools.
|
|
164
|
+
|
|
165
|
+
Returns:
|
|
166
|
+
Tool result or awaitable tool result.
|
|
167
|
+
"""
|
|
168
|
+
...
|
|
169
|
+
|
|
170
|
+
|
|
171
|
+
class AgentTool(ABC):
|
|
172
|
+
"""Abstract base class for all SDK tools.
|
|
173
|
+
|
|
174
|
+
This class defines the interface that all tool implementations must follow. Each tool must provide its name,
|
|
175
|
+
specification, and implement a stream method that executes the tool's functionality.
|
|
176
|
+
"""
|
|
177
|
+
|
|
178
|
+
_is_dynamic: bool
|
|
179
|
+
|
|
180
|
+
def __init__(self) -> None:
|
|
181
|
+
"""Initialize the base agent tool with default dynamic state."""
|
|
182
|
+
self._is_dynamic = False
|
|
183
|
+
|
|
184
|
+
@property
|
|
185
|
+
@abstractmethod
|
|
186
|
+
# pragma: no cover
|
|
187
|
+
def tool_name(self) -> str:
|
|
188
|
+
"""The unique name of the tool used for identification and invocation."""
|
|
189
|
+
pass
|
|
190
|
+
|
|
191
|
+
@property
|
|
192
|
+
@abstractmethod
|
|
193
|
+
# pragma: no cover
|
|
194
|
+
def tool_spec(self) -> ToolSpec:
|
|
195
|
+
"""Tool specification that describes its functionality and parameters."""
|
|
196
|
+
pass
|
|
197
|
+
|
|
198
|
+
@property
|
|
199
|
+
@abstractmethod
|
|
200
|
+
# pragma: no cover
|
|
201
|
+
def tool_type(self) -> str:
|
|
202
|
+
"""The type of the tool implementation (e.g., 'python', 'javascript', 'lambda').
|
|
203
|
+
|
|
204
|
+
Used for categorization and appropriate handling.
|
|
205
|
+
"""
|
|
206
|
+
pass
|
|
207
|
+
|
|
208
|
+
@property
|
|
209
|
+
def supports_hot_reload(self) -> bool:
|
|
210
|
+
"""Whether the tool supports automatic reloading when modified.
|
|
211
|
+
|
|
212
|
+
Returns:
|
|
213
|
+
False by default.
|
|
214
|
+
"""
|
|
215
|
+
return False
|
|
216
|
+
|
|
217
|
+
@abstractmethod
|
|
218
|
+
# pragma: no cover
|
|
219
|
+
def stream(self, tool_use: ToolUse, invocation_state: dict[str, Any], **kwargs: Any) -> ToolGenerator:
|
|
220
|
+
"""Stream tool events and return the final result.
|
|
221
|
+
|
|
222
|
+
Args:
|
|
223
|
+
tool_use: The tool use request containing tool ID and parameters.
|
|
224
|
+
invocation_state: Context for the tool invocation, including agent state.
|
|
225
|
+
**kwargs: Additional keyword arguments for future extensibility.
|
|
226
|
+
|
|
227
|
+
Yields:
|
|
228
|
+
Tool events with the last being the tool result.
|
|
229
|
+
"""
|
|
230
|
+
...
|
|
231
|
+
|
|
232
|
+
@property
|
|
233
|
+
def is_dynamic(self) -> bool:
|
|
234
|
+
"""Whether the tool was dynamically loaded during runtime.
|
|
235
|
+
|
|
236
|
+
Dynamic tools may have different lifecycle management.
|
|
237
|
+
|
|
238
|
+
Returns:
|
|
239
|
+
True if loaded dynamically, False otherwise.
|
|
240
|
+
"""
|
|
241
|
+
return self._is_dynamic
|
|
242
|
+
|
|
243
|
+
def mark_dynamic(self) -> None:
|
|
244
|
+
"""Mark this tool as dynamically loaded."""
|
|
245
|
+
self._is_dynamic = True
|
|
246
|
+
|
|
247
|
+
def get_display_properties(self) -> dict[str, str]:
|
|
248
|
+
"""Get properties to display in UI representations of this tool.
|
|
249
|
+
|
|
250
|
+
Subclasses can extend this to include additional properties.
|
|
251
|
+
|
|
252
|
+
Returns:
|
|
253
|
+
Dictionary of property names and their string values.
|
|
254
|
+
"""
|
|
255
|
+
return {
|
|
256
|
+
"Name": self.tool_name,
|
|
257
|
+
"Type": self.tool_type,
|
|
258
|
+
}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: agentrun-sdk
|
|
3
|
+
Version: 0.1.2
|
|
4
|
+
Summary: AgentRun Operation SDK - A toolkit for agent runtime operations
|
|
5
|
+
Author-email: Your Name <your.email@example.com>
|
|
6
|
+
License: MIT
|
|
7
|
+
Project-URL: Homepage, https://github.com/yourusername/hw-agentrun-sdk-python
|
|
8
|
+
Project-URL: Repository, https://github.com/yourusername/hw-agentrun-sdk-python
|
|
9
|
+
Project-URL: Issues, https://github.com/yourusername/hw-agentrun-sdk-python/issues
|
|
10
|
+
Keywords: agent,runtime,sdk,operations
|
|
11
|
+
Classifier: Development Status :: 3 - Alpha
|
|
12
|
+
Classifier: Intended Audience :: Developers
|
|
13
|
+
Classifier: License :: OSI Approved :: MIT License
|
|
14
|
+
Classifier: Programming Language :: Python :: 3
|
|
15
|
+
Classifier: Programming Language :: Python :: 3.8
|
|
16
|
+
Classifier: Programming Language :: Python :: 3.9
|
|
17
|
+
Classifier: Programming Language :: Python :: 3.10
|
|
18
|
+
Classifier: Programming Language :: Python :: 3.11
|
|
19
|
+
Requires-Python: >=3.8
|
|
20
|
+
Description-Content-Type: text/markdown
|
|
21
|
+
Requires-Dist: typer
|
|
22
|
+
Requires-Dist: rich
|
|
23
|
+
Requires-Dist: pydantic
|
|
24
|
+
Requires-Dist: prompt-toolkit
|
|
25
|
+
|
|
26
|
+
# hw-agentrun-sdk-python
|
|
27
|
+
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
## Getting started
|
|
31
|
+
|
|
32
|
+
### Start server
|
|
33
|
+
```sh
|
|
34
|
+
python agentrun_wrapper_test.py
|
|
35
|
+
```
|
|
36
|
+
|
|
37
|
+
### Start client
|
|
38
|
+
```sh
|
|
39
|
+
python cli.py
|
|
40
|
+
```
|
|
41
|
+
|
|
42
|
+
Input to test:
|
|
43
|
+
```sh
|
|
44
|
+
Example inputs:
|
|
45
|
+
What is the weather now?
|
|
46
|
+
--------------------------------------------------
|
|
47
|
+
|
|
48
|
+
Input: What is the weather now
|
|
49
|
+
Response:
|
|
50
|
+
"The current weather is sunny."
|
|
51
|
+
```
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
agentrun_operation_sdk/cli/__init__.py,sha256=ZjTpGaDfjI4P6AgdDS1N9GuJFqz8Fqu-dmMaJv0_dUk,63
|
|
2
|
+
agentrun_operation_sdk/cli/cli.py,sha256=u1OeNk88NTavLZS7_sMlkR_XrD_flpMt9fOe6gFiU1o,456
|
|
3
|
+
agentrun_operation_sdk/cli/common.py,sha256=B7uw4SwXogDD2ohXjHuX9MfqJYQjhydiQvTdMIdyGK0,764
|
|
4
|
+
agentrun_operation_sdk/cli/runtime/__init__.py,sha256=llva5WqPDkRypokHawBWnHQg3SvTF-R-5Eb2nkDFlH8,61
|
|
5
|
+
agentrun_operation_sdk/cli/runtime/commands.py,sha256=x8mUKwTJsXi1SHHWp4NJi9Xdnh4W68pdXX1vcdsRTZc,8172
|
|
6
|
+
agentrun_operation_sdk/client/client.py,sha256=eYaTe1BIUMeFcAgfKnJZcs5lvDFAYYbp3NA7GDLQnZw,2637
|
|
7
|
+
agentrun_operation_sdk/operations/runtime/__init__.py,sha256=x7nHjUECLcbytDnMncQmouBljonldANmL5-wWSxtcOs,195
|
|
8
|
+
agentrun_operation_sdk/operations/runtime/configure.py,sha256=5eZ5dookSVEVNFTdnTVEQTNoVFZhh88g8dHy96mw4MY,3295
|
|
9
|
+
agentrun_operation_sdk/operations/runtime/launch.py,sha256=aRPMXMofru52fVfsH2GBrXBnyRHmuqO1yel-3tbc5y0,2867
|
|
10
|
+
agentrun_operation_sdk/operations/runtime/models.py,sha256=URKODMKvsy-dq71Id_gxp3OPpkBvWBhGp4hBm7rxRWk,1473
|
|
11
|
+
agentrun_operation_sdk/services/runtime.py,sha256=JFmuxmeAMzr2x2i10U0Le_joEgGq9qpAmCPrBTQXjs0,5539
|
|
12
|
+
agentrun_operation_sdk/utils/logging_config.py,sha256=rOSZLdq7KDVZmlBaaFguTS17btRKyseXD8xBSmUg2eM,2263
|
|
13
|
+
agentrun_operation_sdk/utils/runtime/config.py,sha256=2iKtm1f4h-pJiKTW7PDHifj-emai-Dg_hspV6TZ_2eM,3322
|
|
14
|
+
agentrun_operation_sdk/utils/runtime/container.py,sha256=g5w3r6pkEnFnV-Iu8NeCFab5zwLIU3clc4nyLj8deZg,12912
|
|
15
|
+
agentrun_operation_sdk/utils/runtime/entrypoint.py,sha256=H34a3tSelDU2ClMvHwlRW0Rxij7AXWKAsI9uMbrx-PY,7558
|
|
16
|
+
agentrun_operation_sdk/utils/runtime/schema.py,sha256=R5Y2Khq8xVB-WFtBmC2LoYKoPd1Y9XsK07s3NPgcCVE,2201
|
|
17
|
+
agentrun_sdk/__init__.py,sha256=xgEGL1HLf3zhcRg792HQOOGKdmnVoGvhQtu_EIupm7k,256
|
|
18
|
+
agentrun_sdk/py.typed,sha256=LOqU0Ahtz8WM0FUZLDE_SWQQVvC6lc7srlTwJk3tE4Q,59
|
|
19
|
+
agentrun_sdk/agent/__init__.py,sha256=W3ZggGBDkXzxx8GdGYHGeUm1YHUy8xU8D4mJFaDFSJ8,724
|
|
20
|
+
agentrun_sdk/agent/agent.py,sha256=3DHP_3Fhk6VGgxaI9-kcEaMUN5fX2eXDyGbCNfW6puw,29603
|
|
21
|
+
agentrun_sdk/agent/agent_result.py,sha256=73COxfC-lLYrfFrZZodhzhFVyRHNdI57fEYDzT_Vzd4,1435
|
|
22
|
+
agentrun_sdk/agent/state.py,sha256=8polMncC0GfHi3unqRkdExRUOqMqWBGERrVgGuU1RHs,3027
|
|
23
|
+
agentrun_sdk/agent/conversation_manager/__init__.py,sha256=Bzc7LAv0Ix0z9UL0abkG0UwImuAxcHK5v2P-BxB33go,1191
|
|
24
|
+
agentrun_sdk/agent/conversation_manager/conversation_manager.py,sha256=ZgnPzhFJBkAyV5n5ncjDRg9e4ycOFI4SvVn9KqTbhE4,3609
|
|
25
|
+
agentrun_sdk/agent/conversation_manager/null_conversation_manager.py,sha256=WVACZZV6rMc-aly5qxfU9FFW3TLNJfxgXCyKHMSoxqA,1665
|
|
26
|
+
agentrun_sdk/agent/conversation_manager/sliding_window_conversation_manager.py,sha256=_ZqGpKKO01UdF3db5esMJ21Y8tI34Ip_wwd8szN5nVA,8121
|
|
27
|
+
agentrun_sdk/agent/conversation_manager/summarizing_conversation_manager.py,sha256=U3MBMQO0so-0K6E7e__cmOO3ONOf0oXBt-wEcWxOZQE,10991
|
|
28
|
+
agentrun_sdk/event_loop/__init__.py,sha256=j6mX3s1AJhX11CVr-TWGD1cDjDvjTySs3ATnl2I2SXw,283
|
|
29
|
+
agentrun_sdk/event_loop/event_loop.py,sha256=cjgZS0EA23MdSX_2T9kH8PmzqqUEPcKBFWvsOZjG_K4,19669
|
|
30
|
+
agentrun_sdk/event_loop/streaming.py,sha256=43qzCBsUM-ZOSmVcUOQb4bdtQKxjXnln1DMnftY-9Zo,10703
|
|
31
|
+
agentrun_sdk/experimental/__init__.py,sha256=7_rg0usrAXv-mjyWx5LpCUifRxOOG52RbEJETYuiwXc,143
|
|
32
|
+
agentrun_sdk/experimental/hooks/__init__.py,sha256=SgZUN4-tWKGLEg9I40RenDjLwshyQt9rRkHC6NEhAgc,385
|
|
33
|
+
agentrun_sdk/experimental/hooks/events.py,sha256=RxSVVRPAovs8Rwes-LyNEqntWg28sAAhwBc3I4aoF_Q,4395
|
|
34
|
+
agentrun_sdk/handlers/__init__.py,sha256=REljH2gh_jZ0K1KOuUg9QrOuX-15taWHclYJlUXv9rw,338
|
|
35
|
+
agentrun_sdk/handlers/callback_handler.py,sha256=sCY5Avnmj_aJo8Azu-47p8O98B8TvhHK1WJ3g0hapFs,2471
|
|
36
|
+
agentrun_sdk/hooks/__init__.py,sha256=UQbz6uCcSvCjuH-3XjirbOeqFOIk-k0tNuLIMYHtygo,1648
|
|
37
|
+
agentrun_sdk/hooks/events.py,sha256=6XONrlWaK-s6kZpa6LTsDBRddz90iBdI1JelfZ7lt_8,2665
|
|
38
|
+
agentrun_sdk/hooks/registry.py,sha256=cz1d2Wb2i-lbE_c1RdCqcVHOCCodruPfdnyqVezXtkc,8937
|
|
39
|
+
agentrun_sdk/models/__init__.py,sha256=1_zWbKMNkUNpqd33im47ojYXDYdb40U-xuImqUOlknE,296
|
|
40
|
+
agentrun_sdk/models/anthropic.py,sha256=svyJKBmo2gUu9BH6k-C1ALjzNTVWobZA9r4yp99VdPI,16940
|
|
41
|
+
agentrun_sdk/models/bedrock.py,sha256=drpEgoc3hWJ9KGfKScEKKEhq5tzt_v_PW8U-HMx4S44,27144
|
|
42
|
+
agentrun_sdk/models/litellm.py,sha256=PkLuJbUiVqf3Ufa1Lf6A4pBjHiLrlowpxFMODLWXl5A,8701
|
|
43
|
+
agentrun_sdk/models/llamaapi.py,sha256=WYsSMQL3whPuKO42msy0WkaSlP9KH8sGmPSfzVY4VCs,16834
|
|
44
|
+
agentrun_sdk/models/mistral.py,sha256=VBAOzsHS8yGGaVDl6TRkBTzmuhQtcdUjgsvQR-9eXEs,21389
|
|
45
|
+
agentrun_sdk/models/model.py,sha256=SXXh1P0dzvElTi_pn9rITd-AFiHTcrER0JXGlKa_ogg,3150
|
|
46
|
+
agentrun_sdk/models/ollama.py,sha256=U0P0jwrRnInUy84MRBmIkWCifej7-B_kFPTpQHu44m8,14086
|
|
47
|
+
agentrun_sdk/models/openai.py,sha256=A4zaTbhkioG1HBCRhIchXpZeTBOVEKBOh3pC9vuzYsM,16619
|
|
48
|
+
agentrun_sdk/models/sagemaker.py,sha256=ZFlB-J-m9593EGF9VrW9NyGht9fZDpDBmDUIxxbSm4k,26259
|
|
49
|
+
agentrun_sdk/models/writer.py,sha256=CdqEKZPswkwHVBJy4uYxdsx8TRb_I29QgG-ctulBp60,17628
|
|
50
|
+
agentrun_sdk/multiagent/__init__.py,sha256=UXCSKG0MRLQfBadfBSPN3rfe4ef-bxZDWyW3dP1eero,619
|
|
51
|
+
agentrun_sdk/multiagent/base.py,sha256=lsSpJy0Fb34tIn3EdbuFmh8UxSBhUWU7Kci1c8YwRb4,3381
|
|
52
|
+
agentrun_sdk/multiagent/graph.py,sha256=dbWyE2O_-ngEHc3Oh1HpJluuTzDMF08rK6kHwDRe5TI,23209
|
|
53
|
+
agentrun_sdk/multiagent/swarm.py,sha256=mOotXXnkqxOl-h4d_H3lN9JOJL5cQtJmiUEtZ61603E,27980
|
|
54
|
+
agentrun_sdk/multiagent/a2a/__init__.py,sha256=-CrYHGROR7ZR42VtPksAPHkcMRP5ITtpcSii2wqM2nE,498
|
|
55
|
+
agentrun_sdk/multiagent/a2a/executor.py,sha256=g87vVws_JAKUVEWd9oBXB9sLi4S6ZIFPIldPg1FmInA,6068
|
|
56
|
+
agentrun_sdk/multiagent/a2a/server.py,sha256=TuhT6aE4j4DZWU_iE8iqQKHTpeilWig-8B8Mmtxb1hY,10615
|
|
57
|
+
agentrun_sdk/session/__init__.py,sha256=3ISqVpyZKFzbNgRbV5GAYaMBJI4moY1QIBxkZ9cz4sw,503
|
|
58
|
+
agentrun_sdk/session/file_session_manager.py,sha256=ZRo4CLEPMemz8fbXv7XbDeoH6EFZinjbR2opVFlkrm8,9507
|
|
59
|
+
agentrun_sdk/session/repository_session_manager.py,sha256=CADqpvzsqI760zugo7VhbD8eaJB30yWYkCk-zI8-o2A,6788
|
|
60
|
+
agentrun_sdk/session/s3_session_manager.py,sha256=hYOCTDdFpv1CcqleSsE_-wE9zsqiecpRqaFYaw6LvJc,12249
|
|
61
|
+
agentrun_sdk/session/session_manager.py,sha256=_VqgxtcEA90D3GzGsL5sq3bGnlqKofUnxrH6Q-i0XBE,3340
|
|
62
|
+
agentrun_sdk/session/session_repository.py,sha256=mr91YyUJIhNzTUPcjh8WksLGV2qwY9TjfoTXIw6dOWM,1994
|
|
63
|
+
agentrun_sdk/telemetry/__init__.py,sha256=MO_bA7mEqZ6WEDNqS7ndlXVDi76xOaKw5FH_uru7fFo,462
|
|
64
|
+
agentrun_sdk/telemetry/config.py,sha256=Q27VNLAL_lApYSCuEck3T3ImQ5pXkH-a0M28bYheuIM,7995
|
|
65
|
+
agentrun_sdk/telemetry/metrics.py,sha256=pg7HGwulhfNOVXyYb--y1cp54FIsDAuuanhcwBm1RC4,19506
|
|
66
|
+
agentrun_sdk/telemetry/metrics_constants.py,sha256=qPy7rqZA7ihoZw-qDmNYfxs5Mm77VbGXUns5tHlJG2c,754
|
|
67
|
+
agentrun_sdk/telemetry/tracer.py,sha256=GX1UmjQwERu6kMm62mIYF4qsjnRqjlbX0XOU6LmfY5k,19931
|
|
68
|
+
agentrun_sdk/tools/__init__.py,sha256=4tXpYPTDqGK0rUGLaSl63lnrQOwofx62uaEAMIXLdlw,532
|
|
69
|
+
agentrun_sdk/tools/decorator.py,sha256=vb76X_vLsd_3Y-B6Tj8Xb5qKUe4S4VCGjCC-cHXk8wQ,21895
|
|
70
|
+
agentrun_sdk/tools/executor.py,sha256=YTO8C0S97cGTKtN-WqFqWkTYU-ZcQJlGP4peqXOO5-s,5006
|
|
71
|
+
agentrun_sdk/tools/loader.py,sha256=nmsuoWqAQcWjk4-yvL-uP_Cg_yhEUhQeEHQVJ7lMTd4,6210
|
|
72
|
+
agentrun_sdk/tools/registry.py,sha256=Q8WvR3Da7NcL5-PNBnc6rHHIbvKKna6urcEcxeaM-YU,25470
|
|
73
|
+
agentrun_sdk/tools/structured_output.py,sha256=LINZvFLCq1hdS8UtFV-GR4KRfvNhExlhykEoCkzubVg,16265
|
|
74
|
+
agentrun_sdk/tools/tools.py,sha256=z3HiGA0A4uD8nJ6s8M_1mFplsbYIqx_3epF_C3fdLHk,7082
|
|
75
|
+
agentrun_sdk/tools/watcher.py,sha256=dJV_VlTQzzVk1IVomVXaDoh4j5oKPztUNdLpUEAyOow,5810
|
|
76
|
+
agentrun_sdk/tools/mcp/__init__.py,sha256=IIGTAvp0RS-wZUAFA_PTKc28VO_Z_uGN4I-Sidq6Nrs,422
|
|
77
|
+
agentrun_sdk/tools/mcp/mcp_agent_tool.py,sha256=GmagjYYCzSeIn5MHOU_50lVsNsJryT8jTJQUzAiQkzc,3490
|
|
78
|
+
agentrun_sdk/tools/mcp/mcp_client.py,sha256=jr4tlmi9BSQa12Rhvgtfy1SDcJ9Q5YGvDcolaD2arJI,20098
|
|
79
|
+
agentrun_sdk/tools/mcp/mcp_instrumentation.py,sha256=JpxNuj1MQPatkDhBKYtXF75ohhYAxVItzw-IEWUpZU0,13447
|
|
80
|
+
agentrun_sdk/tools/mcp/mcp_types.py,sha256=7YBGk-oVXMOCPWIUcdSi7vWTreIQp8LFWPCCthfrc_M,2844
|
|
81
|
+
agentrun_sdk/types/__init__.py,sha256=fZE4vXh11u2kwZXJ0CSMxCuK_NsxokNG_G88mhkK398,102
|
|
82
|
+
agentrun_sdk/types/collections.py,sha256=8NxbN-Jostm5OS1787oKS9k9cIP4ht532ft7ezUNW3A,768
|
|
83
|
+
agentrun_sdk/types/content.py,sha256=d0knjz9udJjztCgFOq6SL6fkDAt7v42t5VfMbroKzr8,5326
|
|
84
|
+
agentrun_sdk/types/event_loop.py,sha256=XxSnvGGvAnMAJ3FdrWfDJ80wc4KA_lZF8hkm04YG4RA,1266
|
|
85
|
+
agentrun_sdk/types/exceptions.py,sha256=2KCi4rOGko_BcfCokU0c1v_lLYYx9ed_8z4hgdsdi5k,2864
|
|
86
|
+
agentrun_sdk/types/guardrails.py,sha256=wyM9XQR9G19WrvdWuk8uLwi4KlEJiac_lJS0h0oPeDU,7437
|
|
87
|
+
agentrun_sdk/types/media.py,sha256=bW7GD17xg47yvYM5Gw3TkFa7No6NsZKqL5evl7EeLbQ,2159
|
|
88
|
+
agentrun_sdk/types/session.py,sha256=w9hWfRXasHgR720DJonGgApi3fYXxybyJ_o49aYKtkc,5572
|
|
89
|
+
agentrun_sdk/types/streaming.py,sha256=t-9ny5g4lvTk7wsryrTXqpCz21V75Ah2iVqferw12PQ,6356
|
|
90
|
+
agentrun_sdk/types/tools.py,sha256=3YYZYQm0YugVhOFK2PHyu_QHcJW36sd3Ubb-RIKE--s,7270
|
|
91
|
+
agentrun_sdk/types/traces.py,sha256=MqkB5dZ2wQ-iC0-fqZgv5BqmyTr_G9qam1Wyoblfktg,175
|
|
92
|
+
agentrun_wrapper/__init__.py,sha256=r-qxFXJlddvF_ae-YW-OAQwvsfCVf6sfSFYc4E1T-6s,334
|
|
93
|
+
agentrun_wrapper/py.typed,sha256=LOqU0Ahtz8WM0FUZLDE_SWQQVvC6lc7srlTwJk3tE4Q,59
|
|
94
|
+
agentrun_wrapper/_utils/__init__.py,sha256=bElAUQyzGkpt6MQs5XcCZlPDk8kpZ57x1SqivQ78PG0,285
|
|
95
|
+
agentrun_wrapper/_utils/endpoints.py,sha256=Zn7ViqaoorQyhSOqX4ClJGWaKe28AlitvkXNlC6D37c,651
|
|
96
|
+
agentrun_wrapper/identity/__init__.py,sha256=ERgE9uCtJfBPKwslE9cnu-PINfyx6ZVpRQbFlu3UaZE,167
|
|
97
|
+
agentrun_wrapper/identity/auth.py,sha256=qkjeNuM-oAzK8IV_31hdKsrvvnrfGg6ljUG_DW1G8WM,7296
|
|
98
|
+
agentrun_wrapper/memory/__init__.py,sha256=qpg-U_J0GMKQF3zY4JRKc5Y1XvM5Bt1Ga-LA3ACd3hQ,225
|
|
99
|
+
agentrun_wrapper/memory/client.py,sha256=FHLj0yeLOBbYvbR34fTxJ0Z6Jwlhn_LOALTissHNBPI,67344
|
|
100
|
+
agentrun_wrapper/memory/constants.py,sha256=sxA3FLWoXyiYgyd0r6ozF32XgsN_9IPhByZPyEttV1I,3097
|
|
101
|
+
agentrun_wrapper/memory/controlplane.py,sha256=Ue7LqiuEYD1Gw06SA73MmaNqH3onlSb0qoBh7ivHJcc,24822
|
|
102
|
+
agentrun_wrapper/runtime/__init__.py,sha256=SziwtAvi0Af0F3wpOkjK0FVOJY2djuhxA30KFOZIOZQ,500
|
|
103
|
+
agentrun_wrapper/runtime/app.py,sha256=yK6kjRs52JSdTDbjnohrct9TTUmiF9P4Tbo7km8tC98,20525
|
|
104
|
+
agentrun_wrapper/runtime/context.py,sha256=khSxLX1pn_XDSOaEHf5L591CHvIw7p1jbwc6Tx83Udg,1023
|
|
105
|
+
agentrun_wrapper/runtime/models.py,sha256=KC57eLLlGC-OeLEN9U6u2qIy5WeIqOTAnqlxCtxE8ic,682
|
|
106
|
+
agentrun_wrapper/services/__init__.py,sha256=tgIdVmJjZH_e5MvhGSZv6UuAFwlLx28KZofbMNlVnqU,71
|
|
107
|
+
agentrun_wrapper/services/identity.py,sha256=g8c8U3KpHqb0h2Mj_nqSNc6BZRFfSVdRNSjoJZGUIg8,8176
|
|
108
|
+
agentrun_wrapper/tools/__init__.py,sha256=WgPUpS9FrLnBhDTxnFcz9RG6AeJezyfGl-C931QC30U,259
|
|
109
|
+
agentrun_wrapper/tools/browser_client.py,sha256=l7A_vr8qKlGL1F_IwF1YbUfbm1uIkKt9i-nR6ypNG9E,11831
|
|
110
|
+
agentrun_wrapper/tools/code_interpreter_client.py,sha256=_CsWGtamqmH7jLZf1D6iBbrQaleqsr90PmqhJOdsoxc,6491
|
|
111
|
+
agentrun_sdk-0.1.2.dist-info/METADATA,sha256=mGdYoDSpRLsoaehjPtWOhWBYVtwMY6i4FbVT5vizBNk,1413
|
|
112
|
+
agentrun_sdk-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
113
|
+
agentrun_sdk-0.1.2.dist-info/entry_points.txt,sha256=B8qby5yUxGgYcIHko5cjPj6Uc6FeCqmoKTVU7eLvoik,65
|
|
114
|
+
agentrun_sdk-0.1.2.dist-info/top_level.txt,sha256=BqgDVZ0q1iU5GP2-DBfuVpuSTa06AAk84Gzre5dpqsg,53
|
|
115
|
+
agentrun_sdk-0.1.2.dist-info/RECORD,,
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
"""BedrockAgentCore Runtime SDK - A Python SDK for building and deploying AI agents."""
|
|
2
|
+
|
|
3
|
+
from .runtime import BedrockAgentCoreApp, BedrockAgentCoreContext, RequestContext
|
|
4
|
+
from .runtime.models import PingStatus
|
|
5
|
+
|
|
6
|
+
__all__ = [
|
|
7
|
+
"BedrockAgentCoreApp",
|
|
8
|
+
"RequestContext",
|
|
9
|
+
"BedrockAgentCoreContext",
|
|
10
|
+
"PingStatus",
|
|
11
|
+
]
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
"""Internal utilities package for Bedrock AgentCore SDK.
|
|
2
|
+
|
|
3
|
+
This package contains internal utility modules that are used by other
|
|
4
|
+
components within the Bedrock AgentCore SDK. These utilities are not part of the
|
|
5
|
+
public API and should not be imported directly by external users.
|
|
6
|
+
"""
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
"""Endpoint utilities for BedrockAgentCore services."""
|
|
2
|
+
|
|
3
|
+
import os
|
|
4
|
+
|
|
5
|
+
# Environment-configurable constants with fallback defaults
|
|
6
|
+
DP_ENDPOINT_OVERRIDE = os.getenv("BEDROCK_AGENTCORE_DP_ENDPOINT")
|
|
7
|
+
CP_ENDPOINT_OVERRIDE = os.getenv("BEDROCK_AGENTCORE_CP_ENDPOINT")
|
|
8
|
+
DEFAULT_REGION = os.getenv("AWS_REGION", "us-west-2")
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
def get_data_plane_endpoint(region: str = DEFAULT_REGION) -> str:
|
|
12
|
+
return DP_ENDPOINT_OVERRIDE or f"https://bedrock-agentcore.{region}.amazonaws.com"
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
def get_control_plane_endpoint(region: str = DEFAULT_REGION) -> str:
|
|
16
|
+
return CP_ENDPOINT_OVERRIDE or f"https://bedrock-agentcore-control.{region}.amazonaws.com"
|