@google/adk 2.0.0 → 2.1.0
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.
- package/dist/cjs/a2a/a2a_event.js +0 -29
- package/dist/cjs/a2a/a2a_remote_agent.js +8 -5
- package/dist/cjs/a2a/a2a_remote_agent_utils.js +104 -2
- package/dist/cjs/a2a/agent_card.js +58 -6
- package/dist/cjs/a2a/agent_executor.js +10 -3
- package/dist/cjs/a2a/agent_to_a2a.js +6 -1
- package/dist/cjs/a2a/event_processor_utils.js +1 -1
- package/dist/cjs/a2a/executor_context.js +4 -2
- package/dist/cjs/a2a/index.js +4 -0
- package/dist/cjs/a2a/request_metadata.js +94 -0
- package/dist/cjs/agents/base_agent.js +9 -8
- package/dist/cjs/agents/context.js +7 -0
- package/dist/cjs/agents/functions.js +150 -47
- package/dist/cjs/agents/invocation_context.js +1 -0
- package/dist/cjs/agents/llm_agent.js +6 -5
- package/dist/cjs/agents/loop_agent.js +1 -1
- package/dist/cjs/agents/parallel_agent.js +1 -1
- package/dist/cjs/agents/processors/agent_transfer_llm_request_processor.js +26 -5
- package/dist/cjs/agents/processors/code_execution_request_processor.js +2 -2
- package/dist/cjs/agents/processors/content_processor_utils.js +5 -5
- package/dist/cjs/agents/processors/content_request_processor.js +4 -1
- package/dist/cjs/agents/processors/request_confirmation_llm_request_processor.js +19 -12
- package/dist/cjs/agents/processors/request_input_llm_request_processor.js +3 -7
- package/dist/cjs/agents/readonly_context.js +6 -0
- package/dist/cjs/agents/routed_agent.js +1 -1
- package/dist/cjs/agents/sequential_agent.js +1 -1
- package/dist/cjs/apps/app.js +1 -1
- package/dist/cjs/artifacts/file_artifact_service.js +1 -2
- package/dist/cjs/artifacts/scoped_artifact_service.js +9 -0
- package/dist/cjs/artifacts/session_artifact_service.js +19 -0
- package/dist/cjs/auth/auth_handler.js +44 -8
- package/dist/cjs/auth/oauth2/oauth2_credential_refresher.js +1 -1
- package/dist/cjs/auth/oauth2/oauth2_utils.js +17 -2
- package/dist/cjs/code_executors/base_code_executor.js +1 -1
- package/dist/cjs/code_executors/built_in_code_executor.js +1 -1
- package/dist/cjs/code_executors/code_execution_utils.js +2 -2
- package/dist/cjs/code_executors/container_code_executor.js +186 -0
- package/dist/cjs/code_executors/docker_container.js +282 -0
- package/dist/cjs/code_executors/unsafe_local_code_executor.js +11 -6
- package/dist/cjs/common.js +74 -45
- package/dist/cjs/context/agent_controlled_context_compactor.js +10 -5
- package/dist/cjs/context/anchored_context_compactor.js +41 -18
- package/dist/cjs/context/compaction_utils.js +15 -6
- package/dist/cjs/context/token_based_context_compactor.js +10 -2
- package/dist/cjs/context/trajectory_thought_pruning_compactor.js +17 -10
- package/dist/cjs/context/truncating_context_compactor.js +17 -5
- package/dist/cjs/errors/already_exists_error.js +50 -0
- package/dist/cjs/errors/input_validation_error.js +50 -0
- package/dist/cjs/errors/not_found_error.js +50 -0
- package/dist/cjs/errors/session_not_found_error.js +50 -0
- package/dist/cjs/errors/tool_execution_error.js +68 -0
- package/dist/cjs/events/event.js +1 -1
- package/dist/cjs/events/event_actions.js +7 -0
- package/dist/cjs/examples/base_example_provider.js +1 -1
- package/dist/cjs/index.js +47 -0
- package/dist/cjs/index_web.js +2 -0
- package/dist/cjs/integrations/agent_registry/agent_registry.js +3 -2
- package/dist/cjs/memory/rag_memory_transcript.js +136 -0
- package/dist/cjs/memory/vertex_ai_memory_bank_service.js +2 -2
- package/dist/cjs/memory/vertex_ai_rag_api.js +120 -0
- package/dist/cjs/memory/vertex_ai_rag_memory_service.js +228 -0
- package/dist/cjs/models/apigee_llm.js +1 -1
- package/dist/cjs/models/base_llm.js +1 -1
- package/dist/cjs/models/chrome_prompt_llm.js +304 -0
- package/dist/cjs/models/chrome_prompt_utils.js +229 -0
- package/dist/cjs/models/google_llm.js +9 -4
- package/dist/cjs/models/interactions_utils.js +2 -3
- package/dist/cjs/models/registry.js +2 -0
- package/dist/cjs/plugins/_reflect_retry_utils.js +142 -0
- package/dist/cjs/plugins/reflect_retry_model_plugin.js +274 -0
- package/dist/cjs/plugins/reflect_retry_tool_plugin.js +276 -0
- package/dist/cjs/runner/runner.js +29 -26
- package/dist/cjs/sessions/database_session_service.js +78 -43
- package/dist/cjs/sessions/db/operations.js +229 -22
- package/dist/cjs/sessions/db/schema.js +113 -159
- package/dist/cjs/sessions/registry.js +5 -1
- package/dist/cjs/sessions/vertex_ai_session_service.js +2 -2
- package/dist/cjs/telemetry/setup.js +1 -1
- package/dist/cjs/telemetry/tracing.js +11 -1
- package/dist/cjs/tools/agent_tool.js +4 -5
- package/dist/cjs/tools/base_tool.js +14 -4
- package/dist/cjs/tools/base_toolset.js +4 -4
- package/dist/cjs/tools/built_in_tool.js +75 -0
- package/dist/cjs/tools/enterprise_web_search_tool.js +3 -6
- package/dist/cjs/tools/forwarding_artifact_service.js +18 -13
- package/dist/cjs/tools/function_tool.js +6 -3
- package/dist/cjs/tools/google_maps_grounding_tool.js +3 -6
- package/dist/cjs/tools/google_search_tool.js +3 -6
- package/dist/cjs/tools/load_artifacts_tool.js +47 -23
- package/dist/cjs/tools/load_web_page.js +6 -172
- package/dist/cjs/tools/mcp/load_mcp_resource_tool.js +34 -5
- package/dist/cjs/tools/openapi_tool/auth/auth_helpers.js +11 -2
- package/dist/cjs/tools/openapi_tool/openapi_spec_parser/operation_parser.js +1 -1
- package/dist/cjs/tools/openapi_tool/rest_api_tool.js +16 -18
- package/dist/cjs/tools/preload_memory_tool.js +1 -1
- package/dist/cjs/tools/resume_inputs.js +27 -0
- package/dist/cjs/tools/retrieval/base_retrieval_tool.js +60 -0
- package/dist/cjs/tools/retrieval/files_retrieval.js +95 -0
- package/dist/cjs/tools/retrieval/llama_index_retrieval.js +57 -0
- package/dist/cjs/tools/url_context_tool.js +3 -6
- package/dist/cjs/tools/vertex_ai_search_tool.js +4 -8
- package/dist/cjs/tools/vertex_rag_retrieval_tool.js +3 -10
- package/dist/cjs/utils/crypto_shim.js +16 -0
- package/dist/cjs/utils/gemini_schema_util.js +0 -6
- package/dist/cjs/utils/logger.js +15 -50
- package/dist/cjs/utils/logger_browser.js +86 -0
- package/dist/cjs/utils/logger_node.js +119 -0
- package/dist/cjs/utils/redact_inline_data.js +46 -0
- package/dist/cjs/utils/redact_uri.js +15 -1
- package/dist/cjs/utils/schema.js +1 -1
- package/dist/cjs/utils/ssrf_guard.js +262 -0
- package/dist/cjs/version.js +1 -1
- package/dist/cjs/workflow/base_node.js +1 -1
- package/dist/cjs/workflow/graph.js +1 -1
- package/dist/cjs/workflow/nodes/node_tool.js +2 -11
- package/dist/cjs/workflow/request_input.js +1 -1
- package/dist/cjs/workflow/run_llm_agent_as_node.js +1 -1
- package/dist/cjs/workflow/workflow.js +1 -1
- package/dist/esm/a2a/a2a_event.js +0 -28
- package/dist/esm/a2a/a2a_remote_agent.js +10 -5
- package/dist/esm/a2a/a2a_remote_agent_utils.js +102 -2
- package/dist/esm/a2a/agent_card.js +63 -6
- package/dist/esm/a2a/agent_executor.js +10 -3
- package/dist/esm/a2a/agent_to_a2a.js +8 -2
- package/dist/esm/a2a/event_processor_utils.js +1 -2
- package/dist/esm/a2a/executor_context.js +4 -2
- package/dist/esm/a2a/index.js +7 -1
- package/dist/esm/a2a/request_metadata.js +68 -0
- package/dist/esm/agents/base_agent.js +9 -8
- package/dist/esm/agents/context.js +7 -0
- package/dist/esm/agents/functions.js +152 -48
- package/dist/esm/agents/invocation_context.js +1 -0
- package/dist/esm/agents/llm_agent.js +6 -5
- package/dist/esm/agents/loop_agent.js +1 -1
- package/dist/esm/agents/parallel_agent.js +1 -1
- package/dist/esm/agents/processors/agent_transfer_llm_request_processor.js +26 -5
- package/dist/esm/agents/processors/code_execution_request_processor.js +2 -2
- package/dist/esm/agents/processors/content_processor_utils.js +5 -5
- package/dist/esm/agents/processors/content_request_processor.js +4 -1
- package/dist/esm/agents/processors/request_confirmation_llm_request_processor.js +19 -12
- package/dist/esm/agents/processors/request_input_llm_request_processor.js +3 -7
- package/dist/esm/agents/readonly_context.js +6 -0
- package/dist/esm/agents/routed_agent.js +1 -1
- package/dist/esm/agents/sequential_agent.js +1 -1
- package/dist/esm/apps/app.js +1 -1
- package/dist/esm/artifacts/file_artifact_service.js +1 -2
- package/dist/esm/artifacts/scoped_artifact_service.js +9 -0
- package/dist/esm/artifacts/session_artifact_service.js +12 -0
- package/dist/esm/auth/auth_handler.js +47 -8
- package/dist/esm/auth/oauth2/oauth2_credential_refresher.js +1 -1
- package/dist/esm/auth/oauth2/oauth2_utils.js +15 -2
- package/dist/esm/code_executors/base_code_executor.js +1 -1
- package/dist/esm/code_executors/built_in_code_executor.js +1 -1
- package/dist/esm/code_executors/code_execution_utils.js +2 -2
- package/dist/esm/code_executors/container_code_executor.js +155 -0
- package/dist/esm/code_executors/docker_container.js +265 -0
- package/dist/esm/code_executors/unsafe_local_code_executor.js +11 -6
- package/dist/esm/common.js +73 -38
- package/dist/esm/context/agent_controlled_context_compactor.js +10 -5
- package/dist/esm/context/anchored_context_compactor.js +45 -19
- package/dist/esm/context/compaction_utils.js +13 -5
- package/dist/esm/context/token_based_context_compactor.js +10 -2
- package/dist/esm/context/trajectory_thought_pruning_compactor.js +17 -10
- package/dist/esm/context/truncating_context_compactor.js +17 -5
- package/dist/esm/errors/already_exists_error.js +23 -0
- package/dist/esm/errors/input_validation_error.js +23 -0
- package/dist/{web/auth/refresher/base_credential_refresher.js → esm/errors/not_found_error.js} +9 -6
- package/dist/esm/errors/session_not_found_error.js +23 -0
- package/dist/esm/errors/tool_execution_error.js +40 -0
- package/dist/esm/events/event.js +1 -1
- package/dist/esm/events/event_actions.js +6 -0
- package/dist/esm/examples/base_example_provider.js +1 -1
- package/dist/esm/index.js +37 -0
- package/dist/esm/index_web.js +2 -0
- package/dist/esm/integrations/agent_registry/agent_registry.js +3 -2
- package/dist/esm/memory/rag_memory_transcript.js +106 -0
- package/dist/esm/memory/vertex_ai_memory_bank_service.js +2 -2
- package/dist/esm/memory/vertex_ai_rag_api.js +93 -0
- package/dist/esm/memory/vertex_ai_rag_memory_service.js +208 -0
- package/dist/esm/models/apigee_llm.js +1 -1
- package/dist/esm/models/base_llm.js +1 -1
- package/dist/esm/models/chrome_prompt_llm.js +285 -0
- package/dist/esm/models/chrome_prompt_utils.js +192 -0
- package/dist/esm/models/google_llm.js +9 -4
- package/dist/esm/models/interactions_utils.js +2 -3
- package/dist/esm/models/registry.js +2 -0
- package/dist/esm/plugins/_reflect_retry_utils.js +111 -0
- package/dist/esm/plugins/reflect_retry_model_plugin.js +250 -0
- package/dist/esm/plugins/reflect_retry_tool_plugin.js +254 -0
- package/dist/esm/runner/runner.js +35 -27
- package/dist/esm/sessions/database_session_service.js +58 -25
- package/dist/esm/sessions/db/operations.js +229 -22
- package/dist/esm/sessions/db/schema.js +111 -160
- package/dist/esm/sessions/registry.js +5 -1
- package/dist/esm/sessions/vertex_ai_session_service.js +1 -1
- package/dist/esm/telemetry/setup.js +1 -1
- package/dist/esm/telemetry/tracing.js +11 -1
- package/dist/esm/tools/agent_tool.js +4 -5
- package/dist/esm/tools/base_tool.js +11 -3
- package/dist/esm/tools/base_toolset.js +4 -4
- package/dist/esm/tools/built_in_tool.js +51 -0
- package/dist/esm/tools/enterprise_web_search_tool.js +3 -6
- package/dist/esm/tools/forwarding_artifact_service.js +18 -13
- package/dist/esm/tools/function_tool.js +6 -3
- package/dist/esm/tools/google_maps_grounding_tool.js +3 -6
- package/dist/esm/tools/google_search_tool.js +3 -6
- package/dist/esm/tools/load_artifacts_tool.js +48 -24
- package/dist/esm/tools/load_web_page.js +10 -170
- package/dist/esm/tools/mcp/load_mcp_resource_tool.js +34 -5
- package/dist/esm/tools/openapi_tool/auth/auth_helpers.js +13 -2
- package/dist/esm/tools/openapi_tool/openapi_spec_parser/operation_parser.js +1 -1
- package/dist/esm/tools/openapi_tool/rest_api_tool.js +16 -18
- package/dist/esm/tools/preload_memory_tool.js +1 -1
- package/dist/esm/tools/retrieval/base_retrieval_tool.js +33 -0
- package/dist/esm/tools/retrieval/files_retrieval.js +81 -0
- package/dist/esm/tools/retrieval/llama_index_retrieval.js +30 -0
- package/dist/esm/tools/url_context_tool.js +3 -6
- package/dist/esm/tools/vertex_ai_search_tool.js +4 -8
- package/dist/esm/tools/vertex_rag_retrieval_tool.js +3 -10
- package/dist/esm/utils/crypto_shim.js +14 -0
- package/dist/esm/utils/gemini_schema_util.js +0 -6
- package/dist/esm/utils/logger.js +14 -40
- package/dist/esm/utils/logger_browser.js +58 -0
- package/dist/esm/utils/logger_node.js +81 -0
- package/dist/esm/utils/redact_inline_data.js +19 -0
- package/dist/esm/utils/redact_uri.js +15 -1
- package/dist/esm/utils/schema.js +1 -1
- package/dist/{web/tools/load_web_page.js → esm/utils/ssrf_guard.js} +95 -107
- package/dist/esm/version.js +1 -1
- package/dist/esm/workflow/base_node.js +1 -1
- package/dist/esm/workflow/graph.js +1 -1
- package/dist/esm/workflow/nodes/node_tool.js +2 -11
- package/dist/esm/workflow/request_input.js +1 -1
- package/dist/esm/workflow/run_llm_agent_as_node.js +1 -1
- package/dist/esm/workflow/workflow.js +1 -1
- package/dist/types/a2a/a2a_event.d.ts +7 -10
- package/dist/types/a2a/a2a_remote_agent.d.ts +5 -0
- package/dist/types/a2a/a2a_remote_agent_utils.d.ts +25 -0
- package/dist/types/a2a/agent_card.d.ts +9 -0
- package/dist/types/a2a/agent_executor.d.ts +1 -1
- package/dist/types/a2a/agent_to_a2a.d.ts +2 -0
- package/dist/types/a2a/auth.d.ts +5 -1
- package/dist/types/a2a/executor_context.d.ts +7 -1
- package/dist/types/a2a/index.d.ts +1 -1
- package/dist/types/a2a/request_metadata.d.ts +22 -0
- package/dist/types/agents/base_agent.d.ts +19 -16
- package/dist/types/agents/context.d.ts +8 -0
- package/dist/types/agents/functions.d.ts +72 -6
- package/dist/types/agents/invocation_context.d.ts +11 -3
- package/dist/types/agents/llm_agent.d.ts +9 -7
- package/dist/types/agents/processors/agent_transfer_llm_request_processor.d.ts +16 -1
- package/dist/types/agents/readonly_context.d.ts +4 -0
- package/dist/types/agents/run_config.d.ts +21 -8
- package/dist/types/artifacts/scoped_artifact_service.d.ts +13 -0
- package/dist/types/artifacts/session_artifact_service.d.ts +8 -1
- package/dist/types/auth/auth_credential.d.ts +13 -8
- package/dist/types/auth/auth_handler.d.ts +12 -1
- package/dist/types/auth/oauth2/oauth2_credential_refresher.d.ts +1 -1
- package/dist/types/auth/oauth2/oauth2_utils.d.ts +9 -0
- package/dist/types/auth/refresher/base_credential_refresher.d.ts +1 -1
- package/dist/types/code_executors/code_execution_utils.d.ts +1 -1
- package/dist/types/code_executors/container_code_executor.d.ts +89 -0
- package/dist/types/code_executors/docker_container.d.ts +79 -0
- package/dist/types/common.d.ts +22 -20
- package/dist/types/context/compaction_utils.d.ts +9 -1
- package/dist/types/errors/already_exists_error.d.ts +14 -0
- package/dist/types/errors/input_validation_error.d.ts +14 -0
- package/dist/types/errors/not_found_error.d.ts +14 -0
- package/dist/types/errors/session_not_found_error.d.ts +14 -0
- package/dist/types/errors/tool_execution_error.d.ts +36 -0
- package/dist/types/events/event_actions.d.ts +13 -0
- package/dist/types/index.d.ts +22 -1
- package/dist/types/memory/rag_memory_transcript.d.ts +55 -0
- package/dist/types/memory/vertex_ai_rag_api.d.ts +89 -0
- package/dist/types/memory/vertex_ai_rag_memory_service.d.ts +68 -0
- package/dist/types/models/chrome_prompt_llm.d.ts +210 -0
- package/dist/types/models/chrome_prompt_utils.d.ts +68 -0
- package/dist/types/models/google_llm.d.ts +8 -3
- package/dist/types/models/registry.d.ts +3 -2
- package/dist/types/plugins/_reflect_retry_utils.d.ts +72 -0
- package/dist/types/plugins/base_plugin.d.ts +12 -10
- package/dist/types/plugins/logging_plugin.d.ts +3 -2
- package/dist/types/plugins/reflect_retry_model_plugin.d.ts +132 -0
- package/dist/types/plugins/reflect_retry_tool_plugin.d.ts +187 -0
- package/dist/types/runner/runner.d.ts +7 -8
- package/dist/types/sessions/database_session_service.d.ts +1 -1
- package/dist/types/sessions/db/operations.d.ts +13 -1
- package/dist/types/sessions/db/schema.d.ts +23 -9
- package/dist/types/tools/base_tool.d.ts +11 -0
- package/dist/types/tools/base_toolset.d.ts +3 -4
- package/dist/types/tools/built_in_tool.d.ts +41 -0
- package/dist/types/tools/enterprise_web_search_tool.d.ts +4 -4
- package/dist/types/tools/forwarding_artifact_service.d.ts +23 -10
- package/dist/types/tools/function_tool.d.ts +42 -9
- package/dist/types/tools/google_maps_grounding_tool.d.ts +4 -4
- package/dist/types/tools/google_search_tool.d.ts +4 -4
- package/dist/types/tools/long_running_tool.d.ts +1 -1
- package/dist/types/tools/mcp/mcp_toolset.d.ts +11 -10
- package/dist/types/tools/openapi_tool/auth/auth_helpers.d.ts +5 -0
- package/dist/types/tools/openapi_tool/auth/credential_exchangers/auto_auth_credential_exchanger.d.ts +0 -1
- package/dist/types/tools/openapi_tool/auth/credential_exchangers/service_account_exchanger.d.ts +0 -1
- package/dist/types/tools/preload_memory_tool.d.ts +1 -1
- package/dist/types/tools/resume_inputs.d.ts +25 -0
- package/dist/types/tools/retrieval/base_retrieval_tool.d.ts +21 -0
- package/dist/types/tools/retrieval/files_retrieval.d.ts +50 -0
- package/dist/types/tools/retrieval/llama_index_retrieval.d.ts +62 -0
- package/dist/types/tools/url_context_tool.d.ts +4 -4
- package/dist/types/tools/vertex_ai_search_tool.d.ts +4 -4
- package/dist/types/tools/vertex_rag_retrieval_tool.d.ts +13 -12
- package/dist/types/utils/crypto_shim.d.ts +16 -0
- package/dist/types/utils/gemini_schema_util.d.ts +5 -7
- package/dist/types/utils/logger.d.ts +23 -1
- package/dist/types/utils/logger_browser.d.ts +28 -0
- package/dist/types/utils/logger_node.d.ts +20 -0
- package/dist/types/utils/output_schema_utils.d.ts +3 -3
- package/dist/types/utils/redact_inline_data.d.ts +16 -0
- package/dist/types/utils/redact_uri.d.ts +12 -8
- package/dist/types/utils/ssrf_guard.d.ts +32 -0
- package/dist/types/version.d.ts +1 -1
- package/dist/web/index_web.js +304 -3
- package/dist/web/index_web.js.map +7 -0
- package/package.json +17 -14
- package/dist/web/a2a/a2a_event.js +0 -290
- package/dist/web/a2a/a2a_remote_agent.js +0 -200
- package/dist/web/a2a/a2a_remote_agent_run_processor.js +0 -189
- package/dist/web/a2a/a2a_remote_agent_utils.js +0 -141
- package/dist/web/a2a/agent_card.js +0 -373
- package/dist/web/a2a/agent_executor.js +0 -246
- package/dist/web/a2a/agent_to_a2a.js +0 -126
- package/dist/web/a2a/auth.js +0 -41
- package/dist/web/a2a/event_converter_utils.js +0 -242
- package/dist/web/a2a/event_processor_utils.js +0 -209
- package/dist/web/a2a/executor_context.js +0 -28
- package/dist/web/a2a/index.js +0 -20
- package/dist/web/a2a/metadata_converter_utils.js +0 -115
- package/dist/web/a2a/part_converter_utils.js +0 -201
- package/dist/web/agents/active_streaming_tool.js +0 -25
- package/dist/web/agents/base_agent.js +0 -462
- package/dist/web/agents/context.js +0 -146
- package/dist/web/agents/framework_function_calls.js +0 -34
- package/dist/web/agents/functions.js +0 -462
- package/dist/web/agents/instructions.js +0 -188
- package/dist/web/agents/invocation_context.js +0 -126
- package/dist/web/agents/live_request_queue.js +0 -151
- package/dist/web/agents/llm_agent.js +0 -1514
- package/dist/web/agents/loop_agent.js +0 -131
- package/dist/web/agents/parallel_agent.js +0 -122
- package/dist/web/agents/processors/agent_transfer_llm_request_processor.js +0 -113
- package/dist/web/agents/processors/base_llm_processor.js +0 -21
- package/dist/web/agents/processors/basic_llm_request_processor.js +0 -84
- package/dist/web/agents/processors/code_execution_request_processor.js +0 -381
- package/dist/web/agents/processors/content_processor_utils.js +0 -432
- package/dist/web/agents/processors/content_request_processor.js +0 -73
- package/dist/web/agents/processors/context_compactor_request_processor.js +0 -78
- package/dist/web/agents/processors/identity_llm_request_processor.js +0 -59
- package/dist/web/agents/processors/instructions_llm_request_processor.js +0 -85
- package/dist/web/agents/processors/interactions_request_processor.js +0 -56
- package/dist/web/agents/processors/request_confirmation_llm_request_processor.js +0 -393
- package/dist/web/agents/processors/request_input_llm_request_processor.js +0 -228
- package/dist/web/agents/processors/tool_filter_request_processor.js +0 -72
- package/dist/web/agents/readonly_context.js +0 -60
- package/dist/web/agents/routed_agent.js +0 -161
- package/dist/web/agents/run_config.js +0 -74
- package/dist/web/agents/sequential_agent.js +0 -141
- package/dist/web/agents/transcription_entry.js +0 -7
- package/dist/web/agents/user_input_request.js +0 -135
- package/dist/web/apps/app.js +0 -52
- package/dist/web/apps/resumability_config.js +0 -33
- package/dist/web/artifacts/base_artifact_service.js +0 -7
- package/dist/web/artifacts/file_artifact_service.js +0 -568
- package/dist/web/artifacts/gcs_artifact_service.js +0 -298
- package/dist/web/artifacts/in_memory_artifact_service.js +0 -172
- package/dist/web/artifacts/registry.js +0 -36
- package/dist/web/artifacts/scoped_artifact_service.js +0 -89
- package/dist/web/auth/auth_credential.js +0 -18
- package/dist/web/auth/auth_handler.js +0 -151
- package/dist/web/auth/auth_preprocessor.js +0 -184
- package/dist/web/auth/auth_provider_registry.js +0 -37
- package/dist/web/auth/auth_schemes.js +0 -36
- package/dist/web/auth/auth_tool.js +0 -7
- package/dist/web/auth/base_auth_provider.js +0 -7
- package/dist/web/auth/credential_response_binding.js +0 -91
- package/dist/web/auth/credential_service/base_credential_service.js +0 -7
- package/dist/web/auth/credential_service/in_memory_credential_service.js +0 -50
- package/dist/web/auth/credential_service/session_state_credential_service.js +0 -27
- package/dist/web/auth/exchanger/base_credential_exchanger.js +0 -16
- package/dist/web/auth/exchanger/credential_exchanger_registry.js +0 -35
- package/dist/web/auth/oauth2/oauth2_credential_exchanger.js +0 -194
- package/dist/web/auth/oauth2/oauth2_credential_refresher.js +0 -105
- package/dist/web/auth/oauth2/oauth2_discovery.js +0 -196
- package/dist/web/auth/oauth2/oauth2_utils.js +0 -105
- package/dist/web/auth/refresher/credential_refresher_registry.js +0 -44
- package/dist/web/code_executors/agent_engine_sandbox_code_executor.js +0 -284
- package/dist/web/code_executors/base_code_executor.js +0 -68
- package/dist/web/code_executors/built_in_code_executor.js +0 -50
- package/dist/web/code_executors/code_execution_utils.js +0 -131
- package/dist/web/code_executors/code_executor_context.js +0 -174
- package/dist/web/code_executors/unsafe_local_code_executor.js +0 -221
- package/dist/web/common.js +0 -452
- package/dist/web/context/agent_controlled_context_compactor.js +0 -76
- package/dist/web/context/anchored_context_compactor.js +0 -120
- package/dist/web/context/base_context_compactor.js +0 -7
- package/dist/web/context/compaction_utils.js +0 -42
- package/dist/web/context/summarizers/base_summarizer.js +0 -7
- package/dist/web/context/summarizers/llm_summarizer.js +0 -91
- package/dist/web/context/token_based_context_compactor.js +0 -118
- package/dist/web/context/trajectory_thought_pruning_compactor.js +0 -47
- package/dist/web/context/truncating_context_compactor.js +0 -34
- package/dist/web/environment/base_environment.js +0 -71
- package/dist/web/environment/local_environment.js +0 -165
- package/dist/web/events/compacted_event.js +0 -51
- package/dist/web/events/event.js +0 -208
- package/dist/web/events/event_actions.js +0 -74
- package/dist/web/events/structured_events.js +0 -79
- package/dist/web/examples/base_example_provider.js +0 -32
- package/dist/web/examples/example.js +0 -7
- package/dist/web/examples/example_util.js +0 -83
- package/dist/web/features/feature_registry.js +0 -115
- package/dist/web/index.js +0 -57
- package/dist/web/integrations/agent_registry/agent_registry.js +0 -337
- package/dist/web/integrations/agent_registry/agent_registry_mcp_toolset.js +0 -120
- package/dist/web/integrations/agent_registry/helpers.js +0 -32
- package/dist/web/integrations/agent_registry/types.js +0 -20
- package/dist/web/memory/base_memory_service.js +0 -7
- package/dist/web/memory/in_memory_memory_service.js +0 -89
- package/dist/web/memory/memory_entry.js +0 -7
- package/dist/web/memory/vertex_ai_memory_bank_service.js +0 -496
- package/dist/web/models/apigee_llm.js +0 -227
- package/dist/web/models/base_llm.js +0 -78
- package/dist/web/models/base_llm_connection.js +0 -7
- package/dist/web/models/gemini_llm_connection.js +0 -183
- package/dist/web/models/google_llm.js +0 -396
- package/dist/web/models/interactions_utils.js +0 -745
- package/dist/web/models/llm_request.js +0 -57
- package/dist/web/models/llm_response.js +0 -48
- package/dist/web/models/registry.js +0 -100
- package/dist/web/models/routed_llm.js +0 -112
- package/dist/web/plugins/base_plugin.js +0 -305
- package/dist/web/plugins/global_instruction_plugin.js +0 -71
- package/dist/web/plugins/logging_plugin.js +0 -241
- package/dist/web/plugins/plugin_manager.js +0 -318
- package/dist/web/plugins/security_plugin.js +0 -148
- package/dist/web/runner/in_memory_runner.js +0 -50
- package/dist/web/runner/runner.js +0 -747
- package/dist/web/sessions/base_session_service.js +0 -123
- package/dist/web/sessions/database_session_service.js +0 -440
- package/dist/web/sessions/db/operations.js +0 -124
- package/dist/web/sessions/db/schema.js +0 -216
- package/dist/web/sessions/in_memory_session_service.js +0 -251
- package/dist/web/sessions/registry.js +0 -38
- package/dist/web/sessions/session.js +0 -23
- package/dist/web/sessions/state.js +0 -159
- package/dist/web/sessions/state_write_order.js +0 -74
- package/dist/web/sessions/vertex_ai_session_service.js +0 -546
- package/dist/web/skills/gcp_skill_registry.js +0 -80
- package/dist/web/skills/loader.js +0 -372
- package/dist/web/skills/prompt.js +0 -34
- package/dist/web/skills/skill.js +0 -67
- package/dist/web/skills/skill_registry.js +0 -7
- package/dist/web/telemetry/google_cloud.js +0 -99
- package/dist/web/telemetry/setup.js +0 -89
- package/dist/web/telemetry/tracing.js +0 -304
- package/dist/web/tools/agent_tool.js +0 -150
- package/dist/web/tools/base_tool.js +0 -116
- package/dist/web/tools/base_toolset.js +0 -62
- package/dist/web/tools/consolidate_context_tool.js +0 -55
- package/dist/web/tools/enterprise_web_search_tool.js +0 -60
- package/dist/web/tools/example_tool.js +0 -46
- package/dist/web/tools/exit_loop_tool.js +0 -47
- package/dist/web/tools/finish_task_tool.js +0 -92
- package/dist/web/tools/forwarding_artifact_service.js +0 -51
- package/dist/web/tools/function_tool.js +0 -151
- package/dist/web/tools/get_user_choice_tool.js +0 -27
- package/dist/web/tools/google_maps_grounding_tool.js +0 -59
- package/dist/web/tools/google_search_tool.js +0 -55
- package/dist/web/tools/load_artifacts_tool.js +0 -159
- package/dist/web/tools/load_memory_tool.js +0 -83
- package/dist/web/tools/long_running_tool.js +0 -58
- package/dist/web/tools/mcp/index.js +0 -11
- package/dist/web/tools/mcp/load_mcp_resource_tool.js +0 -111
- package/dist/web/tools/mcp/mcp_session_manager.js +0 -114
- package/dist/web/tools/mcp/mcp_tool.js +0 -47
- package/dist/web/tools/mcp/mcp_toolset.js +0 -144
- package/dist/web/tools/openapi_tool/auth/auth_helpers.js +0 -53
- package/dist/web/tools/openapi_tool/auth/credential_exchangers/auto_auth_credential_exchanger.js +0 -62
- package/dist/web/tools/openapi_tool/auth/credential_exchangers/service_account_exchanger.js +0 -113
- package/dist/web/tools/openapi_tool/openapi_spec_parser/openapi_spec_parser.js +0 -242
- package/dist/web/tools/openapi_tool/openapi_spec_parser/operation_parser.js +0 -200
- package/dist/web/tools/openapi_tool/openapi_spec_parser/tool_auth_handler.js +0 -98
- package/dist/web/tools/openapi_tool/openapi_toolset.js +0 -99
- package/dist/web/tools/openapi_tool/rest_api_tool.js +0 -247
- package/dist/web/tools/preload_memory_tool.js +0 -81
- package/dist/web/tools/request_input_tool.js +0 -29
- package/dist/web/tools/skill/list_skills_tool.js +0 -52
- package/dist/web/tools/skill/load_skill_resource_tool.js +0 -192
- package/dist/web/tools/skill/load_skill_tool.js +0 -96
- package/dist/web/tools/skill/run_skill_inline_script_tool.js +0 -191
- package/dist/web/tools/skill/run_skill_script_tool.js +0 -227
- package/dist/web/tools/skill/search_skills_tool.js +0 -84
- package/dist/web/tools/skill/skill_toolset.js +0 -185
- package/dist/web/tools/tool_confirmation.js +0 -43
- package/dist/web/tools/url_context_tool.js +0 -48
- package/dist/web/tools/vertex_ai_search_tool.js +0 -107
- package/dist/web/tools/vertex_rag_retrieval_tool.js +0 -40
- package/dist/web/utils/async_hooks_shim.js +0 -28
- package/dist/web/utils/async_queue.js +0 -88
- package/dist/web/utils/branch_trie.js +0 -16
- package/dist/web/utils/case_utils.js +0 -28
- package/dist/web/utils/client_labels.js +0 -70
- package/dist/web/utils/crypto_shim.js +0 -18
- package/dist/web/utils/deprecated.js +0 -40
- package/dist/web/utils/env_aware_utils.js +0 -115
- package/dist/web/utils/error_utils.js +0 -101
- package/dist/web/utils/experimental.js +0 -54
- package/dist/web/utils/failover_utils.js +0 -100
- package/dist/web/utils/file_extension_utils.js +0 -55
- package/dist/web/utils/file_utils.js +0 -108
- package/dist/web/utils/gemini_schema_util.js +0 -177
- package/dist/web/utils/genai_schema_to_json.js +0 -80
- package/dist/web/utils/live_connection_utils.js +0 -257
- package/dist/web/utils/logger.js +0 -140
- package/dist/web/utils/model_name.js +0 -75
- package/dist/web/utils/object_notation_utils.js +0 -54
- package/dist/web/utils/optional_peer.js +0 -38
- package/dist/web/utils/output_schema_utils.js +0 -18
- package/dist/web/utils/partial_copy.js +0 -26
- package/dist/web/utils/redact_uri.js +0 -39
- package/dist/web/utils/schema.js +0 -167
- package/dist/web/utils/simple_zod_to_json.js +0 -134
- package/dist/web/utils/streaming_utils.js +0 -438
- package/dist/web/utils/task.js +0 -38
- package/dist/web/utils/variant_utils.js +0 -23
- package/dist/web/utils/vertex_ai_utils.js +0 -27
- package/dist/web/version.js +0 -11
- package/dist/web/workflow/base_node.js +0 -215
- package/dist/web/workflow/branch_path.js +0 -77
- package/dist/web/workflow/dynamic_node_scheduler.js +0 -202
- package/dist/web/workflow/errors.js +0 -140
- package/dist/web/workflow/graph.js +0 -127
- package/dist/web/workflow/index.js +0 -80
- package/dist/web/workflow/node.js +0 -87
- package/dist/web/workflow/node_builders.js +0 -40
- package/dist/web/workflow/node_context.js +0 -236
- package/dist/web/workflow/node_error_event.js +0 -86
- package/dist/web/workflow/node_runner.js +0 -481
- package/dist/web/workflow/node_state.js +0 -43
- package/dist/web/workflow/node_status.js +0 -20
- package/dist/web/workflow/nodes/function_node.js +0 -221
- package/dist/web/workflow/nodes/join_node.js +0 -47
- package/dist/web/workflow/nodes/node_tool.js +0 -131
- package/dist/web/workflow/nodes/parallel_worker.js +0 -120
- package/dist/web/workflow/nodes/tool_node.js +0 -126
- package/dist/web/workflow/request_input.js +0 -36
- package/dist/web/workflow/retry_config.js +0 -43
- package/dist/web/workflow/run_llm_agent_as_node.js +0 -266
- package/dist/web/workflow/run_node_as_invocation.js +0 -170
- package/dist/web/workflow/schedule_dynamic_node.js +0 -26
- package/dist/web/workflow/trigger.js +0 -7
- package/dist/web/workflow/utils/graph_parser.js +0 -188
- package/dist/web/workflow/utils/graph_validation.js +0 -189
- package/dist/web/workflow/utils/hitl_utils.js +0 -239
- package/dist/web/workflow/utils/rehydration_utils.js +0 -342
- package/dist/web/workflow/utils/retry_utils.js +0 -82
- package/dist/web/workflow/utils/workflow_graph_utils.js +0 -115
- package/dist/web/workflow/workflow.js +0 -529
- /package/dist/{web/artifacts/session_artifact_service.js → esm/tools/resume_inputs.js} +0 -0
|
@@ -21,7 +21,7 @@ import { BaseLlm } from "./base_llm.js";
|
|
|
21
21
|
import { GeminiLlmConnection } from "./gemini_llm_connection.js";
|
|
22
22
|
import { generateContentViaInteractions } from "./interactions_utils.js";
|
|
23
23
|
import { createLlmResponse } from "./llm_response.js";
|
|
24
|
-
const GEMINI_MODEL_SYMBOL = Symbol.for("google.adk.geminiModel");
|
|
24
|
+
const GEMINI_MODEL_SYMBOL = /* @__PURE__ */ Symbol.for("google.adk.geminiModel");
|
|
25
25
|
function isGemini(obj) {
|
|
26
26
|
return typeof obj === "object" && obj !== null && GEMINI_MODEL_SYMBOL in obj && obj[GEMINI_MODEL_SYMBOL] === true;
|
|
27
27
|
}
|
|
@@ -66,9 +66,14 @@ const _Gemini = class _Gemini extends (_b = BaseLlm, _a = GEMINI_MODEL_SYMBOL, _
|
|
|
66
66
|
/**
|
|
67
67
|
* Sends a request to the Gemini model.
|
|
68
68
|
*
|
|
69
|
-
* @param llmRequest
|
|
70
|
-
* @param stream
|
|
71
|
-
* @yields LlmResponse
|
|
69
|
+
* @param llmRequest The request to send to the Gemini model.
|
|
70
|
+
* @param stream Whether to make a streaming call. Defaults to `false`.
|
|
71
|
+
* @yields Each {@link LlmResponse} produced by the model. A non-streaming
|
|
72
|
+
* call yields exactly one; a streaming call yields the partial responses
|
|
73
|
+
* as they arrive, then a final aggregated response whenever the stream
|
|
74
|
+
* carried either content or usage/grounding/citation metadata. A
|
|
75
|
+
* metadata-only stream therefore still ends with one, holding the
|
|
76
|
+
* metadata and no `content`; only a stream with neither ends silently.
|
|
72
77
|
*/
|
|
73
78
|
async *generateContentAsync(llmRequest, stream = false, abortSignal) {
|
|
74
79
|
var _a2, _b2, _c;
|
|
@@ -662,13 +662,12 @@ async function* generateContentViaInteractions(apiClient, llmRequest, stream) {
|
|
|
662
662
|
});
|
|
663
663
|
const aggregatedParts = [];
|
|
664
664
|
for await (const event of responses) {
|
|
665
|
-
const
|
|
666
|
-
const interactionId = extractStreamInteractionId(sseEvent);
|
|
665
|
+
const interactionId = extractStreamInteractionId(event);
|
|
667
666
|
if (interactionId) {
|
|
668
667
|
currentInteractionId = interactionId;
|
|
669
668
|
}
|
|
670
669
|
const llmResponse = convertInteractionEventToLlmResponse(
|
|
671
|
-
|
|
670
|
+
event,
|
|
672
671
|
aggregatedParts,
|
|
673
672
|
currentInteractionId
|
|
674
673
|
);
|
|
@@ -9,6 +9,7 @@ var __name = (target, value) => __defProp(target, "name", { value, configurable:
|
|
|
9
9
|
*/
|
|
10
10
|
import { logger } from "../utils/logger.js";
|
|
11
11
|
import { ApigeeLlm } from "./apigee_llm.js";
|
|
12
|
+
import { ChromeBuiltInLlm } from "./chrome_prompt_llm.js";
|
|
12
13
|
import { Gemini } from "./google_llm.js";
|
|
13
14
|
const _LRUCache = class _LRUCache {
|
|
14
15
|
constructor(maxSize) {
|
|
@@ -95,6 +96,7 @@ _LLMRegistry.resolveCache = new LRUCache(32);
|
|
|
95
96
|
let LLMRegistry = _LLMRegistry;
|
|
96
97
|
LLMRegistry.register(Gemini);
|
|
97
98
|
LLMRegistry.register(ApigeeLlm);
|
|
99
|
+
LLMRegistry.register(ChromeBuiltInLlm);
|
|
98
100
|
export {
|
|
99
101
|
LLMRegistry
|
|
100
102
|
};
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
import {createRequire as topLevelCreateRequire} from 'module';
|
|
2
|
+
const require = topLevelCreateRequire(import.meta.url);
|
|
3
|
+
var __defProp = Object.defineProperty;
|
|
4
|
+
var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
|
|
5
|
+
/**
|
|
6
|
+
* @license
|
|
7
|
+
* Copyright 2026 Google LLC
|
|
8
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
9
|
+
*/
|
|
10
|
+
const REFLECT_AND_RETRY_RESPONSE_TYPE = "ERROR_HANDLED_BY_REFLECT_AND_RETRY_PLUGIN";
|
|
11
|
+
const GLOBAL_SCOPE_KEY = "__global_reflect_and_retry_scope__";
|
|
12
|
+
var TrackingScope = /* @__PURE__ */ ((TrackingScope2) => {
|
|
13
|
+
TrackingScope2["INVOCATION"] = "invocation";
|
|
14
|
+
TrackingScope2["GLOBAL"] = "global";
|
|
15
|
+
return TrackingScope2;
|
|
16
|
+
})(TrackingScope || {});
|
|
17
|
+
function resolveScopeKey(scope, invocationId) {
|
|
18
|
+
if (scope === "invocation" /* INVOCATION */) {
|
|
19
|
+
if (!invocationId) {
|
|
20
|
+
throw new Error("invocation_id must be provided for INVOCATION scope");
|
|
21
|
+
}
|
|
22
|
+
return invocationId;
|
|
23
|
+
} else if (scope === "global" /* GLOBAL */) {
|
|
24
|
+
return GLOBAL_SCOPE_KEY;
|
|
25
|
+
}
|
|
26
|
+
throw new Error(`Unknown scope: ${scope}`);
|
|
27
|
+
}
|
|
28
|
+
__name(resolveScopeKey, "resolveScopeKey");
|
|
29
|
+
const _ScopedFailureTracker = class _ScopedFailureTracker {
|
|
30
|
+
constructor() {
|
|
31
|
+
this.scopedFailureCounters = /* @__PURE__ */ new Map();
|
|
32
|
+
this.lockPromise = Promise.resolve();
|
|
33
|
+
}
|
|
34
|
+
async acquireLock() {
|
|
35
|
+
let release;
|
|
36
|
+
const nextLock = new Promise((resolve) => {
|
|
37
|
+
release = resolve;
|
|
38
|
+
});
|
|
39
|
+
const currentLock = this.lockPromise;
|
|
40
|
+
this.lockPromise = this.lockPromise.then(() => nextLock);
|
|
41
|
+
await currentLock;
|
|
42
|
+
return release;
|
|
43
|
+
}
|
|
44
|
+
/**
|
|
45
|
+
* Atomically increments and returns the failure count for an item.
|
|
46
|
+
*
|
|
47
|
+
* @param scopeKey - The scope identifier (e.g. invocation ID or global key).
|
|
48
|
+
* @param itemName - The name of the tool or model.
|
|
49
|
+
* @returns The updated failure count.
|
|
50
|
+
*/
|
|
51
|
+
async increment(scopeKey, itemName) {
|
|
52
|
+
var _a;
|
|
53
|
+
const release = await this.acquireLock();
|
|
54
|
+
try {
|
|
55
|
+
let counter = this.scopedFailureCounters.get(scopeKey);
|
|
56
|
+
if (!counter) {
|
|
57
|
+
counter = /* @__PURE__ */ new Map();
|
|
58
|
+
this.scopedFailureCounters.set(scopeKey, counter);
|
|
59
|
+
}
|
|
60
|
+
const current = ((_a = counter.get(itemName)) != null ? _a : 0) + 1;
|
|
61
|
+
counter.set(itemName, current);
|
|
62
|
+
return current;
|
|
63
|
+
} finally {
|
|
64
|
+
release();
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
/**
|
|
68
|
+
* Atomically resets the failure count for an item and cleans up state.
|
|
69
|
+
*
|
|
70
|
+
* @param scopeKey - The scope identifier.
|
|
71
|
+
* @param itemName - The name of the tool or model.
|
|
72
|
+
*/
|
|
73
|
+
async reset(scopeKey, itemName) {
|
|
74
|
+
const release = await this.acquireLock();
|
|
75
|
+
try {
|
|
76
|
+
const counter = this.scopedFailureCounters.get(scopeKey);
|
|
77
|
+
if (counter) {
|
|
78
|
+
counter.delete(itemName);
|
|
79
|
+
if (counter.size === 0) {
|
|
80
|
+
this.scopedFailureCounters.delete(scopeKey);
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
} finally {
|
|
84
|
+
release();
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
/**
|
|
88
|
+
* Gets the current failure count for an item.
|
|
89
|
+
*
|
|
90
|
+
* @param scopeKey - The scope identifier.
|
|
91
|
+
* @param itemName - The name of the tool or model.
|
|
92
|
+
*/
|
|
93
|
+
async getCount(scopeKey, itemName) {
|
|
94
|
+
var _a, _b;
|
|
95
|
+
const release = await this.acquireLock();
|
|
96
|
+
try {
|
|
97
|
+
return (_b = (_a = this.scopedFailureCounters.get(scopeKey)) == null ? void 0 : _a.get(itemName)) != null ? _b : 0;
|
|
98
|
+
} finally {
|
|
99
|
+
release();
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
};
|
|
103
|
+
__name(_ScopedFailureTracker, "ScopedFailureTracker");
|
|
104
|
+
let ScopedFailureTracker = _ScopedFailureTracker;
|
|
105
|
+
export {
|
|
106
|
+
GLOBAL_SCOPE_KEY,
|
|
107
|
+
REFLECT_AND_RETRY_RESPONSE_TYPE,
|
|
108
|
+
ScopedFailureTracker,
|
|
109
|
+
TrackingScope,
|
|
110
|
+
resolveScopeKey
|
|
111
|
+
};
|
|
@@ -0,0 +1,250 @@
|
|
|
1
|
+
import {createRequire as topLevelCreateRequire} from 'module';
|
|
2
|
+
const require = topLevelCreateRequire(import.meta.url);
|
|
3
|
+
var __defProp = Object.defineProperty;
|
|
4
|
+
var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
|
|
5
|
+
/**
|
|
6
|
+
* @license
|
|
7
|
+
* Copyright 2026 Google LLC
|
|
8
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
9
|
+
*/
|
|
10
|
+
import { FinishReason } from "@google/genai";
|
|
11
|
+
import { v4 as uuidv4 } from "uuid";
|
|
12
|
+
import { FunctionTool } from "../tools/function_tool.js";
|
|
13
|
+
import {
|
|
14
|
+
REFLECT_AND_RETRY_RESPONSE_TYPE,
|
|
15
|
+
resolveScopeKey,
|
|
16
|
+
ScopedFailureTracker,
|
|
17
|
+
TrackingScope
|
|
18
|
+
} from "./_reflect_retry_utils.js";
|
|
19
|
+
import { BasePlugin } from "./base_plugin.js";
|
|
20
|
+
const RESERVED_TOOL_CALL_ERROR_TYPE = "RESERVED_TOOL_CALL";
|
|
21
|
+
const ADK_HANDLE_MODEL_ERROR_TOOL_NAME = "adk_handle_model_error";
|
|
22
|
+
const _ReflectAndRetryModelPlugin = class _ReflectAndRetryModelPlugin extends BasePlugin {
|
|
23
|
+
/**
|
|
24
|
+
* Initializes the {@link ReflectAndRetryModelPlugin}.
|
|
25
|
+
*
|
|
26
|
+
* @param options - Configuration options for the plugin.
|
|
27
|
+
*/
|
|
28
|
+
constructor(options = {}) {
|
|
29
|
+
var _a, _b, _c, _d, _e;
|
|
30
|
+
super((_a = options.name) != null ? _a : "reflect_retry_model_plugin");
|
|
31
|
+
const retries = (_b = options.maxRetries) != null ? _b : 3;
|
|
32
|
+
if (retries < 0) {
|
|
33
|
+
throw new Error("maxRetries must be a non-negative integer.");
|
|
34
|
+
}
|
|
35
|
+
this.maxRetries = retries;
|
|
36
|
+
this.throwExceptionIfRetryExceeded = (_c = options.throwExceptionIfRetryExceeded) != null ? _c : true;
|
|
37
|
+
this.scope = (_d = options.trackingScope) != null ? _d : TrackingScope.INVOCATION;
|
|
38
|
+
this.onModelErrors = (_e = options.onModelErrors) != null ? _e : [
|
|
39
|
+
FinishReason.MALFORMED_FUNCTION_CALL
|
|
40
|
+
];
|
|
41
|
+
this.tracker = new ScopedFailureTracker();
|
|
42
|
+
}
|
|
43
|
+
/**
|
|
44
|
+
* Internal framework reflection tool handler.
|
|
45
|
+
*/
|
|
46
|
+
adkHandleModelError({
|
|
47
|
+
retryCount
|
|
48
|
+
}) {
|
|
49
|
+
const attempt = retryCount != null ? retryCount : 1;
|
|
50
|
+
return {
|
|
51
|
+
reflection_guidance: `
|
|
52
|
+
The call to the model failed.
|
|
53
|
+
|
|
54
|
+
**Reflection Guidance:**
|
|
55
|
+
- This is retry attempt **${attempt}** of **${this.maxRetries}**
|
|
56
|
+
- Analyze the error and the arguments you provided. Do not repeat the exact same call.
|
|
57
|
+
|
|
58
|
+
Formulate a new plan based on your analysis and try a corrected or different approach.
|
|
59
|
+
`.trim()
|
|
60
|
+
};
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* Checks if the model response contains an error matching `onModelErrors`.
|
|
64
|
+
*/
|
|
65
|
+
checkForModelError(llmResponse) {
|
|
66
|
+
if (!llmResponse.errorCode && !llmResponse.finishReason) {
|
|
67
|
+
return false;
|
|
68
|
+
}
|
|
69
|
+
if (llmResponse.finishReason && this.onModelErrors.includes(llmResponse.finishReason)) {
|
|
70
|
+
return true;
|
|
71
|
+
}
|
|
72
|
+
return false;
|
|
73
|
+
}
|
|
74
|
+
/**
|
|
75
|
+
* Retrieves the model name from context.
|
|
76
|
+
*/
|
|
77
|
+
getModelNameFromContext(callbackContext) {
|
|
78
|
+
return callbackContext.agentName || "default_model";
|
|
79
|
+
}
|
|
80
|
+
/**
|
|
81
|
+
* Prepares the LLM request by attaching the reflection tool.
|
|
82
|
+
*/
|
|
83
|
+
async beforeModelCallback({
|
|
84
|
+
llmRequest
|
|
85
|
+
}) {
|
|
86
|
+
this.provideReflectionTool(llmRequest);
|
|
87
|
+
return void 0;
|
|
88
|
+
}
|
|
89
|
+
/**
|
|
90
|
+
* Inspects LLM responses, intercepts model errors or reserved tool calls, and handles retries.
|
|
91
|
+
*/
|
|
92
|
+
async afterModelCallback({
|
|
93
|
+
callbackContext,
|
|
94
|
+
llmResponse
|
|
95
|
+
}) {
|
|
96
|
+
if (this.hasReservedToolCall(llmResponse)) {
|
|
97
|
+
return await this.handleReservedToolCall(callbackContext, llmResponse);
|
|
98
|
+
}
|
|
99
|
+
if (this.checkForModelError(llmResponse)) {
|
|
100
|
+
return await this.handleModelError(callbackContext, llmResponse);
|
|
101
|
+
}
|
|
102
|
+
const scopeKey = this.getModelScopeKey(callbackContext);
|
|
103
|
+
const modelName = this.getModelNameFromContext(callbackContext);
|
|
104
|
+
await this.resetModelFailureCount(scopeKey, modelName);
|
|
105
|
+
return void 0;
|
|
106
|
+
}
|
|
107
|
+
/**
|
|
108
|
+
* Provides the adk_handle_model_error tool for reflection and retries.
|
|
109
|
+
*/
|
|
110
|
+
provideReflectionTool(llmRequest) {
|
|
111
|
+
if (!llmRequest.toolsDict) {
|
|
112
|
+
llmRequest.toolsDict = {};
|
|
113
|
+
}
|
|
114
|
+
llmRequest.toolsDict[ADK_HANDLE_MODEL_ERROR_TOOL_NAME] = new FunctionTool({
|
|
115
|
+
name: ADK_HANDLE_MODEL_ERROR_TOOL_NAME,
|
|
116
|
+
description: "A tool that triggers reflection. Reserved for internal framework use only. Do not call directly.",
|
|
117
|
+
execute: /* @__PURE__ */ __name((args) => this.adkHandleModelError(
|
|
118
|
+
args
|
|
119
|
+
), "execute")
|
|
120
|
+
});
|
|
121
|
+
}
|
|
122
|
+
/**
|
|
123
|
+
* Checks if the model response calls the reserved reflection tool.
|
|
124
|
+
*/
|
|
125
|
+
hasReservedToolCall(llmResponse) {
|
|
126
|
+
var _a;
|
|
127
|
+
if (!((_a = llmResponse.content) == null ? void 0 : _a.parts)) return false;
|
|
128
|
+
for (const part of llmResponse.content.parts) {
|
|
129
|
+
if (part.functionCall && part.functionCall.name === ADK_HANDLE_MODEL_ERROR_TOOL_NAME) {
|
|
130
|
+
return true;
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
return false;
|
|
134
|
+
}
|
|
135
|
+
/**
|
|
136
|
+
* Handles direct calls to the reserved reflection tool.
|
|
137
|
+
*/
|
|
138
|
+
async handleReservedToolCall(callbackContext, _llmResponse) {
|
|
139
|
+
const retryResponse = await this.handleModelRetry(
|
|
140
|
+
callbackContext,
|
|
141
|
+
RESERVED_TOOL_CALL_ERROR_TYPE,
|
|
142
|
+
`Model attempted to call reserved tool ${ADK_HANDLE_MODEL_ERROR_TOOL_NAME} directly. This tool is reserved for framework use only. Do not call it.`,
|
|
143
|
+
FinishReason.OTHER
|
|
144
|
+
);
|
|
145
|
+
if (retryResponse !== void 0) {
|
|
146
|
+
return retryResponse;
|
|
147
|
+
}
|
|
148
|
+
return {
|
|
149
|
+
errorCode: RESERVED_TOOL_CALL_ERROR_TYPE,
|
|
150
|
+
errorMessage: "Model attempted to call reserved tool and retry limit was exceeded."
|
|
151
|
+
};
|
|
152
|
+
}
|
|
153
|
+
/**
|
|
154
|
+
* Handles detected model errors by initiating retry logic.
|
|
155
|
+
*/
|
|
156
|
+
async handleModelError(callbackContext, llmResponse) {
|
|
157
|
+
var _a, _b, _c;
|
|
158
|
+
const retryResponse = await this.handleModelRetry(
|
|
159
|
+
callbackContext,
|
|
160
|
+
(_a = llmResponse.errorCode) != null ? _a : "MODEL_ERROR",
|
|
161
|
+
(_b = llmResponse.errorMessage) != null ? _b : "Model error encountered.",
|
|
162
|
+
(_c = llmResponse.finishReason) != null ? _c : FinishReason.OTHER
|
|
163
|
+
);
|
|
164
|
+
if (retryResponse !== void 0) {
|
|
165
|
+
return retryResponse;
|
|
166
|
+
}
|
|
167
|
+
return llmResponse;
|
|
168
|
+
}
|
|
169
|
+
/**
|
|
170
|
+
* Tracks retry count, generates retry response, and checks against limits.
|
|
171
|
+
*/
|
|
172
|
+
async handleModelRetry(callbackContext, errorType, errorDetails, finishReason) {
|
|
173
|
+
const scopeKey = this.getModelScopeKey(callbackContext);
|
|
174
|
+
const modelName = this.getModelNameFromContext(callbackContext);
|
|
175
|
+
const currentRetries = await this.incrementModelFailureCount(
|
|
176
|
+
scopeKey,
|
|
177
|
+
modelName
|
|
178
|
+
);
|
|
179
|
+
if (currentRetries <= this.maxRetries) {
|
|
180
|
+
return {
|
|
181
|
+
content: {
|
|
182
|
+
role: "model",
|
|
183
|
+
parts: [
|
|
184
|
+
this.generateModelRetryPart(
|
|
185
|
+
currentRetries,
|
|
186
|
+
errorType,
|
|
187
|
+
errorDetails,
|
|
188
|
+
finishReason
|
|
189
|
+
)
|
|
190
|
+
]
|
|
191
|
+
}
|
|
192
|
+
};
|
|
193
|
+
}
|
|
194
|
+
if (this.throwExceptionIfRetryExceeded) {
|
|
195
|
+
throw new Error(
|
|
196
|
+
`The model has failed consecutively ${this.maxRetries} times and the retry limit has been exceeded.`
|
|
197
|
+
);
|
|
198
|
+
}
|
|
199
|
+
return void 0;
|
|
200
|
+
}
|
|
201
|
+
/**
|
|
202
|
+
* Generates a function call part for the model retry tool.
|
|
203
|
+
*/
|
|
204
|
+
generateModelRetryPart(retryCount, errorType, errorDetails, finishReason) {
|
|
205
|
+
return {
|
|
206
|
+
functionCall: {
|
|
207
|
+
id: this.getModelRetryUuid(),
|
|
208
|
+
name: ADK_HANDLE_MODEL_ERROR_TOOL_NAME,
|
|
209
|
+
args: {
|
|
210
|
+
response_type: REFLECT_AND_RETRY_RESPONSE_TYPE,
|
|
211
|
+
error_type: errorType,
|
|
212
|
+
error_details: errorDetails,
|
|
213
|
+
finish_reason: finishReason,
|
|
214
|
+
retry_count: retryCount
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
};
|
|
218
|
+
}
|
|
219
|
+
/**
|
|
220
|
+
* Generates a unique ID for the model retry tool call.
|
|
221
|
+
*/
|
|
222
|
+
getModelRetryUuid() {
|
|
223
|
+
return `${ADK_HANDLE_MODEL_ERROR_TOOL_NAME}_${uuidv4()}`;
|
|
224
|
+
}
|
|
225
|
+
/**
|
|
226
|
+
* Returns the scope key for model failure tracking.
|
|
227
|
+
*/
|
|
228
|
+
getModelScopeKey(callbackContext) {
|
|
229
|
+
return resolveScopeKey(this.scope, callbackContext.invocationId);
|
|
230
|
+
}
|
|
231
|
+
/**
|
|
232
|
+
* Increment the failure count for a model within a scope.
|
|
233
|
+
*/
|
|
234
|
+
async incrementModelFailureCount(scopeKey, itemName) {
|
|
235
|
+
return await this.tracker.increment(scopeKey, itemName);
|
|
236
|
+
}
|
|
237
|
+
/**
|
|
238
|
+
* Reset the failure count for a model within a scope.
|
|
239
|
+
*/
|
|
240
|
+
async resetModelFailureCount(scopeKey, itemName) {
|
|
241
|
+
await this.tracker.reset(scopeKey, itemName);
|
|
242
|
+
}
|
|
243
|
+
};
|
|
244
|
+
__name(_ReflectAndRetryModelPlugin, "ReflectAndRetryModelPlugin");
|
|
245
|
+
let ReflectAndRetryModelPlugin = _ReflectAndRetryModelPlugin;
|
|
246
|
+
export {
|
|
247
|
+
ADK_HANDLE_MODEL_ERROR_TOOL_NAME,
|
|
248
|
+
RESERVED_TOOL_CALL_ERROR_TYPE,
|
|
249
|
+
ReflectAndRetryModelPlugin
|
|
250
|
+
};
|
|
@@ -0,0 +1,254 @@
|
|
|
1
|
+
import {createRequire as topLevelCreateRequire} from 'module';
|
|
2
|
+
const require = topLevelCreateRequire(import.meta.url);
|
|
3
|
+
var __defProp = Object.defineProperty;
|
|
4
|
+
var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
|
|
5
|
+
/**
|
|
6
|
+
* @license
|
|
7
|
+
* Copyright 2026 Google LLC
|
|
8
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
9
|
+
*/
|
|
10
|
+
import {
|
|
11
|
+
REFLECT_AND_RETRY_RESPONSE_TYPE,
|
|
12
|
+
resolveScopeKey,
|
|
13
|
+
ScopedFailureTracker,
|
|
14
|
+
TrackingScope
|
|
15
|
+
} from "./_reflect_retry_utils.js";
|
|
16
|
+
import { BasePlugin } from "./base_plugin.js";
|
|
17
|
+
const _ReflectAndRetryToolPlugin = class _ReflectAndRetryToolPlugin extends BasePlugin {
|
|
18
|
+
/**
|
|
19
|
+
* Initializes the {@link ReflectAndRetryToolPlugin}.
|
|
20
|
+
*
|
|
21
|
+
* @param options - Configuration options for the plugin.
|
|
22
|
+
*/
|
|
23
|
+
constructor(options = {}) {
|
|
24
|
+
var _a, _b, _c, _d;
|
|
25
|
+
super((_a = options.name) != null ? _a : "reflect_retry_tool_plugin");
|
|
26
|
+
const retries = (_b = options.maxRetries) != null ? _b : 3;
|
|
27
|
+
if (retries < 0) {
|
|
28
|
+
throw new Error("maxRetries must be a non-negative integer.");
|
|
29
|
+
}
|
|
30
|
+
this.maxRetries = retries;
|
|
31
|
+
this.throwExceptionIfRetryExceeded = (_c = options.throwExceptionIfRetryExceeded) != null ? _c : true;
|
|
32
|
+
this.scope = (_d = options.trackingScope) != null ? _d : TrackingScope.INVOCATION;
|
|
33
|
+
this.tracker = new ScopedFailureTracker();
|
|
34
|
+
}
|
|
35
|
+
/**
|
|
36
|
+
* Handles successful tool calls or extracts and processes errors.
|
|
37
|
+
*
|
|
38
|
+
* @param params - The tool callback parameters.
|
|
39
|
+
* @returns An optional record containing reflection guidance if an error is
|
|
40
|
+
* detected, or undefined if the tool call was successful or the response is
|
|
41
|
+
* already a reflection message.
|
|
42
|
+
*/
|
|
43
|
+
async afterToolCallback({
|
|
44
|
+
tool,
|
|
45
|
+
toolArgs,
|
|
46
|
+
toolContext,
|
|
47
|
+
result
|
|
48
|
+
}) {
|
|
49
|
+
if (result && typeof result === "object" && result["response_type"] === REFLECT_AND_RETRY_RESPONSE_TYPE) {
|
|
50
|
+
return void 0;
|
|
51
|
+
}
|
|
52
|
+
const error = await this.extractErrorFromResult({
|
|
53
|
+
tool,
|
|
54
|
+
toolArgs,
|
|
55
|
+
toolContext,
|
|
56
|
+
result
|
|
57
|
+
});
|
|
58
|
+
if (error != null) {
|
|
59
|
+
return await this.handleToolError(tool, toolArgs, toolContext, error);
|
|
60
|
+
}
|
|
61
|
+
await this.resetFailuresForTool(toolContext, tool.name);
|
|
62
|
+
return void 0;
|
|
63
|
+
}
|
|
64
|
+
/**
|
|
65
|
+
* Extracts an error from a successful tool result and triggers retry logic.
|
|
66
|
+
*
|
|
67
|
+
* This is useful when a tool call finishes without throwing an exception but
|
|
68
|
+
* the result contains an error payload like `{status: 'error', message: '...'}`
|
|
69
|
+
* that should be handled by the plugin.
|
|
70
|
+
*
|
|
71
|
+
* @param params - Tool execution context and result.
|
|
72
|
+
* @returns The extracted error if any, or undefined if no error was detected.
|
|
73
|
+
*/
|
|
74
|
+
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
75
|
+
async extractErrorFromResult(params) {
|
|
76
|
+
return void 0;
|
|
77
|
+
}
|
|
78
|
+
/**
|
|
79
|
+
* Handles tool exceptions by providing reflection guidance.
|
|
80
|
+
*
|
|
81
|
+
* @param params - The error callback parameters.
|
|
82
|
+
* @returns An optional record containing reflection guidance for the error.
|
|
83
|
+
*/
|
|
84
|
+
async onToolErrorCallback({
|
|
85
|
+
tool,
|
|
86
|
+
toolArgs,
|
|
87
|
+
toolContext,
|
|
88
|
+
error
|
|
89
|
+
}) {
|
|
90
|
+
return await this.handleToolError(tool, toolArgs, toolContext, error);
|
|
91
|
+
}
|
|
92
|
+
/**
|
|
93
|
+
* Central, thread-safe logic for processing tool errors.
|
|
94
|
+
*
|
|
95
|
+
* @param tool - The tool that was called.
|
|
96
|
+
* @param toolArgs - The arguments passed to the tool.
|
|
97
|
+
* @param toolContext - The context of the tool call.
|
|
98
|
+
* @param error - The error to be handled.
|
|
99
|
+
* @returns An optional record containing reflection guidance for the error.
|
|
100
|
+
*/
|
|
101
|
+
async handleToolError(tool, toolArgs, toolContext, error) {
|
|
102
|
+
if (this.maxRetries === 0) {
|
|
103
|
+
if (this.throwExceptionIfRetryExceeded) {
|
|
104
|
+
throw this.ensureException(error);
|
|
105
|
+
}
|
|
106
|
+
return this.getToolRetryExceedMsg(tool, toolArgs, error);
|
|
107
|
+
}
|
|
108
|
+
const scopeKey = this.getScopeKey(toolContext);
|
|
109
|
+
const currentRetries = await this.tracker.increment(scopeKey, tool.name);
|
|
110
|
+
if (currentRetries <= this.maxRetries) {
|
|
111
|
+
return this.createToolReflectionResponse(
|
|
112
|
+
tool,
|
|
113
|
+
toolArgs,
|
|
114
|
+
error,
|
|
115
|
+
currentRetries
|
|
116
|
+
);
|
|
117
|
+
}
|
|
118
|
+
if (this.throwExceptionIfRetryExceeded) {
|
|
119
|
+
throw this.ensureException(error);
|
|
120
|
+
} else {
|
|
121
|
+
return this.getToolRetryExceedMsg(tool, toolArgs, error);
|
|
122
|
+
}
|
|
123
|
+
}
|
|
124
|
+
/**
|
|
125
|
+
* Returns a unique key for the state tracker based on the configured scope.
|
|
126
|
+
*
|
|
127
|
+
* This method can be overridden in a subclass to implement custom scoping
|
|
128
|
+
* logic, for example tracking failures on a per-user or per-session basis.
|
|
129
|
+
*
|
|
130
|
+
* @param toolContext - The tool context.
|
|
131
|
+
* @returns The resolved scope key.
|
|
132
|
+
*/
|
|
133
|
+
getScopeKey(toolContext) {
|
|
134
|
+
return resolveScopeKey(this.scope, toolContext.invocationId);
|
|
135
|
+
}
|
|
136
|
+
/**
|
|
137
|
+
* Atomically resets the failure count for a tool and cleans up state.
|
|
138
|
+
*
|
|
139
|
+
* @param toolContext - The tool context.
|
|
140
|
+
* @param toolName - The name of the tool.
|
|
141
|
+
*/
|
|
142
|
+
async resetFailuresForTool(toolContext, toolName) {
|
|
143
|
+
const scopeKey = this.getScopeKey(toolContext);
|
|
144
|
+
await this.tracker.reset(scopeKey, toolName);
|
|
145
|
+
}
|
|
146
|
+
/**
|
|
147
|
+
* Ensures the given error is an Error instance, wrapping if not.
|
|
148
|
+
*
|
|
149
|
+
* @param error - The error object.
|
|
150
|
+
* @returns An Error instance.
|
|
151
|
+
*/
|
|
152
|
+
ensureException(error) {
|
|
153
|
+
return error instanceof Error ? error : new Error(String(error));
|
|
154
|
+
}
|
|
155
|
+
/**
|
|
156
|
+
* Formats error details for inclusion in the reflection message.
|
|
157
|
+
*
|
|
158
|
+
* @param error - The error object.
|
|
159
|
+
* @returns Formatted error string.
|
|
160
|
+
*/
|
|
161
|
+
formatErrorDetails(error) {
|
|
162
|
+
if (error instanceof Error) {
|
|
163
|
+
const name = error.name || "Error";
|
|
164
|
+
return `${name}: ${error.message}`;
|
|
165
|
+
}
|
|
166
|
+
return String(error);
|
|
167
|
+
}
|
|
168
|
+
/**
|
|
169
|
+
* Generates structured reflection guidance for tool failures.
|
|
170
|
+
*
|
|
171
|
+
* @param tool - The tool that failed.
|
|
172
|
+
* @param toolArgs - The arguments supplied to the tool.
|
|
173
|
+
* @param error - The error encountered.
|
|
174
|
+
* @param retryCount - The current retry count.
|
|
175
|
+
* @returns The structured tool failure response record.
|
|
176
|
+
*/
|
|
177
|
+
createToolReflectionResponse(tool, toolArgs, error, retryCount) {
|
|
178
|
+
const argsSummary = JSON.stringify(toolArgs != null ? toolArgs : {}, null, 2);
|
|
179
|
+
const errorDetails = this.formatErrorDetails(error);
|
|
180
|
+
const reflectionMessage = `
|
|
181
|
+
The call to tool \`${tool.name}\` failed.
|
|
182
|
+
|
|
183
|
+
**Error Details:**
|
|
184
|
+
\`\`\`
|
|
185
|
+
${errorDetails}
|
|
186
|
+
\`\`\`
|
|
187
|
+
|
|
188
|
+
**Tool Arguments Used:**
|
|
189
|
+
\`\`\`json
|
|
190
|
+
${argsSummary}
|
|
191
|
+
\`\`\`
|
|
192
|
+
|
|
193
|
+
**Reflection Guidance:**
|
|
194
|
+
This is retry attempt **${retryCount} of ${this.maxRetries}**. Analyze the error and the arguments you provided. Do not repeat the exact same call. Consider the following before your next attempt:
|
|
195
|
+
|
|
196
|
+
1. **Invalid Parameters**: Does the error suggest that one or more arguments are incorrect, badly formatted, or missing? Review the tool's schema and your arguments.
|
|
197
|
+
2. **State or Preconditions**: Did a previous step fail or not produce the necessary state/resource for this tool to succeed?
|
|
198
|
+
3. **Alternative Approach**: Is this the right tool for the job? Could another tool or a different sequence of steps achieve the goal?
|
|
199
|
+
4. **Simplify the Task**: Can you break the problem down into smaller, simpler steps?
|
|
200
|
+
5. **Wrong Function Name**: Does the error indicates the tool is not found? Please check again and only use available tools.
|
|
201
|
+
|
|
202
|
+
Formulate a new plan based on your analysis and try a corrected or different approach.
|
|
203
|
+
`.trim();
|
|
204
|
+
const response = {
|
|
205
|
+
response_type: REFLECT_AND_RETRY_RESPONSE_TYPE,
|
|
206
|
+
error_type: error instanceof Error ? error.name || "Error" : "ToolError",
|
|
207
|
+
error_details: error instanceof Error ? error.message : String(error),
|
|
208
|
+
retry_count: retryCount,
|
|
209
|
+
reflection_guidance: reflectionMessage
|
|
210
|
+
};
|
|
211
|
+
return response;
|
|
212
|
+
}
|
|
213
|
+
/**
|
|
214
|
+
* Generates guidance when the maximum retry limit is exceeded.
|
|
215
|
+
*
|
|
216
|
+
* @param tool - The tool that failed.
|
|
217
|
+
* @param toolArgs - The arguments supplied to the tool.
|
|
218
|
+
* @param error - The error encountered.
|
|
219
|
+
* @returns The structured retry-exceeded failure response record.
|
|
220
|
+
*/
|
|
221
|
+
getToolRetryExceedMsg(tool, toolArgs, error) {
|
|
222
|
+
const errorDetails = this.formatErrorDetails(error);
|
|
223
|
+
const argsSummary = JSON.stringify(toolArgs != null ? toolArgs : {}, null, 2);
|
|
224
|
+
const reflectionMessage = `
|
|
225
|
+
The tool \`${tool.name}\` has failed consecutively ${this.maxRetries} times and the retry limit has been exceeded.
|
|
226
|
+
|
|
227
|
+
**Last Error:**
|
|
228
|
+
\`\`\`
|
|
229
|
+
${errorDetails}
|
|
230
|
+
\`\`\`
|
|
231
|
+
|
|
232
|
+
**Last Arguments Used:**
|
|
233
|
+
\`\`\`json
|
|
234
|
+
${argsSummary}
|
|
235
|
+
\`\`\`
|
|
236
|
+
|
|
237
|
+
**Final Instruction:**
|
|
238
|
+
**Do not attempt to use the \`${tool.name}\` tool again for this task.** You must now try a different approach. Acknowledge the failure and devise a new strategy, potentially using other available tools or informing the user that the task cannot be completed.
|
|
239
|
+
`.trim();
|
|
240
|
+
const response = {
|
|
241
|
+
response_type: REFLECT_AND_RETRY_RESPONSE_TYPE,
|
|
242
|
+
error_type: error instanceof Error ? error.name || "Error" : "ToolError",
|
|
243
|
+
error_details: error instanceof Error ? error.message : String(error),
|
|
244
|
+
retry_count: this.maxRetries,
|
|
245
|
+
reflection_guidance: reflectionMessage
|
|
246
|
+
};
|
|
247
|
+
return response;
|
|
248
|
+
}
|
|
249
|
+
};
|
|
250
|
+
__name(_ReflectAndRetryToolPlugin, "ReflectAndRetryToolPlugin");
|
|
251
|
+
let ReflectAndRetryToolPlugin = _ReflectAndRetryToolPlugin;
|
|
252
|
+
export {
|
|
253
|
+
ReflectAndRetryToolPlugin
|
|
254
|
+
};
|