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.
- augments_adk-0.3.0/LICENSE +21 -0
- augments_adk-0.3.0/PKG-INFO +420 -0
- augments_adk-0.3.0/README.md +247 -0
- augments_adk-0.3.0/pyproject.toml +684 -0
- augments_adk-0.3.0/setup.cfg +4 -0
- augments_adk-0.3.0/src/augments/adk/__init__.py +330 -0
- augments_adk-0.3.0/src/augments/adk/a2a/__init__.py +138 -0
- augments_adk-0.3.0/src/augments/adk/a2a/a2a_agent.py +585 -0
- augments_adk-0.3.0/src/augments/adk/a2a/a2a_client.py +1140 -0
- augments_adk-0.3.0/src/augments/adk/a2a/a2a_continuation_token.py +134 -0
- augments_adk-0.3.0/src/augments/adk/a2a/a2a_runner.py +241 -0
- augments_adk-0.3.0/src/augments/adk/a2a/adapters.py +229 -0
- augments_adk-0.3.0/src/augments/adk/a2a/app_factory.py +154 -0
- augments_adk-0.3.0/src/augments/adk/a2a/converters.py +491 -0
- augments_adk-0.3.0/src/augments/adk/a2a/exceptions.py +165 -0
- augments_adk-0.3.0/src/augments/adk/a2a/executor.py +513 -0
- augments_adk-0.3.0/src/augments/adk/a2a/postgres_task_store.py +257 -0
- augments_adk-0.3.0/src/augments/adk/a2a/server.py +195 -0
- augments_adk-0.3.0/src/augments/adk/a2a/task_store.py +526 -0
- augments_adk-0.3.0/src/augments/adk/agents/__init__.py +73 -0
- augments_adk-0.3.0/src/augments/adk/agents/agent.py +975 -0
- augments_adk-0.3.0/src/augments/adk/agents/agent_guardrails.py +749 -0
- augments_adk-0.3.0/src/augments/adk/agents/middleware.py +104 -0
- augments_adk-0.3.0/src/augments/adk/audit/__init__.py +19 -0
- augments_adk-0.3.0/src/augments/adk/audit/event.py +88 -0
- augments_adk-0.3.0/src/augments/adk/audit/sink.py +98 -0
- augments_adk-0.3.0/src/augments/adk/audit/sinks/__init__.py +3 -0
- augments_adk-0.3.0/src/augments/adk/audit/sinks/postgres.py +134 -0
- augments_adk-0.3.0/src/augments/adk/audit/sinks/s3.py +85 -0
- augments_adk-0.3.0/src/augments/adk/budgets/__init__.py +14 -0
- augments_adk-0.3.0/src/augments/adk/budgets/budget.py +86 -0
- augments_adk-0.3.0/src/augments/adk/budgets/ledger.py +92 -0
- augments_adk-0.3.0/src/augments/adk/budgets/ledgers/__init__.py +1 -0
- augments_adk-0.3.0/src/augments/adk/budgets/ledgers/postgres.py +147 -0
- augments_adk-0.3.0/src/augments/adk/budgets/ledgers/redis.py +134 -0
- augments_adk-0.3.0/src/augments/adk/cli/__init__.py +51 -0
- augments_adk-0.3.0/src/augments/adk/cli/__main__.py +6 -0
- augments_adk-0.3.0/src/augments/adk/cli/chat.py +152 -0
- augments_adk-0.3.0/src/augments/adk/cli/deploy.py +486 -0
- augments_adk-0.3.0/src/augments/adk/cli/errors.py +41 -0
- augments_adk-0.3.0/src/augments/adk/cli/loading.py +194 -0
- augments_adk-0.3.0/src/augments/adk/cli/new.py +62 -0
- augments_adk-0.3.0/src/augments/adk/cli/options.py +118 -0
- augments_adk-0.3.0/src/augments/adk/cli/run.py +277 -0
- augments_adk-0.3.0/src/augments/adk/cli/schema.py +46 -0
- augments_adk-0.3.0/src/augments/adk/cli/serve.py +389 -0
- augments_adk-0.3.0/src/augments/adk/cli/sessions.py +147 -0
- augments_adk-0.3.0/src/augments/adk/cli/templates.py +148 -0
- augments_adk-0.3.0/src/augments/adk/cli/validate.py +88 -0
- augments_adk-0.3.0/src/augments/adk/config/__init__.py +39 -0
- augments_adk-0.3.0/src/augments/adk/config/assembler.py +142 -0
- augments_adk-0.3.0/src/augments/adk/config/dump.py +240 -0
- augments_adk-0.3.0/src/augments/adk/config/guardrails.py +43 -0
- augments_adk-0.3.0/src/augments/adk/config/hosted_tools.py +97 -0
- augments_adk-0.3.0/src/augments/adk/config/loader.py +113 -0
- augments_adk-0.3.0/src/augments/adk/config/providers.py +246 -0
- augments_adk-0.3.0/src/augments/adk/config/resolver.py +278 -0
- augments_adk-0.3.0/src/augments/adk/config/schema.py +82 -0
- augments_adk-0.3.0/src/augments/adk/config/topology.py +430 -0
- augments_adk-0.3.0/src/augments/adk/context/__init__.py +42 -0
- augments_adk-0.3.0/src/augments/adk/context/compaction.py +277 -0
- augments_adk-0.3.0/src/augments/adk/context/context_config.py +275 -0
- augments_adk-0.3.0/src/augments/adk/context/context_editing.py +239 -0
- augments_adk-0.3.0/src/augments/adk/context/context_manager.py +442 -0
- augments_adk-0.3.0/src/augments/adk/context/directives.py +248 -0
- augments_adk-0.3.0/src/augments/adk/context/token_counter.py +66 -0
- augments_adk-0.3.0/src/augments/adk/databases/__init__.py +5 -0
- augments_adk-0.3.0/src/augments/adk/databases/connections/__init__.py +5 -0
- augments_adk-0.3.0/src/augments/adk/databases/connections/sqlite/__init__.py +5 -0
- augments_adk-0.3.0/src/augments/adk/databases/connections/sqlite/sqlite_database_connection.py +246 -0
- augments_adk-0.3.0/src/augments/adk/deploy/__init__.py +59 -0
- augments_adk-0.3.0/src/augments/adk/deploy/artifacts.py +39 -0
- augments_adk-0.3.0/src/augments/adk/deploy/aws_ecr.py +70 -0
- augments_adk-0.3.0/src/augments/adk/deploy/aws_manifests.py +165 -0
- augments_adk-0.3.0/src/augments/adk/deploy/cloudrun_manifests.py +66 -0
- augments_adk-0.3.0/src/augments/adk/deploy/commands.py +183 -0
- augments_adk-0.3.0/src/augments/adk/deploy/context.py +72 -0
- augments_adk-0.3.0/src/augments/adk/deploy/helm_chart.py +240 -0
- augments_adk-0.3.0/src/augments/adk/deploy/k8s_manifests.py +238 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/__init__.py +42 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/apprunner.py +79 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/aws_lambda.py +92 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/base.py +37 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/cloudrun.py +91 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/docker.py +71 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/ecs.py +100 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/gke.py +85 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/helm.py +68 -0
- augments_adk-0.3.0/src/augments/adk/deploy/targets/k8s.py +61 -0
- augments_adk-0.3.0/src/augments/adk/deploy/templates.py +126 -0
- augments_adk-0.3.0/src/augments/adk/exceptions/__init__.py +141 -0
- augments_adk-0.3.0/src/augments/adk/exceptions/exceptions.py +1309 -0
- augments_adk-0.3.0/src/augments/adk/flows/__init__.py +198 -0
- augments_adk-0.3.0/src/augments/adk/flows/agent_bridge.py +265 -0
- augments_adk-0.3.0/src/augments/adk/flows/approval_policy.py +101 -0
- augments_adk-0.3.0/src/augments/adk/flows/checkpoint.py +583 -0
- augments_adk-0.3.0/src/augments/adk/flows/combinators.py +194 -0
- augments_adk-0.3.0/src/augments/adk/flows/config.py +93 -0
- augments_adk-0.3.0/src/augments/adk/flows/decorators.py +422 -0
- augments_adk-0.3.0/src/augments/adk/flows/deferred.py +247 -0
- augments_adk-0.3.0/src/augments/adk/flows/definition.py +370 -0
- augments_adk-0.3.0/src/augments/adk/flows/events.py +323 -0
- augments_adk-0.3.0/src/augments/adk/flows/exceptions.py +322 -0
- augments_adk-0.3.0/src/augments/adk/flows/executable.py +143 -0
- augments_adk-0.3.0/src/augments/adk/flows/executor.py +1481 -0
- augments_adk-0.3.0/src/augments/adk/flows/flow.py +642 -0
- augments_adk-0.3.0/src/augments/adk/flows/flow_wrappers.py +486 -0
- augments_adk-0.3.0/src/augments/adk/flows/registry.py +347 -0
- augments_adk-0.3.0/src/augments/adk/flows/result.py +367 -0
- augments_adk-0.3.0/src/augments/adk/flows/sqlite_worker_backend.py +385 -0
- augments_adk-0.3.0/src/augments/adk/flows/step_cache_policy.py +110 -0
- augments_adk-0.3.0/src/augments/adk/flows/step_context.py +103 -0
- augments_adk-0.3.0/src/augments/adk/flows/step_guardrails.py +135 -0
- augments_adk-0.3.0/src/augments/adk/flows/step_rate_limit.py +82 -0
- augments_adk-0.3.0/src/augments/adk/flows/triggers.py +88 -0
- augments_adk-0.3.0/src/augments/adk/flows/worker_backend.py +424 -0
- augments_adk-0.3.0/src/augments/adk/graphs/__init__.py +182 -0
- augments_adk-0.3.0/src/augments/adk/graphs/adapters.py +1103 -0
- augments_adk-0.3.0/src/augments/adk/graphs/builder.py +623 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointer.py +209 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/__init__.py +40 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/hooks.py +100 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/in_memory.py +126 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/postgres.py +283 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/redis.py +256 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/s3.py +254 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/sqlite.py +194 -0
- augments_adk-0.3.0/src/augments/adk/graphs/checkpointers/tiered.py +321 -0
- augments_adk-0.3.0/src/augments/adk/graphs/config.py +177 -0
- augments_adk-0.3.0/src/augments/adk/graphs/events.py +365 -0
- augments_adk-0.3.0/src/augments/adk/graphs/graph.py +407 -0
- augments_adk-0.3.0/src/augments/adk/graphs/hooks.py +424 -0
- augments_adk-0.3.0/src/augments/adk/graphs/interrupt.py +420 -0
- augments_adk-0.3.0/src/augments/adk/graphs/join.py +224 -0
- augments_adk-0.3.0/src/augments/adk/graphs/merge.py +212 -0
- augments_adk-0.3.0/src/augments/adk/graphs/nested_snapshot.py +156 -0
- augments_adk-0.3.0/src/augments/adk/graphs/node.py +253 -0
- augments_adk-0.3.0/src/augments/adk/graphs/node_input.py +193 -0
- augments_adk-0.3.0/src/augments/adk/graphs/result.py +528 -0
- augments_adk-0.3.0/src/augments/adk/graphs/state.py +659 -0
- augments_adk-0.3.0/src/augments/adk/guardrails/__init__.py +69 -0
- augments_adk-0.3.0/src/augments/adk/guardrails/injection.py +161 -0
- augments_adk-0.3.0/src/augments/adk/guardrails/language.py +280 -0
- augments_adk-0.3.0/src/augments/adk/guardrails/pii.py +154 -0
- augments_adk-0.3.0/src/augments/adk/guardrails/scan.py +83 -0
- augments_adk-0.3.0/src/augments/adk/guardrails/semantic.py +291 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/__init__.py +66 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff.py +363 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_collapse_mode.py +36 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_config.py +251 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_filters.py +248 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_helpers.py +267 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_input_data.py +90 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_prompt.py +29 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_route.py +336 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_strategy.py +27 -0
- augments_adk-0.3.0/src/augments/adk/handoffs/handoff_target.py +354 -0
- augments_adk-0.3.0/src/augments/adk/hooks/__init__.py +9 -0
- augments_adk-0.3.0/src/augments/adk/hooks/hooks.py +1031 -0
- augments_adk-0.3.0/src/augments/adk/llms/__init__.py +75 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/__init__.py +35 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_boundary.py +149 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_cache_applicator.py +224 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_config.py +147 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_converter.py +1001 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_model.py +967 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_reasoning_resolver.py +107 -0
- augments_adk-0.3.0/src/augments/adk/llms/anthropic/anthropic_retry.py +104 -0
- augments_adk-0.3.0/src/augments/adk/llms/cost.py +39 -0
- augments_adk-0.3.0/src/augments/adk/llms/embedder.py +135 -0
- augments_adk-0.3.0/src/augments/adk/llms/gemini/__init__.py +38 -0
- augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_boundary.py +108 -0
- augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_config.py +122 -0
- augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_converter.py +686 -0
- augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_model.py +759 -0
- augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_reasoning_resolver.py +82 -0
- augments_adk-0.3.0/src/augments/adk/llms/gemini/gemini_retry.py +77 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/__init__.py +13 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_cache_applicator.py +60 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_converter.py +1028 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_embedder.py +117 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_model.py +1626 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_provider.py +77 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_retry.py +121 -0
- augments_adk-0.3.0/src/augments/adk/llms/litellm/litellm_token_counter.py +33 -0
- augments_adk-0.3.0/src/augments/adk/llms/llm.py +236 -0
- augments_adk-0.3.0/src/augments/adk/llms/llm_config.py +292 -0
- augments_adk-0.3.0/src/augments/adk/llms/llm_middleware.py +421 -0
- augments_adk-0.3.0/src/augments/adk/llms/llm_stream_middleware.py +494 -0
- augments_adk-0.3.0/src/augments/adk/llms/llm_usage.py +19 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/__init__.py +50 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_boundary.py +151 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_chatcompletions_config.py +119 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_chatcompletions_converter.py +624 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_chatcompletions_model.py +608 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_responses_config.py +129 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_responses_converter.py +760 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_responses_model.py +529 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_retry.py +92 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_stt.py +292 -0
- augments_adk-0.3.0/src/augments/adk/llms/openai/openai_tts.py +124 -0
- augments_adk-0.3.0/src/augments/adk/llms/retry.py +88 -0
- augments_adk-0.3.0/src/augments/adk/llms/routing/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/llms/routing/cheapest_first.py +51 -0
- augments_adk-0.3.0/src/augments/adk/llms/routing/latency_first.py +47 -0
- augments_adk-0.3.0/src/augments/adk/llms/routing/router.py +88 -0
- augments_adk-0.3.0/src/augments/adk/llms/stream_error.py +72 -0
- augments_adk-0.3.0/src/augments/adk/mcp/__init__.py +116 -0
- augments_adk-0.3.0/src/augments/adk/mcp/auth.py +55 -0
- augments_adk-0.3.0/src/augments/adk/mcp/conversion.py +251 -0
- augments_adk-0.3.0/src/augments/adk/mcp/elicitation.py +94 -0
- augments_adk-0.3.0/src/augments/adk/mcp/exceptions.py +66 -0
- augments_adk-0.3.0/src/augments/adk/mcp/extras.py +212 -0
- augments_adk-0.3.0/src/augments/adk/mcp/filters.py +59 -0
- augments_adk-0.3.0/src/augments/adk/mcp/http.py +425 -0
- augments_adk-0.3.0/src/augments/adk/mcp/manager.py +310 -0
- augments_adk-0.3.0/src/augments/adk/mcp/mcp_server.py +587 -0
- augments_adk-0.3.0/src/augments/adk/mcp/notifications.py +88 -0
- augments_adk-0.3.0/src/augments/adk/mcp/otel.py +56 -0
- augments_adk-0.3.0/src/augments/adk/mcp/run_hooks_bridge.py +82 -0
- augments_adk-0.3.0/src/augments/adk/mcp/sampling.py +365 -0
- augments_adk-0.3.0/src/augments/adk/mcp/schema_resolver.py +301 -0
- augments_adk-0.3.0/src/augments/adk/mcp/sse.py +189 -0
- augments_adk-0.3.0/src/augments/adk/mcp/stdio.py +231 -0
- augments_adk-0.3.0/src/augments/adk/memory/__init__.py +42 -0
- augments_adk-0.3.0/src/augments/adk/memory/distill.py +114 -0
- augments_adk-0.3.0/src/augments/adk/memory/extractor.py +209 -0
- augments_adk-0.3.0/src/augments/adk/memory/in_memory.py +332 -0
- augments_adk-0.3.0/src/augments/adk/memory/memory.py +254 -0
- augments_adk-0.3.0/src/augments/adk/memory/memory_config.py +58 -0
- augments_adk-0.3.0/src/augments/adk/memory/memory_types.py +119 -0
- augments_adk-0.3.0/src/augments/adk/memory/sqlite_memory.py +585 -0
- augments_adk-0.3.0/src/augments/adk/memory/stores/__init__.py +11 -0
- augments_adk-0.3.0/src/augments/adk/memory/stores/chroma.py +263 -0
- augments_adk-0.3.0/src/augments/adk/memory/stores/in_memory.py +176 -0
- augments_adk-0.3.0/src/augments/adk/memory/stores/pgvector.py +340 -0
- augments_adk-0.3.0/src/augments/adk/memory/stores/pinecone.py +254 -0
- augments_adk-0.3.0/src/augments/adk/memory/stores/qdrant.py +257 -0
- augments_adk-0.3.0/src/augments/adk/memory/vector_memory.py +190 -0
- augments_adk-0.3.0/src/augments/adk/memory/vector_store.py +149 -0
- augments_adk-0.3.0/src/augments/adk/orchestration/__init__.py +35 -0
- augments_adk-0.3.0/src/augments/adk/orchestration/executable.py +266 -0
- augments_adk-0.3.0/src/augments/adk/permissions/__init__.py +24 -0
- augments_adk-0.3.0/src/augments/adk/prompts/__init__.py +7 -0
- augments_adk-0.3.0/src/augments/adk/prompts/system_prompt.py +203 -0
- augments_adk-0.3.0/src/augments/adk/py.typed +0 -0
- augments_adk-0.3.0/src/augments/adk/rag/__init__.py +60 -0
- augments_adk-0.3.0/src/augments/adk/rag/chunking.py +166 -0
- augments_adk-0.3.0/src/augments/adk/rag/document.py +67 -0
- augments_adk-0.3.0/src/augments/adk/rag/index.py +150 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/__init__.py +40 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/base.py +81 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/directory.py +73 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/docx.py +73 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/github.py +136 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/pdf.py +71 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/plaintext.py +84 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/registry.py +145 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/structured.py +122 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/website.py +90 -0
- augments_adk-0.3.0/src/augments/adk/rag/loaders/youtube.py +161 -0
- augments_adk-0.3.0/src/augments/adk/run/__init__.py +97 -0
- augments_adk-0.3.0/src/augments/adk/run/agent_middleware.py +445 -0
- augments_adk-0.3.0/src/augments/adk/run/config.py +809 -0
- augments_adk-0.3.0/src/augments/adk/run/context.py +193 -0
- augments_adk-0.3.0/src/augments/adk/run/cost.py +199 -0
- augments_adk-0.3.0/src/augments/adk/run/demo.py +155 -0
- augments_adk-0.3.0/src/augments/adk/run/governance.py +158 -0
- augments_adk-0.3.0/src/augments/adk/run/graph_loop.py +2160 -0
- augments_adk-0.3.0/src/augments/adk/run/guardrails_executor.py +696 -0
- augments_adk-0.3.0/src/augments/adk/run/handoffs_executor.py +685 -0
- augments_adk-0.3.0/src/augments/adk/run/llm_calls.py +1462 -0
- augments_adk-0.3.0/src/augments/adk/run/loop.py +1622 -0
- augments_adk-0.3.0/src/augments/adk/run/messages.py +160 -0
- augments_adk-0.3.0/src/augments/adk/run/next_step.py +157 -0
- augments_adk-0.3.0/src/augments/adk/run/node_reliability.py +205 -0
- augments_adk-0.3.0/src/augments/adk/run/profile.py +1136 -0
- augments_adk-0.3.0/src/augments/adk/run/resumption.py +778 -0
- augments_adk-0.3.0/src/augments/adk/run/runner.py +5235 -0
- augments_adk-0.3.0/src/augments/adk/run/state.py +536 -0
- augments_adk-0.3.0/src/augments/adk/run/stream.py +579 -0
- augments_adk-0.3.0/src/augments/adk/run/swarm_loop.py +908 -0
- augments_adk-0.3.0/src/augments/adk/run/swarm_loop_streamed.py +816 -0
- augments_adk-0.3.0/src/augments/adk/run/swarm_resume.py +288 -0
- augments_adk-0.3.0/src/augments/adk/run/tools_executor.py +1580 -0
- augments_adk-0.3.0/src/augments/adk/run/turn_resolution.py +598 -0
- augments_adk-0.3.0/src/augments/adk/run/types.py +13 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/__init__.py +22 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/agent.py +122 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/apply_diff.py +387 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/apply_patch.py +299 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/__init__.py +34 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/base.py +225 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/capabilities.py +45 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/compaction.py +264 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/filesystem.py +34 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/memory.py +272 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/shell.py +87 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/capabilities/skills.py +498 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/__init__.py +30 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/base.py +196 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/docker/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/docker/docker_client.py +401 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/docker/docker_session.py +669 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/__init__.py +6 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/blaxel/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/blaxel/blaxel_client.py +174 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/cloudflare/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/cloudflare/cloudflare_client.py +174 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/daytona/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/daytona/daytona_client.py +174 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/e2b/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/e2b/e2b_client.py +208 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/modal/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/modal/modal_client.py +184 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/remote_vm/__init__.py +32 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/remote_vm/remote_vm_client.py +355 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/remote_vm/remote_vm_session.py +446 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/runloop/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/runloop/runloop_client.py +174 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/vercel/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/hosted/vercel/vercel_client.py +185 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/k8s/__init__.py +20 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/k8s/k8s_client.py +419 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/k8s/k8s_session.py +767 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/local/__init__.py +21 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/local/subprocess_client.py +512 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/clients/session.py +317 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/config.py +136 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/editor.py +102 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/guardrails/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/guardrails/command_guardrail.py +216 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/manifest_render.py +315 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/__init__.py +76 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/consolidation.py +76 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/interface.py +135 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/manager.py +372 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/prompts.py +188 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/rollout_extraction.py +133 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/rollouts.py +166 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/memory/storage.py +480 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/observability/__init__.py +26 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/observability/audit_sink.py +137 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/observability/observability.py +177 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/policy/__init__.py +44 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/policy/mounts.py +620 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/policy/network_policy.py +109 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/policy/resource_limits.py +70 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/remote_mount_policy.py +92 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/__init__.py +42 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/capability_lifecycle.py +165 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/concurrency_guard.py +83 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/iac_runner.py +293 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/instructions_composer.py +79 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/runner_integration/lifecycle.py +443 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/selector.py +83 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/__init__.py +145 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/archive_extraction.py +628 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/concurrency.py +102 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/dependencies.py +328 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/events.py +239 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/manager.py +284 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/__init__.py +178 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/git.py +185 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/inline.py +80 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/local.py +297 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/metadata.py +86 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/materialization/paths.py +56 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/op_codes.py +82 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/pty_types.py +182 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/runtime_helpers.py +495 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/sinks.py +371 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/tar_workspace.py +46 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/utils.py +65 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/session/workspace_payloads.py +209 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/__init__.py +26 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/gcs_store.py +259 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/local_store.py +153 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/s3_store.py +297 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/snapshot/store.py +49 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/snapshot_defaults.py +153 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/tools/__init__.py +20 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/tools/apply_patch_tool.py +56 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/tools/load_skill_tool.py +55 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/tools/run_command_tool.py +119 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/tools/view_image_tool.py +81 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/__init__.py +100 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/checksums.py +94 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/deep_merge.py +38 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/github.py +155 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/iterator_io.py +141 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/parse_utils.py +166 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/retry.py +229 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/tar_utils.py +466 -0
- augments_adk-0.3.0/src/augments/adk/sandbox/utils/token_truncation.py +251 -0
- augments_adk-0.3.0/src/augments/adk/schemas/__init__.py +34 -0
- augments_adk-0.3.0/src/augments/adk/schemas/agent_output_schema.py +584 -0
- augments_adk-0.3.0/src/augments/adk/schemas/function_schema.py +496 -0
- augments_adk-0.3.0/src/augments/adk/schemas/utils.py +637 -0
- augments_adk-0.3.0/src/augments/adk/serving/__init__.py +79 -0
- augments_adk-0.3.0/src/augments/adk/serving/app.py +111 -0
- augments_adk-0.3.0/src/augments/adk/serving/health.py +76 -0
- augments_adk-0.3.0/src/augments/adk/serving/rest.py +352 -0
- augments_adk-0.3.0/src/augments/adk/serving/serializers.py +175 -0
- augments_adk-0.3.0/src/augments/adk/session/__init__.py +26 -0
- augments_adk-0.3.0/src/augments/adk/session/buffered_session.py +264 -0
- augments_adk-0.3.0/src/augments/adk/session/multi_sessions.py +136 -0
- augments_adk-0.3.0/src/augments/adk/session/postgres_multi_sessions.py +345 -0
- augments_adk-0.3.0/src/augments/adk/session/postgres_session.py +414 -0
- augments_adk-0.3.0/src/augments/adk/session/session.py +88 -0
- augments_adk-0.3.0/src/augments/adk/session/session_event.py +75 -0
- augments_adk-0.3.0/src/augments/adk/session/session_settings.py +18 -0
- augments_adk-0.3.0/src/augments/adk/session/sqlite_multi_sessions.py +593 -0
- augments_adk-0.3.0/src/augments/adk/session/sqlite_session.py +629 -0
- augments_adk-0.3.0/src/augments/adk/session/state.py +224 -0
- augments_adk-0.3.0/src/augments/adk/skills/__init__.py +25 -0
- augments_adk-0.3.0/src/augments/adk/skills/activation.py +29 -0
- augments_adk-0.3.0/src/augments/adk/skills/discovery.py +617 -0
- augments_adk-0.3.0/src/augments/adk/skills/skill.py +270 -0
- augments_adk-0.3.0/src/augments/adk/skills/skill_prompt.py +58 -0
- augments_adk-0.3.0/src/augments/adk/skills/skill_rendering.py +111 -0
- augments_adk-0.3.0/src/augments/adk/skills/skill_set.py +58 -0
- augments_adk-0.3.0/src/augments/adk/skills/sources/__init__.py +15 -0
- augments_adk-0.3.0/src/augments/adk/skills/sources/base.py +41 -0
- augments_adk-0.3.0/src/augments/adk/skills/sources/directory.py +249 -0
- augments_adk-0.3.0/src/augments/adk/skills/sources/remote.py +122 -0
- augments_adk-0.3.0/src/augments/adk/status/__init__.py +36 -0
- augments_adk-0.3.0/src/augments/adk/status/hooks.py +279 -0
- augments_adk-0.3.0/src/augments/adk/status/store.py +446 -0
- augments_adk-0.3.0/src/augments/adk/status/types.py +179 -0
- augments_adk-0.3.0/src/augments/adk/swarms/__init__.py +169 -0
- augments_adk-0.3.0/src/augments/adk/swarms/builder.py +323 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointer.py +154 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/__init__.py +38 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/hooks.py +133 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/in_memory.py +93 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/postgres.py +261 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/redis.py +247 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/s3.py +247 -0
- augments_adk-0.3.0/src/augments/adk/swarms/checkpointers/tiered.py +252 -0
- augments_adk-0.3.0/src/augments/adk/swarms/config.py +137 -0
- augments_adk-0.3.0/src/augments/adk/swarms/events.py +207 -0
- augments_adk-0.3.0/src/augments/adk/swarms/hooks.py +329 -0
- augments_adk-0.3.0/src/augments/adk/swarms/interrupt.py +121 -0
- augments_adk-0.3.0/src/augments/adk/swarms/policy.py +615 -0
- augments_adk-0.3.0/src/augments/adk/swarms/result.py +409 -0
- augments_adk-0.3.0/src/augments/adk/swarms/shared_context.py +333 -0
- augments_adk-0.3.0/src/augments/adk/swarms/shared_context_strategy.py +54 -0
- augments_adk-0.3.0/src/augments/adk/swarms/state.py +703 -0
- augments_adk-0.3.0/src/augments/adk/swarms/stop_reason.py +63 -0
- augments_adk-0.3.0/src/augments/adk/swarms/swarm.py +445 -0
- augments_adk-0.3.0/src/augments/adk/swarms/swarm_prompt.py +53 -0
- augments_adk-0.3.0/src/augments/adk/swarms/termination.py +328 -0
- augments_adk-0.3.0/src/augments/adk/swarms/yield_signal.py +89 -0
- augments_adk-0.3.0/src/augments/adk/tasks/__init__.py +60 -0
- augments_adk-0.3.0/src/augments/adk/tasks/task.py +361 -0
- augments_adk-0.3.0/src/augments/adk/tasks/task_filters.py +135 -0
- augments_adk-0.3.0/src/augments/adk/tasks/task_group.py +219 -0
- augments_adk-0.3.0/src/augments/adk/tasks/task_input_data.py +95 -0
- augments_adk-0.3.0/src/augments/adk/tasks/task_output.py +239 -0
- augments_adk-0.3.0/src/augments/adk/tasks/task_pipeline.py +182 -0
- augments_adk-0.3.0/src/augments/adk/tasks/task_pipeline_state.py +172 -0
- augments_adk-0.3.0/src/augments/adk/tasks/topology.py +183 -0
- augments_adk-0.3.0/src/augments/adk/tools/__init__.py +233 -0
- augments_adk-0.3.0/src/augments/adk/tools/builtin/__init__.py +94 -0
- augments_adk-0.3.0/src/augments/adk/tools/builtin/builtin_tool.py +170 -0
- augments_adk-0.3.0/src/augments/adk/tools/builtin/document_search_tool.py +417 -0
- augments_adk-0.3.0/src/augments/adk/tools/builtin/jit_context_aware_tool.py +748 -0
- augments_adk-0.3.0/src/augments/adk/tools/builtin/memory_tool.py +254 -0
- augments_adk-0.3.0/src/augments/adk/tools/deferred_tool.py +380 -0
- augments_adk-0.3.0/src/augments/adk/tools/function_tool.py +1358 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/__init__.py +56 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/code_execution_tool.py +37 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/computer_tool.py +275 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/exceptions.py +50 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/file_search_tool.py +41 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/hosted_tool.py +55 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/image_generation_tool.py +38 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/mcp_tool.py +117 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/url_context_tool.py +27 -0
- augments_adk-0.3.0/src/augments/adk/tools/hosted/web_search_tool.py +86 -0
- augments_adk-0.3.0/src/augments/adk/tools/local/__init__.py +36 -0
- augments_adk-0.3.0/src/augments/adk/tools/local/apply_patch_tool.py +84 -0
- augments_adk-0.3.0/src/augments/adk/tools/local/shell_tool.py +110 -0
- augments_adk-0.3.0/src/augments/adk/tools/token_budget.py +44 -0
- augments_adk-0.3.0/src/augments/adk/tools/tool_context.py +249 -0
- augments_adk-0.3.0/src/augments/adk/tools/tool_guardrails.py +419 -0
- augments_adk-0.3.0/src/augments/adk/tools/tool_middleware.py +447 -0
- augments_adk-0.3.0/src/augments/adk/tools/tool_output_trimmer.py +161 -0
- augments_adk-0.3.0/src/augments/adk/tools/tool_search.py +370 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/__init__.py +52 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/abstract.py +179 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/clone.py +43 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/combined.py +81 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/filtered.py +99 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/function_toolset.py +78 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/mcp_toolset.py +271 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/prefixed.py +77 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/renamed.py +66 -0
- augments_adk-0.3.0/src/augments/adk/tools/toolsets/wrapper.py +81 -0
- augments_adk-0.3.0/src/augments/adk/tracing/__init__.py +118 -0
- augments_adk-0.3.0/src/augments/adk/tracing/exporters/__init__.py +8 -0
- augments_adk-0.3.0/src/augments/adk/tracing/exporters/helicone.py +52 -0
- augments_adk-0.3.0/src/augments/adk/tracing/exporters/langsmith.py +61 -0
- augments_adk-0.3.0/src/augments/adk/tracing/exporters/logfire.py +50 -0
- augments_adk-0.3.0/src/augments/adk/tracing/exporters/phoenix.py +36 -0
- augments_adk-0.3.0/src/augments/adk/tracing/logging.py +54 -0
- augments_adk-0.3.0/src/augments/adk/tracing/metrics/__init__.py +7 -0
- augments_adk-0.3.0/src/augments/adk/tracing/metrics/instruments.py +155 -0
- augments_adk-0.3.0/src/augments/adk/tracing/metrics/setup.py +53 -0
- augments_adk-0.3.0/src/augments/adk/tracing/metrics/tracer.py +188 -0
- augments_adk-0.3.0/src/augments/adk/tracing/multi_tracer.py +258 -0
- augments_adk-0.3.0/src/augments/adk/tracing/openinference/__init__.py +21 -0
- augments_adk-0.3.0/src/augments/adk/tracing/openinference/conventions.py +196 -0
- augments_adk-0.3.0/src/augments/adk/tracing/otel/__init__.py +24 -0
- augments_adk-0.3.0/src/augments/adk/tracing/otel/otel_span.py +149 -0
- augments_adk-0.3.0/src/augments/adk/tracing/otel/otel_tracer.py +695 -0
- augments_adk-0.3.0/src/augments/adk/tracing/otel/setup.py +212 -0
- augments_adk-0.3.0/src/augments/adk/tracing/spans.py +645 -0
- augments_adk-0.3.0/src/augments/adk/tracing/tracer.py +204 -0
- augments_adk-0.3.0/src/augments/adk/types/__init__.py +135 -0
- augments_adk-0.3.0/src/augments/adk/types/agents/__init__.py +17 -0
- augments_adk-0.3.0/src/augments/adk/types/agents/agent_as_tool_types.py +102 -0
- augments_adk-0.3.0/src/augments/adk/types/common/__init__.py +9 -0
- augments_adk-0.3.0/src/augments/adk/types/common/metadata.py +21 -0
- augments_adk-0.3.0/src/augments/adk/types/common/typing.py +5 -0
- augments_adk-0.3.0/src/augments/adk/types/config/__init__.py +87 -0
- augments_adk-0.3.0/src/augments/adk/types/config/agent_config.py +197 -0
- augments_adk-0.3.0/src/augments/adk/types/config/graph_config.py +85 -0
- augments_adk-0.3.0/src/augments/adk/types/config/guardrail_config.py +42 -0
- augments_adk-0.3.0/src/augments/adk/types/config/llm_config.py +594 -0
- augments_adk-0.3.0/src/augments/adk/types/config/prompt_config.py +22 -0
- augments_adk-0.3.0/src/augments/adk/types/config/references.py +80 -0
- augments_adk-0.3.0/src/augments/adk/types/config/swarm_config.py +168 -0
- augments_adk-0.3.0/src/augments/adk/types/config/tool_config.py +43 -0
- augments_adk-0.3.0/src/augments/adk/types/config/topology_config.py +65 -0
- augments_adk-0.3.0/src/augments/adk/types/guardrails/__init__.py +7 -0
- augments_adk-0.3.0/src/augments/adk/types/guardrails/action.py +56 -0
- augments_adk-0.3.0/src/augments/adk/types/input/__init__.py +84 -0
- augments_adk-0.3.0/src/augments/adk/types/input/llm_input_audio.py +34 -0
- augments_adk-0.3.0/src/augments/adk/types/input/llm_input_easy_message.py +36 -0
- augments_adk-0.3.0/src/augments/adk/types/input/llm_input_image.py +44 -0
- augments_adk-0.3.0/src/augments/adk/types/input/llm_input_message.py +45 -0
- augments_adk-0.3.0/src/augments/adk/types/input/llm_input_text.py +27 -0
- augments_adk-0.3.0/src/augments/adk/types/intents/__init__.py +46 -0
- augments_adk-0.3.0/src/augments/adk/types/items/__init__.py +45 -0
- augments_adk-0.3.0/src/augments/adk/types/items/items.py +1348 -0
- augments_adk-0.3.0/src/augments/adk/types/llms/__init__.py +18 -0
- augments_adk-0.3.0/src/augments/adk/types/llms/effort.py +21 -0
- augments_adk-0.3.0/src/augments/adk/types/llms/retry_policy.py +129 -0
- augments_adk-0.3.0/src/augments/adk/types/multimodal/__init__.py +38 -0
- augments_adk-0.3.0/src/augments/adk/types/multimodal/file.py +311 -0
- augments_adk-0.3.0/src/augments/adk/types/output/__init__.py +38 -0
- augments_adk-0.3.0/src/augments/adk/types/output/function_tool_call_result.py +74 -0
- augments_adk-0.3.0/src/augments/adk/types/output/function_tool_call_result_param.py +42 -0
- augments_adk-0.3.0/src/augments/adk/types/output/llm_response_function_tool_call_param.py +54 -0
- augments_adk-0.3.0/src/augments/adk/types/output/llm_response_message_param.py +42 -0
- augments_adk-0.3.0/src/augments/adk/types/output/llm_response_provider_item_param.py +44 -0
- augments_adk-0.3.0/src/augments/adk/types/output/llm_response_reasoning_param.py +67 -0
- augments_adk-0.3.0/src/augments/adk/types/output/llm_response_refusal_param.py +24 -0
- augments_adk-0.3.0/src/augments/adk/types/output/llm_response_text_param.py +28 -0
- augments_adk-0.3.0/src/augments/adk/types/output/reasoning_content_text_param.py +24 -0
- augments_adk-0.3.0/src/augments/adk/types/output/reasoning_summary_text_param.py +24 -0
- augments_adk-0.3.0/src/augments/adk/types/permissions/__init__.py +21 -0
- augments_adk-0.3.0/src/augments/adk/types/permissions/permission_types.py +482 -0
- augments_adk-0.3.0/src/augments/adk/types/responses/__init__.py +23 -0
- augments_adk-0.3.0/src/augments/adk/types/responses/llm_response.py +511 -0
- augments_adk-0.3.0/src/augments/adk/types/run/__init__.py +5 -0
- augments_adk-0.3.0/src/augments/adk/types/run/guardrail_audit.py +92 -0
- augments_adk-0.3.0/src/augments/adk/types/run/run_result.py +270 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/__init__.py +133 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/cost.py +172 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/entries.py +420 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/exec_result.py +187 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/iac.py +78 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/manifest.py +288 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/mounts.py +569 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/network.py +121 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/permissions.py +231 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/resource_limits.py +89 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/session_state.py +51 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/snapshot.py +164 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/usage.py +162 -0
- augments_adk-0.3.0/src/augments/adk/types/sandbox/workspace_paths.py +155 -0
- augments_adk-0.3.0/src/augments/adk/types/session/__init__.py +5 -0
- augments_adk-0.3.0/src/augments/adk/types/session/store.py +97 -0
- augments_adk-0.3.0/src/augments/adk/types/tokens/__init__.py +17 -0
- augments_adk-0.3.0/src/augments/adk/types/tokens/llm_usage.py +195 -0
- augments_adk-0.3.0/src/augments/adk/types/tokens/tokens.py +38 -0
- augments_adk-0.3.0/src/augments/adk/types/tools/__init__.py +121 -0
- augments_adk-0.3.0/src/augments/adk/types/tools/approval_policy.py +23 -0
- augments_adk-0.3.0/src/augments/adk/types/tools/builtin_tool_types.py +595 -0
- augments_adk-0.3.0/src/augments/adk/types/tools/tool_output_types.py +83 -0
- augments_adk-0.3.0/src/augments/adk/types/tools/tool_rate_limit.py +91 -0
- augments_adk-0.3.0/src/augments/adk/types/tools/tool_stream_event.py +60 -0
- augments_adk-0.3.0/src/augments/adk/types/tools/tool_use_behavior.py +89 -0
- augments_adk-0.3.0/src/augments/adk/types/tracing/__init__.py +49 -0
- augments_adk-0.3.0/src/augments/adk/types/tracing/convention.py +16 -0
- augments_adk-0.3.0/src/augments/adk/types/tracing/span_data.py +637 -0
- augments_adk-0.3.0/src/augments/adk/utils/__init__.py +7 -0
- augments_adk-0.3.0/src/augments/adk/utils/snake_case.py +39 -0
- augments_adk-0.3.0/src/augments/adk/utils/typedef.py +9 -0
- augments_adk-0.3.0/src/augments/adk/verbose/__init__.py +165 -0
- augments_adk-0.3.0/src/augments/adk/verbose/config.py +554 -0
- augments_adk-0.3.0/src/augments/adk/verbose/hooks.py +2252 -0
- augments_adk-0.3.0/src/augments/adk/verbose/mode.py +203 -0
- augments_adk-0.3.0/src/augments/adk/verbose/panel_renderer.py +1053 -0
- augments_adk-0.3.0/src/augments/adk/verbose/renderer.py +371 -0
- augments_adk-0.3.0/src/augments/adk/verbose/run_bridge.py +82 -0
- augments_adk-0.3.0/src/augments/adk/verbose/state.py +299 -0
- augments_adk-0.3.0/src/augments/adk/visualization/__init__.py +40 -0
- augments_adk-0.3.0/src/augments/adk/visualization/dot.py +454 -0
- augments_adk-0.3.0/src/augments/adk/visualization/helpers.py +208 -0
- augments_adk-0.3.0/src/augments/adk/visualization/mermaid.py +462 -0
- augments_adk-0.3.0/src/augments/adk/visualization/render.py +120 -0
- augments_adk-0.3.0/src/augments/adk/voice/__init__.py +106 -0
- augments_adk-0.3.0/src/augments/adk/voice/audio.py +196 -0
- augments_adk-0.3.0/src/augments/adk/voice/events.py +80 -0
- augments_adk-0.3.0/src/augments/adk/voice/exceptions.py +32 -0
- augments_adk-0.3.0/src/augments/adk/voice/pipeline.py +139 -0
- augments_adk-0.3.0/src/augments/adk/voice/pipeline_config.py +44 -0
- augments_adk-0.3.0/src/augments/adk/voice/result.py +214 -0
- augments_adk-0.3.0/src/augments/adk/voice/splitter.py +58 -0
- augments_adk-0.3.0/src/augments/adk/voice/stt.py +162 -0
- augments_adk-0.3.0/src/augments/adk/voice/tts.py +76 -0
- augments_adk-0.3.0/src/augments/adk/voice/workflow.py +125 -0
- augments_adk-0.3.0/src/augments/adk/workflows/__init__.py +30 -0
- augments_adk-0.3.0/src/augments/adk/workflows/engine.py +189 -0
- augments_adk-0.3.0/src/augments/adk/workflows/restate/__init__.py +23 -0
- augments_adk-0.3.0/src/augments/adk/workflows/restate/activity.py +140 -0
- augments_adk-0.3.0/src/augments/adk/workflows/restate/engine.py +117 -0
- augments_adk-0.3.0/src/augments/adk/workflows/restate/llm.py +250 -0
- augments_adk-0.3.0/src/augments/adk/workflows/restate/service.py +139 -0
- augments_adk-0.3.0/src/augments/adk/workflows/restate/tools.py +79 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/__init__.py +54 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/activity.py +254 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/determinism.py +80 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/engine.py +122 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/llm.py +515 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/mcp.py +166 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/plugin.py +222 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/routing.py +111 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/serialization.py +273 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/streaming.py +185 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/tools.py +319 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/tracing.py +110 -0
- augments_adk-0.3.0/src/augments/adk/workflows/temporal/workflow.py +237 -0
- augments_adk-0.3.0/src/augments_adk.egg-info/PKG-INFO +420 -0
- augments_adk-0.3.0/src/augments_adk.egg-info/SOURCES.txt +651 -0
- augments_adk-0.3.0/src/augments_adk.egg-info/dependency_links.txt +1 -0
- augments_adk-0.3.0/src/augments_adk.egg-info/entry_points.txt +2 -0
- augments_adk-0.3.0/src/augments_adk.egg-info/requires.txt +194 -0
- 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.
|