augments-adk 0.3.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 (653) hide show
  1. augments_adk-0.3.0/LICENSE +21 -0
  2. augments_adk-0.3.0/PKG-INFO +420 -0
  3. augments_adk-0.3.0/README.md +247 -0
  4. augments_adk-0.3.0/pyproject.toml +684 -0
  5. augments_adk-0.3.0/setup.cfg +4 -0
  6. augments_adk-0.3.0/src/augments/adk/__init__.py +330 -0
  7. augments_adk-0.3.0/src/augments/adk/a2a/__init__.py +138 -0
  8. augments_adk-0.3.0/src/augments/adk/a2a/a2a_agent.py +585 -0
  9. augments_adk-0.3.0/src/augments/adk/a2a/a2a_client.py +1140 -0
  10. augments_adk-0.3.0/src/augments/adk/a2a/a2a_continuation_token.py +134 -0
  11. augments_adk-0.3.0/src/augments/adk/a2a/a2a_runner.py +241 -0
  12. augments_adk-0.3.0/src/augments/adk/a2a/adapters.py +229 -0
  13. augments_adk-0.3.0/src/augments/adk/a2a/app_factory.py +154 -0
  14. augments_adk-0.3.0/src/augments/adk/a2a/converters.py +491 -0
  15. augments_adk-0.3.0/src/augments/adk/a2a/exceptions.py +165 -0
  16. augments_adk-0.3.0/src/augments/adk/a2a/executor.py +513 -0
  17. augments_adk-0.3.0/src/augments/adk/a2a/postgres_task_store.py +257 -0
  18. augments_adk-0.3.0/src/augments/adk/a2a/server.py +195 -0
  19. augments_adk-0.3.0/src/augments/adk/a2a/task_store.py +526 -0
  20. augments_adk-0.3.0/src/augments/adk/agents/__init__.py +73 -0
  21. augments_adk-0.3.0/src/augments/adk/agents/agent.py +975 -0
  22. augments_adk-0.3.0/src/augments/adk/agents/agent_guardrails.py +749 -0
  23. augments_adk-0.3.0/src/augments/adk/agents/middleware.py +104 -0
  24. augments_adk-0.3.0/src/augments/adk/audit/__init__.py +19 -0
  25. augments_adk-0.3.0/src/augments/adk/audit/event.py +88 -0
  26. augments_adk-0.3.0/src/augments/adk/audit/sink.py +98 -0
  27. augments_adk-0.3.0/src/augments/adk/audit/sinks/__init__.py +3 -0
  28. augments_adk-0.3.0/src/augments/adk/audit/sinks/postgres.py +134 -0
  29. augments_adk-0.3.0/src/augments/adk/audit/sinks/s3.py +85 -0
  30. augments_adk-0.3.0/src/augments/adk/budgets/__init__.py +14 -0
  31. augments_adk-0.3.0/src/augments/adk/budgets/budget.py +86 -0
  32. augments_adk-0.3.0/src/augments/adk/budgets/ledger.py +92 -0
  33. augments_adk-0.3.0/src/augments/adk/budgets/ledgers/__init__.py +1 -0
  34. augments_adk-0.3.0/src/augments/adk/budgets/ledgers/postgres.py +147 -0
  35. augments_adk-0.3.0/src/augments/adk/budgets/ledgers/redis.py +134 -0
  36. augments_adk-0.3.0/src/augments/adk/cli/__init__.py +51 -0
  37. augments_adk-0.3.0/src/augments/adk/cli/__main__.py +6 -0
  38. augments_adk-0.3.0/src/augments/adk/cli/chat.py +152 -0
  39. augments_adk-0.3.0/src/augments/adk/cli/deploy.py +486 -0
  40. augments_adk-0.3.0/src/augments/adk/cli/errors.py +41 -0
  41. augments_adk-0.3.0/src/augments/adk/cli/loading.py +194 -0
  42. augments_adk-0.3.0/src/augments/adk/cli/new.py +62 -0
  43. augments_adk-0.3.0/src/augments/adk/cli/options.py +118 -0
  44. augments_adk-0.3.0/src/augments/adk/cli/run.py +277 -0
  45. augments_adk-0.3.0/src/augments/adk/cli/schema.py +46 -0
  46. augments_adk-0.3.0/src/augments/adk/cli/serve.py +389 -0
  47. augments_adk-0.3.0/src/augments/adk/cli/sessions.py +147 -0
  48. augments_adk-0.3.0/src/augments/adk/cli/templates.py +148 -0
  49. augments_adk-0.3.0/src/augments/adk/cli/validate.py +88 -0
  50. augments_adk-0.3.0/src/augments/adk/config/__init__.py +39 -0
  51. augments_adk-0.3.0/src/augments/adk/config/assembler.py +142 -0
  52. augments_adk-0.3.0/src/augments/adk/config/dump.py +240 -0
  53. augments_adk-0.3.0/src/augments/adk/config/guardrails.py +43 -0
  54. augments_adk-0.3.0/src/augments/adk/config/hosted_tools.py +97 -0
  55. augments_adk-0.3.0/src/augments/adk/config/loader.py +113 -0
  56. augments_adk-0.3.0/src/augments/adk/config/providers.py +246 -0
  57. augments_adk-0.3.0/src/augments/adk/config/resolver.py +278 -0
  58. augments_adk-0.3.0/src/augments/adk/config/schema.py +82 -0
  59. augments_adk-0.3.0/src/augments/adk/config/topology.py +430 -0
  60. augments_adk-0.3.0/src/augments/adk/context/__init__.py +42 -0
  61. augments_adk-0.3.0/src/augments/adk/context/compaction.py +277 -0
  62. augments_adk-0.3.0/src/augments/adk/context/context_config.py +275 -0
  63. augments_adk-0.3.0/src/augments/adk/context/context_editing.py +239 -0
  64. augments_adk-0.3.0/src/augments/adk/context/context_manager.py +442 -0
  65. augments_adk-0.3.0/src/augments/adk/context/directives.py +248 -0
  66. augments_adk-0.3.0/src/augments/adk/context/token_counter.py +66 -0
  67. augments_adk-0.3.0/src/augments/adk/databases/__init__.py +5 -0
  68. augments_adk-0.3.0/src/augments/adk/databases/connections/__init__.py +5 -0
  69. augments_adk-0.3.0/src/augments/adk/databases/connections/sqlite/__init__.py +5 -0
  70. augments_adk-0.3.0/src/augments/adk/databases/connections/sqlite/sqlite_database_connection.py +246 -0
  71. augments_adk-0.3.0/src/augments/adk/deploy/__init__.py +59 -0
  72. augments_adk-0.3.0/src/augments/adk/deploy/artifacts.py +39 -0
  73. augments_adk-0.3.0/src/augments/adk/deploy/aws_ecr.py +70 -0
  74. augments_adk-0.3.0/src/augments/adk/deploy/aws_manifests.py +165 -0
  75. augments_adk-0.3.0/src/augments/adk/deploy/cloudrun_manifests.py +66 -0
  76. augments_adk-0.3.0/src/augments/adk/deploy/commands.py +183 -0
  77. augments_adk-0.3.0/src/augments/adk/deploy/context.py +72 -0
  78. augments_adk-0.3.0/src/augments/adk/deploy/helm_chart.py +240 -0
  79. augments_adk-0.3.0/src/augments/adk/deploy/k8s_manifests.py +238 -0
  80. augments_adk-0.3.0/src/augments/adk/deploy/targets/__init__.py +42 -0
  81. augments_adk-0.3.0/src/augments/adk/deploy/targets/apprunner.py +79 -0
  82. augments_adk-0.3.0/src/augments/adk/deploy/targets/aws_lambda.py +92 -0
  83. augments_adk-0.3.0/src/augments/adk/deploy/targets/base.py +37 -0
  84. augments_adk-0.3.0/src/augments/adk/deploy/targets/cloudrun.py +91 -0
  85. augments_adk-0.3.0/src/augments/adk/deploy/targets/docker.py +71 -0
  86. augments_adk-0.3.0/src/augments/adk/deploy/targets/ecs.py +100 -0
  87. augments_adk-0.3.0/src/augments/adk/deploy/targets/gke.py +85 -0
  88. augments_adk-0.3.0/src/augments/adk/deploy/targets/helm.py +68 -0
  89. augments_adk-0.3.0/src/augments/adk/deploy/targets/k8s.py +61 -0
  90. augments_adk-0.3.0/src/augments/adk/deploy/templates.py +126 -0
  91. augments_adk-0.3.0/src/augments/adk/exceptions/__init__.py +141 -0
  92. augments_adk-0.3.0/src/augments/adk/exceptions/exceptions.py +1309 -0
  93. augments_adk-0.3.0/src/augments/adk/flows/__init__.py +198 -0
  94. augments_adk-0.3.0/src/augments/adk/flows/agent_bridge.py +265 -0
  95. augments_adk-0.3.0/src/augments/adk/flows/approval_policy.py +101 -0
  96. augments_adk-0.3.0/src/augments/adk/flows/checkpoint.py +583 -0
  97. augments_adk-0.3.0/src/augments/adk/flows/combinators.py +194 -0
  98. augments_adk-0.3.0/src/augments/adk/flows/config.py +93 -0
  99. augments_adk-0.3.0/src/augments/adk/flows/decorators.py +422 -0
  100. augments_adk-0.3.0/src/augments/adk/flows/deferred.py +247 -0
  101. augments_adk-0.3.0/src/augments/adk/flows/definition.py +370 -0
  102. augments_adk-0.3.0/src/augments/adk/flows/events.py +323 -0
  103. augments_adk-0.3.0/src/augments/adk/flows/exceptions.py +322 -0
  104. augments_adk-0.3.0/src/augments/adk/flows/executable.py +143 -0
  105. augments_adk-0.3.0/src/augments/adk/flows/executor.py +1481 -0
  106. augments_adk-0.3.0/src/augments/adk/flows/flow.py +642 -0
  107. augments_adk-0.3.0/src/augments/adk/flows/flow_wrappers.py +486 -0
  108. augments_adk-0.3.0/src/augments/adk/flows/registry.py +347 -0
  109. augments_adk-0.3.0/src/augments/adk/flows/result.py +367 -0
  110. augments_adk-0.3.0/src/augments/adk/flows/sqlite_worker_backend.py +385 -0
  111. augments_adk-0.3.0/src/augments/adk/flows/step_cache_policy.py +110 -0
  112. augments_adk-0.3.0/src/augments/adk/flows/step_context.py +103 -0
  113. augments_adk-0.3.0/src/augments/adk/flows/step_guardrails.py +135 -0
  114. augments_adk-0.3.0/src/augments/adk/flows/step_rate_limit.py +82 -0
  115. augments_adk-0.3.0/src/augments/adk/flows/triggers.py +88 -0
  116. augments_adk-0.3.0/src/augments/adk/flows/worker_backend.py +424 -0
  117. augments_adk-0.3.0/src/augments/adk/graphs/__init__.py +182 -0
  118. augments_adk-0.3.0/src/augments/adk/graphs/adapters.py +1103 -0
  119. augments_adk-0.3.0/src/augments/adk/graphs/builder.py +623 -0
  120. augments_adk-0.3.0/src/augments/adk/graphs/checkpointer.py +209 -0
  121. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/__init__.py +40 -0
  122. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/hooks.py +100 -0
  123. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/in_memory.py +126 -0
  124. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/postgres.py +283 -0
  125. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/redis.py +256 -0
  126. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/s3.py +254 -0
  127. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/sqlite.py +194 -0
  128. augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/tiered.py +321 -0
  129. augments_adk-0.3.0/src/augments/adk/graphs/config.py +177 -0
  130. augments_adk-0.3.0/src/augments/adk/graphs/events.py +365 -0
  131. augments_adk-0.3.0/src/augments/adk/graphs/graph.py +407 -0
  132. augments_adk-0.3.0/src/augments/adk/graphs/hooks.py +424 -0
  133. augments_adk-0.3.0/src/augments/adk/graphs/interrupt.py +420 -0
  134. augments_adk-0.3.0/src/augments/adk/graphs/join.py +224 -0
  135. augments_adk-0.3.0/src/augments/adk/graphs/merge.py +212 -0
  136. augments_adk-0.3.0/src/augments/adk/graphs/nested_snapshot.py +156 -0
  137. augments_adk-0.3.0/src/augments/adk/graphs/node.py +253 -0
  138. augments_adk-0.3.0/src/augments/adk/graphs/node_input.py +193 -0
  139. augments_adk-0.3.0/src/augments/adk/graphs/result.py +528 -0
  140. augments_adk-0.3.0/src/augments/adk/graphs/state.py +659 -0
  141. augments_adk-0.3.0/src/augments/adk/guardrails/__init__.py +69 -0
  142. augments_adk-0.3.0/src/augments/adk/guardrails/injection.py +161 -0
  143. augments_adk-0.3.0/src/augments/adk/guardrails/language.py +280 -0
  144. augments_adk-0.3.0/src/augments/adk/guardrails/pii.py +154 -0
  145. augments_adk-0.3.0/src/augments/adk/guardrails/scan.py +83 -0
  146. augments_adk-0.3.0/src/augments/adk/guardrails/semantic.py +291 -0
  147. augments_adk-0.3.0/src/augments/adk/handoffs/__init__.py +66 -0
  148. augments_adk-0.3.0/src/augments/adk/handoffs/handoff.py +363 -0
  149. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_collapse_mode.py +36 -0
  150. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_config.py +251 -0
  151. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_filters.py +248 -0
  152. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_helpers.py +267 -0
  153. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_input_data.py +90 -0
  154. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_prompt.py +29 -0
  155. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_route.py +336 -0
  156. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_strategy.py +27 -0
  157. augments_adk-0.3.0/src/augments/adk/handoffs/handoff_target.py +354 -0
  158. augments_adk-0.3.0/src/augments/adk/hooks/__init__.py +9 -0
  159. augments_adk-0.3.0/src/augments/adk/hooks/hooks.py +1031 -0
  160. augments_adk-0.3.0/src/augments/adk/llms/__init__.py +75 -0
  161. augments_adk-0.3.0/src/augments/adk/llms/anthropic/__init__.py +35 -0
  162. augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_boundary.py +149 -0
  163. augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_cache_applicator.py +224 -0
  164. augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_config.py +147 -0
  165. augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_converter.py +1001 -0
  166. augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_model.py +967 -0
  167. augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_reasoning_resolver.py +107 -0
  168. augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_retry.py +104 -0
  169. augments_adk-0.3.0/src/augments/adk/llms/cost.py +39 -0
  170. augments_adk-0.3.0/src/augments/adk/llms/embedder.py +135 -0
  171. augments_adk-0.3.0/src/augments/adk/llms/gemini/__init__.py +38 -0
  172. augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_boundary.py +108 -0
  173. augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_config.py +122 -0
  174. augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_converter.py +686 -0
  175. augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_model.py +759 -0
  176. augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_reasoning_resolver.py +82 -0
  177. augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_retry.py +77 -0
  178. augments_adk-0.3.0/src/augments/adk/llms/litellm/__init__.py +13 -0
  179. augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_cache_applicator.py +60 -0
  180. augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_converter.py +1028 -0
  181. augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_embedder.py +117 -0
  182. augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_model.py +1626 -0
  183. augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_provider.py +77 -0
  184. augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_retry.py +121 -0
  185. augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_token_counter.py +33 -0
  186. augments_adk-0.3.0/src/augments/adk/llms/llm.py +236 -0
  187. augments_adk-0.3.0/src/augments/adk/llms/llm_config.py +292 -0
  188. augments_adk-0.3.0/src/augments/adk/llms/llm_middleware.py +421 -0
  189. augments_adk-0.3.0/src/augments/adk/llms/llm_stream_middleware.py +494 -0
  190. augments_adk-0.3.0/src/augments/adk/llms/llm_usage.py +19 -0
  191. augments_adk-0.3.0/src/augments/adk/llms/openai/__init__.py +50 -0
  192. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_boundary.py +151 -0
  193. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_chatcompletions_config.py +119 -0
  194. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_chatcompletions_converter.py +624 -0
  195. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_chatcompletions_model.py +608 -0
  196. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_responses_config.py +129 -0
  197. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_responses_converter.py +760 -0
  198. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_responses_model.py +529 -0
  199. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_retry.py +92 -0
  200. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_stt.py +292 -0
  201. augments_adk-0.3.0/src/augments/adk/llms/openai/openai_tts.py +124 -0
  202. augments_adk-0.3.0/src/augments/adk/llms/retry.py +88 -0
  203. augments_adk-0.3.0/src/augments/adk/llms/routing/__init__.py +15 -0
  204. augments_adk-0.3.0/src/augments/adk/llms/routing/cheapest_first.py +51 -0
  205. augments_adk-0.3.0/src/augments/adk/llms/routing/latency_first.py +47 -0
  206. augments_adk-0.3.0/src/augments/adk/llms/routing/router.py +88 -0
  207. augments_adk-0.3.0/src/augments/adk/llms/stream_error.py +72 -0
  208. augments_adk-0.3.0/src/augments/adk/mcp/__init__.py +116 -0
  209. augments_adk-0.3.0/src/augments/adk/mcp/auth.py +55 -0
  210. augments_adk-0.3.0/src/augments/adk/mcp/conversion.py +251 -0
  211. augments_adk-0.3.0/src/augments/adk/mcp/elicitation.py +94 -0
  212. augments_adk-0.3.0/src/augments/adk/mcp/exceptions.py +66 -0
  213. augments_adk-0.3.0/src/augments/adk/mcp/extras.py +212 -0
  214. augments_adk-0.3.0/src/augments/adk/mcp/filters.py +59 -0
  215. augments_adk-0.3.0/src/augments/adk/mcp/http.py +425 -0
  216. augments_adk-0.3.0/src/augments/adk/mcp/manager.py +310 -0
  217. augments_adk-0.3.0/src/augments/adk/mcp/mcp_server.py +587 -0
  218. augments_adk-0.3.0/src/augments/adk/mcp/notifications.py +88 -0
  219. augments_adk-0.3.0/src/augments/adk/mcp/otel.py +56 -0
  220. augments_adk-0.3.0/src/augments/adk/mcp/run_hooks_bridge.py +82 -0
  221. augments_adk-0.3.0/src/augments/adk/mcp/sampling.py +365 -0
  222. augments_adk-0.3.0/src/augments/adk/mcp/schema_resolver.py +301 -0
  223. augments_adk-0.3.0/src/augments/adk/mcp/sse.py +189 -0
  224. augments_adk-0.3.0/src/augments/adk/mcp/stdio.py +231 -0
  225. augments_adk-0.3.0/src/augments/adk/memory/__init__.py +42 -0
  226. augments_adk-0.3.0/src/augments/adk/memory/distill.py +114 -0
  227. augments_adk-0.3.0/src/augments/adk/memory/extractor.py +209 -0
  228. augments_adk-0.3.0/src/augments/adk/memory/in_memory.py +332 -0
  229. augments_adk-0.3.0/src/augments/adk/memory/memory.py +254 -0
  230. augments_adk-0.3.0/src/augments/adk/memory/memory_config.py +58 -0
  231. augments_adk-0.3.0/src/augments/adk/memory/memory_types.py +119 -0
  232. augments_adk-0.3.0/src/augments/adk/memory/sqlite_memory.py +585 -0
  233. augments_adk-0.3.0/src/augments/adk/memory/stores/__init__.py +11 -0
  234. augments_adk-0.3.0/src/augments/adk/memory/stores/chroma.py +263 -0
  235. augments_adk-0.3.0/src/augments/adk/memory/stores/in_memory.py +176 -0
  236. augments_adk-0.3.0/src/augments/adk/memory/stores/pgvector.py +340 -0
  237. augments_adk-0.3.0/src/augments/adk/memory/stores/pinecone.py +254 -0
  238. augments_adk-0.3.0/src/augments/adk/memory/stores/qdrant.py +257 -0
  239. augments_adk-0.3.0/src/augments/adk/memory/vector_memory.py +190 -0
  240. augments_adk-0.3.0/src/augments/adk/memory/vector_store.py +149 -0
  241. augments_adk-0.3.0/src/augments/adk/orchestration/__init__.py +35 -0
  242. augments_adk-0.3.0/src/augments/adk/orchestration/executable.py +266 -0
  243. augments_adk-0.3.0/src/augments/adk/permissions/__init__.py +24 -0
  244. augments_adk-0.3.0/src/augments/adk/prompts/__init__.py +7 -0
  245. augments_adk-0.3.0/src/augments/adk/prompts/system_prompt.py +203 -0
  246. augments_adk-0.3.0/src/augments/adk/py.typed +0 -0
  247. augments_adk-0.3.0/src/augments/adk/rag/__init__.py +60 -0
  248. augments_adk-0.3.0/src/augments/adk/rag/chunking.py +166 -0
  249. augments_adk-0.3.0/src/augments/adk/rag/document.py +67 -0
  250. augments_adk-0.3.0/src/augments/adk/rag/index.py +150 -0
  251. augments_adk-0.3.0/src/augments/adk/rag/loaders/__init__.py +40 -0
  252. augments_adk-0.3.0/src/augments/adk/rag/loaders/base.py +81 -0
  253. augments_adk-0.3.0/src/augments/adk/rag/loaders/directory.py +73 -0
  254. augments_adk-0.3.0/src/augments/adk/rag/loaders/docx.py +73 -0
  255. augments_adk-0.3.0/src/augments/adk/rag/loaders/github.py +136 -0
  256. augments_adk-0.3.0/src/augments/adk/rag/loaders/pdf.py +71 -0
  257. augments_adk-0.3.0/src/augments/adk/rag/loaders/plaintext.py +84 -0
  258. augments_adk-0.3.0/src/augments/adk/rag/loaders/registry.py +145 -0
  259. augments_adk-0.3.0/src/augments/adk/rag/loaders/structured.py +122 -0
  260. augments_adk-0.3.0/src/augments/adk/rag/loaders/website.py +90 -0
  261. augments_adk-0.3.0/src/augments/adk/rag/loaders/youtube.py +161 -0
  262. augments_adk-0.3.0/src/augments/adk/run/__init__.py +97 -0
  263. augments_adk-0.3.0/src/augments/adk/run/agent_middleware.py +445 -0
  264. augments_adk-0.3.0/src/augments/adk/run/config.py +809 -0
  265. augments_adk-0.3.0/src/augments/adk/run/context.py +193 -0
  266. augments_adk-0.3.0/src/augments/adk/run/cost.py +199 -0
  267. augments_adk-0.3.0/src/augments/adk/run/demo.py +155 -0
  268. augments_adk-0.3.0/src/augments/adk/run/governance.py +158 -0
  269. augments_adk-0.3.0/src/augments/adk/run/graph_loop.py +2160 -0
  270. augments_adk-0.3.0/src/augments/adk/run/guardrails_executor.py +696 -0
  271. augments_adk-0.3.0/src/augments/adk/run/handoffs_executor.py +685 -0
  272. augments_adk-0.3.0/src/augments/adk/run/llm_calls.py +1462 -0
  273. augments_adk-0.3.0/src/augments/adk/run/loop.py +1622 -0
  274. augments_adk-0.3.0/src/augments/adk/run/messages.py +160 -0
  275. augments_adk-0.3.0/src/augments/adk/run/next_step.py +157 -0
  276. augments_adk-0.3.0/src/augments/adk/run/node_reliability.py +205 -0
  277. augments_adk-0.3.0/src/augments/adk/run/profile.py +1136 -0
  278. augments_adk-0.3.0/src/augments/adk/run/resumption.py +778 -0
  279. augments_adk-0.3.0/src/augments/adk/run/runner.py +5235 -0
  280. augments_adk-0.3.0/src/augments/adk/run/state.py +536 -0
  281. augments_adk-0.3.0/src/augments/adk/run/stream.py +579 -0
  282. augments_adk-0.3.0/src/augments/adk/run/swarm_loop.py +908 -0
  283. augments_adk-0.3.0/src/augments/adk/run/swarm_loop_streamed.py +816 -0
  284. augments_adk-0.3.0/src/augments/adk/run/swarm_resume.py +288 -0
  285. augments_adk-0.3.0/src/augments/adk/run/tools_executor.py +1580 -0
  286. augments_adk-0.3.0/src/augments/adk/run/turn_resolution.py +598 -0
  287. augments_adk-0.3.0/src/augments/adk/run/types.py +13 -0
  288. augments_adk-0.3.0/src/augments/adk/sandbox/__init__.py +22 -0
  289. augments_adk-0.3.0/src/augments/adk/sandbox/agent.py +122 -0
  290. augments_adk-0.3.0/src/augments/adk/sandbox/apply_diff.py +387 -0
  291. augments_adk-0.3.0/src/augments/adk/sandbox/apply_patch.py +299 -0
  292. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/__init__.py +34 -0
  293. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/base.py +225 -0
  294. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/capabilities.py +45 -0
  295. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/compaction.py +264 -0
  296. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/filesystem.py +34 -0
  297. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/memory.py +272 -0
  298. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/shell.py +87 -0
  299. augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/skills.py +498 -0
  300. augments_adk-0.3.0/src/augments/adk/sandbox/clients/__init__.py +30 -0
  301. augments_adk-0.3.0/src/augments/adk/sandbox/clients/base.py +196 -0
  302. augments_adk-0.3.0/src/augments/adk/sandbox/clients/docker/__init__.py +15 -0
  303. augments_adk-0.3.0/src/augments/adk/sandbox/clients/docker/docker_client.py +401 -0
  304. augments_adk-0.3.0/src/augments/adk/sandbox/clients/docker/docker_session.py +669 -0
  305. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/__init__.py +6 -0
  306. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/blaxel/__init__.py +15 -0
  307. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/blaxel/blaxel_client.py +174 -0
  308. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/cloudflare/__init__.py +15 -0
  309. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/cloudflare/cloudflare_client.py +174 -0
  310. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/daytona/__init__.py +15 -0
  311. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/daytona/daytona_client.py +174 -0
  312. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/e2b/__init__.py +15 -0
  313. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/e2b/e2b_client.py +208 -0
  314. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/modal/__init__.py +15 -0
  315. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/modal/modal_client.py +184 -0
  316. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/remote_vm/__init__.py +32 -0
  317. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/remote_vm/remote_vm_client.py +355 -0
  318. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/remote_vm/remote_vm_session.py +446 -0
  319. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/runloop/__init__.py +15 -0
  320. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/runloop/runloop_client.py +174 -0
  321. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/vercel/__init__.py +15 -0
  322. augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/vercel/vercel_client.py +185 -0
  323. augments_adk-0.3.0/src/augments/adk/sandbox/clients/k8s/__init__.py +20 -0
  324. augments_adk-0.3.0/src/augments/adk/sandbox/clients/k8s/k8s_client.py +419 -0
  325. augments_adk-0.3.0/src/augments/adk/sandbox/clients/k8s/k8s_session.py +767 -0
  326. augments_adk-0.3.0/src/augments/adk/sandbox/clients/local/__init__.py +21 -0
  327. augments_adk-0.3.0/src/augments/adk/sandbox/clients/local/subprocess_client.py +512 -0
  328. augments_adk-0.3.0/src/augments/adk/sandbox/clients/session.py +317 -0
  329. augments_adk-0.3.0/src/augments/adk/sandbox/config.py +136 -0
  330. augments_adk-0.3.0/src/augments/adk/sandbox/editor.py +102 -0
  331. augments_adk-0.3.0/src/augments/adk/sandbox/guardrails/__init__.py +15 -0
  332. augments_adk-0.3.0/src/augments/adk/sandbox/guardrails/command_guardrail.py +216 -0
  333. augments_adk-0.3.0/src/augments/adk/sandbox/manifest_render.py +315 -0
  334. augments_adk-0.3.0/src/augments/adk/sandbox/memory/__init__.py +76 -0
  335. augments_adk-0.3.0/src/augments/adk/sandbox/memory/consolidation.py +76 -0
  336. augments_adk-0.3.0/src/augments/adk/sandbox/memory/interface.py +135 -0
  337. augments_adk-0.3.0/src/augments/adk/sandbox/memory/manager.py +372 -0
  338. augments_adk-0.3.0/src/augments/adk/sandbox/memory/prompts.py +188 -0
  339. augments_adk-0.3.0/src/augments/adk/sandbox/memory/rollout_extraction.py +133 -0
  340. augments_adk-0.3.0/src/augments/adk/sandbox/memory/rollouts.py +166 -0
  341. augments_adk-0.3.0/src/augments/adk/sandbox/memory/storage.py +480 -0
  342. augments_adk-0.3.0/src/augments/adk/sandbox/observability/__init__.py +26 -0
  343. augments_adk-0.3.0/src/augments/adk/sandbox/observability/audit_sink.py +137 -0
  344. augments_adk-0.3.0/src/augments/adk/sandbox/observability/observability.py +177 -0
  345. augments_adk-0.3.0/src/augments/adk/sandbox/policy/__init__.py +44 -0
  346. augments_adk-0.3.0/src/augments/adk/sandbox/policy/mounts.py +620 -0
  347. augments_adk-0.3.0/src/augments/adk/sandbox/policy/network_policy.py +109 -0
  348. augments_adk-0.3.0/src/augments/adk/sandbox/policy/resource_limits.py +70 -0
  349. augments_adk-0.3.0/src/augments/adk/sandbox/remote_mount_policy.py +92 -0
  350. augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/__init__.py +42 -0
  351. augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/capability_lifecycle.py +165 -0
  352. augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/concurrency_guard.py +83 -0
  353. augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/iac_runner.py +293 -0
  354. augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/instructions_composer.py +79 -0
  355. augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/lifecycle.py +443 -0
  356. augments_adk-0.3.0/src/augments/adk/sandbox/selector.py +83 -0
  357. augments_adk-0.3.0/src/augments/adk/sandbox/session/__init__.py +145 -0
  358. augments_adk-0.3.0/src/augments/adk/sandbox/session/archive_extraction.py +628 -0
  359. augments_adk-0.3.0/src/augments/adk/sandbox/session/concurrency.py +102 -0
  360. augments_adk-0.3.0/src/augments/adk/sandbox/session/dependencies.py +328 -0
  361. augments_adk-0.3.0/src/augments/adk/sandbox/session/events.py +239 -0
  362. augments_adk-0.3.0/src/augments/adk/sandbox/session/manager.py +284 -0
  363. augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/__init__.py +178 -0
  364. augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/git.py +185 -0
  365. augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/inline.py +80 -0
  366. augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/local.py +297 -0
  367. augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/metadata.py +86 -0
  368. augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/paths.py +56 -0
  369. augments_adk-0.3.0/src/augments/adk/sandbox/session/op_codes.py +82 -0
  370. augments_adk-0.3.0/src/augments/adk/sandbox/session/pty_types.py +182 -0
  371. augments_adk-0.3.0/src/augments/adk/sandbox/session/runtime_helpers.py +495 -0
  372. augments_adk-0.3.0/src/augments/adk/sandbox/session/sinks.py +371 -0
  373. augments_adk-0.3.0/src/augments/adk/sandbox/session/tar_workspace.py +46 -0
  374. augments_adk-0.3.0/src/augments/adk/sandbox/session/utils.py +65 -0
  375. augments_adk-0.3.0/src/augments/adk/sandbox/session/workspace_payloads.py +209 -0
  376. augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/__init__.py +26 -0
  377. augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/gcs_store.py +259 -0
  378. augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/local_store.py +153 -0
  379. augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/s3_store.py +297 -0
  380. augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/store.py +49 -0
  381. augments_adk-0.3.0/src/augments/adk/sandbox/snapshot_defaults.py +153 -0
  382. augments_adk-0.3.0/src/augments/adk/sandbox/tools/__init__.py +20 -0
  383. augments_adk-0.3.0/src/augments/adk/sandbox/tools/apply_patch_tool.py +56 -0
  384. augments_adk-0.3.0/src/augments/adk/sandbox/tools/load_skill_tool.py +55 -0
  385. augments_adk-0.3.0/src/augments/adk/sandbox/tools/run_command_tool.py +119 -0
  386. augments_adk-0.3.0/src/augments/adk/sandbox/tools/view_image_tool.py +81 -0
  387. augments_adk-0.3.0/src/augments/adk/sandbox/utils/__init__.py +100 -0
  388. augments_adk-0.3.0/src/augments/adk/sandbox/utils/checksums.py +94 -0
  389. augments_adk-0.3.0/src/augments/adk/sandbox/utils/deep_merge.py +38 -0
  390. augments_adk-0.3.0/src/augments/adk/sandbox/utils/github.py +155 -0
  391. augments_adk-0.3.0/src/augments/adk/sandbox/utils/iterator_io.py +141 -0
  392. augments_adk-0.3.0/src/augments/adk/sandbox/utils/parse_utils.py +166 -0
  393. augments_adk-0.3.0/src/augments/adk/sandbox/utils/retry.py +229 -0
  394. augments_adk-0.3.0/src/augments/adk/sandbox/utils/tar_utils.py +466 -0
  395. augments_adk-0.3.0/src/augments/adk/sandbox/utils/token_truncation.py +251 -0
  396. augments_adk-0.3.0/src/augments/adk/schemas/__init__.py +34 -0
  397. augments_adk-0.3.0/src/augments/adk/schemas/agent_output_schema.py +584 -0
  398. augments_adk-0.3.0/src/augments/adk/schemas/function_schema.py +496 -0
  399. augments_adk-0.3.0/src/augments/adk/schemas/utils.py +637 -0
  400. augments_adk-0.3.0/src/augments/adk/serving/__init__.py +79 -0
  401. augments_adk-0.3.0/src/augments/adk/serving/app.py +111 -0
  402. augments_adk-0.3.0/src/augments/adk/serving/health.py +76 -0
  403. augments_adk-0.3.0/src/augments/adk/serving/rest.py +352 -0
  404. augments_adk-0.3.0/src/augments/adk/serving/serializers.py +175 -0
  405. augments_adk-0.3.0/src/augments/adk/session/__init__.py +26 -0
  406. augments_adk-0.3.0/src/augments/adk/session/buffered_session.py +264 -0
  407. augments_adk-0.3.0/src/augments/adk/session/multi_sessions.py +136 -0
  408. augments_adk-0.3.0/src/augments/adk/session/postgres_multi_sessions.py +345 -0
  409. augments_adk-0.3.0/src/augments/adk/session/postgres_session.py +414 -0
  410. augments_adk-0.3.0/src/augments/adk/session/session.py +88 -0
  411. augments_adk-0.3.0/src/augments/adk/session/session_event.py +75 -0
  412. augments_adk-0.3.0/src/augments/adk/session/session_settings.py +18 -0
  413. augments_adk-0.3.0/src/augments/adk/session/sqlite_multi_sessions.py +593 -0
  414. augments_adk-0.3.0/src/augments/adk/session/sqlite_session.py +629 -0
  415. augments_adk-0.3.0/src/augments/adk/session/state.py +224 -0
  416. augments_adk-0.3.0/src/augments/adk/skills/__init__.py +25 -0
  417. augments_adk-0.3.0/src/augments/adk/skills/activation.py +29 -0
  418. augments_adk-0.3.0/src/augments/adk/skills/discovery.py +617 -0
  419. augments_adk-0.3.0/src/augments/adk/skills/skill.py +270 -0
  420. augments_adk-0.3.0/src/augments/adk/skills/skill_prompt.py +58 -0
  421. augments_adk-0.3.0/src/augments/adk/skills/skill_rendering.py +111 -0
  422. augments_adk-0.3.0/src/augments/adk/skills/skill_set.py +58 -0
  423. augments_adk-0.3.0/src/augments/adk/skills/sources/__init__.py +15 -0
  424. augments_adk-0.3.0/src/augments/adk/skills/sources/base.py +41 -0
  425. augments_adk-0.3.0/src/augments/adk/skills/sources/directory.py +249 -0
  426. augments_adk-0.3.0/src/augments/adk/skills/sources/remote.py +122 -0
  427. augments_adk-0.3.0/src/augments/adk/status/__init__.py +36 -0
  428. augments_adk-0.3.0/src/augments/adk/status/hooks.py +279 -0
  429. augments_adk-0.3.0/src/augments/adk/status/store.py +446 -0
  430. augments_adk-0.3.0/src/augments/adk/status/types.py +179 -0
  431. augments_adk-0.3.0/src/augments/adk/swarms/__init__.py +169 -0
  432. augments_adk-0.3.0/src/augments/adk/swarms/builder.py +323 -0
  433. augments_adk-0.3.0/src/augments/adk/swarms/checkpointer.py +154 -0
  434. augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/__init__.py +38 -0
  435. augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/hooks.py +133 -0
  436. augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/in_memory.py +93 -0
  437. augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/postgres.py +261 -0
  438. augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/redis.py +247 -0
  439. augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/s3.py +247 -0
  440. augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/tiered.py +252 -0
  441. augments_adk-0.3.0/src/augments/adk/swarms/config.py +137 -0
  442. augments_adk-0.3.0/src/augments/adk/swarms/events.py +207 -0
  443. augments_adk-0.3.0/src/augments/adk/swarms/hooks.py +329 -0
  444. augments_adk-0.3.0/src/augments/adk/swarms/interrupt.py +121 -0
  445. augments_adk-0.3.0/src/augments/adk/swarms/policy.py +615 -0
  446. augments_adk-0.3.0/src/augments/adk/swarms/result.py +409 -0
  447. augments_adk-0.3.0/src/augments/adk/swarms/shared_context.py +333 -0
  448. augments_adk-0.3.0/src/augments/adk/swarms/shared_context_strategy.py +54 -0
  449. augments_adk-0.3.0/src/augments/adk/swarms/state.py +703 -0
  450. augments_adk-0.3.0/src/augments/adk/swarms/stop_reason.py +63 -0
  451. augments_adk-0.3.0/src/augments/adk/swarms/swarm.py +445 -0
  452. augments_adk-0.3.0/src/augments/adk/swarms/swarm_prompt.py +53 -0
  453. augments_adk-0.3.0/src/augments/adk/swarms/termination.py +328 -0
  454. augments_adk-0.3.0/src/augments/adk/swarms/yield_signal.py +89 -0
  455. augments_adk-0.3.0/src/augments/adk/tasks/__init__.py +60 -0
  456. augments_adk-0.3.0/src/augments/adk/tasks/task.py +361 -0
  457. augments_adk-0.3.0/src/augments/adk/tasks/task_filters.py +135 -0
  458. augments_adk-0.3.0/src/augments/adk/tasks/task_group.py +219 -0
  459. augments_adk-0.3.0/src/augments/adk/tasks/task_input_data.py +95 -0
  460. augments_adk-0.3.0/src/augments/adk/tasks/task_output.py +239 -0
  461. augments_adk-0.3.0/src/augments/adk/tasks/task_pipeline.py +182 -0
  462. augments_adk-0.3.0/src/augments/adk/tasks/task_pipeline_state.py +172 -0
  463. augments_adk-0.3.0/src/augments/adk/tasks/topology.py +183 -0
  464. augments_adk-0.3.0/src/augments/adk/tools/__init__.py +233 -0
  465. augments_adk-0.3.0/src/augments/adk/tools/builtin/__init__.py +94 -0
  466. augments_adk-0.3.0/src/augments/adk/tools/builtin/builtin_tool.py +170 -0
  467. augments_adk-0.3.0/src/augments/adk/tools/builtin/document_search_tool.py +417 -0
  468. augments_adk-0.3.0/src/augments/adk/tools/builtin/jit_context_aware_tool.py +748 -0
  469. augments_adk-0.3.0/src/augments/adk/tools/builtin/memory_tool.py +254 -0
  470. augments_adk-0.3.0/src/augments/adk/tools/deferred_tool.py +380 -0
  471. augments_adk-0.3.0/src/augments/adk/tools/function_tool.py +1358 -0
  472. augments_adk-0.3.0/src/augments/adk/tools/hosted/__init__.py +56 -0
  473. augments_adk-0.3.0/src/augments/adk/tools/hosted/code_execution_tool.py +37 -0
  474. augments_adk-0.3.0/src/augments/adk/tools/hosted/computer_tool.py +275 -0
  475. augments_adk-0.3.0/src/augments/adk/tools/hosted/exceptions.py +50 -0
  476. augments_adk-0.3.0/src/augments/adk/tools/hosted/file_search_tool.py +41 -0
  477. augments_adk-0.3.0/src/augments/adk/tools/hosted/hosted_tool.py +55 -0
  478. augments_adk-0.3.0/src/augments/adk/tools/hosted/image_generation_tool.py +38 -0
  479. augments_adk-0.3.0/src/augments/adk/tools/hosted/mcp_tool.py +117 -0
  480. augments_adk-0.3.0/src/augments/adk/tools/hosted/url_context_tool.py +27 -0
  481. augments_adk-0.3.0/src/augments/adk/tools/hosted/web_search_tool.py +86 -0
  482. augments_adk-0.3.0/src/augments/adk/tools/local/__init__.py +36 -0
  483. augments_adk-0.3.0/src/augments/adk/tools/local/apply_patch_tool.py +84 -0
  484. augments_adk-0.3.0/src/augments/adk/tools/local/shell_tool.py +110 -0
  485. augments_adk-0.3.0/src/augments/adk/tools/token_budget.py +44 -0
  486. augments_adk-0.3.0/src/augments/adk/tools/tool_context.py +249 -0
  487. augments_adk-0.3.0/src/augments/adk/tools/tool_guardrails.py +419 -0
  488. augments_adk-0.3.0/src/augments/adk/tools/tool_middleware.py +447 -0
  489. augments_adk-0.3.0/src/augments/adk/tools/tool_output_trimmer.py +161 -0
  490. augments_adk-0.3.0/src/augments/adk/tools/tool_search.py +370 -0
  491. augments_adk-0.3.0/src/augments/adk/tools/toolsets/__init__.py +52 -0
  492. augments_adk-0.3.0/src/augments/adk/tools/toolsets/abstract.py +179 -0
  493. augments_adk-0.3.0/src/augments/adk/tools/toolsets/clone.py +43 -0
  494. augments_adk-0.3.0/src/augments/adk/tools/toolsets/combined.py +81 -0
  495. augments_adk-0.3.0/src/augments/adk/tools/toolsets/filtered.py +99 -0
  496. augments_adk-0.3.0/src/augments/adk/tools/toolsets/function_toolset.py +78 -0
  497. augments_adk-0.3.0/src/augments/adk/tools/toolsets/mcp_toolset.py +271 -0
  498. augments_adk-0.3.0/src/augments/adk/tools/toolsets/prefixed.py +77 -0
  499. augments_adk-0.3.0/src/augments/adk/tools/toolsets/renamed.py +66 -0
  500. augments_adk-0.3.0/src/augments/adk/tools/toolsets/wrapper.py +81 -0
  501. augments_adk-0.3.0/src/augments/adk/tracing/__init__.py +118 -0
  502. augments_adk-0.3.0/src/augments/adk/tracing/exporters/__init__.py +8 -0
  503. augments_adk-0.3.0/src/augments/adk/tracing/exporters/helicone.py +52 -0
  504. augments_adk-0.3.0/src/augments/adk/tracing/exporters/langsmith.py +61 -0
  505. augments_adk-0.3.0/src/augments/adk/tracing/exporters/logfire.py +50 -0
  506. augments_adk-0.3.0/src/augments/adk/tracing/exporters/phoenix.py +36 -0
  507. augments_adk-0.3.0/src/augments/adk/tracing/logging.py +54 -0
  508. augments_adk-0.3.0/src/augments/adk/tracing/metrics/__init__.py +7 -0
  509. augments_adk-0.3.0/src/augments/adk/tracing/metrics/instruments.py +155 -0
  510. augments_adk-0.3.0/src/augments/adk/tracing/metrics/setup.py +53 -0
  511. augments_adk-0.3.0/src/augments/adk/tracing/metrics/tracer.py +188 -0
  512. augments_adk-0.3.0/src/augments/adk/tracing/multi_tracer.py +258 -0
  513. augments_adk-0.3.0/src/augments/adk/tracing/openinference/__init__.py +21 -0
  514. augments_adk-0.3.0/src/augments/adk/tracing/openinference/conventions.py +196 -0
  515. augments_adk-0.3.0/src/augments/adk/tracing/otel/__init__.py +24 -0
  516. augments_adk-0.3.0/src/augments/adk/tracing/otel/otel_span.py +149 -0
  517. augments_adk-0.3.0/src/augments/adk/tracing/otel/otel_tracer.py +695 -0
  518. augments_adk-0.3.0/src/augments/adk/tracing/otel/setup.py +212 -0
  519. augments_adk-0.3.0/src/augments/adk/tracing/spans.py +645 -0
  520. augments_adk-0.3.0/src/augments/adk/tracing/tracer.py +204 -0
  521. augments_adk-0.3.0/src/augments/adk/types/__init__.py +135 -0
  522. augments_adk-0.3.0/src/augments/adk/types/agents/__init__.py +17 -0
  523. augments_adk-0.3.0/src/augments/adk/types/agents/agent_as_tool_types.py +102 -0
  524. augments_adk-0.3.0/src/augments/adk/types/common/__init__.py +9 -0
  525. augments_adk-0.3.0/src/augments/adk/types/common/metadata.py +21 -0
  526. augments_adk-0.3.0/src/augments/adk/types/common/typing.py +5 -0
  527. augments_adk-0.3.0/src/augments/adk/types/config/__init__.py +87 -0
  528. augments_adk-0.3.0/src/augments/adk/types/config/agent_config.py +197 -0
  529. augments_adk-0.3.0/src/augments/adk/types/config/graph_config.py +85 -0
  530. augments_adk-0.3.0/src/augments/adk/types/config/guardrail_config.py +42 -0
  531. augments_adk-0.3.0/src/augments/adk/types/config/llm_config.py +594 -0
  532. augments_adk-0.3.0/src/augments/adk/types/config/prompt_config.py +22 -0
  533. augments_adk-0.3.0/src/augments/adk/types/config/references.py +80 -0
  534. augments_adk-0.3.0/src/augments/adk/types/config/swarm_config.py +168 -0
  535. augments_adk-0.3.0/src/augments/adk/types/config/tool_config.py +43 -0
  536. augments_adk-0.3.0/src/augments/adk/types/config/topology_config.py +65 -0
  537. augments_adk-0.3.0/src/augments/adk/types/guardrails/__init__.py +7 -0
  538. augments_adk-0.3.0/src/augments/adk/types/guardrails/action.py +56 -0
  539. augments_adk-0.3.0/src/augments/adk/types/input/__init__.py +84 -0
  540. augments_adk-0.3.0/src/augments/adk/types/input/llm_input_audio.py +34 -0
  541. augments_adk-0.3.0/src/augments/adk/types/input/llm_input_easy_message.py +36 -0
  542. augments_adk-0.3.0/src/augments/adk/types/input/llm_input_image.py +44 -0
  543. augments_adk-0.3.0/src/augments/adk/types/input/llm_input_message.py +45 -0
  544. augments_adk-0.3.0/src/augments/adk/types/input/llm_input_text.py +27 -0
  545. augments_adk-0.3.0/src/augments/adk/types/intents/__init__.py +46 -0
  546. augments_adk-0.3.0/src/augments/adk/types/items/__init__.py +45 -0
  547. augments_adk-0.3.0/src/augments/adk/types/items/items.py +1348 -0
  548. augments_adk-0.3.0/src/augments/adk/types/llms/__init__.py +18 -0
  549. augments_adk-0.3.0/src/augments/adk/types/llms/effort.py +21 -0
  550. augments_adk-0.3.0/src/augments/adk/types/llms/retry_policy.py +129 -0
  551. augments_adk-0.3.0/src/augments/adk/types/multimodal/__init__.py +38 -0
  552. augments_adk-0.3.0/src/augments/adk/types/multimodal/file.py +311 -0
  553. augments_adk-0.3.0/src/augments/adk/types/output/__init__.py +38 -0
  554. augments_adk-0.3.0/src/augments/adk/types/output/function_tool_call_result.py +74 -0
  555. augments_adk-0.3.0/src/augments/adk/types/output/function_tool_call_result_param.py +42 -0
  556. augments_adk-0.3.0/src/augments/adk/types/output/llm_response_function_tool_call_param.py +54 -0
  557. augments_adk-0.3.0/src/augments/adk/types/output/llm_response_message_param.py +42 -0
  558. augments_adk-0.3.0/src/augments/adk/types/output/llm_response_provider_item_param.py +44 -0
  559. augments_adk-0.3.0/src/augments/adk/types/output/llm_response_reasoning_param.py +67 -0
  560. augments_adk-0.3.0/src/augments/adk/types/output/llm_response_refusal_param.py +24 -0
  561. augments_adk-0.3.0/src/augments/adk/types/output/llm_response_text_param.py +28 -0
  562. augments_adk-0.3.0/src/augments/adk/types/output/reasoning_content_text_param.py +24 -0
  563. augments_adk-0.3.0/src/augments/adk/types/output/reasoning_summary_text_param.py +24 -0
  564. augments_adk-0.3.0/src/augments/adk/types/permissions/__init__.py +21 -0
  565. augments_adk-0.3.0/src/augments/adk/types/permissions/permission_types.py +482 -0
  566. augments_adk-0.3.0/src/augments/adk/types/responses/__init__.py +23 -0
  567. augments_adk-0.3.0/src/augments/adk/types/responses/llm_response.py +511 -0
  568. augments_adk-0.3.0/src/augments/adk/types/run/__init__.py +5 -0
  569. augments_adk-0.3.0/src/augments/adk/types/run/guardrail_audit.py +92 -0
  570. augments_adk-0.3.0/src/augments/adk/types/run/run_result.py +270 -0
  571. augments_adk-0.3.0/src/augments/adk/types/sandbox/__init__.py +133 -0
  572. augments_adk-0.3.0/src/augments/adk/types/sandbox/cost.py +172 -0
  573. augments_adk-0.3.0/src/augments/adk/types/sandbox/entries.py +420 -0
  574. augments_adk-0.3.0/src/augments/adk/types/sandbox/exec_result.py +187 -0
  575. augments_adk-0.3.0/src/augments/adk/types/sandbox/iac.py +78 -0
  576. augments_adk-0.3.0/src/augments/adk/types/sandbox/manifest.py +288 -0
  577. augments_adk-0.3.0/src/augments/adk/types/sandbox/mounts.py +569 -0
  578. augments_adk-0.3.0/src/augments/adk/types/sandbox/network.py +121 -0
  579. augments_adk-0.3.0/src/augments/adk/types/sandbox/permissions.py +231 -0
  580. augments_adk-0.3.0/src/augments/adk/types/sandbox/resource_limits.py +89 -0
  581. augments_adk-0.3.0/src/augments/adk/types/sandbox/session_state.py +51 -0
  582. augments_adk-0.3.0/src/augments/adk/types/sandbox/snapshot.py +164 -0
  583. augments_adk-0.3.0/src/augments/adk/types/sandbox/usage.py +162 -0
  584. augments_adk-0.3.0/src/augments/adk/types/sandbox/workspace_paths.py +155 -0
  585. augments_adk-0.3.0/src/augments/adk/types/session/__init__.py +5 -0
  586. augments_adk-0.3.0/src/augments/adk/types/session/store.py +97 -0
  587. augments_adk-0.3.0/src/augments/adk/types/tokens/__init__.py +17 -0
  588. augments_adk-0.3.0/src/augments/adk/types/tokens/llm_usage.py +195 -0
  589. augments_adk-0.3.0/src/augments/adk/types/tokens/tokens.py +38 -0
  590. augments_adk-0.3.0/src/augments/adk/types/tools/__init__.py +121 -0
  591. augments_adk-0.3.0/src/augments/adk/types/tools/approval_policy.py +23 -0
  592. augments_adk-0.3.0/src/augments/adk/types/tools/builtin_tool_types.py +595 -0
  593. augments_adk-0.3.0/src/augments/adk/types/tools/tool_output_types.py +83 -0
  594. augments_adk-0.3.0/src/augments/adk/types/tools/tool_rate_limit.py +91 -0
  595. augments_adk-0.3.0/src/augments/adk/types/tools/tool_stream_event.py +60 -0
  596. augments_adk-0.3.0/src/augments/adk/types/tools/tool_use_behavior.py +89 -0
  597. augments_adk-0.3.0/src/augments/adk/types/tracing/__init__.py +49 -0
  598. augments_adk-0.3.0/src/augments/adk/types/tracing/convention.py +16 -0
  599. augments_adk-0.3.0/src/augments/adk/types/tracing/span_data.py +637 -0
  600. augments_adk-0.3.0/src/augments/adk/utils/__init__.py +7 -0
  601. augments_adk-0.3.0/src/augments/adk/utils/snake_case.py +39 -0
  602. augments_adk-0.3.0/src/augments/adk/utils/typedef.py +9 -0
  603. augments_adk-0.3.0/src/augments/adk/verbose/__init__.py +165 -0
  604. augments_adk-0.3.0/src/augments/adk/verbose/config.py +554 -0
  605. augments_adk-0.3.0/src/augments/adk/verbose/hooks.py +2252 -0
  606. augments_adk-0.3.0/src/augments/adk/verbose/mode.py +203 -0
  607. augments_adk-0.3.0/src/augments/adk/verbose/panel_renderer.py +1053 -0
  608. augments_adk-0.3.0/src/augments/adk/verbose/renderer.py +371 -0
  609. augments_adk-0.3.0/src/augments/adk/verbose/run_bridge.py +82 -0
  610. augments_adk-0.3.0/src/augments/adk/verbose/state.py +299 -0
  611. augments_adk-0.3.0/src/augments/adk/visualization/__init__.py +40 -0
  612. augments_adk-0.3.0/src/augments/adk/visualization/dot.py +454 -0
  613. augments_adk-0.3.0/src/augments/adk/visualization/helpers.py +208 -0
  614. augments_adk-0.3.0/src/augments/adk/visualization/mermaid.py +462 -0
  615. augments_adk-0.3.0/src/augments/adk/visualization/render.py +120 -0
  616. augments_adk-0.3.0/src/augments/adk/voice/__init__.py +106 -0
  617. augments_adk-0.3.0/src/augments/adk/voice/audio.py +196 -0
  618. augments_adk-0.3.0/src/augments/adk/voice/events.py +80 -0
  619. augments_adk-0.3.0/src/augments/adk/voice/exceptions.py +32 -0
  620. augments_adk-0.3.0/src/augments/adk/voice/pipeline.py +139 -0
  621. augments_adk-0.3.0/src/augments/adk/voice/pipeline_config.py +44 -0
  622. augments_adk-0.3.0/src/augments/adk/voice/result.py +214 -0
  623. augments_adk-0.3.0/src/augments/adk/voice/splitter.py +58 -0
  624. augments_adk-0.3.0/src/augments/adk/voice/stt.py +162 -0
  625. augments_adk-0.3.0/src/augments/adk/voice/tts.py +76 -0
  626. augments_adk-0.3.0/src/augments/adk/voice/workflow.py +125 -0
  627. augments_adk-0.3.0/src/augments/adk/workflows/__init__.py +30 -0
  628. augments_adk-0.3.0/src/augments/adk/workflows/engine.py +189 -0
  629. augments_adk-0.3.0/src/augments/adk/workflows/restate/__init__.py +23 -0
  630. augments_adk-0.3.0/src/augments/adk/workflows/restate/activity.py +140 -0
  631. augments_adk-0.3.0/src/augments/adk/workflows/restate/engine.py +117 -0
  632. augments_adk-0.3.0/src/augments/adk/workflows/restate/llm.py +250 -0
  633. augments_adk-0.3.0/src/augments/adk/workflows/restate/service.py +139 -0
  634. augments_adk-0.3.0/src/augments/adk/workflows/restate/tools.py +79 -0
  635. augments_adk-0.3.0/src/augments/adk/workflows/temporal/__init__.py +54 -0
  636. augments_adk-0.3.0/src/augments/adk/workflows/temporal/activity.py +254 -0
  637. augments_adk-0.3.0/src/augments/adk/workflows/temporal/determinism.py +80 -0
  638. augments_adk-0.3.0/src/augments/adk/workflows/temporal/engine.py +122 -0
  639. augments_adk-0.3.0/src/augments/adk/workflows/temporal/llm.py +515 -0
  640. augments_adk-0.3.0/src/augments/adk/workflows/temporal/mcp.py +166 -0
  641. augments_adk-0.3.0/src/augments/adk/workflows/temporal/plugin.py +222 -0
  642. augments_adk-0.3.0/src/augments/adk/workflows/temporal/routing.py +111 -0
  643. augments_adk-0.3.0/src/augments/adk/workflows/temporal/serialization.py +273 -0
  644. augments_adk-0.3.0/src/augments/adk/workflows/temporal/streaming.py +185 -0
  645. augments_adk-0.3.0/src/augments/adk/workflows/temporal/tools.py +319 -0
  646. augments_adk-0.3.0/src/augments/adk/workflows/temporal/tracing.py +110 -0
  647. augments_adk-0.3.0/src/augments/adk/workflows/temporal/workflow.py +237 -0
  648. augments_adk-0.3.0/src/augments_adk.egg-info/PKG-INFO +420 -0
  649. augments_adk-0.3.0/src/augments_adk.egg-info/SOURCES.txt +651 -0
  650. augments_adk-0.3.0/src/augments_adk.egg-info/dependency_links.txt +1 -0
  651. augments_adk-0.3.0/src/augments_adk.egg-info/entry_points.txt +2 -0
  652. augments_adk-0.3.0/src/augments_adk.egg-info/requires.txt +194 -0
  653. augments_adk-0.3.0/src/augments_adk.egg-info/top_level.txt +1 -0
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2026 augments-labs
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,420 @@
1
+ Metadata-Version: 2.4
2
+ Name: augments-adk
3
+ Version: 0.3.0
4
+ Summary: Augments ADK — orchestrate a complex system of agents that performs real-world actions, provider-agnostic across 100+ LLMs
5
+ Author-email: Augments Labs <support@augmentslabs.com>
6
+ License: MIT
7
+ Project-URL: Homepage, https://github.com/augments-labs/augments-adk-python
8
+ Project-URL: Repository, https://github.com/augments-labs/augments-adk-python
9
+ Project-URL: Bug Tracker, https://github.com/augments-labs/augments-adk-python/issues
10
+ Project-URL: Documentation, https://github.com/augments-labs/augments-adk-python/tree/main/docs
11
+ Project-URL: Changelog, https://github.com/augments-labs/augments-adk-python/releases
12
+ Keywords: agents,llm,multi-agent,agentic,litellm,openai,anthropic,gemini,mcp,opentelemetry,swarm,graph,handoff,tool-use,streaming
13
+ Classifier: Development Status :: 3 - Alpha
14
+ Classifier: Intended Audience :: Developers
15
+ Classifier: Intended Audience :: Science/Research
16
+ Classifier: License :: OSI Approved :: MIT License
17
+ Classifier: Operating System :: OS Independent
18
+ Classifier: Programming Language :: Python
19
+ Classifier: Programming Language :: Python :: 3
20
+ Classifier: Programming Language :: Python :: 3 :: Only
21
+ Classifier: Programming Language :: Python :: 3.12
22
+ Classifier: Programming Language :: Python :: 3.13
23
+ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
24
+ Classifier: Topic :: Software Development :: Libraries
25
+ Classifier: Topic :: Software Development :: Libraries :: Python Modules
26
+ Classifier: Typing :: Typed
27
+ Requires-Python: <3.14,>=3.12
28
+ Description-Content-Type: text/markdown
29
+ License-File: LICENSE
30
+ Requires-Dist: litellm<2,>=1.80.9
31
+ Requires-Dist: pydantic<3,>=2.12.5
32
+ Requires-Dist: griffe>=1.15.0
33
+ Requires-Dist: aiosqlite>=0.21.0
34
+ Requires-Dist: httpx>=0.27.0
35
+ Requires-Dist: typing-extensions<5,>=4.15.0
36
+ Requires-Dist: click<9,>=8.1.7
37
+ Provides-Extra: anthropic
38
+ Requires-Dist: anthropic<1,>=0.75.0; extra == "anthropic"
39
+ Provides-Extra: openai
40
+ Requires-Dist: openai<3,>=2.0.0; extra == "openai"
41
+ Provides-Extra: voice
42
+ Requires-Dist: augments-adk[openai]; extra == "voice"
43
+ Requires-Dist: websockets>=13.0; extra == "voice"
44
+ Requires-Dist: numpy>=1.26; extra == "voice"
45
+ Provides-Extra: gemini
46
+ Requires-Dist: google-genai>=1.0.0; extra == "gemini"
47
+ Provides-Extra: mcp
48
+ Requires-Dist: mcp<3,>=2.0.0; extra == "mcp"
49
+ Requires-Dist: httpx2>=2.5.0; extra == "mcp"
50
+ Provides-Extra: a2a
51
+ Requires-Dist: a2a-sdk[http-server]>=1.0.0; extra == "a2a"
52
+ Provides-Extra: server
53
+ Requires-Dist: starlette>=0.37; extra == "server"
54
+ Requires-Dist: sse-starlette>=2.0; extra == "server"
55
+ Provides-Extra: serve
56
+ Requires-Dist: augments-adk[a2a,server]; extra == "serve"
57
+ Requires-Dist: uvicorn>=0.30; extra == "serve"
58
+ Provides-Extra: otel
59
+ Requires-Dist: opentelemetry-api>=1.39.1; extra == "otel"
60
+ Requires-Dist: opentelemetry-sdk>=1.39.1; extra == "otel"
61
+ Requires-Dist: opentelemetry-exporter-otlp>=1.39.1; extra == "otel"
62
+ Provides-Extra: viz
63
+ Requires-Dist: graphviz>=0.20.2; extra == "viz"
64
+ Provides-Extra: mermaid
65
+ Requires-Dist: mermaid-py>=0.5.0; extra == "mermaid"
66
+ Provides-Extra: verbose
67
+ Requires-Dist: rich>=14.2.0; extra == "verbose"
68
+ Provides-Extra: yaml
69
+ Requires-Dist: pyyaml>=6.0; extra == "yaml"
70
+ Provides-Extra: guardrails-lingua
71
+ Requires-Dist: lingua-language-detector>=2.0.0; extra == "guardrails-lingua"
72
+ Provides-Extra: guardrails
73
+ Requires-Dist: augments-adk[guardrails-lingua]; extra == "guardrails"
74
+ Provides-Extra: rag-pdf
75
+ Requires-Dist: pymupdf>=1.24; extra == "rag-pdf"
76
+ Provides-Extra: rag-docx
77
+ Requires-Dist: python-docx>=1.1; extra == "rag-docx"
78
+ Provides-Extra: rag-web
79
+ Requires-Dist: requests>=2.31; extra == "rag-web"
80
+ Requires-Dist: beautifulsoup4>=4.12; extra == "rag-web"
81
+ Provides-Extra: rag-github
82
+ Requires-Dist: PyGithub>=2.1; extra == "rag-github"
83
+ Provides-Extra: rag-youtube
84
+ Requires-Dist: youtube-transcript-api>=1.0; extra == "rag-youtube"
85
+ Requires-Dist: pytube>=15.0; extra == "rag-youtube"
86
+ Provides-Extra: rag
87
+ Requires-Dist: augments-adk[rag-docx,rag-github,rag-pdf,rag-web,rag-youtube]; extra == "rag"
88
+ Provides-Extra: sandbox
89
+ Provides-Extra: sandbox-docker
90
+ Requires-Dist: docker>=7.0.0; extra == "sandbox-docker"
91
+ Provides-Extra: sandbox-k8s
92
+ Requires-Dist: kubernetes>=30.0.0; extra == "sandbox-k8s"
93
+ Provides-Extra: sandbox-remote-vm
94
+ Requires-Dist: httpx>=0.27.0; extra == "sandbox-remote-vm"
95
+ Provides-Extra: sandbox-e2b
96
+ Requires-Dist: augments-adk[sandbox-remote-vm]; extra == "sandbox-e2b"
97
+ Provides-Extra: sandbox-vercel
98
+ Requires-Dist: augments-adk[sandbox-remote-vm]; extra == "sandbox-vercel"
99
+ Provides-Extra: sandbox-modal
100
+ Requires-Dist: augments-adk[sandbox-remote-vm]; extra == "sandbox-modal"
101
+ Provides-Extra: sandbox-daytona
102
+ Requires-Dist: augments-adk[sandbox-remote-vm]; extra == "sandbox-daytona"
103
+ Provides-Extra: sandbox-cloudflare
104
+ Requires-Dist: augments-adk[sandbox-remote-vm]; extra == "sandbox-cloudflare"
105
+ Provides-Extra: sandbox-blaxel
106
+ Requires-Dist: augments-adk[sandbox-remote-vm]; extra == "sandbox-blaxel"
107
+ Provides-Extra: sandbox-runloop
108
+ Requires-Dist: augments-adk[sandbox-remote-vm]; extra == "sandbox-runloop"
109
+ Provides-Extra: sandbox-s3
110
+ Requires-Dist: boto3>=1.34.0; extra == "sandbox-s3"
111
+ Provides-Extra: sandbox-gcs
112
+ Requires-Dist: google-cloud-storage>=2.16.0; extra == "sandbox-gcs"
113
+ Provides-Extra: checkpointer-postgres
114
+ Requires-Dist: psycopg[binary,pool]>=3.2; extra == "checkpointer-postgres"
115
+ Provides-Extra: cost-ledger-postgres
116
+ Requires-Dist: psycopg[binary,pool]>=3.2; extra == "cost-ledger-postgres"
117
+ Provides-Extra: memory-pgvector
118
+ Requires-Dist: psycopg[binary,pool]>=3.2; extra == "memory-pgvector"
119
+ Requires-Dist: pgvector>=0.3; extra == "memory-pgvector"
120
+ Provides-Extra: memory-chroma
121
+ Requires-Dist: chromadb>=0.5; extra == "memory-chroma"
122
+ Provides-Extra: memory-pinecone
123
+ Requires-Dist: pinecone>=8.0; extra == "memory-pinecone"
124
+ Provides-Extra: memory-qdrant
125
+ Requires-Dist: qdrant-client>=1.12; extra == "memory-qdrant"
126
+ Provides-Extra: cost-ledger-redis
127
+ Requires-Dist: redis>=5.0; extra == "cost-ledger-redis"
128
+ Provides-Extra: checkpointer-redis
129
+ Requires-Dist: redis>=5.0; extra == "checkpointer-redis"
130
+ Provides-Extra: checkpointer-s3
131
+ Requires-Dist: boto3>=1.34.0; extra == "checkpointer-s3"
132
+ Provides-Extra: audit-s3
133
+ Requires-Dist: boto3>=1.34.0; extra == "audit-s3"
134
+ Provides-Extra: audit-postgres
135
+ Requires-Dist: psycopg[binary,pool]>=3.2; extra == "audit-postgres"
136
+ Provides-Extra: a2a-postgres
137
+ Requires-Dist: psycopg[binary,pool]>=3.2; extra == "a2a-postgres"
138
+ Provides-Extra: session-postgres
139
+ Requires-Dist: psycopg[binary,pool]>=3.2; extra == "session-postgres"
140
+ Provides-Extra: temporal
141
+ Requires-Dist: temporalio>=1.24.0; extra == "temporal"
142
+ Provides-Extra: restate
143
+ Requires-Dist: restate-sdk>=0.18.0; extra == "restate"
144
+ Provides-Extra: all
145
+ Requires-Dist: augments-adk[a2a-postgres,anthropic,checkpointer-postgres,checkpointer-redis,checkpointer-s3,cost-ledger-postgres,cost-ledger-redis,gemini,guardrails,mcp,memory-chroma,memory-pgvector,memory-pinecone,memory-qdrant,openai,otel,rag,serve,server,session-postgres,temporal,verbose,viz,voice]; extra == "all"
146
+ Provides-Extra: test
147
+ Requires-Dist: pytest<10,>=9.0.0; extra == "test"
148
+ Requires-Dist: pytest-asyncio>=1.0.0; extra == "test"
149
+ Requires-Dist: pytest-cov>=7.0.0; extra == "test"
150
+ Requires-Dist: pytest-mock>=3.14.0; extra == "test"
151
+ Requires-Dist: pytest-timeout>=2.4.0; extra == "test"
152
+ Requires-Dist: pytest-xdist>=3.6.0; extra == "test"
153
+ Requires-Dist: pytest-rerunfailures>=14.0; extra == "test"
154
+ Requires-Dist: coverage[toml]>=7.6.0; extra == "test"
155
+ Requires-Dist: parameterized>=0.9.0; extra == "test"
156
+ Requires-Dist: inline-snapshot>=0.20.7; extra == "test"
157
+ Requires-Dist: fakeredis[lua]>=2.21; extra == "test"
158
+ Requires-Dist: moto>=5.0; extra == "test"
159
+ Requires-Dist: pytest-postgresql>=6.0; extra == "test"
160
+ Provides-Extra: dev
161
+ Requires-Dist: augments-adk[all,test]; extra == "dev"
162
+ Requires-Dist: ruff==0.16.8; extra == "dev"
163
+ Requires-Dist: mypy>=1.15.0; extra == "dev"
164
+ Requires-Dist: pyright>=1.1.408; extra == "dev"
165
+ Requires-Dist: types-PyYAML>=6.0; extra == "dev"
166
+ Requires-Dist: pre-commit>=3.5.0; extra == "dev"
167
+ Requires-Dist: bandit[toml]>=1.7; extra == "dev"
168
+ Requires-Dist: pip-audit>=2.7; extra == "dev"
169
+ Requires-Dist: python-dotenv>=1.2.2; extra == "dev"
170
+ Requires-Dist: commitizen>=4.4.0; extra == "dev"
171
+ Requires-Dist: build>=1.2.0; extra == "dev"
172
+ Dynamic: license-file
173
+
174
+ <div align="center">
175
+ <picture>
176
+ <source media="(prefers-color-scheme: dark)"
177
+ srcset="https://raw.githubusercontent.com/augments-labs/augments-adk-python/main/docs/images/logo-dark.png">
178
+ <img src="https://raw.githubusercontent.com/augments-labs/augments-adk-python/main/docs/images/logo-light.png"
179
+ alt="Augments" width="256">
180
+ </picture>
181
+ </div>
182
+
183
+ # Augments Agent Development Kit (ADK)
184
+
185
+ A provider-agnostic Python framework to orchestrate complex
186
+ systems of agents that perform real-world actions, across 100+ LLMs via litellm.
187
+
188
+ ## The concept
189
+
190
+ An agent is a model that stopped talking and started doing: it calls tools,
191
+ changes state, and leaves side effects in the world. One agent is useful. A
192
+ set of specialists that cannot coordinate is a liability.
193
+
194
+ Augmenting a model means adding capability without giving up control. This
195
+ ADK adds three things, and each one is explicit and yours to configure:
196
+
197
+ - **Structure** — explicit orchestration. Graphs for state machines, flows
198
+ for pipelines, swarms for open-ended exploration, handoffs for delegation.
199
+ You write the structure; nothing is inferred behind your back.
200
+ - **Specialists** — agents scoped to one job, each carrying its own tools,
201
+ guardrails, and budget. An `Agent` is configuration, never a hidden runtime.
202
+ - **Control** — the `Runner`. Every run travels one execution path, where
203
+ turns, retries, token budgets, and interrupts are *enforced* rather than
204
+ suggested.
205
+
206
+ The framework never injects a prompt, a tool, or a token you did not ask for,
207
+ and every cost-bearing default starts bounded. Decisions, tool I/O, and token
208
+ spend come back as structured traces, so what the agents actually did is
209
+ readable after the fact.
210
+
211
+ ## Design tenets
212
+
213
+ 1. **Explicit over magical.** If you can't step through it in a debugger, it
214
+ doesn't belong in the orchestration path.
215
+ 2. **One obvious way.** Fewer knobs, sharper edges — the ADK has opinions.
216
+ 3. **Everything is inspectable.** Decisions, tool I/O, and token costs are
217
+ structured traces, not anecdotes.
218
+ 4. **Benchmarks or it didn't happen.** Claims ship with eval evidence or not
219
+ at all — including this framework's own.
220
+
221
+ ## Core Concepts
222
+
223
+ - [**Agents**](docs/agents/) — Autonomous entities with tools, guardrails, and handoffs
224
+ - [**Tools**](docs/tools/) — Function wrappers with schema validation and guardrails
225
+ - [**Handoffs**](docs/handoffs/) — Agent-to-agent routing (LLM-orchestrated or code-orchestrated)
226
+ - [**Guardrails**](docs/guardrails/) — Pre/post execution validation at agent and tool level
227
+ - [**Memory**](docs/memory/) — Persistent knowledge across sessions
228
+ - [**Skills**](docs/skills/) — Reusable capability packages (instructions + tools + governance)
229
+ - [**Tracing**](docs/tracing/) — OpenTelemetry observability
230
+
231
+ ## Installation
232
+
233
+ ### Use it in your own project
234
+
235
+ ```bash
236
+ uv add augments-adk # or: pip install augments-adk
237
+ ```
238
+
239
+ The core install is deliberately lean — `litellm`, `pydantic`, `griffe`, `aiosqlite`, `typing-extensions` — and every optional provider / exporter / UI enhancement is gated behind its own extra:
240
+
241
+ ```bash
242
+ pip install 'augments-adk[anthropic]' # native Anthropic SDK path
243
+ pip install 'augments-adk[otel]' # OpenTelemetry tracing bridge
244
+ pip install 'augments-adk[mcp]' # Model Context Protocol client
245
+ pip install 'augments-adk[viz]' # Agent graph visualization (graphviz)
246
+ pip install 'augments-adk[verbose]' # Rich-backed panel/line verbose renderer (ANSI fallback without it)
247
+ pip install 'augments-adk[all]' # all of the above
248
+ ```
249
+
250
+ ### Work on the ADK itself
251
+
252
+ Prerequisites: Python 3.12+ and [uv](https://docs.astral.sh/uv/) (uv installs the interpreter itself if you don't have it).
253
+
254
+ ```bash
255
+ # 1. Clone the repository
256
+ git clone https://github.com/augments-labs/augments-adk-python.git
257
+ cd augments-adk-python
258
+
259
+ # 2. Build .venv from the committed lockfile — everything + test + lint + typecheck
260
+ uv sync --extra dev
261
+
262
+ # 3. Run anything inside it
263
+ uv run augments --help
264
+ uv run pytest
265
+ ```
266
+
267
+ `uv sync` resolves from `uv.lock`, so every contributor and every CI job installs byte-identical versions. Activate the environment directly (`source .venv/bin/activate`) if you would rather not prefix commands with `uv run`, and swap `--extra dev` for any other extra (`--extra anthropic`, `--extra all`, or none at all) to work against a leaner surface.
268
+
269
+ Conda works too, if you prefer it:
270
+
271
+ ```bash
272
+ conda env create -f environment.yaml # also runs `pip install -e '.[dev]'`
273
+ conda activate augments-adk-python
274
+ ```
275
+
276
+ Either way the install is editable: `augments.adk` is importable from the `src/` layout defined in `pyproject.toml`, and source changes take effect immediately without reinstalling.
277
+
278
+ ### API Keys
279
+
280
+ Set the API keys for the LLM providers you want to use:
281
+
282
+ ```bash
283
+ export ANTHROPIC_API_KEY="your-key"
284
+ export OPENAI_API_KEY="your-key"
285
+ export GEMINI_API_KEY="your-key"
286
+ ```
287
+
288
+ ### Verify Installation
289
+
290
+ ```bash
291
+ python -c "from augments.adk import Agent, Runner; print('OK')"
292
+ ```
293
+
294
+ ## Quick Start
295
+
296
+ ```python
297
+ import asyncio
298
+ import logging
299
+
300
+ from augments.adk import Agent, Runner
301
+
302
+ logger = logging.getLogger(__name__)
303
+
304
+ agent = Agent(
305
+ name="Assistant",
306
+ system_prompt="You are a helpful assistant.",
307
+ )
308
+
309
+ result = asyncio.run(Runner.arun(agent, "Hello!"))
310
+ logger.info(result.final_output)
311
+ ```
312
+
313
+ ## Command-Line Interface
314
+
315
+ The `augments` console script drives agents from the terminal — scaffold a
316
+ project, validate its config without spending a token, then run or chat:
317
+
318
+ ```bash
319
+ augments new my_agent # scaffold config + tools + schema
320
+ augments validate my_agent/agent.json # strict schema check, no tokens
321
+ augments run my_agent/agent.json "hello" # one-shot run (config or --agent module:var)
322
+ augments chat my_agent/agent.json # interactive REPL, optional --session-db
323
+ augments serve my_agent/agent.json # REST + health over HTTP ([serve] extra)
324
+ ```
325
+
326
+ `run` auto-dispatches agents, swarms, graphs, and topologies; every
327
+ cost-affecting behavior (sessions, verbose rendering, tracing, env
328
+ files) stays off until you pass its flag. See `docs/cli/cli.md` for the
329
+ full command reference.
330
+
331
+ ## Deployment
332
+
333
+ Serve an agent over HTTP, then ship the container to any cloud. The
334
+ framework imports no server runtime and no cloud SDK — every piece is
335
+ opt-in, and you keep control of the runtime.
336
+
337
+ ```bash
338
+ pip install 'augments-adk[serve]'
339
+
340
+ # Serve locally: REST (POST /run, POST /run_sse) + health (/healthz, /readyz).
341
+ augments serve --agent my_agent.app:agent --host 0.0.0.0 --port 8000
342
+
343
+ # Generate the deployment artifacts you own (Dockerfile + manifests):
344
+ augments deploy init --target k8s --agent my_agent.app:agent --image my-agent:latest
345
+
346
+ # Or build and ship to a target via your installed CLIs:
347
+ augments deploy build --agent my_agent.app:agent --image my-agent:latest --push
348
+ augments deploy cloud-run --agent my_agent.app:agent --image gcr.io/PROJECT/my-agent --project PROJECT --region REGION
349
+ augments deploy gke --agent my_agent.app:agent --image IMAGE --project P --region R --cluster C
350
+ augments deploy ecs --agent my_agent.app:agent --image ACCT.dkr.ecr.R.amazonaws.com/my-agent --region R --execution-role-arn ARN
351
+ ```
352
+
353
+ `augments deploy` targets `docker`, `k8s`, `gke`, `helm`, `cloudrun`,
354
+ `ecs`, `app-runner`, and `lambda`. The generated image satisfies the
355
+ universal container contract (binds `0.0.0.0:$PORT`, config from env,
356
+ non-root, `/healthz` + `/readyz` probes), so the same image runs
357
+ everywhere. The generated `requirements.txt` installs `augments-adk`
358
+ from PyPI; edit it if you need a pin, a vendored wheel, or a VCS URL
359
+ instead.
360
+
361
+ A single replica works out of the box on the default per-pod SQLite
362
+ stores. For multi-replica (horizontally-scaled) deployments, back A2A
363
+ tasks and REST sessions with Postgres so state is shared across pods —
364
+ `augments serve --task-dsn "$PG_DSN" --session-dsn "$PG_DSN"` (install
365
+ `augments-adk[a2a-postgres,session-postgres]`). The AWS deploy commands
366
+ also accept `--push` to log in to ECR and build/push the image for you.
367
+ See [`docs/deploy/`](docs/deploy/) for the full guide.
368
+
369
+ ## Running Examples
370
+
371
+ All examples are runnable from the project root:
372
+
373
+ ```bash
374
+ python examples/agent_patterns/agents_as_tools.py
375
+ python examples/handoffs/llm_orchestrated.py
376
+ python examples/tools/tool_guardrails.py
377
+ ```
378
+
379
+ ## Project Structure
380
+
381
+ ```
382
+ src/augments/adk/ # Source code (namespace package)
383
+ tests/ # Unit and integration tests
384
+ examples/ # Single-file runnable examples (one concept each)
385
+ docs/ # Usage documentation
386
+ configs/ # Logging and other configs
387
+ ```
388
+
389
+ ## Key Dependencies
390
+
391
+ **Core**: `litellm` | `pydantic` | `griffe` | `aiosqlite` | `typing-extensions`
392
+ **Optional extras**: `anthropic` (`.[anthropic]`) | `mcp` (`.[mcp]`) | `opentelemetry-*` (`.[otel]`) | `graphviz` (`.[viz]`) | `rich` (`.[verbose]`)
393
+
394
+ ## Acknowledgements
395
+
396
+ This ADK draws on prior art and ongoing work from across the
397
+ multi-agent ecosystem:
398
+
399
+ - [**LangGraph**](https://github.com/langchain-ai/langgraph) —
400
+ state-machine multi-agent orchestration; influence on the
401
+ `graphs/` subsystem.
402
+ - [**CrewAI**](https://github.com/crewAIInc/crewAI) — multi-agent
403
+ collaboration patterns.
404
+ - [**OpenAI Swarm**](https://github.com/openai/swarm) — swarm cycle
405
+ pattern (reference shape for `swarms/`).
406
+ - [**OpenAI Agents SDK**](https://github.com/openai/openai-agents-python) —
407
+ Runner design and handoff mechanism reference.
408
+ - [**Anthropic Claude Agent SDK**](https://github.com/anthropics/anthropic-quickstarts) —
409
+ Anthropic-native provider design.
410
+ - [**LiteLLM**](https://github.com/BerriAI/litellm) — provider-agnostic
411
+ LLM abstraction over 100+ models.
412
+ - [**Model Context Protocol**](https://modelcontextprotocol.io/) —
413
+ tool-integration substrate.
414
+ - [**Pydantic**](https://docs.pydantic.dev/) — typed validation.
415
+ - [**Temporal**](https://temporal.io/) — durable execution backbone.
416
+ - [**OpenTelemetry**](https://opentelemetry.io/) and
417
+ [**OpenInference**](https://github.com/Arize-ai/openinference) —
418
+ observability conventions.
419
+
420
+ Inclusion here records influence, not endorsement.
@@ -0,0 +1,247 @@
1
+ <div align="center">
2
+ <picture>
3
+ <source media="(prefers-color-scheme: dark)"
4
+ srcset="https://raw.githubusercontent.com/augments-labs/augments-adk-python/main/docs/images/logo-dark.png">
5
+ <img src="https://raw.githubusercontent.com/augments-labs/augments-adk-python/main/docs/images/logo-light.png"
6
+ alt="Augments" width="256">
7
+ </picture>
8
+ </div>
9
+
10
+ # Augments Agent Development Kit (ADK)
11
+
12
+ A provider-agnostic Python framework to orchestrate complex
13
+ systems of agents that perform real-world actions, across 100+ LLMs via litellm.
14
+
15
+ ## The concept
16
+
17
+ An agent is a model that stopped talking and started doing: it calls tools,
18
+ changes state, and leaves side effects in the world. One agent is useful. A
19
+ set of specialists that cannot coordinate is a liability.
20
+
21
+ Augmenting a model means adding capability without giving up control. This
22
+ ADK adds three things, and each one is explicit and yours to configure:
23
+
24
+ - **Structure** — explicit orchestration. Graphs for state machines, flows
25
+ for pipelines, swarms for open-ended exploration, handoffs for delegation.
26
+ You write the structure; nothing is inferred behind your back.
27
+ - **Specialists** — agents scoped to one job, each carrying its own tools,
28
+ guardrails, and budget. An `Agent` is configuration, never a hidden runtime.
29
+ - **Control** — the `Runner`. Every run travels one execution path, where
30
+ turns, retries, token budgets, and interrupts are *enforced* rather than
31
+ suggested.
32
+
33
+ The framework never injects a prompt, a tool, or a token you did not ask for,
34
+ and every cost-bearing default starts bounded. Decisions, tool I/O, and token
35
+ spend come back as structured traces, so what the agents actually did is
36
+ readable after the fact.
37
+
38
+ ## Design tenets
39
+
40
+ 1. **Explicit over magical.** If you can't step through it in a debugger, it
41
+ doesn't belong in the orchestration path.
42
+ 2. **One obvious way.** Fewer knobs, sharper edges — the ADK has opinions.
43
+ 3. **Everything is inspectable.** Decisions, tool I/O, and token costs are
44
+ structured traces, not anecdotes.
45
+ 4. **Benchmarks or it didn't happen.** Claims ship with eval evidence or not
46
+ at all — including this framework's own.
47
+
48
+ ## Core Concepts
49
+
50
+ - [**Agents**](docs/agents/) — Autonomous entities with tools, guardrails, and handoffs
51
+ - [**Tools**](docs/tools/) — Function wrappers with schema validation and guardrails
52
+ - [**Handoffs**](docs/handoffs/) — Agent-to-agent routing (LLM-orchestrated or code-orchestrated)
53
+ - [**Guardrails**](docs/guardrails/) — Pre/post execution validation at agent and tool level
54
+ - [**Memory**](docs/memory/) — Persistent knowledge across sessions
55
+ - [**Skills**](docs/skills/) — Reusable capability packages (instructions + tools + governance)
56
+ - [**Tracing**](docs/tracing/) — OpenTelemetry observability
57
+
58
+ ## Installation
59
+
60
+ ### Use it in your own project
61
+
62
+ ```bash
63
+ uv add augments-adk # or: pip install augments-adk
64
+ ```
65
+
66
+ The core install is deliberately lean — `litellm`, `pydantic`, `griffe`, `aiosqlite`, `typing-extensions` — and every optional provider / exporter / UI enhancement is gated behind its own extra:
67
+
68
+ ```bash
69
+ pip install 'augments-adk[anthropic]' # native Anthropic SDK path
70
+ pip install 'augments-adk[otel]' # OpenTelemetry tracing bridge
71
+ pip install 'augments-adk[mcp]' # Model Context Protocol client
72
+ pip install 'augments-adk[viz]' # Agent graph visualization (graphviz)
73
+ pip install 'augments-adk[verbose]' # Rich-backed panel/line verbose renderer (ANSI fallback without it)
74
+ pip install 'augments-adk[all]' # all of the above
75
+ ```
76
+
77
+ ### Work on the ADK itself
78
+
79
+ Prerequisites: Python 3.12+ and [uv](https://docs.astral.sh/uv/) (uv installs the interpreter itself if you don't have it).
80
+
81
+ ```bash
82
+ # 1. Clone the repository
83
+ git clone https://github.com/augments-labs/augments-adk-python.git
84
+ cd augments-adk-python
85
+
86
+ # 2. Build .venv from the committed lockfile — everything + test + lint + typecheck
87
+ uv sync --extra dev
88
+
89
+ # 3. Run anything inside it
90
+ uv run augments --help
91
+ uv run pytest
92
+ ```
93
+
94
+ `uv sync` resolves from `uv.lock`, so every contributor and every CI job installs byte-identical versions. Activate the environment directly (`source .venv/bin/activate`) if you would rather not prefix commands with `uv run`, and swap `--extra dev` for any other extra (`--extra anthropic`, `--extra all`, or none at all) to work against a leaner surface.
95
+
96
+ Conda works too, if you prefer it:
97
+
98
+ ```bash
99
+ conda env create -f environment.yaml # also runs `pip install -e '.[dev]'`
100
+ conda activate augments-adk-python
101
+ ```
102
+
103
+ Either way the install is editable: `augments.adk` is importable from the `src/` layout defined in `pyproject.toml`, and source changes take effect immediately without reinstalling.
104
+
105
+ ### API Keys
106
+
107
+ Set the API keys for the LLM providers you want to use:
108
+
109
+ ```bash
110
+ export ANTHROPIC_API_KEY="your-key"
111
+ export OPENAI_API_KEY="your-key"
112
+ export GEMINI_API_KEY="your-key"
113
+ ```
114
+
115
+ ### Verify Installation
116
+
117
+ ```bash
118
+ python -c "from augments.adk import Agent, Runner; print('OK')"
119
+ ```
120
+
121
+ ## Quick Start
122
+
123
+ ```python
124
+ import asyncio
125
+ import logging
126
+
127
+ from augments.adk import Agent, Runner
128
+
129
+ logger = logging.getLogger(__name__)
130
+
131
+ agent = Agent(
132
+ name="Assistant",
133
+ system_prompt="You are a helpful assistant.",
134
+ )
135
+
136
+ result = asyncio.run(Runner.arun(agent, "Hello!"))
137
+ logger.info(result.final_output)
138
+ ```
139
+
140
+ ## Command-Line Interface
141
+
142
+ The `augments` console script drives agents from the terminal — scaffold a
143
+ project, validate its config without spending a token, then run or chat:
144
+
145
+ ```bash
146
+ augments new my_agent # scaffold config + tools + schema
147
+ augments validate my_agent/agent.json # strict schema check, no tokens
148
+ augments run my_agent/agent.json "hello" # one-shot run (config or --agent module:var)
149
+ augments chat my_agent/agent.json # interactive REPL, optional --session-db
150
+ augments serve my_agent/agent.json # REST + health over HTTP ([serve] extra)
151
+ ```
152
+
153
+ `run` auto-dispatches agents, swarms, graphs, and topologies; every
154
+ cost-affecting behavior (sessions, verbose rendering, tracing, env
155
+ files) stays off until you pass its flag. See `docs/cli/cli.md` for the
156
+ full command reference.
157
+
158
+ ## Deployment
159
+
160
+ Serve an agent over HTTP, then ship the container to any cloud. The
161
+ framework imports no server runtime and no cloud SDK — every piece is
162
+ opt-in, and you keep control of the runtime.
163
+
164
+ ```bash
165
+ pip install 'augments-adk[serve]'
166
+
167
+ # Serve locally: REST (POST /run, POST /run_sse) + health (/healthz, /readyz).
168
+ augments serve --agent my_agent.app:agent --host 0.0.0.0 --port 8000
169
+
170
+ # Generate the deployment artifacts you own (Dockerfile + manifests):
171
+ augments deploy init --target k8s --agent my_agent.app:agent --image my-agent:latest
172
+
173
+ # Or build and ship to a target via your installed CLIs:
174
+ augments deploy build --agent my_agent.app:agent --image my-agent:latest --push
175
+ augments deploy cloud-run --agent my_agent.app:agent --image gcr.io/PROJECT/my-agent --project PROJECT --region REGION
176
+ augments deploy gke --agent my_agent.app:agent --image IMAGE --project P --region R --cluster C
177
+ augments deploy ecs --agent my_agent.app:agent --image ACCT.dkr.ecr.R.amazonaws.com/my-agent --region R --execution-role-arn ARN
178
+ ```
179
+
180
+ `augments deploy` targets `docker`, `k8s`, `gke`, `helm`, `cloudrun`,
181
+ `ecs`, `app-runner`, and `lambda`. The generated image satisfies the
182
+ universal container contract (binds `0.0.0.0:$PORT`, config from env,
183
+ non-root, `/healthz` + `/readyz` probes), so the same image runs
184
+ everywhere. The generated `requirements.txt` installs `augments-adk`
185
+ from PyPI; edit it if you need a pin, a vendored wheel, or a VCS URL
186
+ instead.
187
+
188
+ A single replica works out of the box on the default per-pod SQLite
189
+ stores. For multi-replica (horizontally-scaled) deployments, back A2A
190
+ tasks and REST sessions with Postgres so state is shared across pods —
191
+ `augments serve --task-dsn "$PG_DSN" --session-dsn "$PG_DSN"` (install
192
+ `augments-adk[a2a-postgres,session-postgres]`). The AWS deploy commands
193
+ also accept `--push` to log in to ECR and build/push the image for you.
194
+ See [`docs/deploy/`](docs/deploy/) for the full guide.
195
+
196
+ ## Running Examples
197
+
198
+ All examples are runnable from the project root:
199
+
200
+ ```bash
201
+ python examples/agent_patterns/agents_as_tools.py
202
+ python examples/handoffs/llm_orchestrated.py
203
+ python examples/tools/tool_guardrails.py
204
+ ```
205
+
206
+ ## Project Structure
207
+
208
+ ```
209
+ src/augments/adk/ # Source code (namespace package)
210
+ tests/ # Unit and integration tests
211
+ examples/ # Single-file runnable examples (one concept each)
212
+ docs/ # Usage documentation
213
+ configs/ # Logging and other configs
214
+ ```
215
+
216
+ ## Key Dependencies
217
+
218
+ **Core**: `litellm` | `pydantic` | `griffe` | `aiosqlite` | `typing-extensions`
219
+ **Optional extras**: `anthropic` (`.[anthropic]`) | `mcp` (`.[mcp]`) | `opentelemetry-*` (`.[otel]`) | `graphviz` (`.[viz]`) | `rich` (`.[verbose]`)
220
+
221
+ ## Acknowledgements
222
+
223
+ This ADK draws on prior art and ongoing work from across the
224
+ multi-agent ecosystem:
225
+
226
+ - [**LangGraph**](https://github.com/langchain-ai/langgraph) —
227
+ state-machine multi-agent orchestration; influence on the
228
+ `graphs/` subsystem.
229
+ - [**CrewAI**](https://github.com/crewAIInc/crewAI) — multi-agent
230
+ collaboration patterns.
231
+ - [**OpenAI Swarm**](https://github.com/openai/swarm) — swarm cycle
232
+ pattern (reference shape for `swarms/`).
233
+ - [**OpenAI Agents SDK**](https://github.com/openai/openai-agents-python) —
234
+ Runner design and handoff mechanism reference.
235
+ - [**Anthropic Claude Agent SDK**](https://github.com/anthropics/anthropic-quickstarts) —
236
+ Anthropic-native provider design.
237
+ - [**LiteLLM**](https://github.com/BerriAI/litellm) — provider-agnostic
238
+ LLM abstraction over 100+ models.
239
+ - [**Model Context Protocol**](https://modelcontextprotocol.io/) —
240
+ tool-integration substrate.
241
+ - [**Pydantic**](https://docs.pydantic.dev/) — typed validation.
242
+ - [**Temporal**](https://temporal.io/) — durable execution backbone.
243
+ - [**OpenTelemetry**](https://opentelemetry.io/) and
244
+ [**OpenInference**](https://github.com/Arize-ai/openinference) —
245
+ observability conventions.
246
+
247
+ Inclusion here records influence, not endorsement.