capo-agent-registry 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 (106) hide show
  1. capo_agent_registry-0.1.0/LICENSE +21 -0
  2. capo_agent_registry-0.1.0/PKG-INFO +103 -0
  3. capo_agent_registry-0.1.0/README.md +75 -0
  4. capo_agent_registry-0.1.0/pyproject.toml +62 -0
  5. capo_agent_registry-0.1.0/setup.cfg +4 -0
  6. capo_agent_registry-0.1.0/src/capo_agent_registry/__init__.py +73 -0
  7. capo_agent_registry-0.1.0/src/capo_agent_registry/_async.py +25 -0
  8. capo_agent_registry-0.1.0/src/capo_agent_registry/_auth/_identity.py +16 -0
  9. capo_agent_registry-0.1.0/src/capo_agent_registry/_auth/_providers.py +886 -0
  10. capo_agent_registry-0.1.0/src/capo_agent_registry/_auth/_signers.py +125 -0
  11. capo_agent_registry-0.1.0/src/capo_agent_registry/_auth/_sigv4.py +579 -0
  12. capo_agent_registry-0.1.0/src/capo_agent_registry/_auth/_zapros_handler.py +80 -0
  13. capo_agent_registry-0.1.0/src/capo_agent_registry/_body.py +221 -0
  14. capo_agent_registry-0.1.0/src/capo_agent_registry/_iter.py +113 -0
  15. capo_agent_registry-0.1.0/src/capo_agent_registry/_operations/agent_registry/batch_get_discoverable_registry_record.py +169 -0
  16. capo_agent_registry-0.1.0/src/capo_agent_registry/_operations/agent_registry/list_discoverable_registry_records.py +172 -0
  17. capo_agent_registry-0.1.0/src/capo_agent_registry/_operations/agent_registry/search_discoverable_registry_records.py +168 -0
  18. capo_agent_registry-0.1.0/src/capo_agent_registry/_pagination.py +21 -0
  19. capo_agent_registry-0.1.0/src/capo_agent_registry/_protocol/__init__.py +1 -0
  20. capo_agent_registry-0.1.0/src/capo_agent_registry/_protocol/errors.py +93 -0
  21. capo_agent_registry-0.1.0/src/capo_agent_registry/_protocol/eventstream.py +240 -0
  22. capo_agent_registry-0.1.0/src/capo_agent_registry/_protocol/serialize.py +47 -0
  23. capo_agent_registry-0.1.0/src/capo_agent_registry/_protocol/xml.py +33 -0
  24. capo_agent_registry-0.1.0/src/capo_agent_registry/_rule_engine/__init__.py +0 -0
  25. capo_agent_registry-0.1.0/src/capo_agent_registry/_rule_engine/_aws_partition.py +160 -0
  26. capo_agent_registry-0.1.0/src/capo_agent_registry/_rule_engine/_endpoint_rule_set.py +38 -0
  27. capo_agent_registry-0.1.0/src/capo_agent_registry/_rule_engine/_endpoint_runtime.py +389 -0
  28. capo_agent_registry-0.1.0/src/capo_agent_registry/_services/_aws_config.py +152 -0
  29. capo_agent_registry-0.1.0/src/capo_agent_registry/_services/_pipeline.py +243 -0
  30. capo_agent_registry-0.1.0/src/capo_agent_registry/_services/agent_registry.py +330 -0
  31. capo_agent_registry-0.1.0/src/capo_agent_registry/_services/async_agent_registry.py +333 -0
  32. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/__init__.py +33 -0
  33. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/_base.py +94 -0
  34. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/access_denied_exception.py +51 -0
  35. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/internal_server_exception.py +51 -0
  36. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/resource_not_found_exception.py +51 -0
  37. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/throttling_exception.py +49 -0
  38. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/unauthorized_exception.py +51 -0
  39. capo_agent_registry-0.1.0/src/capo_agent_registry/errors/validation_exception.py +90 -0
  40. capo_agent_registry-0.1.0/src/capo_agent_registry/py.typed +0 -0
  41. capo_agent_registry-0.1.0/src/capo_agent_registry/types/_prelude/blob.py +12 -0
  42. capo_agent_registry-0.1.0/src/capo_agent_registry/types/_prelude/timestamp.py +17 -0
  43. capo_agent_registry-0.1.0/src/capo_agent_registry/types/a2a_agent_card_descriptor.py +52 -0
  44. capo_agent_registry-0.1.0/src/capo_agent_registry/types/ag_ui_descriptor.py +36 -0
  45. capo_agent_registry-0.1.0/src/capo_agent_registry/types/agent_skills_additional_data.py +42 -0
  46. capo_agent_registry-0.1.0/src/capo_agent_registry/types/agent_skills_definition_descriptor.py +58 -0
  47. capo_agent_registry-0.1.0/src/capo_agent_registry/types/agent_skills_md_descriptor.py +52 -0
  48. capo_agent_registry-0.1.0/src/capo_agent_registry/types/batch_get_discoverable_registry_record_error.py +71 -0
  49. capo_agent_registry-0.1.0/src/capo_agent_registry/types/batch_get_discoverable_registry_record_error_code.py +18 -0
  50. capo_agent_registry-0.1.0/src/capo_agent_registry/types/batch_get_discoverable_registry_record_error_list.py +39 -0
  51. capo_agent_registry-0.1.0/src/capo_agent_registry/types/batch_get_discoverable_registry_record_request.py +47 -0
  52. capo_agent_registry-0.1.0/src/capo_agent_registry/types/batch_get_discoverable_registry_record_response.py +67 -0
  53. capo_agent_registry-0.1.0/src/capo_agent_registry/types/custom_descriptor.py +28 -0
  54. capo_agent_registry-0.1.0/src/capo_agent_registry/types/data_schema_version.py +6 -0
  55. capo_agent_registry-0.1.0/src/capo_agent_registry/types/date_timestamp.py +21 -0
  56. capo_agent_registry-0.1.0/src/capo_agent_registry/types/description.py +6 -0
  57. capo_agent_registry-0.1.0/src/capo_agent_registry/types/descriptor_data.py +6 -0
  58. capo_agent_registry-0.1.0/src/capo_agent_registry/types/descriptor_source.py +42 -0
  59. capo_agent_registry-0.1.0/src/capo_agent_registry/types/descriptor_source_from_url.py +31 -0
  60. capo_agent_registry-0.1.0/src/capo_agent_registry/types/descriptor_source_url.py +6 -0
  61. capo_agent_registry-0.1.0/src/capo_agent_registry/types/descriptor_type_list.py +14 -0
  62. capo_agent_registry-0.1.0/src/capo_agent_registry/types/descriptors.py +129 -0
  63. capo_agent_registry-0.1.0/src/capo_agent_registry/types/discoverable_filter_values.py +19 -0
  64. capo_agent_registry-0.1.0/src/capo_agent_registry/types/discoverable_registry_record_summary.py +181 -0
  65. capo_agent_registry-0.1.0/src/capo_agent_registry/types/discoverable_registry_record_summary_list.py +39 -0
  66. capo_agent_registry-0.1.0/src/capo_agent_registry/types/filter_value.py +6 -0
  67. capo_agent_registry-0.1.0/src/capo_agent_registry/types/http_descriptor.py +36 -0
  68. capo_agent_registry-0.1.0/src/capo_agent_registry/types/list_discoverable_registry_records_request.py +57 -0
  69. capo_agent_registry-0.1.0/src/capo_agent_registry/types/list_discoverable_registry_records_response.py +51 -0
  70. capo_agent_registry-0.1.0/src/capo_agent_registry/types/mcp_server_additional_data.py +38 -0
  71. capo_agent_registry-0.1.0/src/capo_agent_registry/types/mcp_server_descriptor.py +73 -0
  72. capo_agent_registry-0.1.0/src/capo_agent_registry/types/mcp_tools_descriptor.py +37 -0
  73. capo_agent_registry-0.1.0/src/capo_agent_registry/types/metadata_filter_expression.py +5 -0
  74. capo_agent_registry-0.1.0/src/capo_agent_registry/types/non_blank_string.py +5 -0
  75. capo_agent_registry-0.1.0/src/capo_agent_registry/types/record_identifier.py +6 -0
  76. capo_agent_registry-0.1.0/src/capo_agent_registry/types/record_type.py +21 -0
  77. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_arn.py +6 -0
  78. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_id_list.py +19 -0
  79. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_identifier.py +6 -0
  80. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_arn.py +6 -0
  81. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_display_name.py +6 -0
  82. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_filter.py +63 -0
  83. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_filter_list.py +35 -0
  84. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_filter_name.py +18 -0
  85. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_id.py +6 -0
  86. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_id_list.py +19 -0
  87. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_name.py +6 -0
  88. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_status.py +25 -0
  89. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_summary.py +162 -0
  90. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_summary_list.py +35 -0
  91. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_record_version.py +6 -0
  92. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_records_entry.py +49 -0
  93. capo_agent_registry-0.1.0/src/capo_agent_registry/types/registry_records_entry_list.py +35 -0
  94. capo_agent_registry-0.1.0/src/capo_agent_registry/types/search_discoverable_registry_records_request.py +69 -0
  95. capo_agent_registry-0.1.0/src/capo_agent_registry/types/search_discoverable_registry_records_response.py +45 -0
  96. capo_agent_registry-0.1.0/src/capo_agent_registry/types/search_query.py +5 -0
  97. capo_agent_registry-0.1.0/src/capo_agent_registry/types/validation_exception_field.py +33 -0
  98. capo_agent_registry-0.1.0/src/capo_agent_registry/types/validation_exception_field_list.py +35 -0
  99. capo_agent_registry-0.1.0/src/capo_agent_registry/types/validation_exception_reason.py +20 -0
  100. capo_agent_registry-0.1.0/src/capo_agent_registry.egg-info/PKG-INFO +103 -0
  101. capo_agent_registry-0.1.0/src/capo_agent_registry.egg-info/SOURCES.txt +104 -0
  102. capo_agent_registry-0.1.0/src/capo_agent_registry.egg-info/dependency_links.txt +1 -0
  103. capo_agent_registry-0.1.0/src/capo_agent_registry.egg-info/requires.txt +9 -0
  104. capo_agent_registry-0.1.0/src/capo_agent_registry.egg-info/top_level.txt +1 -0
  105. capo_agent_registry-0.1.0/tests/test_body.py +225 -0
  106. capo_agent_registry-0.1.0/tests/test_sigv4.py +758 -0
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2026 Karen Petrosyan
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1,103 @@
1
+ Metadata-Version: 2.4
2
+ Name: capo-agent-registry
3
+ Version: 0.1.0
4
+ Summary: Python SDK for Agent Registry.
5
+ Classifier: Development Status :: 4 - Beta
6
+ Classifier: Intended Audience :: Developers
7
+ Classifier: License :: OSI Approved :: MIT License
8
+ Classifier: Operating System :: OS Independent
9
+ Classifier: Programming Language :: Python :: 3
10
+ Classifier: Programming Language :: Python :: 3.10
11
+ Classifier: Programming Language :: Python :: 3.11
12
+ Classifier: Programming Language :: Python :: 3.12
13
+ Classifier: Programming Language :: Python :: 3.13
14
+ Classifier: Programming Language :: Python :: 3.14
15
+ Classifier: Topic :: Software Development :: Code Generators
16
+ Classifier: Typing :: Typed
17
+ Requires-Python: >=3.10
18
+ Description-Content-Type: text/markdown
19
+ License-File: LICENSE
20
+ Requires-Dist: zapros>=0.13.0
21
+ Provides-Extra: anyio
22
+ Requires-Dist: anyio>=3.0; extra == "anyio"
23
+ Provides-Extra: sso
24
+ Requires-Dist: capo-sso>=0.2.0; extra == "sso"
25
+ Requires-Dist: capo-sso-oidc>=0.2.0; extra == "sso"
26
+ Requires-Dist: capo-sts>=0.3.0; extra == "sso"
27
+ Dynamic: license-file
28
+
29
+ # Getting Started
30
+
31
+ ## Installation
32
+
33
+ ```
34
+ pip install capo-agent-registry
35
+ ```
36
+
37
+ ## Usage
38
+
39
+ ```python
40
+ from capo_agent_registry import AsyncAgentRegistryClient
41
+
42
+
43
+ async def main():
44
+ async with AsyncAgentRegistryClient() as agent_registry:
45
+ # Example: call the batch_get_discoverable_registry_record operation
46
+ response = await agent_registry.batch_get_discoverable_registry_record()
47
+ print(response["registry_records"])
48
+ ```
49
+
50
+ ## Pagination
51
+
52
+ Some operations in this SDK support pagination. If the operation supports pagination it will have an `iter_` prefixed method that returns an async iterator.
53
+
54
+ ```python
55
+ from capo_agent_registry import AsyncAgentRegistryClient
56
+
57
+
58
+ async def main():
59
+ async with AsyncAgentRegistryClient() as agent_registry:
60
+ # Example: paginate over list_discoverable_registry_records
61
+ async for item in agent_registry.iter_list_discoverable_registry_records():
62
+ print(item)
63
+ ```
64
+
65
+ ## Error Handling
66
+
67
+ The SDK raises exceptions for errors returned by the API. Catch them to handle failures gracefully.
68
+
69
+ ```python
70
+ from capo_agent_registry import AsyncAgentRegistryClient
71
+ from capo_agent_registry.error import AccessDeniedException
72
+
73
+
74
+ async def main():
75
+ async with AsyncAgentRegistryClient() as agent_registry:
76
+ try:
77
+ await agent_registry.batch_get_discoverable_registry_record()
78
+ except AccessDeniedException as e:
79
+ print(f"Error: {e}")
80
+ print(e.data) # additional error data
81
+ ```
82
+
83
+ ## Retrying
84
+
85
+ The SDK retries failed operations automatically. Retry behaviour follows the Smithy specification: errors are retried based on their `is_retryable` and `is_throttling_error` attributes. Throttling errors use a longer base delay. Network-level failures (connection errors and timeouts) are also retried. Non-retryable errors, such as client errors without the `@retryable` trait, are raised immediately without further attempts.
86
+
87
+ The number of attempts defaults to 3 and can be changed at the client level via `retry_max_attempts`, or per call via `config_overrides`.
88
+
89
+ ```python
90
+ from capo_agent_registry import AsyncAgentRegistryClient
91
+
92
+
93
+ async def main():
94
+ async with AsyncAgentRegistryClient() as agent_registry:
95
+ # Default: 3 attempts for every operation
96
+ response = await agent_registry.batch_get_discoverable_registry_record()
97
+
98
+ # Override per operation
99
+ response = await agent_registry.batch_get_discoverable_registry_record(config_overrides={"retry_max_attempts": 5})
100
+
101
+ # Disable retries for this call
102
+ response = await agent_registry.batch_get_discoverable_registry_record(config_overrides={"retry_max_attempts": 1})
103
+ ```
@@ -0,0 +1,75 @@
1
+ # Getting Started
2
+
3
+ ## Installation
4
+
5
+ ```
6
+ pip install capo-agent-registry
7
+ ```
8
+
9
+ ## Usage
10
+
11
+ ```python
12
+ from capo_agent_registry import AsyncAgentRegistryClient
13
+
14
+
15
+ async def main():
16
+ async with AsyncAgentRegistryClient() as agent_registry:
17
+ # Example: call the batch_get_discoverable_registry_record operation
18
+ response = await agent_registry.batch_get_discoverable_registry_record()
19
+ print(response["registry_records"])
20
+ ```
21
+
22
+ ## Pagination
23
+
24
+ Some operations in this SDK support pagination. If the operation supports pagination it will have an `iter_` prefixed method that returns an async iterator.
25
+
26
+ ```python
27
+ from capo_agent_registry import AsyncAgentRegistryClient
28
+
29
+
30
+ async def main():
31
+ async with AsyncAgentRegistryClient() as agent_registry:
32
+ # Example: paginate over list_discoverable_registry_records
33
+ async for item in agent_registry.iter_list_discoverable_registry_records():
34
+ print(item)
35
+ ```
36
+
37
+ ## Error Handling
38
+
39
+ The SDK raises exceptions for errors returned by the API. Catch them to handle failures gracefully.
40
+
41
+ ```python
42
+ from capo_agent_registry import AsyncAgentRegistryClient
43
+ from capo_agent_registry.error import AccessDeniedException
44
+
45
+
46
+ async def main():
47
+ async with AsyncAgentRegistryClient() as agent_registry:
48
+ try:
49
+ await agent_registry.batch_get_discoverable_registry_record()
50
+ except AccessDeniedException as e:
51
+ print(f"Error: {e}")
52
+ print(e.data) # additional error data
53
+ ```
54
+
55
+ ## Retrying
56
+
57
+ The SDK retries failed operations automatically. Retry behaviour follows the Smithy specification: errors are retried based on their `is_retryable` and `is_throttling_error` attributes. Throttling errors use a longer base delay. Network-level failures (connection errors and timeouts) are also retried. Non-retryable errors, such as client errors without the `@retryable` trait, are raised immediately without further attempts.
58
+
59
+ The number of attempts defaults to 3 and can be changed at the client level via `retry_max_attempts`, or per call via `config_overrides`.
60
+
61
+ ```python
62
+ from capo_agent_registry import AsyncAgentRegistryClient
63
+
64
+
65
+ async def main():
66
+ async with AsyncAgentRegistryClient() as agent_registry:
67
+ # Default: 3 attempts for every operation
68
+ response = await agent_registry.batch_get_discoverable_registry_record()
69
+
70
+ # Override per operation
71
+ response = await agent_registry.batch_get_discoverable_registry_record(config_overrides={"retry_max_attempts": 5})
72
+
73
+ # Disable retries for this call
74
+ response = await agent_registry.batch_get_discoverable_registry_record(config_overrides={"retry_max_attempts": 1})
75
+ ```
@@ -0,0 +1,62 @@
1
+ [project]
2
+ name = "capo-agent-registry"
3
+ version = "0.1.0"
4
+ description = "Python SDK for Agent Registry."
5
+ readme = "README.md"
6
+ requires-python = ">=3.10"
7
+ classifiers = [
8
+ "Development Status :: 4 - Beta",
9
+ "Intended Audience :: Developers",
10
+ "License :: OSI Approved :: MIT License",
11
+ "Operating System :: OS Independent",
12
+ "Programming Language :: Python :: 3",
13
+ "Programming Language :: Python :: 3.10",
14
+ "Programming Language :: Python :: 3.11",
15
+ "Programming Language :: Python :: 3.12",
16
+ "Programming Language :: Python :: 3.13",
17
+ "Programming Language :: Python :: 3.14",
18
+ "Topic :: Software Development :: Code Generators",
19
+ "Typing :: Typed",
20
+ ]
21
+ dependencies = [
22
+ "zapros>=0.13.0",
23
+ ]
24
+
25
+ [project.optional-dependencies]
26
+ # Async file request bodies via Body.async_from_path
27
+ anyio = ["anyio>=3.0"]
28
+ # IAM Identity Center (SSO), assume-role and web-identity credentials providers
29
+ sso = [
30
+ "capo-sso>=0.2.0",
31
+ "capo-sso-oidc>=0.2.0",
32
+ "capo-sts>=0.3.0",
33
+ ]
34
+
35
+ [tool.pytest.ini_options]
36
+ pythonpath = ["src"]
37
+ testpaths = ["tests"]
38
+
39
+ [tool.ty.environment]
40
+ python-version = "3.14"
41
+ root = ["./src"]
42
+
43
+ [tool.ty.src]
44
+ include = ["src", "tests"]
45
+
46
+ [tool.ty.rules]
47
+ missing-typed-dict-key = "ignore"
48
+ unused-type-ignore-comment = "ignore"
49
+
50
+ [tool.ruff.lint]
51
+ select = ["E", "F", "I"]
52
+ ignore = ["E501", "E741"]
53
+
54
+ [dependency-groups]
55
+ dev = [
56
+ "inline-snapshot>=0.33.0",
57
+ # TODO: fix for 0.0.69
58
+ "ty==0.0.68",
59
+ "pytest>=9.0.3",
60
+ "ruff>=0.15.14",
61
+ "trio",
62
+ ]
@@ -0,0 +1,4 @@
1
+ [egg_info]
2
+ tag_build =
3
+ tag_date = 0
4
+
@@ -0,0 +1,73 @@
1
+ from __future__ import annotations
2
+
3
+ from . import _iter as _iter
4
+ from . import _protocol as _protocol
5
+ from ._auth._identity import Credentials as Credentials
6
+ from ._auth._identity import Identity as Identity
7
+ from ._auth._providers import (
8
+ AssumeRoleCredentialsProvider as AssumeRoleCredentialsProvider,
9
+ )
10
+ from ._auth._providers import (
11
+ AssumeRoleError as AssumeRoleError,
12
+ )
13
+ from ._auth._providers import (
14
+ CachedProvider as CachedProvider,
15
+ )
16
+ from ._auth._providers import (
17
+ ChainedProvider as ChainedProvider,
18
+ )
19
+ from ._auth._providers import (
20
+ CredentialsProvider as CredentialsProvider,
21
+ )
22
+ from ._auth._providers import (
23
+ EnvCredentialsProvider as EnvCredentialsProvider,
24
+ )
25
+ from ._auth._providers import (
26
+ IdentityNotFound as IdentityNotFound,
27
+ )
28
+ from ._auth._providers import (
29
+ IdentityProvider as IdentityProvider,
30
+ )
31
+ from ._auth._providers import (
32
+ MissingDependencyError as MissingDependencyError,
33
+ )
34
+ from ._auth._providers import (
35
+ ProfileCredentialsProvider as ProfileCredentialsProvider,
36
+ )
37
+ from ._auth._providers import (
38
+ SsoCredentialsProvider as SsoCredentialsProvider,
39
+ )
40
+ from ._auth._providers import (
41
+ SSOError as SSOError,
42
+ )
43
+ from ._auth._providers import (
44
+ StaticAwsCredentialsProvider as StaticAwsCredentialsProvider,
45
+ )
46
+ from ._auth._providers import (
47
+ WebIdentityCredentialsProvider as WebIdentityCredentialsProvider,
48
+ )
49
+ from ._auth._signers import Signer as Signer
50
+ from ._auth._signers import SigV4Signer as SigV4Signer
51
+ from ._body import Body as Body
52
+ from ._services._pipeline import (
53
+ AsyncOperationOptions as AsyncOperationOptions,
54
+ )
55
+ from ._services._pipeline import (
56
+ AsyncOperationRequest as AsyncOperationRequest,
57
+ )
58
+ from ._services._pipeline import (
59
+ AsyncOperationResponse as AsyncOperationResponse,
60
+ )
61
+ from ._services._pipeline import (
62
+ OperationOptions as OperationOptions,
63
+ )
64
+ from ._services._pipeline import (
65
+ OperationRequest as OperationRequest,
66
+ )
67
+ from ._services._pipeline import (
68
+ OperationResponse as OperationResponse,
69
+ )
70
+ from ._services.agent_registry import AgentRegistryClient as AgentRegistryClient
71
+ from ._services.async_agent_registry import (
72
+ AsyncAgentRegistryClient as AsyncAgentRegistryClient,
73
+ )
@@ -0,0 +1,25 @@
1
+ from __future__ import annotations
2
+
3
+ import asyncio
4
+ from typing import TYPE_CHECKING
5
+
6
+ if TYPE_CHECKING:
7
+ import trio
8
+ else:
9
+ try:
10
+ import trio
11
+ except ImportError:
12
+ trio = None
13
+
14
+
15
+ def in_trio_run() -> bool:
16
+ if trio is None:
17
+ return False
18
+ return trio.lowlevel.in_trio_run()
19
+
20
+
21
+ async def anysleep(delay: float) -> None:
22
+ if in_trio_run():
23
+ await trio.sleep(delay)
24
+ else:
25
+ await asyncio.sleep(delay)
@@ -0,0 +1,16 @@
1
+ from __future__ import annotations
2
+
3
+ from datetime import datetime
4
+ from typing import TypedDict
5
+
6
+ from typing_extensions import NotRequired
7
+
8
+
9
+ class Identity(TypedDict):
10
+ expiration: NotRequired[datetime | None]
11
+
12
+
13
+ class Credentials(Identity):
14
+ access_key: str
15
+ secret_key: str
16
+ session_token: NotRequired[str | None]