@google/adk 2.3.0 → 2.5.0

package/dist/cjs/telemetry/tracing.js

@@ -1,350 +0,0 @@
-/**
-  * @license
-  * Copyright 2026 Google LLC
-  * SPDX-License-Identifier: Apache-2.0
-  */
-
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var tracing_exports = {};
-__export(tracing_exports, {
-  runAsyncGeneratorWithOtelContext: () => runAsyncGeneratorWithOtelContext,
-  traceAgentInvocation: () => traceAgentInvocation,
-  traceCallLlm: () => traceCallLlm,
-  traceMergedToolCalls: () => traceMergedToolCalls,
-  traceNodeExecution: () => traceNodeExecution,
-  traceSendData: () => traceSendData,
-  traceToolCall: () => traceToolCall,
-  traceWorkflowInvocation: () => traceWorkflowInvocation,
-  tracer: () => tracer
-});
-module.exports = __toCommonJS(tracing_exports);
-var import_api = require("@opentelemetry/api");
-var import_version = require("../version.js");
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-var GenAiAttr = /* @__PURE__ */ ((GenAiAttr2) => {
-  GenAiAttr2["AGENT_DESCRIPTION"] = "gen_ai.agent.description";
-  GenAiAttr2["AGENT_NAME"] = "gen_ai.agent.name";
-  GenAiAttr2["CONVERSATION_ID"] = "gen_ai.conversation.id";
-  GenAiAttr2["OPERATION_NAME"] = "gen_ai.operation.name";
-  GenAiAttr2["SYSTEM"] = "gen_ai.system";
-  GenAiAttr2["TOOL_CALL_ID"] = "gen_ai.tool.call.id";
-  GenAiAttr2["TOOL_DESCRIPTION"] = "gen_ai.tool.description";
-  GenAiAttr2["TOOL_NAME"] = "gen_ai.tool.name";
-  GenAiAttr2["TOOL_TYPE"] = "gen_ai.tool.type";
-  GenAiAttr2["REQUEST_MODEL"] = "gen_ai.request.model";
-  GenAiAttr2["REQUEST_TOP_P"] = "gen_ai.request.top_p";
-  GenAiAttr2["REQUEST_MAX_TOKENS"] = "gen_ai.request.max_tokens";
-  GenAiAttr2["USAGE_INPUT_TOKENS"] = "gen_ai.usage.input_tokens";
-  GenAiAttr2["USAGE_OUTPUT_TOKENS"] = "gen_ai.usage.output_tokens";
-  GenAiAttr2["RESPONSE_FINISH_REASONS"] = "gen_ai.response.finish_reasons";
-  return GenAiAttr2;
-})(GenAiAttr || {});
-var AdkAttr = /* @__PURE__ */ ((AdkAttr2) => {
-  AdkAttr2["WORKFLOW_NAME"] = "adk.workflow.name";
-  AdkAttr2["NODE_PATH"] = "adk.node.path";
-  AdkAttr2["NODE_RUN_ID"] = "adk.node.run_id";
-  AdkAttr2["NODE_ATTEMPT"] = "adk.node.attempt";
-  AdkAttr2["NODE_STATUS"] = "adk.node.status";
-  AdkAttr2["NODE_INTERRUPT_COUNT"] = "adk.node.interrupt_count";
-  return AdkAttr2;
-})(AdkAttr || {});
-var GcpAttr = /* @__PURE__ */ ((GcpAttr2) => {
-  GcpAttr2["INVOCATION_ID"] = "gcp.vertex.agent.invocation_id";
-  GcpAttr2["SESSION_ID"] = "gcp.vertex.agent.session_id";
-  GcpAttr2["EVENT_ID"] = "gcp.vertex.agent.event_id";
-  GcpAttr2["LLM_REQUEST"] = "gcp.vertex.agent.llm_request";
-  GcpAttr2["LLM_RESPONSE"] = "gcp.vertex.agent.llm_response";
-  GcpAttr2["TOOL_CALL_ARGS"] = "gcp.vertex.agent.tool_call_args";
-  GcpAttr2["TOOL_RESPONSE"] = "gcp.vertex.agent.tool_response";
-  GcpAttr2["DATA"] = "gcp.vertex.agent.data";
-  return GcpAttr2;
-})(GcpAttr || {});
-const GEN_AI_SYSTEM_VALUE = "gcp.vertex.agent";
-const tracer = import_api.trace.getTracer(GEN_AI_SYSTEM_VALUE, import_version.version);
-function safeJsonSerialize(obj) {
-  try {
-    return JSON.stringify(obj);
-  } catch (_e) {
-    return "<not serializable>";
-  }
-}
-__name(safeJsonSerialize, "safeJsonSerialize");
-function traceAgentInvocation({
-  agent,
-  invocationContext
-}) {
-  const span = import_api.trace.getActiveSpan();
-  if (!span) return;
-  span.setAttributes({
-    ["gen_ai.operation.name" /* OPERATION_NAME */]: "invoke_agent",
-    // Conditionally Required
-    ["gen_ai.agent.description" /* AGENT_DESCRIPTION */]: agent.description,
-    ["gen_ai.agent.name" /* AGENT_NAME */]: agent.name,
-    ["gen_ai.conversation.id" /* CONVERSATION_ID */]: invocationContext.session.id
-  });
-}
-__name(traceAgentInvocation, "traceAgentInvocation");
-function traceWorkflowInvocation({
-  workflowName,
-  nodePath,
-  sessionId
-}) {
-  const span = import_api.trace.getActiveSpan();
-  if (!span) return;
-  span.setAttributes({
-    ["gen_ai.operation.name" /* OPERATION_NAME */]: "invoke_workflow",
-    ["gen_ai.conversation.id" /* CONVERSATION_ID */]: sessionId,
-    ["adk.workflow.name" /* WORKFLOW_NAME */]: workflowName,
-    ["adk.node.path" /* NODE_PATH */]: nodePath
-  });
-}
-__name(traceWorkflowInvocation, "traceWorkflowInvocation");
-function traceNodeExecution({
-  nodePath,
-  runId,
-  attempt,
-  status,
-  interruptCount
-}) {
-  const span = import_api.trace.getActiveSpan();
-  if (!span) return;
-  span.setAttributes({
-    ["gen_ai.operation.name" /* OPERATION_NAME */]: "execute_node",
-    ["adk.node.path" /* NODE_PATH */]: nodePath,
-    ["adk.node.run_id" /* NODE_RUN_ID */]: runId,
-    ["adk.node.attempt" /* NODE_ATTEMPT */]: attempt,
-    ["adk.node.status" /* NODE_STATUS */]: status,
-    ["adk.node.interrupt_count" /* NODE_INTERRUPT_COUNT */]: interruptCount
-  });
-}
-__name(traceNodeExecution, "traceNodeExecution");
-function traceToolCall({
-  tool,
-  args,
-  functionResponseEvent,
-  invocationContext
-}) {
-  var _a, _b;
-  const span = import_api.trace.getActiveSpan();
-  if (!span) return;
-  span.setAttributes({
-    ["gen_ai.operation.name" /* OPERATION_NAME */]: "execute_tool",
-    ["gen_ai.system" /* SYSTEM */]: GEN_AI_SYSTEM_VALUE,
-    ["gen_ai.tool.description" /* TOOL_DESCRIPTION */]: tool.description || "",
-    ["gen_ai.tool.name" /* TOOL_NAME */]: tool.name,
-    // e.g. FunctionTool
-    ["gen_ai.tool.type" /* TOOL_TYPE */]: tool.constructor.name,
-    ["gcp.vertex.agent.invocation_id" /* INVOCATION_ID */]: invocationContext.invocationId,
-    // Setting empty llm request and response (as UI expect these) while not
-    // applicable for tool_response.
-    ["gcp.vertex.agent.llm_request" /* LLM_REQUEST */]: "{}",
-    ["gcp.vertex.agent.llm_response" /* LLM_RESPONSE */]: "{}",
-    ["gcp.vertex.agent.tool_call_args" /* TOOL_CALL_ARGS */]: shouldAddRequestResponseToSpans() ? safeJsonSerialize(args) : "{}"
-  });
-  let toolCallId = "<not specified>";
-  let toolResponse = "<not specified>";
-  if ((_a = functionResponseEvent.content) == null ? void 0 : _a.parts) {
-    const responseParts = functionResponseEvent.content.parts;
-    const functionResponse = (_b = responseParts[0]) == null ? void 0 : _b.functionResponse;
-    if (functionResponse == null ? void 0 : functionResponse.id) {
-      toolCallId = functionResponse.id;
-    }
-    if (functionResponse == null ? void 0 : functionResponse.response) {
-      toolResponse = functionResponse.response;
-    }
-  }
-  if (typeof toolResponse !== "object" || toolResponse === null) {
-    toolResponse = { result: toolResponse };
-  }
-  span.setAttributes({
-    ["gen_ai.tool.call.id" /* TOOL_CALL_ID */]: toolCallId,
-    ["gcp.vertex.agent.event_id" /* EVENT_ID */]: functionResponseEvent.id,
-    ["gcp.vertex.agent.tool_response" /* TOOL_RESPONSE */]: shouldAddRequestResponseToSpans() ? safeJsonSerialize(toolResponse) : "{}"
-  });
-}
-__name(traceToolCall, "traceToolCall");
-function traceMergedToolCalls({
-  responseEventId,
-  functionResponseEvent,
-  invocationContext
-}) {
-  const span = import_api.trace.getActiveSpan();
-  if (!span) return;
-  span.setAttributes({
-    ["gen_ai.operation.name" /* OPERATION_NAME */]: "execute_tool",
-    ["gen_ai.system" /* SYSTEM */]: GEN_AI_SYSTEM_VALUE,
-    ["gen_ai.tool.name" /* TOOL_NAME */]: "(merged tools)",
-    ["gen_ai.tool.description" /* TOOL_DESCRIPTION */]: "(merged tools)",
-    ["gen_ai.tool.call.id" /* TOOL_CALL_ID */]: responseEventId,
-    ["gcp.vertex.agent.invocation_id" /* INVOCATION_ID */]: invocationContext.invocationId,
-    ["gcp.vertex.agent.tool_call_args" /* TOOL_CALL_ARGS */]: "N/A",
-    ["gcp.vertex.agent.event_id" /* EVENT_ID */]: responseEventId,
-    // Setting empty llm request and response (as UI expect these) while not
-    // applicable for tool_response.
-    ["gcp.vertex.agent.llm_request" /* LLM_REQUEST */]: "{}",
-    ["gcp.vertex.agent.llm_response" /* LLM_RESPONSE */]: "{}"
-  });
-  span.setAttribute(
-    "gcp.vertex.agent.tool_response" /* TOOL_RESPONSE */,
-    shouldAddRequestResponseToSpans() ? safeJsonSerialize(functionResponseEvent) : "{}"
-  );
-}
-__name(traceMergedToolCalls, "traceMergedToolCalls");
-function traceCallLlm({
-  invocationContext,
-  eventId,
-  llmRequest,
-  llmResponse
-}) {
-  var _a, _b, _c, _d;
-  const span = import_api.trace.getActiveSpan();
-  if (!span) return;
-  span.setAttributes({
-    ["gen_ai.system" /* SYSTEM */]: GEN_AI_SYSTEM_VALUE,
-    ["gen_ai.request.model" /* REQUEST_MODEL */]: llmRequest.model,
-    ...((_a = invocationContext.agent) == null ? void 0 : _a.name) ? { ["gen_ai.agent.name" /* AGENT_NAME */]: invocationContext.agent.name } : {},
-    ["gcp.vertex.agent.invocation_id" /* INVOCATION_ID */]: invocationContext.invocationId,
-    ["gcp.vertex.agent.session_id" /* SESSION_ID */]: invocationContext.session.id,
-    ["gcp.vertex.agent.event_id" /* EVENT_ID */]: eventId,
-    // Consider removing once GenAI SDK provides a way to record this info.
-    ["gcp.vertex.agent.llm_request" /* LLM_REQUEST */]: shouldAddRequestResponseToSpans() ? safeJsonSerialize(buildLlmRequestForTrace(llmRequest)) : "{}"
-  });
-  if ((_b = llmRequest.config) == null ? void 0 : _b.topP) {
-    span.setAttribute("gen_ai.request.top_p" /* REQUEST_TOP_P */, llmRequest.config.topP);
-  }
-  if (((_c = llmRequest.config) == null ? void 0 : _c.maxOutputTokens) !== void 0) {
-    span.setAttribute(
-      "gen_ai.request.max_tokens" /* REQUEST_MAX_TOKENS */,
-      llmRequest.config.maxOutputTokens
-    );
-  }
-  span.setAttribute(
-    "gcp.vertex.agent.llm_response" /* LLM_RESPONSE */,
-    shouldAddRequestResponseToSpans() ? safeJsonSerialize(llmResponse) : "{}"
-  );
-  if (llmResponse.usageMetadata) {
-    span.setAttribute(
-      "gen_ai.usage.input_tokens" /* USAGE_INPUT_TOKENS */,
-      llmResponse.usageMetadata.promptTokenCount || 0
-    );
-  }
-  if ((_d = llmResponse.usageMetadata) == null ? void 0 : _d.candidatesTokenCount) {
-    span.setAttribute(
-      "gen_ai.usage.output_tokens" /* USAGE_OUTPUT_TOKENS */,
-      llmResponse.usageMetadata.candidatesTokenCount
-    );
-  }
-  if (llmResponse.finishReason) {
-    const finishReasonValue = typeof llmResponse.finishReason === "string" ? llmResponse.finishReason.toLowerCase() : String(llmResponse.finishReason).toLowerCase();
-    span.setAttribute("gen_ai.response.finish_reasons" /* RESPONSE_FINISH_REASONS */, [finishReasonValue]);
-  }
-}
-__name(traceCallLlm, "traceCallLlm");
-function traceSendData({
-  invocationContext,
-  eventId,
-  data
-}) {
-  const span = import_api.trace.getActiveSpan();
-  if (!span) return;
-  span.setAttributes({
-    ["gcp.vertex.agent.invocation_id" /* INVOCATION_ID */]: invocationContext.invocationId,
-    ["gcp.vertex.agent.event_id" /* EVENT_ID */]: eventId
-  });
-  span.setAttribute(
-    "gcp.vertex.agent.data" /* DATA */,
-    shouldAddRequestResponseToSpans() ? safeJsonSerialize(data) : "{}"
-  );
-}
-__name(traceSendData, "traceSendData");
-function redactHttpOptions(httpOptions) {
-  const redacted = { ...httpOptions };
-  delete redacted.headers;
-  delete redacted.extraBody;
-  return redacted;
-}
-__name(redactHttpOptions, "redactHttpOptions");
-function buildLlmRequestForTrace(llmRequest) {
-  const result = {
-    model: llmRequest.model,
-    contents: []
-  };
-  if (llmRequest.config) {
-    const { responseSchema: _responseSchema, ...cleanConfig } = llmRequest.config;
-    if (cleanConfig.httpOptions) {
-      cleanConfig.httpOptions = redactHttpOptions(cleanConfig.httpOptions);
-    }
-    result.config = cleanConfig;
-  }
-  result.contents = llmRequest.contents.map((content) => {
-    var _a;
-    return {
-      role: content.role,
-      parts: ((_a = content.parts) == null ? void 0 : _a.filter((part) => !part.inlineData)) || []
-    };
-  });
-  return result;
-}
-__name(buildLlmRequestForTrace, "buildLlmRequestForTrace");
-function bindOtelContextToAsyncGenerator(ctx, generator) {
-  return {
-    // Bind the next() method to execute within the provided context
-    next: import_api.context.bind(ctx, generator.next.bind(generator)),
-    // Bind the return() method to execute within the provided context
-    return: import_api.context.bind(ctx, generator.return.bind(generator)),
-    // Bind the throw() method to execute within the provided context
-    throw: import_api.context.bind(ctx, generator.throw.bind(generator)),
-    // Ensure the async iterator symbol also returns a context-bound generator
-    [Symbol.asyncIterator]() {
-      return bindOtelContextToAsyncGenerator(
-        ctx,
-        generator[Symbol.asyncIterator]()
-      );
-    }
-  };
-}
-__name(bindOtelContextToAsyncGenerator, "bindOtelContextToAsyncGenerator");
-function runAsyncGeneratorWithOtelContext(otelContext, generatorFnContext, generatorFn) {
-  const generator = generatorFn.call(generatorFnContext);
-  return bindOtelContextToAsyncGenerator(otelContext, generator);
-}
-__name(runAsyncGeneratorWithOtelContext, "runAsyncGeneratorWithOtelContext");
-function shouldAddRequestResponseToSpans() {
-  const envValue = process.env.ADK_CAPTURE_MESSAGE_CONTENT_IN_SPANS || "true";
-  return envValue === "true" || envValue === "1";
-}
-__name(shouldAddRequestResponseToSpans, "shouldAddRequestResponseToSpans");
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  runAsyncGeneratorWithOtelContext,
-  traceAgentInvocation,
-  traceCallLlm,
-  traceMergedToolCalls,
-  traceNodeExecution,
-  traceSendData,
-  traceToolCall,
-  traceWorkflowInvocation,
-  tracer
-});

