agentops-rpc 0.1.0__tar.gz

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.
Files changed (35) hide show
  1. agentops_rpc-0.1.0/LICENSE +53 -0
  2. agentops_rpc-0.1.0/PKG-INFO +102 -0
  3. agentops_rpc-0.1.0/README.md +89 -0
  4. agentops_rpc-0.1.0/pyproject.toml +25 -0
  5. agentops_rpc-0.1.0/setup.cfg +4 -0
  6. agentops_rpc-0.1.0/src/agentops_rpc/__init__.py +393 -0
  7. agentops_rpc-0.1.0/src/agentops_rpc/a2a.py +75 -0
  8. agentops_rpc-0.1.0/src/agentops_rpc/agent_create.py +176 -0
  9. agentops_rpc-0.1.0/src/agentops_rpc/agents.py +405 -0
  10. agentops_rpc-0.1.0/src/agentops_rpc/artifacts.py +15 -0
  11. agentops_rpc-0.1.0/src/agentops_rpc/credentials.py +87 -0
  12. agentops_rpc-0.1.0/src/agentops_rpc/events.py +138 -0
  13. agentops_rpc-0.1.0/src/agentops_rpc/grader.py +63 -0
  14. agentops_rpc-0.1.0/src/agentops_rpc/hosted_agents.py +107 -0
  15. agentops_rpc-0.1.0/src/agentops_rpc/insights.py +52 -0
  16. agentops_rpc-0.1.0/src/agentops_rpc/integrations.py +143 -0
  17. agentops_rpc-0.1.0/src/agentops_rpc/issues.py +96 -0
  18. agentops_rpc-0.1.0/src/agentops_rpc/knowledge_base.py +128 -0
  19. agentops_rpc-0.1.0/src/agentops_rpc/quality.py +78 -0
  20. agentops_rpc-0.1.0/src/agentops_rpc/runs.py +239 -0
  21. agentops_rpc-0.1.0/src/agentops_rpc/schedules.py +38 -0
  22. agentops_rpc-0.1.0/src/agentops_rpc/scores.py +86 -0
  23. agentops_rpc-0.1.0/src/agentops_rpc/session.py +39 -0
  24. agentops_rpc-0.1.0/src/agentops_rpc/triggers.py +110 -0
  25. agentops_rpc-0.1.0/src/agentops_rpc/types.py +91 -0
  26. agentops_rpc-0.1.0/src/agentops_rpc/usage.py +22 -0
  27. agentops_rpc-0.1.0/src/agentops_rpc/worker_catalog.py +106 -0
  28. agentops_rpc-0.1.0/src/agentops_rpc.egg-info/PKG-INFO +102 -0
  29. agentops_rpc-0.1.0/src/agentops_rpc.egg-info/SOURCES.txt +33 -0
  30. agentops_rpc-0.1.0/src/agentops_rpc.egg-info/dependency_links.txt +1 -0
  31. agentops_rpc-0.1.0/src/agentops_rpc.egg-info/requires.txt +1 -0
  32. agentops_rpc-0.1.0/src/agentops_rpc.egg-info/top_level.txt +1 -0
  33. agentops_rpc-0.1.0/tests/test_agent_create_models.py +129 -0
  34. agentops_rpc-0.1.0/tests/test_integrations_schemas.py +69 -0
  35. agentops_rpc-0.1.0/tests/test_use_case_model.py +87 -0
