@google/adk 0.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/LICENSE +202 -0
- package/README.md +9 -0
- package/dist/cjs/agents/active_streaming_tool.js +44 -0
- package/dist/cjs/agents/base_agent.js +245 -0
- package/dist/cjs/agents/base_llm_processor.js +44 -0
- package/dist/cjs/agents/callback_context.js +98 -0
- package/dist/cjs/agents/content_processor_utils.js +299 -0
- package/dist/cjs/agents/functions.js +394 -0
- package/dist/cjs/agents/instructions.js +110 -0
- package/dist/cjs/agents/invocation_context.js +109 -0
- package/dist/cjs/agents/live_request_queue.js +136 -0
- package/dist/cjs/agents/llm_agent.js +859 -0
- package/dist/cjs/agents/loop_agent.js +68 -0
- package/dist/cjs/agents/parallel_agent.js +78 -0
- package/dist/cjs/agents/readonly_context.js +68 -0
- package/dist/cjs/agents/run_config.js +74 -0
- package/dist/cjs/agents/sequential_agent.js +84 -0
- package/dist/cjs/agents/transcription_entry.js +27 -0
- package/dist/cjs/artifacts/base_artifact_service.js +27 -0
- package/dist/cjs/artifacts/in_memory_artifact_service.js +119 -0
- package/dist/cjs/auth/auth_credential.js +46 -0
- package/dist/cjs/auth/auth_handler.js +92 -0
- package/dist/cjs/auth/auth_schemes.js +62 -0
- package/dist/cjs/auth/auth_tool.js +27 -0
- package/dist/cjs/auth/credential_service/base_credential_service.js +27 -0
- package/dist/cjs/auth/credential_service/in_memory_credential_service.js +63 -0
- package/dist/cjs/code_executors/base_code_executor.js +76 -0
- package/dist/cjs/code_executors/built_in_code_executor.js +58 -0
- package/dist/cjs/code_executors/code_execution_utils.js +142 -0
- package/dist/cjs/code_executors/code_executor_context.js +198 -0
- package/dist/cjs/common.js +161 -0
- package/dist/cjs/events/event.js +107 -0
- package/dist/cjs/events/event_actions.js +83 -0
- package/dist/cjs/examples/base_example_provider.js +40 -0
- package/dist/cjs/examples/example.js +27 -0
- package/dist/cjs/examples/example_util.js +107 -0
- package/dist/cjs/index.js +40 -0
- package/dist/cjs/index.js.map +7 -0
- package/dist/cjs/index_web.js +33 -0
- package/dist/cjs/memory/base_memory_service.js +27 -0
- package/dist/cjs/memory/in_memory_memory_service.js +97 -0
- package/dist/cjs/memory/memory_entry.js +27 -0
- package/dist/cjs/models/base_llm.js +77 -0
- package/dist/cjs/models/base_llm_connection.js +27 -0
- package/dist/cjs/models/gemini_llm_connection.js +132 -0
- package/dist/cjs/models/google_llm.js +321 -0
- package/dist/cjs/models/llm_request.js +82 -0
- package/dist/cjs/models/llm_response.js +71 -0
- package/dist/cjs/models/registry.js +121 -0
- package/dist/cjs/package.json +1 -0
- package/dist/cjs/plugins/base_plugin.js +236 -0
- package/dist/cjs/plugins/logging_plugin.js +222 -0
- package/dist/cjs/plugins/plugin_manager.js +239 -0
- package/dist/cjs/plugins/security_plugin.js +153 -0
- package/dist/cjs/runner/in_memory_runner.js +58 -0
- package/dist/cjs/runner/runner.js +276 -0
- package/dist/cjs/sessions/base_session_service.js +71 -0
- package/dist/cjs/sessions/in_memory_session_service.js +184 -0
- package/dist/cjs/sessions/session.js +48 -0
- package/dist/cjs/sessions/state.js +101 -0
- package/dist/cjs/tools/agent_tool.js +134 -0
- package/dist/cjs/tools/base_tool.js +107 -0
- package/dist/cjs/tools/base_toolset.js +76 -0
- package/dist/cjs/tools/forwarding_artifact_service.js +71 -0
- package/dist/cjs/tools/function_tool.js +101 -0
- package/dist/cjs/tools/google_search_tool.js +76 -0
- package/dist/cjs/tools/long_running_tool.js +63 -0
- package/dist/cjs/tools/mcp/mcp_session_manager.js +65 -0
- package/dist/cjs/tools/mcp/mcp_tool.js +65 -0
- package/dist/cjs/tools/mcp/mcp_toolset.js +61 -0
- package/dist/cjs/tools/tool_confirmation.js +49 -0
- package/dist/cjs/tools/tool_context.js +129 -0
- package/dist/cjs/utils/deep_clone.js +44 -0
- package/dist/cjs/utils/env_aware_utils.js +83 -0
- package/dist/cjs/utils/gemini_schema_util.js +88 -0
- package/dist/cjs/utils/logger.js +121 -0
- package/dist/cjs/utils/model_name.js +64 -0
- package/dist/cjs/utils/simple_zod_to_json.js +191 -0
- package/dist/cjs/utils/variant_utils.js +55 -0
- package/dist/cjs/version.js +39 -0
- package/dist/esm/agents/active_streaming_tool.js +14 -0
- package/dist/esm/agents/base_agent.js +214 -0
- package/dist/esm/agents/base_llm_processor.js +13 -0
- package/dist/esm/agents/callback_context.js +68 -0
- package/dist/esm/agents/content_processor_utils.js +268 -0
- package/dist/esm/agents/functions.js +353 -0
- package/dist/esm/agents/instructions.js +80 -0
- package/dist/esm/agents/invocation_context.js +78 -0
- package/dist/esm/agents/live_request_queue.js +106 -0
- package/dist/esm/agents/llm_agent.js +828 -0
- package/dist/esm/agents/loop_agent.js +38 -0
- package/dist/esm/agents/parallel_agent.js +48 -0
- package/dist/esm/agents/readonly_context.js +38 -0
- package/dist/esm/agents/run_config.js +43 -0
- package/dist/esm/agents/sequential_agent.js +54 -0
- package/dist/esm/agents/transcription_entry.js +5 -0
- package/dist/esm/artifacts/base_artifact_service.js +5 -0
- package/dist/esm/artifacts/in_memory_artifact_service.js +89 -0
- package/dist/esm/auth/auth_credential.js +16 -0
- package/dist/esm/auth/auth_handler.js +62 -0
- package/dist/esm/auth/auth_schemes.js +31 -0
- package/dist/esm/auth/auth_tool.js +5 -0
- package/dist/esm/auth/credential_service/base_credential_service.js +5 -0
- package/dist/esm/auth/credential_service/in_memory_credential_service.js +33 -0
- package/dist/esm/code_executors/base_code_executor.js +46 -0
- package/dist/esm/code_executors/built_in_code_executor.js +28 -0
- package/dist/esm/code_executors/code_execution_utils.js +108 -0
- package/dist/esm/code_executors/code_executor_context.js +168 -0
- package/dist/esm/common.js +85 -0
- package/dist/esm/events/event.js +72 -0
- package/dist/esm/events/event_actions.js +52 -0
- package/dist/esm/examples/base_example_provider.js +10 -0
- package/dist/esm/examples/example.js +5 -0
- package/dist/esm/examples/example_util.js +76 -0
- package/dist/esm/index.js +40 -0
- package/dist/esm/index.js.map +7 -0
- package/dist/esm/index_web.js +6 -0
- package/dist/esm/memory/base_memory_service.js +5 -0
- package/dist/esm/memory/in_memory_memory_service.js +67 -0
- package/dist/esm/memory/memory_entry.js +5 -0
- package/dist/esm/models/base_llm.js +47 -0
- package/dist/esm/models/base_llm_connection.js +5 -0
- package/dist/esm/models/gemini_llm_connection.js +102 -0
- package/dist/esm/models/google_llm.js +291 -0
- package/dist/esm/models/llm_request.js +50 -0
- package/dist/esm/models/llm_response.js +41 -0
- package/dist/esm/models/registry.js +91 -0
- package/dist/esm/plugins/base_plugin.js +206 -0
- package/dist/esm/plugins/logging_plugin.js +192 -0
- package/dist/esm/plugins/plugin_manager.js +209 -0
- package/dist/esm/plugins/security_plugin.js +119 -0
- package/dist/esm/runner/in_memory_runner.js +28 -0
- package/dist/esm/runner/runner.js +246 -0
- package/dist/esm/sessions/base_session_service.js +41 -0
- package/dist/esm/sessions/in_memory_session_service.js +154 -0
- package/dist/esm/sessions/session.js +18 -0
- package/dist/esm/sessions/state.js +71 -0
- package/dist/esm/tools/agent_tool.js +104 -0
- package/dist/esm/tools/base_tool.js +77 -0
- package/dist/esm/tools/base_toolset.js +46 -0
- package/dist/esm/tools/forwarding_artifact_service.js +41 -0
- package/dist/esm/tools/function_tool.js +71 -0
- package/dist/esm/tools/google_search_tool.js +46 -0
- package/dist/esm/tools/long_running_tool.js +33 -0
- package/dist/esm/tools/mcp/mcp_session_manager.js +35 -0
- package/dist/esm/tools/mcp/mcp_tool.js +35 -0
- package/dist/esm/tools/mcp/mcp_toolset.js +31 -0
- package/dist/esm/tools/tool_confirmation.js +19 -0
- package/dist/esm/tools/tool_context.js +99 -0
- package/dist/esm/utils/deep_clone.js +14 -0
- package/dist/esm/utils/env_aware_utils.js +49 -0
- package/dist/esm/utils/gemini_schema_util.js +58 -0
- package/dist/esm/utils/logger.js +89 -0
- package/dist/esm/utils/model_name.js +31 -0
- package/dist/esm/utils/simple_zod_to_json.js +160 -0
- package/dist/esm/utils/variant_utils.js +24 -0
- package/dist/esm/version.js +9 -0
- package/dist/types/agents/active_streaming_tool.d.ts +29 -0
- package/dist/types/agents/base_agent.d.ts +167 -0
- package/dist/types/agents/base_llm_processor.d.ts +27 -0
- package/dist/types/agents/callback_context.d.ts +42 -0
- package/dist/types/agents/content_processor_utils.d.ts +36 -0
- package/dist/types/agents/functions.d.ts +90 -0
- package/dist/types/agents/instructions.d.ts +32 -0
- package/dist/types/agents/invocation_context.d.ts +155 -0
- package/dist/types/agents/live_request_queue.d.ts +67 -0
- package/dist/types/agents/llm_agent.d.ts +333 -0
- package/dist/types/agents/loop_agent.d.ts +31 -0
- package/dist/types/agents/parallel_agent.d.ts +21 -0
- package/dist/types/agents/readonly_context.d.ts +31 -0
- package/dist/types/agents/run_config.d.ts +76 -0
- package/dist/types/agents/sequential_agent.d.ts +26 -0
- package/dist/types/agents/transcription_entry.d.ts +17 -0
- package/dist/types/artifacts/base_artifact_service.d.ts +127 -0
- package/dist/types/artifacts/in_memory_artifact_service.d.ts +18 -0
- package/dist/types/auth/auth_credential.d.ts +227 -0
- package/dist/types/auth/auth_handler.d.ts +27 -0
- package/dist/types/auth/auth_schemes.d.ts +36 -0
- package/dist/types/auth/auth_tool.d.ts +51 -0
- package/dist/types/auth/credential_service/base_credential_service.d.ts +27 -0
- package/dist/types/auth/credential_service/in_memory_credential_service.d.ts +19 -0
- package/dist/types/code_executors/base_code_executor.d.ts +60 -0
- package/dist/types/code_executors/built_in_code_executor.d.ts +13 -0
- package/dist/types/code_executors/code_execution_utils.d.ts +99 -0
- package/dist/types/code_executors/code_executor_context.d.ts +92 -0
- package/dist/types/common.d.ts +51 -0
- package/dist/types/events/event.d.ts +81 -0
- package/dist/types/events/event_actions.d.ts +74 -0
- package/dist/types/examples/base_example_provider.d.ts +20 -0
- package/dist/types/examples/example.d.ts +19 -0
- package/dist/types/examples/example_util.d.ts +13 -0
- package/dist/types/index.d.ts +9 -0
- package/dist/types/index_web.d.ts +6 -0
- package/dist/types/memory/base_memory_service.d.ts +47 -0
- package/dist/types/memory/in_memory_memory_service.d.ts +18 -0
- package/dist/types/memory/memory_entry.d.ts +24 -0
- package/dist/types/models/base_llm.d.ts +46 -0
- package/dist/types/models/base_llm_connection.d.ts +51 -0
- package/dist/types/models/gemini_llm_connection.d.ts +54 -0
- package/dist/types/models/google_llm.d.ts +88 -0
- package/dist/types/models/llm_request.d.ts +49 -0
- package/dist/types/models/llm_response.d.ts +79 -0
- package/dist/types/models/registry.d.ts +45 -0
- package/dist/types/plugins/base_plugin.d.ts +310 -0
- package/dist/types/plugins/logging_plugin.d.ts +104 -0
- package/dist/types/plugins/plugin_manager.d.ts +155 -0
- package/dist/types/plugins/security_plugin.d.ts +60 -0
- package/dist/types/runner/in_memory_runner.d.ts +15 -0
- package/dist/types/runner/runner.d.ts +80 -0
- package/dist/types/sessions/base_session_service.d.ts +129 -0
- package/dist/types/sessions/in_memory_session_service.d.ts +32 -0
- package/dist/types/sessions/session.d.ts +46 -0
- package/dist/types/sessions/state.d.ts +57 -0
- package/dist/types/tools/agent_tool.d.ts +37 -0
- package/dist/types/tools/base_tool.d.ts +84 -0
- package/dist/types/tools/base_toolset.d.ts +64 -0
- package/dist/types/tools/forwarding_artifact_service.d.ts +21 -0
- package/dist/types/tools/function_tool.d.ts +48 -0
- package/dist/types/tools/google_search_tool.d.ts +18 -0
- package/dist/types/tools/long_running_tool.d.ts +18 -0
- package/dist/types/tools/mcp/mcp_session_manager.d.ts +57 -0
- package/dist/types/tools/mcp/mcp_tool.d.ts +30 -0
- package/dist/types/tools/mcp/mcp_toolset.d.ts +39 -0
- package/dist/types/tools/tool_confirmation.d.ts +25 -0
- package/dist/types/tools/tool_context.d.ts +63 -0
- package/dist/types/utils/deep_clone.d.ts +1 -0
- package/dist/types/utils/env_aware_utils.d.ts +31 -0
- package/dist/types/utils/gemini_schema_util.d.ts +23 -0
- package/dist/types/utils/logger.d.ts +41 -0
- package/dist/types/utils/model_name.d.ts +34 -0
- package/dist/types/utils/simple_zod_to_json.d.ts +12 -0
- package/dist/types/utils/variant_utils.d.ts +24 -0
- package/dist/types/version.d.ts +6 -0
- package/dist/web/agents/active_streaming_tool.js +14 -0
- package/dist/web/agents/base_agent.js +265 -0
- package/dist/web/agents/base_llm_processor.js +13 -0
- package/dist/web/agents/callback_context.js +68 -0
- package/dist/web/agents/content_processor_utils.js +268 -0
- package/dist/web/agents/functions.js +353 -0
- package/dist/web/agents/instructions.js +80 -0
- package/dist/web/agents/invocation_context.js +78 -0
- package/dist/web/agents/live_request_queue.js +124 -0
- package/dist/web/agents/llm_agent.js +973 -0
- package/dist/web/agents/loop_agent.js +71 -0
- package/dist/web/agents/parallel_agent.js +83 -0
- package/dist/web/agents/readonly_context.js +38 -0
- package/dist/web/agents/run_config.js +43 -0
- package/dist/web/agents/sequential_agent.js +99 -0
- package/dist/web/agents/transcription_entry.js +5 -0
- package/dist/web/artifacts/base_artifact_service.js +5 -0
- package/dist/web/artifacts/in_memory_artifact_service.js +89 -0
- package/dist/web/auth/auth_credential.js +16 -0
- package/dist/web/auth/auth_handler.js +62 -0
- package/dist/web/auth/auth_schemes.js +31 -0
- package/dist/web/auth/auth_tool.js +5 -0
- package/dist/web/auth/credential_service/base_credential_service.js +5 -0
- package/dist/web/auth/credential_service/in_memory_credential_service.js +33 -0
- package/dist/web/code_executors/base_code_executor.js +46 -0
- package/dist/web/code_executors/built_in_code_executor.js +28 -0
- package/dist/web/code_executors/code_execution_utils.js +105 -0
- package/dist/web/code_executors/code_executor_context.js +168 -0
- package/dist/web/common.js +85 -0
- package/dist/web/events/event.js +90 -0
- package/dist/web/events/event_actions.js +67 -0
- package/dist/web/examples/base_example_provider.js +10 -0
- package/dist/web/examples/example.js +5 -0
- package/dist/web/examples/example_util.js +75 -0
- package/dist/web/index.js +13 -0
- package/dist/web/index.js.map +7 -0
- package/dist/web/index_web.js +6 -0
- package/dist/web/memory/base_memory_service.js +5 -0
- package/dist/web/memory/in_memory_memory_service.js +67 -0
- package/dist/web/memory/memory_entry.js +5 -0
- package/dist/web/models/base_llm.js +47 -0
- package/dist/web/models/base_llm_connection.js +5 -0
- package/dist/web/models/gemini_llm_connection.js +120 -0
- package/dist/web/models/google_llm.js +332 -0
- package/dist/web/models/llm_request.js +50 -0
- package/dist/web/models/llm_response.js +41 -0
- package/dist/web/models/registry.js +91 -0
- package/dist/web/plugins/base_plugin.js +206 -0
- package/dist/web/plugins/logging_plugin.js +192 -0
- package/dist/web/plugins/plugin_manager.js +209 -0
- package/dist/web/plugins/security_plugin.js +119 -0
- package/dist/web/runner/in_memory_runner.js +28 -0
- package/dist/web/runner/runner.js +277 -0
- package/dist/web/sessions/base_session_service.js +41 -0
- package/dist/web/sessions/in_memory_session_service.js +154 -0
- package/dist/web/sessions/session.js +18 -0
- package/dist/web/sessions/state.js +87 -0
- package/dist/web/tools/agent_tool.js +118 -0
- package/dist/web/tools/base_tool.js +77 -0
- package/dist/web/tools/base_toolset.js +46 -0
- package/dist/web/tools/forwarding_artifact_service.js +41 -0
- package/dist/web/tools/function_tool.js +71 -0
- package/dist/web/tools/google_search_tool.js +46 -0
- package/dist/web/tools/long_running_tool.js +50 -0
- package/dist/web/tools/mcp/mcp_session_manager.js +35 -0
- package/dist/web/tools/mcp/mcp_tool.js +35 -0
- package/dist/web/tools/mcp/mcp_toolset.js +31 -0
- package/dist/web/tools/tool_confirmation.js +19 -0
- package/dist/web/tools/tool_context.js +99 -0
- package/dist/web/utils/deep_clone.js +14 -0
- package/dist/web/utils/env_aware_utils.js +49 -0
- package/dist/web/utils/gemini_schema_util.js +58 -0
- package/dist/web/utils/logger.js +89 -0
- package/dist/web/utils/model_name.js +31 -0
- package/dist/web/utils/simple_zod_to_json.js +174 -0
- package/dist/web/utils/variant_utils.js +24 -0
- package/dist/web/version.js +9 -0
- package/package.json +61 -0
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var forwarding_artifact_service_exports = {};
|
|
26
|
+
__export(forwarding_artifact_service_exports, {
|
|
27
|
+
ForwardingArtifactService: () => ForwardingArtifactService
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(forwarding_artifact_service_exports);
|
|
30
|
+
/**
|
|
31
|
+
* @license
|
|
32
|
+
* Copyright 2025 Google LLC
|
|
33
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
34
|
+
*/
|
|
35
|
+
class ForwardingArtifactService {
|
|
36
|
+
constructor(toolContext) {
|
|
37
|
+
this.toolContext = toolContext;
|
|
38
|
+
this.invocationContext = toolContext.invocationContext;
|
|
39
|
+
}
|
|
40
|
+
// TODO - b/425992518: Remove unnecessary parameters. We should rethink the
|
|
41
|
+
// abstraction layer to make it more clear.
|
|
42
|
+
async saveArtifact(request) {
|
|
43
|
+
return this.toolContext.saveArtifact(request.filename, request.artifact);
|
|
44
|
+
}
|
|
45
|
+
async loadArtifact(request) {
|
|
46
|
+
return this.toolContext.loadArtifact(request.filename, request.version);
|
|
47
|
+
}
|
|
48
|
+
async listArtifactKeys(request) {
|
|
49
|
+
return this.toolContext.listArtifacts();
|
|
50
|
+
}
|
|
51
|
+
async deleteArtifact(request) {
|
|
52
|
+
if (!this.toolContext.invocationContext.artifactService) {
|
|
53
|
+
throw new Error("Artifact service is not initialized.");
|
|
54
|
+
}
|
|
55
|
+
return this.toolContext.invocationContext.artifactService.deleteArtifact(
|
|
56
|
+
request
|
|
57
|
+
);
|
|
58
|
+
}
|
|
59
|
+
async listVersions(request) {
|
|
60
|
+
if (!this.toolContext.invocationContext.artifactService) {
|
|
61
|
+
throw new Error("Artifact service is not initialized.");
|
|
62
|
+
}
|
|
63
|
+
return this.toolContext.invocationContext.artifactService.listVersions(
|
|
64
|
+
request
|
|
65
|
+
);
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
69
|
+
0 && (module.exports = {
|
|
70
|
+
ForwardingArtifactService
|
|
71
|
+
});
|
|
@@ -0,0 +1,101 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var function_tool_exports = {};
|
|
26
|
+
__export(function_tool_exports, {
|
|
27
|
+
FunctionTool: () => FunctionTool
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(function_tool_exports);
|
|
30
|
+
var import_genai = require("@google/genai");
|
|
31
|
+
var import_zod = require("zod");
|
|
32
|
+
var import_simple_zod_to_json = require("../utils/simple_zod_to_json.js");
|
|
33
|
+
var import_base_tool = require("./base_tool.js");
|
|
34
|
+
/**
|
|
35
|
+
* @license
|
|
36
|
+
* Copyright 2025 Google LLC
|
|
37
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
38
|
+
*/
|
|
39
|
+
function toSchema(parameters) {
|
|
40
|
+
if (parameters === void 0) {
|
|
41
|
+
return { type: import_genai.Type.OBJECT, properties: {} };
|
|
42
|
+
}
|
|
43
|
+
if ((0, import_simple_zod_to_json.isZodObject)(parameters)) {
|
|
44
|
+
return (0, import_simple_zod_to_json.zodObjectToSchema)(parameters);
|
|
45
|
+
}
|
|
46
|
+
return parameters;
|
|
47
|
+
}
|
|
48
|
+
class FunctionTool extends import_base_tool.BaseTool {
|
|
49
|
+
/**
|
|
50
|
+
* The constructor acts as the user-friendly factory.
|
|
51
|
+
* @param options The configuration for the tool.
|
|
52
|
+
*/
|
|
53
|
+
constructor(options) {
|
|
54
|
+
var _a;
|
|
55
|
+
const name = (_a = options.name) != null ? _a : options.execute.name;
|
|
56
|
+
if (!name) {
|
|
57
|
+
throw new Error(
|
|
58
|
+
"Tool name cannot be empty. Either name the `execute` function or provide a `name`."
|
|
59
|
+
);
|
|
60
|
+
}
|
|
61
|
+
super({
|
|
62
|
+
name,
|
|
63
|
+
description: options.description,
|
|
64
|
+
isLongRunning: options.isLongRunning
|
|
65
|
+
});
|
|
66
|
+
this.execute = options.execute;
|
|
67
|
+
this.parameters = options.parameters;
|
|
68
|
+
}
|
|
69
|
+
/**
|
|
70
|
+
* Provide a schema for the function.
|
|
71
|
+
*/
|
|
72
|
+
_getDeclaration() {
|
|
73
|
+
return {
|
|
74
|
+
name: this.name,
|
|
75
|
+
description: this.description,
|
|
76
|
+
parameters: toSchema(this.parameters)
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
/**
|
|
80
|
+
* Logic for running the tool.
|
|
81
|
+
*/
|
|
82
|
+
async runAsync(req) {
|
|
83
|
+
try {
|
|
84
|
+
let validatedArgs = req.args;
|
|
85
|
+
if (this.parameters instanceof import_zod.ZodObject) {
|
|
86
|
+
validatedArgs = this.parameters.parse(req.args);
|
|
87
|
+
}
|
|
88
|
+
return await this.execute(
|
|
89
|
+
validatedArgs,
|
|
90
|
+
req.toolContext
|
|
91
|
+
);
|
|
92
|
+
} catch (error) {
|
|
93
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
94
|
+
throw new Error(`Error in tool '${this.name}': ${errorMessage}`);
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
99
|
+
0 && (module.exports = {
|
|
100
|
+
FunctionTool
|
|
101
|
+
});
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var google_search_tool_exports = {};
|
|
26
|
+
__export(google_search_tool_exports, {
|
|
27
|
+
GOOGLE_SEARCH: () => GOOGLE_SEARCH
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(google_search_tool_exports);
|
|
30
|
+
var import_model_name = require("../utils/model_name.js");
|
|
31
|
+
var import_base_tool = require("./base_tool.js");
|
|
32
|
+
/**
|
|
33
|
+
* @license
|
|
34
|
+
* Copyright 2025 Google LLC
|
|
35
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
36
|
+
*/
|
|
37
|
+
class GoogleSearchTool extends import_base_tool.BaseTool {
|
|
38
|
+
constructor() {
|
|
39
|
+
super({ name: "google_search", description: "Google Search Tool" });
|
|
40
|
+
}
|
|
41
|
+
runAsync(request) {
|
|
42
|
+
return Promise.resolve();
|
|
43
|
+
}
|
|
44
|
+
async processLlmRequest({ toolContext, llmRequest }) {
|
|
45
|
+
if (!llmRequest.model) {
|
|
46
|
+
return;
|
|
47
|
+
}
|
|
48
|
+
llmRequest.config = llmRequest.config || {};
|
|
49
|
+
llmRequest.config.tools = llmRequest.config.tools || [];
|
|
50
|
+
if ((0, import_model_name.isGemini1Model)(llmRequest.model)) {
|
|
51
|
+
if (llmRequest.config.tools.length > 0) {
|
|
52
|
+
throw new Error(
|
|
53
|
+
"Google search tool can not be used with other tools in Gemini 1.x."
|
|
54
|
+
);
|
|
55
|
+
}
|
|
56
|
+
llmRequest.config.tools.push({
|
|
57
|
+
googleSearchRetrieval: {}
|
|
58
|
+
});
|
|
59
|
+
return;
|
|
60
|
+
}
|
|
61
|
+
if ((0, import_model_name.isGeminiModel)(llmRequest.model)) {
|
|
62
|
+
llmRequest.config.tools.push({
|
|
63
|
+
googleSearch: {}
|
|
64
|
+
});
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
throw new Error(
|
|
68
|
+
`Google search tool is not supported for model ${llmRequest.model}`
|
|
69
|
+
);
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
const GOOGLE_SEARCH = new GoogleSearchTool();
|
|
73
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
74
|
+
0 && (module.exports = {
|
|
75
|
+
GOOGLE_SEARCH
|
|
76
|
+
});
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var long_running_tool_exports = {};
|
|
26
|
+
__export(long_running_tool_exports, {
|
|
27
|
+
LongRunningFunctionTool: () => LongRunningFunctionTool
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(long_running_tool_exports);
|
|
30
|
+
var import_function_tool = require("./function_tool.js");
|
|
31
|
+
/**
|
|
32
|
+
* @license
|
|
33
|
+
* Copyright 2025 Google LLC
|
|
34
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
35
|
+
*/
|
|
36
|
+
const LONG_RUNNING_INSTRUCTION = `
|
|
37
|
+
|
|
38
|
+
NOTE: This is a long-running operation. Do not call this tool again if it has already returned some intermediate or pending status.`;
|
|
39
|
+
class LongRunningFunctionTool extends import_function_tool.FunctionTool {
|
|
40
|
+
/**
|
|
41
|
+
* The constructor acts as the user-friendly factory.
|
|
42
|
+
* @param options The configuration for the tool.
|
|
43
|
+
*/
|
|
44
|
+
constructor(options) {
|
|
45
|
+
super({ ...options, isLongRunning: true });
|
|
46
|
+
}
|
|
47
|
+
/**
|
|
48
|
+
* Provide a schema for the function.
|
|
49
|
+
*/
|
|
50
|
+
_getDeclaration() {
|
|
51
|
+
const declaration = super._getDeclaration();
|
|
52
|
+
if (declaration.description) {
|
|
53
|
+
declaration.description += LONG_RUNNING_INSTRUCTION;
|
|
54
|
+
} else {
|
|
55
|
+
declaration.description = LONG_RUNNING_INSTRUCTION.trimStart();
|
|
56
|
+
}
|
|
57
|
+
return declaration;
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
61
|
+
0 && (module.exports = {
|
|
62
|
+
LongRunningFunctionTool
|
|
63
|
+
});
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var mcp_session_manager_exports = {};
|
|
26
|
+
__export(mcp_session_manager_exports, {
|
|
27
|
+
MCPSessionManager: () => MCPSessionManager
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(mcp_session_manager_exports);
|
|
30
|
+
var import_client = require("@modelcontextprotocol/sdk/client/index.js");
|
|
31
|
+
var import_stdio = require("@modelcontextprotocol/sdk/client/stdio.js");
|
|
32
|
+
var import_streamableHttp = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
|
|
33
|
+
/**
|
|
34
|
+
* @license
|
|
35
|
+
* Copyright 2025 Google LLC
|
|
36
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
37
|
+
*/
|
|
38
|
+
class MCPSessionManager {
|
|
39
|
+
constructor(connectionParams) {
|
|
40
|
+
this.connectionParams = connectionParams;
|
|
41
|
+
}
|
|
42
|
+
async createSession() {
|
|
43
|
+
const client = new import_client.Client({ name: "MCPClient", version: "1.0.0" });
|
|
44
|
+
switch (this.connectionParams.type) {
|
|
45
|
+
case "StdioConnectionParams":
|
|
46
|
+
await client.connect(
|
|
47
|
+
new import_stdio.StdioClientTransport(this.connectionParams.serverParams)
|
|
48
|
+
);
|
|
49
|
+
break;
|
|
50
|
+
case "StreamableHTTPConnectionParams":
|
|
51
|
+
await client.connect(new import_streamableHttp.StreamableHTTPClientTransport(
|
|
52
|
+
new URL(this.connectionParams.url)
|
|
53
|
+
));
|
|
54
|
+
break;
|
|
55
|
+
default:
|
|
56
|
+
const _exhaustiveCheck = this.connectionParams;
|
|
57
|
+
break;
|
|
58
|
+
}
|
|
59
|
+
return client;
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
63
|
+
0 && (module.exports = {
|
|
64
|
+
MCPSessionManager
|
|
65
|
+
});
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var mcp_tool_exports = {};
|
|
26
|
+
__export(mcp_tool_exports, {
|
|
27
|
+
MCPTool: () => MCPTool
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(mcp_tool_exports);
|
|
30
|
+
var import_gemini_schema_util = require("../../utils/gemini_schema_util.js");
|
|
31
|
+
var import_base_tool = require("../base_tool.js");
|
|
32
|
+
/**
|
|
33
|
+
* @license
|
|
34
|
+
* Copyright 2025 Google LLC
|
|
35
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
36
|
+
*/
|
|
37
|
+
class MCPTool extends import_base_tool.BaseTool {
|
|
38
|
+
constructor(mcpTool, mcpSessionManager) {
|
|
39
|
+
super({ name: mcpTool.name, description: mcpTool.description || "" });
|
|
40
|
+
this.mcpTool = mcpTool;
|
|
41
|
+
this.mcpSessionManager = mcpSessionManager;
|
|
42
|
+
}
|
|
43
|
+
_getDeclaration() {
|
|
44
|
+
let declaration;
|
|
45
|
+
declaration = {
|
|
46
|
+
name: this.mcpTool.name,
|
|
47
|
+
description: this.mcpTool.description,
|
|
48
|
+
parameters: (0, import_gemini_schema_util.toGeminiSchema)(this.mcpTool.inputSchema),
|
|
49
|
+
// TODO: need revisit, refer to this
|
|
50
|
+
// https://modelcontextprotocol.io/specification/2025-06-18/server/tools#tool-result
|
|
51
|
+
response: (0, import_gemini_schema_util.toGeminiSchema)(this.mcpTool.outputSchema)
|
|
52
|
+
};
|
|
53
|
+
return declaration;
|
|
54
|
+
}
|
|
55
|
+
async runAsync(request) {
|
|
56
|
+
const session = await this.mcpSessionManager.createSession();
|
|
57
|
+
const callRequest = {};
|
|
58
|
+
callRequest.params = { name: this.mcpTool.name, arguments: request.args };
|
|
59
|
+
return await session.callTool(callRequest.params);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
63
|
+
0 && (module.exports = {
|
|
64
|
+
MCPTool
|
|
65
|
+
});
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var mcp_toolset_exports = {};
|
|
26
|
+
__export(mcp_toolset_exports, {
|
|
27
|
+
MCPToolset: () => MCPToolset
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(mcp_toolset_exports);
|
|
30
|
+
var import_logger = require("../../utils/logger.js");
|
|
31
|
+
var import_base_toolset = require("../base_toolset.js");
|
|
32
|
+
var import_mcp_session_manager = require("./mcp_session_manager.js");
|
|
33
|
+
var import_mcp_tool = require("./mcp_tool.js");
|
|
34
|
+
/**
|
|
35
|
+
* @license
|
|
36
|
+
* Copyright 2025 Google LLC
|
|
37
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
38
|
+
*/
|
|
39
|
+
class MCPToolset extends import_base_toolset.BaseToolset {
|
|
40
|
+
constructor(connectionParams, toolFilter = []) {
|
|
41
|
+
super(toolFilter);
|
|
42
|
+
this.mcpSessionManager = new import_mcp_session_manager.MCPSessionManager(connectionParams);
|
|
43
|
+
}
|
|
44
|
+
async getTools(context) {
|
|
45
|
+
const session = await this.mcpSessionManager.createSession();
|
|
46
|
+
const listResult = await session.listTools();
|
|
47
|
+
import_logger.logger.debug(`number of tools: ${listResult.tools.length}`);
|
|
48
|
+
for (const tool of listResult.tools) {
|
|
49
|
+
import_logger.logger.debug(`tool: ${tool.name}`);
|
|
50
|
+
}
|
|
51
|
+
return listResult.tools.map(
|
|
52
|
+
(tool) => new import_mcp_tool.MCPTool(tool, this.mcpSessionManager)
|
|
53
|
+
);
|
|
54
|
+
}
|
|
55
|
+
async close() {
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
59
|
+
0 && (module.exports = {
|
|
60
|
+
MCPToolset
|
|
61
|
+
});
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var tool_confirmation_exports = {};
|
|
26
|
+
__export(tool_confirmation_exports, {
|
|
27
|
+
ToolConfirmation: () => ToolConfirmation
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(tool_confirmation_exports);
|
|
30
|
+
/**
|
|
31
|
+
* @license
|
|
32
|
+
* Copyright 2025 Google LLC
|
|
33
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
34
|
+
*/
|
|
35
|
+
class ToolConfirmation {
|
|
36
|
+
constructor({
|
|
37
|
+
hint,
|
|
38
|
+
confirmed,
|
|
39
|
+
payload
|
|
40
|
+
}) {
|
|
41
|
+
this.hint = hint != null ? hint : "";
|
|
42
|
+
this.confirmed = confirmed;
|
|
43
|
+
this.payload = payload;
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
47
|
+
0 && (module.exports = {
|
|
48
|
+
ToolConfirmation
|
|
49
|
+
});
|
|
@@ -0,0 +1,129 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
"use strict";
|
|
8
|
+
var __defProp = Object.defineProperty;
|
|
9
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
10
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
11
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
12
|
+
var __export = (target, all) => {
|
|
13
|
+
for (var name in all)
|
|
14
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
15
|
+
};
|
|
16
|
+
var __copyProps = (to, from, except, desc) => {
|
|
17
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
18
|
+
for (let key of __getOwnPropNames(from))
|
|
19
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
20
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
21
|
+
}
|
|
22
|
+
return to;
|
|
23
|
+
};
|
|
24
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
25
|
+
var tool_context_exports = {};
|
|
26
|
+
__export(tool_context_exports, {
|
|
27
|
+
ToolContext: () => ToolContext
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(tool_context_exports);
|
|
30
|
+
var import_callback_context = require("../agents/callback_context.js");
|
|
31
|
+
var import_auth_handler = require("../auth/auth_handler.js");
|
|
32
|
+
var import_tool_confirmation = require("../tools/tool_confirmation.js");
|
|
33
|
+
/**
|
|
34
|
+
* @license
|
|
35
|
+
* Copyright 2025 Google LLC
|
|
36
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
37
|
+
*/
|
|
38
|
+
class ToolContext extends import_callback_context.CallbackContext {
|
|
39
|
+
/**
|
|
40
|
+
* @param params.invocationContext The invocation context of the tool.
|
|
41
|
+
* @param params.eventActions The event actions of the current tool call.
|
|
42
|
+
* @param params.functionCallId The function call id of the current tool call.
|
|
43
|
+
* This id was returned in the function call event from LLM to identify a
|
|
44
|
+
* function call. If LLM didn't return this id, ADK will assign one to it.
|
|
45
|
+
* This id is used to map function call response to the original function
|
|
46
|
+
* call.
|
|
47
|
+
* @param params.toolConfirmation The tool confirmation of the current tool
|
|
48
|
+
* call.
|
|
49
|
+
*/
|
|
50
|
+
constructor({
|
|
51
|
+
invocationContext,
|
|
52
|
+
eventActions,
|
|
53
|
+
functionCallId,
|
|
54
|
+
toolConfirmation
|
|
55
|
+
}) {
|
|
56
|
+
super({ invocationContext, eventActions });
|
|
57
|
+
this.functionCallId = functionCallId;
|
|
58
|
+
this.toolConfirmation = toolConfirmation;
|
|
59
|
+
}
|
|
60
|
+
get actions() {
|
|
61
|
+
return this.eventActions;
|
|
62
|
+
}
|
|
63
|
+
requestCredential(authConfig) {
|
|
64
|
+
if (!this.functionCallId) {
|
|
65
|
+
throw new Error("functionCallId is not set.");
|
|
66
|
+
}
|
|
67
|
+
const authHandler = new import_auth_handler.AuthHandler(authConfig);
|
|
68
|
+
this.eventActions.requestedAuthConfigs[this.functionCallId] = authHandler.generateAuthRequest();
|
|
69
|
+
}
|
|
70
|
+
/**
|
|
71
|
+
* Gets the auth credential for the given auth config.
|
|
72
|
+
*
|
|
73
|
+
* @param authConfig The auth config to get the auth credential for.
|
|
74
|
+
* @return The auth credential for the given auth config.
|
|
75
|
+
*/
|
|
76
|
+
getAuthResponse(authConfig) {
|
|
77
|
+
const authHandler = new import_auth_handler.AuthHandler(authConfig);
|
|
78
|
+
return authHandler.getAuthResponse(this.state);
|
|
79
|
+
}
|
|
80
|
+
/**
|
|
81
|
+
* Lists the filenames of the artifacts attached to the current session.
|
|
82
|
+
*
|
|
83
|
+
* @return A promise that resolves to a list of artifact filenames.
|
|
84
|
+
*/
|
|
85
|
+
listArtifacts() {
|
|
86
|
+
if (!this.invocationContext.artifactService) {
|
|
87
|
+
throw new Error("Artifact service is not initialized.");
|
|
88
|
+
}
|
|
89
|
+
return this.invocationContext.artifactService.listArtifactKeys({
|
|
90
|
+
appName: this.invocationContext.session.appName,
|
|
91
|
+
userId: this.invocationContext.session.userId,
|
|
92
|
+
sessionId: this.invocationContext.session.id
|
|
93
|
+
});
|
|
94
|
+
}
|
|
95
|
+
/**
|
|
96
|
+
* Searches the memory of the current user.
|
|
97
|
+
*
|
|
98
|
+
* @param query The query to search memory for.
|
|
99
|
+
* @return A promise that resolves to SearchMemoryResponse containing the
|
|
100
|
+
* matching memories.
|
|
101
|
+
*/
|
|
102
|
+
searchMemory(query) {
|
|
103
|
+
if (!this.invocationContext.memoryService) {
|
|
104
|
+
throw new Error("Memory service is not initialized.");
|
|
105
|
+
}
|
|
106
|
+
return this.invocationContext.memoryService.searchMemory({
|
|
107
|
+
appName: this.invocationContext.session.appName,
|
|
108
|
+
userId: this.invocationContext.session.userId,
|
|
109
|
+
query
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
/**
|
|
113
|
+
* Requests confirmation for the current tool call.
|
|
114
|
+
*/
|
|
115
|
+
requestConfirmation({ hint, payload }) {
|
|
116
|
+
if (!this.functionCallId) {
|
|
117
|
+
throw new Error("functionCallId is not set.");
|
|
118
|
+
}
|
|
119
|
+
this.eventActions.requestedToolConfirmations[this.functionCallId] = new import_tool_confirmation.ToolConfirmation({
|
|
120
|
+
hint,
|
|
121
|
+
confirmed: false,
|
|
122
|
+
payload
|
|
123
|
+
});
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
127
|
+
0 && (module.exports = {
|
|
128
|
+
ToolContext
|
|
129
|
+
});
|