package/dist/cjs/tools/crewai_tool.js

@@ -1,132 +0,0 @@
-/**
-  * @license
-  * Copyright 2026 Google LLC
-  * SPDX-License-Identifier: Apache-2.0
-  */
-
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var crewai_tool_exports = {};
-__export(crewai_tool_exports, {
-  CrewaiTool: () => CrewaiTool,
-  isCrewaiTool: () => isCrewaiTool
-});
-module.exports = __toCommonJS(crewai_tool_exports);
-var import_schema = require("../utils/schema.js");
-var import_simple_zod_to_json = require("../utils/simple_zod_to_json.js");
-var import_variant_utils = require("../utils/variant_utils.js");
-var import_function_tool = require("./function_tool.js");
-/**
- * @license
- * Copyright 2026 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-var _a, _b;
-const CREWAI_TOOL_SIGNATURE_SYMBOL = /* @__PURE__ */ Symbol.for("google.adk.crewaiTool");
-function isCrewaiTool(obj) {
-  return typeof obj === "object" && obj !== null && CREWAI_TOOL_SIGNATURE_SYMBOL in obj && obj[CREWAI_TOOL_SIGNATURE_SYMBOL] === true;
-}
-__name(isCrewaiTool, "isCrewaiTool");
-function isCrewaiToolConfig(value) {
-  return typeof value === "object" && value !== null && "tool" in value && typeof value.tool === "object" && value.tool !== null && typeof value.tool.run === "function";
-}
-__name(isCrewaiToolConfig, "isCrewaiToolConfig");
-function normalizeCrewaiSchemaJson(argsSchema) {
-  if (!argsSchema) {
-    return {};
-  }
-  if (typeof argsSchema.modelJsonSchema === "function") {
-    return argsSchema.modelJsonSchema();
-  }
-  if (typeof argsSchema.jsonSchema === "function") {
-    return argsSchema.jsonSchema();
-  }
-  if ((0, import_simple_zod_to_json.isZodSchema)(argsSchema)) {
-    const raw = structuredClone((0, import_schema.toJsonSchema)(argsSchema));
-    const requiredSet = new Set(
-      Array.isArray(raw["required"]) ? raw["required"] : []
-    );
-    const properties = raw["properties"];
-    if (properties && typeof properties === "object") {
-      for (const [key, prop] of Object.entries(properties)) {
-        if (!prop || typeof prop !== "object") {
-          continue;
-        }
-        if (Array.isArray(prop["type"])) {
-          const types = prop["type"];
-          delete prop["type"];
-          prop["anyOf"] = types.map((t) => ({ type: t }));
-        } else if (!requiredSet.has(key) && !("default" in prop) && !("anyOf" in prop) && typeof prop["type"] === "string") {
-          const propType = prop["type"];
-          delete prop["type"];
-          prop["anyOf"] = [{ type: propType }, { type: "null" }];
-        }
-      }
-    }
-    return raw;
-  }
-  return structuredClone(argsSchema);
-}
-__name(normalizeCrewaiSchemaJson, "normalizeCrewaiSchemaJson");
-const _CrewaiTool = class _CrewaiTool extends (_b = import_function_tool.FunctionTool, _a = CREWAI_TOOL_SIGNATURE_SYMBOL, _b) {
-  constructor(toolOrConfig, options) {
-    const config = isCrewaiToolConfig(toolOrConfig) ? toolOrConfig : {
-      tool: toolOrConfig,
-      name: options == null ? void 0 : options.name,
-      description: options == null ? void 0 : options.description
-    };
-    const resolvedName = config.name || (config.tool.name ? config.tool.name.replace(/ /g, "_").toLowerCase() : config.tool.run.name || "crewai_tool");
-    const resolvedDescription = config.description || config.tool.description || "";
-    super({
-      name: resolvedName,
-      description: resolvedDescription,
-      parameters: (0, import_simple_zod_to_json.isZodObject)(config.tool.argsSchema) ? config.tool.argsSchema : void 0,
-      execute: /* @__PURE__ */ __name(async (args, toolContext) => {
-        return await config.tool.run(
-          args != null ? args : {},
-          toolContext
-        );
-      }, "execute")
-    });
-    /** A unique symbol to identify ADK CrewaiTool class. */
-    this[_a] = true;
-    this.tool = config.tool;
-  }
-  /**
-   * Builds the `FunctionDeclaration` for the wrapped CrewAI tool.
-   */
-  _getDeclaration() {
-    const schemaJson = normalizeCrewaiSchemaJson(this.tool.argsSchema);
-    return (0, import_function_tool.buildFunctionDeclarationForParamsForCrewai)(
-      this.apiVariant === import_variant_utils.GoogleLLMVariant.VERTEX_AI,
-      this.name,
-      this.description,
-      this.tool.run,
-      schemaJson
-    );
-  }
-};
-__name(_CrewaiTool, "CrewaiTool");
-let CrewaiTool = _CrewaiTool;
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  CrewaiTool,
-  isCrewaiTool
-});