@@ -0,0 +1,53 @@
1
+ Komodor AgentOps License
2
+
3
+ The Komodor AgentOps SDK, together with its associated libraries, documentation,
4
+ and files (the "Software"), is provided by Komodor Ltd. ("Komodor") for integration
5
+ with the Komodor service (the "Service"). By installing, copying, or using the
6
+ Software, you agree to these terms.
7
+
8
+ © 2020 - Komodor Ltd.
9
+
10
+ Grant. Subject to your compliance with these terms, Komodor grants you a
11
+ non-exclusive, non-transferable, non-sublicensable, worldwide, revocable, and
12
+ limited license to (a) install and use the Software to develop, test, and operate
13
+ integrations of your applications and services with the Service, or to use
14
+ applications and services that embed the Software, and (b) reproduce
15
+ and distribute verbatim copies of the Software, but only as embedded within your own
16
+ applications, services, or container images, and not on a standalone basis. No
17
+ separate agreement with Komodor is required; anyone who obtains the Software
18
+ receives this license. If you have a separate written agreement with Komodor
19
+ covering the Software, that agreement controls to the extent of any conflict.
20
+
21
+ You may develop and operate your own software that imports, calls, subclasses,
22
+ configures, or otherwise depends on the Software through its published interfaces;
23
+ doing so is permitted use and does not, by itself, create a modified version or
24
+ derivative work of the Software.
25
+
26
+ Restrictions. You shall not: (a) distribute the Software on a standalone basis, or
27
+ resell, lease, or sublicense it; (b) offer the Software itself, or substantially its
28
+ standalone functionality, to third parties as a hosted or managed service — this
29
+ does not restrict you from operating or offering your own applications or services
30
+ that embed the Software; (c) modify the Software or create a derivative work of its
31
+ files or code; (d) remove, alter, or obscure any copyright, license, or proprietary
32
+ notice, all of which you must retain in every copy you distribute; or (e) claim any
33
+ proprietary interest in the Software, or contest or infringe Komodor's intellectual
34
+ property rights in it.
35
+
36
+ Ownership. The Software is licensed, not sold. Komodor retains all right, title, and
37
+ interest in and to the Software and all intellectual property rights in it. No
38
+ rights are granted except as expressly stated here, and no patent rights are
39
+ granted. This license does not permit use of Komodor's names or trademarks, except
40
+ for reasonable and customary use in describing the origin of the Software.
41
+
42
+ Termination. This license terminates automatically if you breach it and do not cure
43
+ the breach within thirty (30) days of Komodor's written notice. On termination you
44
+ must stop installing, using, and distributing the Software, except that copies
45
+ already embedded in products you distributed to end users before termination may
46
+ remain in use. The Restrictions and this "No warranty; no liability" section survive termination.
47
+
48
+ No warranty; no liability. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
49
+ KIND, EXPRESS OR IMPLIED, INCLUDING THE IMPLIED WARRANTIES OF MERCHANTABILITY,
50
+ FITNESS FOR A PARTICULAR PURPOSE, TITLE, AND NON-INFRINGEMENT. IN NO EVENT SHALL
51
+ KOMODOR OR THE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES, OR OTHER
52
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT, OR OTHERWISE, ARISING FROM, OUT
53
+ OF, OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
@@ -0,0 +1,102 @@
1
+ Metadata-Version: 2.4
2
+ Name: agentops-rpc
3
+ Version: 0.1.0
4
+ Summary: Shared Pydantic wire types for AgentOps SDK ↔ controlplane RPC.
5
+ Author: Komodor Ltd.
6
+ License-Expression: LicenseRef-Komodor-Proprietary
7
+ Project-URL: Homepage, https://agentops.komodor.com
8
+ Requires-Python: >=3.11
9
+ Description-Content-Type: text/markdown
10
+ License-File: LICENSE
11
+ Requires-Dist: pydantic>=2.0
12
+ Dynamic: license-file
13
+
14
+ # agentops-rpc
15
+
16
+ Shared Pydantic wire types for the AgentOps SDK and controlplane.
17
+
18
+ Both the SDK (`komodor-agentops`) and the controlplane import models from this
19
+ package, giving build-time (mypy) and runtime (Pydantic v2) type safety across
20
+ the wire boundary.
21
+
22
+ ## Install
23
+
24
+ ```bash
25
+ pip install agentops-rpc
26
+ ```
27
+
28
+ ## Quick Start
29
+
30
+ ```python
31
+ from agentops_rpc import RunEventIn, IngestRunEventsRequest, HOOK_EVENT_TYPE_MAP
32
+
33
+ event = RunEventIn(
34
+ event_type="span.started",
35
+ span_id="span_abc",
36
+ kind="tool",
37
+ name="Bash",
38
+ hook_event_type="tool.start",
39
+ )
40
+ event.model_dump() # JSON-safe dict
41
+
42
+ request = IngestRunEventsRequest(worker_id="wrk_1", events=[event])
43
+ ```
44
+
45
+ ## Exported Models
46
+
47
+ ### types.py -- Literal type aliases
48
+
49
+ | Type | Values |
50
+ |------|--------|
51
+ | `EventType` | `run.started`, `run.completed`, `run.failed`, `span.started`, `span.ended`, `message.created`, `log.created`, `token.created`, `custom.created`, `output.updated` |
52
+ | `RunStatus` | `queued`, `claimed`, `running`, `succeeded`, `failed` |
53
+ | `SpanKind` | `agent`, `llm`, `tool`, `chain`, `generation` |
54
+ | `HookEventName` | `SessionStart`, `UserPromptSubmit`, `PreToolUse`, `PostToolUse`, `PostToolUseFailure`, `SubagentStart`, `SubagentStop`, `Notification`, `Stop`, `SessionEnd`, `PermissionRequest`, `PreCompact`, `PostCompact` |
55
+ | `HookEventType` | `session.start`, `prompt.submit`, `tool.start`, `tool.end`, `tool.error`, `subagent.start`, `subagent.end`, `notification`, `session.stop`, `compaction.start`, `compaction.end` |
56
+ | `AgentStatus` | `online`, `offline` |
57
+ | `RunTransport` | `polling`, `api_sync`, `api_sse`, `a2a` |
58
+
59
+ ### events.py -- Event ingest models
60
+
61
+ - `RunEventIn` -- unified event sent SDK -> controlplane
62
+ - `TraceEventIn`, `LogEventIn` -- structured events for DB tables
63
+ - `IngestRunEventsRequest` / `IngestRunEventsResponse`
64
+ - `BatchEventsRequest`
65
+ - `TraceEventResponse`, `LogEventResponse`, `RunMessageResponse`
66
+
67
+ ### runs.py -- Run lifecycle models
68
+
69
+ - `CreateRunRequest`, `RunSummary`, `RunDetail`, `RunJob`
70
+ - `PollNextRunRequest`, `ClaimRunResponse`
71
+ - `StartRunRequest`, `StartRunDirectRequest`
72
+ - `CompleteRunRequest`, `FailRunRequest`, `UpdateRunOutputRequest`
73
+ - `AddRunMessageRequest`
74
+ - `InvokeAgentRequest` / `InvokeAgentResponse`
75
+ - `WorkerInvokeRequest` / `WorkerInvokeResponse`
76
+
77
+ ### agents.py -- Agent registration models
78
+
79
+ - `AgentManifest`, `HeartbeatRequest`, `HeartbeatResponse`
80
+ - `AgentInstanceResponse`
81
+
82
+ ### hook_types.py -- Hook event mapping
83
+
84
+ - `HOOK_EVENT_TYPE_MAP` -- maps Claude Code `HookEventName` to `HookEventType`
85
+
86
+ ### a2a.py -- A2A protocol wire types
87
+
88
+ - `A2AMessage`, `A2AMessagePart`
89
+ - `A2AMessageRequest`, `A2ATaskResponse`
90
+
91
+ ## Type Safety Guarantees
92
+
93
+ | Layer | Protection |
94
+ |-------|-----------|
95
+ | Build-time (mypy) | SDK and controlplane import the same Pydantic models -- field mismatches caught at type-check |
96
+ | Runtime (Pydantic) | Every event validated at construction and at API boundary |
97
+ | Wire (JSON) | `model_dump()` / `model_validate()` guarantee JSON matches schema |
98
+ | Hook events | `HookEventType` literal -- typos fail both type-check and runtime |
99
+
100
+ ## Dependencies
101
+
102
+ Only `pydantic>=2.0`. Zero internal dependencies.
@@ -0,0 +1,89 @@
1
+ # agentops-rpc
2
+
3
+ Shared Pydantic wire types for the AgentOps SDK and controlplane.
4
+
5
+ Both the SDK (`komodor-agentops`) and the controlplane import models from this
6
+ package, giving build-time (mypy) and runtime (Pydantic v2) type safety across
7
+ the wire boundary.
8
+
9
+ ## Install
10
+
11
+ ```bash
12
+ pip install agentops-rpc
13
+ ```
14
+
15
+ ## Quick Start
16
+
17
+ ```python
18
+ from agentops_rpc import RunEventIn, IngestRunEventsRequest, HOOK_EVENT_TYPE_MAP
19
+
20
+ event = RunEventIn(
21
+ event_type="span.started",
22
+ span_id="span_abc",
23
+ kind="tool",
24
+ name="Bash",
25
+ hook_event_type="tool.start",
26
+ )
27
+ event.model_dump() # JSON-safe dict
28
+
29
+ request = IngestRunEventsRequest(worker_id="wrk_1", events=[event])
30
+ ```
31
+
32
+ ## Exported Models
33
+
34
+ ### types.py -- Literal type aliases
35
+
36
+ | Type | Values |
37
+ |------|--------|
38
+ | `EventType` | `run.started`, `run.completed`, `run.failed`, `span.started`, `span.ended`, `message.created`, `log.created`, `token.created`, `custom.created`, `output.updated` |
39
+ | `RunStatus` | `queued`, `claimed`, `running`, `succeeded`, `failed` |
40
+ | `SpanKind` | `agent`, `llm`, `tool`, `chain`, `generation` |
41
+ | `HookEventName` | `SessionStart`, `UserPromptSubmit`, `PreToolUse`, `PostToolUse`, `PostToolUseFailure`, `SubagentStart`, `SubagentStop`, `Notification`, `Stop`, `SessionEnd`, `PermissionRequest`, `PreCompact`, `PostCompact` |
42
+ | `HookEventType` | `session.start`, `prompt.submit`, `tool.start`, `tool.end`, `tool.error`, `subagent.start`, `subagent.end`, `notification`, `session.stop`, `compaction.start`, `compaction.end` |
43
+ | `AgentStatus` | `online`, `offline` |
44
+ | `RunTransport` | `polling`, `api_sync`, `api_sse`, `a2a` |
45
+
46
+ ### events.py -- Event ingest models
47
+
48
+ - `RunEventIn` -- unified event sent SDK -> controlplane
49
+ - `TraceEventIn`, `LogEventIn` -- structured events for DB tables
50
+ - `IngestRunEventsRequest` / `IngestRunEventsResponse`
51
+ - `BatchEventsRequest`
52
+ - `TraceEventResponse`, `LogEventResponse`, `RunMessageResponse`
53
+
54
+ ### runs.py -- Run lifecycle models
55
+
56
+ - `CreateRunRequest`, `RunSummary`, `RunDetail`, `RunJob`
57
+ - `PollNextRunRequest`, `ClaimRunResponse`
58
+ - `StartRunRequest`, `StartRunDirectRequest`
59
+ - `CompleteRunRequest`, `FailRunRequest`, `UpdateRunOutputRequest`
60
+ - `AddRunMessageRequest`
61
+ - `InvokeAgentRequest` / `InvokeAgentResponse`
62
+ - `WorkerInvokeRequest` / `WorkerInvokeResponse`
63
+
64
+ ### agents.py -- Agent registration models
65
+
66
+ - `AgentManifest`, `HeartbeatRequest`, `HeartbeatResponse`
67
+ - `AgentInstanceResponse`
68
+
69
+ ### hook_types.py -- Hook event mapping
70
+
71
+ - `HOOK_EVENT_TYPE_MAP` -- maps Claude Code `HookEventName` to `HookEventType`
72
+
73
+ ### a2a.py -- A2A protocol wire types
74
+
75
+ - `A2AMessage`, `A2AMessagePart`
76
+ - `A2AMessageRequest`, `A2ATaskResponse`
77
+
78
+ ## Type Safety Guarantees
79
+
80
+ | Layer | Protection |
81
+ |-------|-----------|
82
+ | Build-time (mypy) | SDK and controlplane import the same Pydantic models -- field mismatches caught at type-check |
83
+ | Runtime (Pydantic) | Every event validated at construction and at API boundary |
84
+ | Wire (JSON) | `model_dump()` / `model_validate()` guarantee JSON matches schema |
85
+ | Hook events | `HookEventType` literal -- typos fail both type-check and runtime |
86
+
87
+ ## Dependencies
88
+
89
+ Only `pydantic>=2.0`. Zero internal dependencies.
@@ -0,0 +1,25 @@
1
+ [project]
2
+ name = "agentops-rpc"
3
+ version = "0.1.0"
4
+ description = "Shared Pydantic wire types for AgentOps SDK ↔ controlplane RPC."
5
+ readme = "README.md"
6
+ license = "LicenseRef-Komodor-Proprietary"
7
+ license-files = ["LICENSE"]
8
+ authors = [{ name = "Komodor Ltd." }]
9
+ requires-python = ">=3.11"
10
+ dependencies = [
11
+ "pydantic>=2.0",
12
+ ]
13
+
14
+ [project.urls]
15
+ Homepage = "https://agentops.komodor.com"
16
+
17
+ [build-system]
18
+ requires = ["setuptools>=77"]
19
+ build-backend = "setuptools.build_meta"
20
+
21
+ [tool.setuptools.package-dir]
22
+ "" = "src"
23
+
24
+ [tool.setuptools.packages.find]
25
+ where = ["src"]
@@ -0,0 +1,4 @@
1
+ [egg_info]
2
+ tag_build =
3
+ tag_date = 0
4
+
@@ -0,0 +1,393 @@
1
+ from __future__ import annotations
2
+
3
+ from .a2a import A2AMessage, A2AMessagePart, A2AMessageRequest, A2ATask, A2ATaskResponse
4
+ from .agent_create import (
5
+ CreateAgentMcpServer,
6
+ CreateAgentRequest,
7
+ CreateAgentResponse,
8
+ CreateAgentSkill,
9
+ CreateWorkerTokenRequest,
10
+ CreateWorkerTokenResponse,
11
+ WorkerTokenStatusResponse,
12
+ WorkerTokenSummaryResponse,
13
+ )
14
+ from .agents import (
15
+ BUILTIN_USE_CASE_SLUGS,
16
+ BUILTIN_USE_CASES,
17
+ CONNECTIVITY_CHECK_USE_CASE,
18
+ LABEL_MAX_LENGTH,
19
+ LABEL_RULE_HINT,
20
+ PUSH_CAPABILITY,
21
+ REVIEW_CAPABILITY,
22
+ SAMPLE_RUN_PROMPT,
23
+ SAMPLE_RUN_USE_CASE,
24
+ USE_CASE_INPUT_KEY,
25
+ USE_CASE_SLUG_HINT,
26
+ USE_CASE_SLUG_MAX_LENGTH,
27
+ AgentGenerationInfo,
28
+ AgentInstanceResponse,
29
+ AgentManifest,
30
+ BuiltinUseCaseSlug,
31
+ CreateAgentDraftRequest,
32
+ HeartbeatRequest,
33
+ HeartbeatResponse,
34
+ ScheduleTriggerManifest,
35
+ SkillDetail,
36
+ SkillReference,
37
+ SkillSourceAgent,
38
+ SkillSummary,
39
+ SlackKeywordTriggerManifest,
40
+ UpdateAgentDraftRequest,
41
+ UpdateAgentLabelsRequest,
42
+ UseCase,
43
+ WorkerInstanceInfo,
44
+ label_violations,
45
+ )
46
+ from .artifacts import ArtifactDetail, ArtifactSummary
47
+ from .credentials import (
48
+ SECRETS_METADATA_KEY,
49
+ BindCredentialRequest,
50
+ CreateCredentialRequest,
51
+ CredentialBinding,
52
+ CredentialMetadata,
53
+ CredentialStatus,
54
+ ReplaceCredentialValueRequest,
55
+ UpdateCredentialMetadataRequest,
56
+ WorkerSecretsResponse,
57
+ )
58
+ from .events import (
59
+ HANDLER_FAILED_LOG_PREFIX,
60
+ BatchEventsRequest,
61
+ IngestRunEventsRequest,
62
+ IngestRunEventsResponse,
63
+ LogEventIn,
64
+ LogEventResponse,
65
+ RunEventIn,
66
+ RunMessageResponse,
67
+ TraceEventIn,
68
+ TraceEventResponse,
69
+ )
70
+ from .grader import (
71
+ GRADER_CAPABILITY,
72
+ GradeContext,
73
+ GradeResult,
74
+ )
75
+ from .hosted_agents import (
76
+ HostedAgentCreateRequest,
77
+ HostedAgentImage,
78
+ HostedAgentMcpServer,
79
+ HostedAgentResponse,
80
+ HostedAgentSkill,
81
+ HostedAgentUpdateRequest,
82
+ )
83
+ from .insights import (
84
+ FailureCategoryCount,
85
+ IssueRecommendation,
86
+ RunIssue,
87
+ SaveIssueRecommendationRequest,
88
+ WorkspaceIssuesResponse,
89
+ )
90
+ from .integrations import (
91
+ CreateIntegrationConnectionRequest,
92
+ CreateIntegrationConnectionResponse,
93
+ IntegrationAuthMode,
94
+ IntegrationCatalogEntry,
95
+ IntegrationConnection,
96
+ IntegrationConnectionDetail,
97
+ IntegrationCredentialField,
98
+ IntegrationFieldType,
99
+ IntegrationOAuthMetadata,
100
+ IntegrationProviderCredentials,
101
+ IntegrationStatus,
102
+ StartIntegrationAppRequest,
103
+ StartIntegrationAppResponse,
104
+ StartIntegrationOAuthRequest,
105
+ StartIntegrationOAuthResponse,
106
+ TestIntegrationConnectionResponse,
107
+ )
108
+ from .issues import (
109
+ IssueDetail,
110
+ IssueGrouping,
111
+ IssueListResponse,
112
+ IssueLog,
113
+ IssueRca,
114
+ IssueSignal,
115
+ IssueSource,
116
+ IssueTimelineEvent,
117
+ IssueWorker,
118
+ UpdateIssueRequest,
119
+ )
120
+ from .knowledge_base import (
121
+ AttachAgentToKBRequest,
122
+ CreateKnowledgeBaseRequest,
123
+ KBAgentGrant,
124
+ KBChunkSummary,
125
+ KBDocumentDetail,
126
+ KBDocumentMetadata,
127
+ KBDocumentOrigin,
128
+ KBDocumentStatus,
129
+ KBDocumentSummary,
130
+ KBSearchFilters,
131
+ KBSearchHit,
132
+ KBSearchRequest,
133
+ KBSearchResponse,
134
+ KnowledgeBaseDetail,
135
+ KnowledgeBaseSummary,
136
+ )
137
+ from .quality import (
138
+ AgentQualityRollup,
139
+ AutoScoreSeries,
140
+ RunQualityRow,
141
+ SkillQualityRollup,
142
+ UngradedQueueResponse,
143
+ )
144
+ from .runs import (
145
+ ActivityBucket,
146
+ AddRunMessageRequest,
147
+ AgentActivitySummary,
148
+ ClaimRunResponse,
149
+ CompleteRunRequest,
150
+ CreateRunRequest,
151
+ FailRunRequest,
152
+ InvokeAgentRequest,
153
+ InvokeAgentResponse,
154
+ ModelActivity,
155
+ PollNextRunRequest,
156
+ RunDetail,
157
+ RunJob,
158
+ RunSummary,
159
+ RunUsage,
160
+ StartRunDirectRequest,
161
+ StartRunRequest,
162
+ TopCostRunSummary,
163
+ UpdateRunOutputRequest,
164
+ WorkerInvokeAgent,
165
+ WorkerInvokeRequest,
166
+ WorkerInvokeResponse,
167
+ WorkspaceActivityResponse,
168
+ )
169
+ from .schedules import CreateScheduleRequest, ScheduleResponse, UpdateScheduleRequest
170
+ from .scores import (
171
+ CreateScoreRequest,
172
+ ScoreListResponse,
173
+ ScoreResponse,
174
+ ScoreTargetType,
175
+ ScoreValue,
176
+ UpdateScoreRequest,
177
+ )
178
+ from .session import SESSION_SNAPSHOT_METADATA_KEY, SessionSnapshot, SnapshotComponent, SnapshotComponentKind
179
+ from .types import (
180
+ ACTIVE_RUN_STATUSES,
181
+ OBSERVABILITY_EVENT_TYPES,
182
+ OWNED_RUN_STATUSES,
183
+ TERMINAL_RUN_STATUSES,
184
+ AgentStatus,
185
+ DeployStatus,
186
+ EventType,
187
+ HookEventName,
188
+ HookEventType,
189
+ LogLevel,
190
+ MessageRole,
191
+ MessageType,
192
+ RunChannel,
193
+ RunEventType,
194
+ RunStatus,
195
+ RunTransport,
196
+ SpanKind,
197
+ TraceEventType,
198
+ )
199
+ from .worker_catalog import (
200
+ WorkerCatalogCredential,
201
+ WorkerCatalogDeployRequest,
202
+ WorkerCatalogEntryResponse,
203
+ WorkerCatalogIntegration,
204
+ WorkerCatalogSelfHostedDeployRequest,
205
+ WorkerCatalogSelfHostedDeployResponse,
206
+ )
207
+
208
+ __all__ = [
209
+ "ACTIVE_RUN_STATUSES",
210
+ "HANDLER_FAILED_LOG_PREFIX",
211
+ "OWNED_RUN_STATUSES",
212
+ "TERMINAL_RUN_STATUSES",
213
+ "ArtifactDetail",
214
+ "ArtifactSummary",
215
+ "A2AMessage",
216
+ "A2AMessagePart",
217
+ "A2AMessageRequest",
218
+ "A2ATask",
219
+ "A2ATaskResponse",
220
+ "AddRunMessageRequest",
221
+ "AgentActivitySummary",
222
+ "AgentInstanceResponse",
223
+ "AgentGenerationInfo",
224
+ "AgentManifest",
225
+ "AgentStatus",
226
+ "DeployStatus",
227
+ "PUSH_CAPABILITY",
228
+ "REVIEW_CAPABILITY",
229
+ "UseCase",
230
+ "BuiltinUseCaseSlug",
231
+ "BUILTIN_USE_CASES",
232
+ "BUILTIN_USE_CASE_SLUGS",
233
+ "CONNECTIVITY_CHECK_USE_CASE",
234
+ "SAMPLE_RUN_USE_CASE",
235
+ "SAMPLE_RUN_PROMPT",
236
+ "USE_CASE_INPUT_KEY",
237
+ "USE_CASE_SLUG_HINT",
238
+ "USE_CASE_SLUG_MAX_LENGTH",
239
+ "ActivityBucket",
240
+ "BatchEventsRequest",
241
+ "ClaimRunResponse",
242
+ "CompleteRunRequest",
243
+ "BindCredentialRequest",
244
+ "CreateAgentDraftRequest",
245
+ "CreateAgentMcpServer",
246
+ "CreateAgentRequest",
247
+ "CreateAgentResponse",
248
+ "CreateAgentSkill",
249
+ "CreateWorkerTokenRequest",
250
+ "CreateWorkerTokenResponse",
251
+ "WorkerTokenStatusResponse",
252
+ "WorkerTokenSummaryResponse",
253
+ "CreateCredentialRequest",
254
+ "CreateRunRequest",
255
+ "CredentialBinding",
256
+ "CredentialMetadata",
257
+ "CredentialStatus",
258
+ "ReplaceCredentialValueRequest",
259
+ "WorkerSecretsResponse",
260
+ "WorkerInstanceInfo",
261
+ "SECRETS_METADATA_KEY",
262
+ "SESSION_SNAPSHOT_METADATA_KEY",
263
+ "SessionSnapshot",
264
+ "SnapshotComponent",
265
+ "SnapshotComponentKind",
266
+ "UpdateCredentialMetadataRequest",
267
+ "EventType",
268
+ "FailRunRequest",
269
+ "GRADER_CAPABILITY",
270
+ "GradeContext",
271
+ "GradeResult",
272
+ "HeartbeatRequest",
273
+ "HeartbeatResponse",
274
+ "HookEventName",
275
+ "HookEventType",
276
+ "HostedAgentCreateRequest",
277
+ "HostedAgentImage",
278
+ "HostedAgentMcpServer",
279
+ "HostedAgentResponse",
280
+ "HostedAgentSkill",
281
+ "HostedAgentUpdateRequest",
282
+ "WorkerCatalogCredential",
283
+ "WorkerCatalogDeployRequest",
284
+ "WorkerCatalogEntryResponse",
285
+ "WorkerCatalogIntegration",
286
+ "WorkerCatalogSelfHostedDeployRequest",
287
+ "WorkerCatalogSelfHostedDeployResponse",
288
+ "IngestRunEventsRequest",
289
+ "IngestRunEventsResponse",
290
+ "InvokeAgentRequest",
291
+ "InvokeAgentResponse",
292
+ "LogEventIn",
293
+ "LogEventResponse",
294
+ "LogLevel",
295
+ "MessageRole",
296
+ "MessageType",
297
+ "ModelActivity",
298
+ "OBSERVABILITY_EVENT_TYPES",
299
+ "PollNextRunRequest",
300
+ "RunDetail",
301
+ "RunEventIn",
302
+ "RunChannel",
303
+ "RunEventType",
304
+ "RunJob",
305
+ "RunMessageResponse",
306
+ "RunStatus",
307
+ "RunSummary",
308
+ "RunTransport",
309
+ "RunUsage",
310
+ "ScheduleResponse",
311
+ "ScheduleTriggerManifest",
312
+ "SlackKeywordTriggerManifest",
313
+ "CreateIntegrationConnectionRequest",
314
+ "CreateIntegrationConnectionResponse",
315
+ "CreateScheduleRequest",
316
+ "IntegrationAuthMode",
317
+ "IntegrationCatalogEntry",
318
+ "IntegrationConnection",
319
+ "IntegrationConnectionDetail",
320
+ "IntegrationCredentialField",
321
+ "IntegrationFieldType",
322
+ "IntegrationOAuthMetadata",
323
+ "IntegrationProviderCredentials",
324
+ "IntegrationStatus",
325
+ "StartIntegrationAppRequest",
326
+ "StartIntegrationAppResponse",
327
+ "StartIntegrationOAuthRequest",
328
+ "StartIntegrationOAuthResponse",
329
+ "TestIntegrationConnectionResponse",
330
+ "UpdateScheduleRequest",
331
+ "SkillDetail",
332
+ "SkillReference",
333
+ "SkillSourceAgent",
334
+ "SkillSummary",
335
+ "SpanKind",
336
+ "StartRunDirectRequest",
337
+ "StartRunRequest",
338
+ "TraceEventIn",
339
+ "TraceEventResponse",
340
+ "TraceEventType",
341
+ "TopCostRunSummary",
342
+ "AttachAgentToKBRequest",
343
+ "CreateKnowledgeBaseRequest",
344
+ "KBAgentGrant",
345
+ "KBChunkSummary",
346
+ "KBDocumentDetail",
347
+ "KBDocumentMetadata",
348
+ "KBDocumentOrigin",
349
+ "KBDocumentStatus",
350
+ "KBDocumentSummary",
351
+ "KBSearchFilters",
352
+ "KBSearchHit",
353
+ "KBSearchRequest",
354
+ "KBSearchResponse",
355
+ "KnowledgeBaseDetail",
356
+ "KnowledgeBaseSummary",
357
+ "UpdateAgentDraftRequest",
358
+ "UpdateAgentLabelsRequest",
359
+ "LABEL_MAX_LENGTH",
360
+ "LABEL_RULE_HINT",
361
+ "label_violations",
362
+ "UpdateRunOutputRequest",
363
+ "WorkerInvokeAgent",
364
+ "WorkerInvokeRequest",
365
+ "WorkerInvokeResponse",
366
+ "WorkspaceActivityResponse",
367
+ "FailureCategoryCount",
368
+ "IssueRecommendation",
369
+ "RunIssue",
370
+ "SaveIssueRecommendationRequest",
371
+ "WorkspaceIssuesResponse",
372
+ "IssueDetail",
373
+ "IssueGrouping",
374
+ "IssueListResponse",
375
+ "IssueLog",
376
+ "IssueRca",
377
+ "IssueSignal",
378
+ "IssueSource",
379
+ "IssueTimelineEvent",
380
+ "IssueWorker",
381
+ "UpdateIssueRequest",
382
+ "CreateScoreRequest",
383
+ "ScoreListResponse",
384
+ "ScoreResponse",
385
+ "ScoreTargetType",
386
+ "ScoreValue",
387
+ "UpdateScoreRequest",
388
+ "AgentQualityRollup",
389
+ "AutoScoreSeries",
390
+ "RunQualityRow",
391
+ "SkillQualityRollup",
392
+ "UngradedQueueResponse",
393
+ ]