@iqai/adk 0.1.21 → 0.1.22

package/dist/index.js

@@ -53,7 +53,7 @@ var init_logger = __esm({
       }
       info(message, ...args) {
         const time = (/* @__PURE__ */ new Date()).toLocaleTimeString();
-        console.info(
+        console.debug(
           this.colorize(`[${time}] \u2139\uFE0F [${this.name}] ${message}`),
           ...args
         );
@@ -229,7 +229,7 @@ var init_base_tool = __esm({
        * @param context The context of the tool
        * @returns The result of running the tool
        */
-      async runAsync(args, context) {
+      async runAsync(args, context4) {
         throw new Error(`${this.constructor.name} runAsync is not implemented`);
       }
       /**
@@ -253,6 +253,12 @@ var init_base_tool = __esm({
           if (!toolWithFunctionDeclarations.functionDeclarations) {
             toolWithFunctionDeclarations.functionDeclarations = [];
           }
+          const alreadyExists = toolWithFunctionDeclarations.functionDeclarations.some(
+            (fd) => _optionalChain([fd, 'optionalAccess', _2 => _2.name]) === functionDeclaration.name
+          );
+          if (alreadyExists) {
+            return;
+          }
           toolWithFunctionDeclarations.functionDeclarations.push(
             functionDeclaration
           );
@@ -281,7 +287,7 @@ var init_base_tool = __esm({
        * @param context Tool execution context
        * @returns Result of the tool execution or error information
        */
-      async safeExecute(args, context) {
+      async safeExecute(args, context4) {
         if (!this.validateArguments(args)) {
           return {
             error: "Invalid arguments",
@@ -302,7 +308,7 @@ var init_base_tool = __esm({
               );
               await new Promise((resolve) => setTimeout(resolve, delay));
             }
-            const result = await this.runAsync(args, context);
+            const result = await this.runAsync(args, context4);
             return { result };
           } catch (error) {
             lastError = error instanceof Error ? error : new Error(String(error));
@@ -312,7 +318,7 @@ var init_base_tool = __esm({
         }
         return {
           error: "Execution failed",
-          message: _optionalChain([lastError, 'optionalAccess', _2 => _2.message]) || "Unknown error occurred",
+          message: _optionalChain([lastError, 'optionalAccess', _3 => _3.message]) || "Unknown error occurred",
           tool: this.name
         };
       }
@@ -484,23 +490,23 @@ var init_function_tool = __esm({
        * @param options Optional configuration for the tool
        */
       constructor(func, options) {
-        const name = _optionalChain([options, 'optionalAccess', _3 => _3.name]) || func.name;
-        const description = _optionalChain([options, 'optionalAccess', _4 => _4.description]) || _optionalChain([(func.toString().match(/\/\*\*([\s\S]*?)\*\//) || []), 'access', _5 => _5[1], 'optionalAccess', _6 => _6.trim, 'call', _7 => _7()]) || "";
+        const name = _optionalChain([options, 'optionalAccess', _4 => _4.name]) || func.name;
+        const description = _optionalChain([options, 'optionalAccess', _5 => _5.description]) || _optionalChain([(func.toString().match(/\/\*\*([\s\S]*?)\*\//) || []), 'access', _6 => _6[1], 'optionalAccess', _7 => _7.trim, 'call', _8 => _8()]) || "";
         super({
           name,
           description,
-          isLongRunning: _optionalChain([options, 'optionalAccess', _8 => _8.isLongRunning]) || false,
-          shouldRetryOnFailure: _optionalChain([options, 'optionalAccess', _9 => _9.shouldRetryOnFailure]) || false,
-          maxRetryAttempts: _optionalChain([options, 'optionalAccess', _10 => _10.maxRetryAttempts]) || 3
+          isLongRunning: _optionalChain([options, 'optionalAccess', _9 => _9.isLongRunning]) || false,
+          shouldRetryOnFailure: _optionalChain([options, 'optionalAccess', _10 => _10.shouldRetryOnFailure]) || false,
+          maxRetryAttempts: _optionalChain([options, 'optionalAccess', _11 => _11.maxRetryAttempts]) || 3
         });_class3.prototype.__init6.call(this);_class3.prototype.__init7.call(this);;
         this.func = func;
         this.mandatoryArgs = this.getMandatoryArgs(func);
-        this.parameterTypes = _optionalChain([options, 'optionalAccess', _11 => _11.parameterTypes]) || {};
+        this.parameterTypes = _optionalChain([options, 'optionalAccess', _12 => _12.parameterTypes]) || {};
       }
       /**
        * Executes the wrapped function with the provided arguments.
        */
-      async runAsync(args, context) {
+      async runAsync(args, context4) {
         try {
           const missingArgs = this.getMissingMandatoryArgs(args);
           if (missingArgs.length > 0) {
@@ -513,13 +519,13 @@ You could retry calling this tool, but it is IMPORTANT for you to provide all th
           }
           const argsToCall = { ...args };
           if (this.functionAcceptsToolContext()) {
-            argsToCall.toolContext = context;
+            argsToCall.toolContext = context4;
           }
           const funcParams = this.getFunctionParameters();
           const argValues = [];
           for (const paramName of funcParams) {
             if (paramName === "toolContext" && this.functionAcceptsToolContext()) {
-              argValues.push(context);
+              argValues.push(context4);
             } else if (paramName in argsToCall) {
               const convertedValue = this.convertArgumentType(
                 argsToCall[paramName],
@@ -549,7 +555,7 @@ You could retry calling this tool, but it is IMPORTANT for you to provide all th
           description: this.description,
           ignoreParams: ["toolContext"]
         });
-        if (Object.keys(this.parameterTypes).length > 0 && _optionalChain([declaration, 'access', _12 => _12.parameters, 'optionalAccess', _13 => _13.properties])) {
+        if (Object.keys(this.parameterTypes).length > 0 && _optionalChain([declaration, 'access', _13 => _13.parameters, 'optionalAccess', _14 => _14.properties])) {
           for (const [paramName, paramType] of Object.entries(
             this.parameterTypes
           )) {
@@ -647,9 +653,9 @@ You could retry calling this tool, but it is IMPORTANT for you to provide all th
           return this.parameterTypes[paramName].toLowerCase();
         }
         const declaration = this.getDeclaration();
-        if (_optionalChain([declaration, 'optionalAccess', _14 => _14.parameters, 'optionalAccess', _15 => _15.properties])) {
+        if (_optionalChain([declaration, 'optionalAccess', _15 => _15.parameters, 'optionalAccess', _16 => _16.properties])) {
           const paramSchema = declaration.parameters.properties[paramName];
-          if (_optionalChain([paramSchema, 'optionalAccess', _16 => _16.type])) {
+          if (_optionalChain([paramSchema, 'optionalAccess', _17 => _17.type])) {
             return paramSchema.type.toLowerCase();
           }
         }
@@ -735,11 +741,11 @@ var LlmRequest = class {
    */
   
   constructor(data) {
-    this.model = _optionalChain([data, 'optionalAccess', _17 => _17.model]);
-    this.contents = _nullishCoalesce(_optionalChain([data, 'optionalAccess', _18 => _18.contents]), () => ( []));
-    this.config = _optionalChain([data, 'optionalAccess', _19 => _19.config]);
-    this.liveConnectConfig = _nullishCoalesce(_optionalChain([data, 'optionalAccess', _20 => _20.liveConnectConfig]), () => ( {}));
-    this.toolsDict = _nullishCoalesce(_optionalChain([data, 'optionalAccess', _21 => _21.toolsDict]), () => ( {}));
+    this.model = _optionalChain([data, 'optionalAccess', _18 => _18.model]);
+    this.contents = _nullishCoalesce(_optionalChain([data, 'optionalAccess', _19 => _19.contents]), () => ( []));
+    this.config = _optionalChain([data, 'optionalAccess', _20 => _20.config]);
+    this.liveConnectConfig = _nullishCoalesce(_optionalChain([data, 'optionalAccess', _21 => _21.liveConnectConfig]), () => ( {}));
+    this.toolsDict = _nullishCoalesce(_optionalChain([data, 'optionalAccess', _22 => _22.toolsDict]), () => ( {}));
   }
   /**
    * Appends instructions to the system instruction.
@@ -760,10 +766,10 @@ ${instructions.join("\n\n")}`;
    * @param tools The tools to append.
    */
   appendTools(tools) {
-    if (!_optionalChain([tools, 'optionalAccess', _22 => _22.length])) return;
+    if (!_optionalChain([tools, 'optionalAccess', _23 => _23.length])) return;
     const declarations = [];
     for (const tool of tools) {
-      const declaration = _optionalChain([tool, 'access', _23 => _23.getDeclaration, 'optionalCall', _24 => _24()]);
+      const declaration = _optionalChain([tool, 'access', _24 => _24.getDeclaration, 'optionalCall', _25 => _25()]);
       if (declaration) {
         declarations.push(declaration);
         this.toolsDict[tool.name] = tool;
@@ -790,7 +796,7 @@ ${instructions.join("\n\n")}`;
    * @returns The system instruction as a string, or undefined if not set.
    */
   getSystemInstructionText() {
-    if (!_optionalChain([this, 'access', _25 => _25.config, 'optionalAccess', _26 => _26.systemInstruction])) {
+    if (!_optionalChain([this, 'access', _26 => _26.config, 'optionalAccess', _27 => _27.systemInstruction])) {
       return void 0;
     }
     const systemInstruction = this.config.systemInstruction;
@@ -818,7 +824,7 @@ ${instructions.join("\n\n")}`;
     if (Array.isArray(content)) {
       return content.map((part) => part.text || "").filter(Boolean).join("");
     }
-    if (_optionalChain([content, 'optionalAccess', _27 => _27.parts])) {
+    if (_optionalChain([content, 'optionalAccess', _28 => _28.parts])) {
       return content.parts.map((part) => part.text || "").filter(Boolean).join("");
     }
     return String(content || "");
@@ -956,6 +962,7 @@ init_logger();
 
 
 
+
 var _api = require('@opentelemetry/api');
 var _autoinstrumentationsnode = require('@opentelemetry/auto-instrumentations-node');
 var _exportertraceotlphttp = require('@opentelemetry/exporter-trace-otlp-http');
@@ -994,13 +1001,24 @@ var TelemetryService = (_class4 = class {
     this.sdk = new (0, _sdknode.NodeSDK)({
       resource,
       traceExporter,
-      instrumentations: [_autoinstrumentationsnode.getNodeAutoInstrumentations.call(void 0, )]
+      instrumentations: [
+        _autoinstrumentationsnode.getNodeAutoInstrumentations.call(void 0, {
+          // Follow Python ADK approach: let all HTTP instrumentation through.
+          // This provides transparency and aligns with standard OpenTelemetry behavior.
+          // High-level LLM tracing is provided through dedicated ADK spans.
+          "@opentelemetry/instrumentation-http": {
+            ignoreIncomingRequestHook: (req) => {
+              return true;
+            }
+          }
+        })
+      ]
     });
     try {
       this.sdk.start();
       this.isInitialized = true;
       this.tracer = _api.trace.getTracer("iqai-adk", config.appVersion || "0.1.0");
-      _api.diag.info("OpenTelemetry SDK started successfully.");
+      _api.diag.debug("OpenTelemetry SDK started successfully.");
     } catch (error) {
       _api.diag.error("Error starting OpenTelemetry SDK:", error);
       throw error;
@@ -1043,7 +1061,7 @@ var TelemetryService = (_class4 = class {
       });
       await Promise.race([this.sdk.shutdown(), timeoutPromise]);
       this.isInitialized = false;
-      _api.diag.info("Telemetry terminated successfully.");
+      _api.diag.debug("Telemetry terminated successfully.");
     } catch (error) {
       if (error instanceof Error && error.message.includes("timeout")) {
         _api.diag.warn("Telemetry shutdown timed out, some traces may be lost");
@@ -1063,7 +1081,7 @@ var TelemetryService = (_class4 = class {
     if (!span) return;
     let toolCallId = "<not specified>";
     let toolResponse = "<not specified>";
-    if (_optionalChain([functionResponseEvent, 'access', _28 => _28.content, 'optionalAccess', _29 => _29.parts]) && functionResponseEvent.content.parts.length > 0) {
+    if (_optionalChain([functionResponseEvent, 'access', _29 => _29.content, 'optionalAccess', _30 => _30.parts]) && functionResponseEvent.content.parts.length > 0) {
       const functionResponse = functionResponseEvent.content.parts[0].functionResponse;
       if (functionResponse) {
         toolCallId = functionResponse.id || "<not specified>";
@@ -1071,7 +1089,7 @@ var TelemetryService = (_class4 = class {
       }
     }
     span.setAttributes({
-      "gen_ai.system.name": "iqai-adk",
+      "gen_ai.system": "iqai-adk",
       "gen_ai.operation.name": "execute_tool",
       "gen_ai.tool.name": tool.name,
       "gen_ai.tool.description": tool.description,
@@ -1085,7 +1103,7 @@ var TelemetryService = (_class4 = class {
       ...process.env.NODE_ENV && {
         "deployment.environment.name": process.env.NODE_ENV
       },
-      // Tool-specific data
+      // ADK-specific attributes (matching Python namespace pattern)
       "adk.tool_call_args": this._safeJsonStringify(args),
       "adk.event_id": functionResponseEvent.invocationId,
       "adk.tool_response": this._safeJsonStringify(toolResponse),
@@ -1101,9 +1119,8 @@ var TelemetryService = (_class4 = class {
     if (!span) return;
     const requestData = this._buildLlmRequestForTrace(llmRequest);
     span.setAttributes({
-      // Standard OpenTelemetry attributes
-      "gen_ai.system.name": "iqai-adk",
-      "gen_ai.operation.name": "generate",
+      // Standard OpenTelemetry attributes (following Python pattern)
+      "gen_ai.system": "iqai-adk",
       "gen_ai.request.model": llmRequest.model,
       // Session and user tracking (maps to Langfuse sessionId, userId)
       "session.id": invocationContext.session.id,
@@ -1116,15 +1133,21 @@ var TelemetryService = (_class4 = class {
       "gen_ai.request.max_tokens": llmRequest.config.maxOutputTokens || 0,
       "gen_ai.request.temperature": llmRequest.config.temperature || 0,
       "gen_ai.request.top_p": llmRequest.config.topP || 0,
-      // Legacy ADK attributes (keep for backward compatibility)
       "adk.system_name": "iqai-adk",
       "adk.request_model": llmRequest.model,
-      "adk.invocation_id": invocationContext.session.id,
+      // ADK-specific attributes (matching Python namespace pattern)
+      "adk.invocation_id": invocationContext.invocationId,
       "adk.session_id": invocationContext.session.id,
       "adk.event_id": eventId,
       "adk.llm_request": this._safeJsonStringify(requestData),
       "adk.llm_response": this._safeJsonStringify(llmResponse)
     });
+    if (llmResponse.usageMetadata) {
+      span.setAttributes({
+        "gen_ai.usage.input_tokens": llmResponse.usageMetadata.promptTokenCount || 0,
+        "gen_ai.usage.output_tokens": llmResponse.usageMetadata.candidatesTokenCount || 0
+      });
+    }
     span.addEvent("gen_ai.content.prompt", {
       "gen_ai.prompt": this._safeJsonStringify(requestData.messages)
     });
@@ -1137,9 +1160,14 @@ var TelemetryService = (_class4 = class {
    */
   async *traceAsyncGenerator(spanName, generator) {
     const span = this.tracer.startSpan(spanName);
+    const spanContext = _api.trace.setSpan(_api.context.active(), span);
     try {
-      for await (const item of generator) {
-        yield item;
+      while (true) {
+        const result = await _api.context.with(spanContext, () => generator.next());
+        if (result.done) {
+          break;
+        }
+        yield result.value;
       }
     } catch (error) {
       span.recordException(error);
@@ -1171,7 +1199,7 @@ var TelemetryService = (_class4 = class {
       contents: []
     };
     for (const content of llmRequest.contents || []) {
-      const parts = _optionalChain([content, 'access', _30 => _30.parts, 'optionalAccess', _31 => _31.filter, 'call', _32 => _32((part) => !part.inlineData)]) || [];
+      const parts = _optionalChain([content, 'access', _31 => _31.parts, 'optionalAccess', _32 => _32.filter, 'call', _33 => _33((part) => !part.inlineData)]) || [];
       result.contents.push({
         role: content.role,
         parts
@@ -1226,7 +1254,7 @@ var traceLlmCall = (invocationContext, eventId, llmRequest, llmResponse) => tele
 // src/models/base-llm.ts
 var BaseLlm = (_class5 = class {
   /**
-   * The name of the LLM, e.g. gemini-1.5-flash or gemini-1.5-flash-001.
+   * The name of the LLM, e.g. gemini-2.5-flash or gemini-2.5-flash-001.
    */
   
   __init11() {this.logger = new Logger({ name: "BaseLlm" })}
@@ -1265,14 +1293,14 @@ var BaseLlm = (_class5 = class {
             "gen_ai.system.name": "iqai-adk",
             "gen_ai.operation.name": "generate",
             "gen_ai.request.model": this.model,
-            "gen_ai.request.max_tokens": _optionalChain([llmRequest, 'access', _33 => _33.config, 'optionalAccess', _34 => _34.maxOutputTokens]) || 0,
-            "gen_ai.request.temperature": _optionalChain([llmRequest, 'access', _35 => _35.config, 'optionalAccess', _36 => _36.temperature]) || 0,
-            "gen_ai.request.top_p": _optionalChain([llmRequest, 'access', _37 => _37.config, 'optionalAccess', _38 => _38.topP]) || 0,
+            "gen_ai.request.max_tokens": _optionalChain([llmRequest, 'access', _34 => _34.config, 'optionalAccess', _35 => _35.maxOutputTokens]) || 0,
+            "gen_ai.request.temperature": _optionalChain([llmRequest, 'access', _36 => _36.config, 'optionalAccess', _37 => _37.temperature]) || 0,
+            "gen_ai.request.top_p": _optionalChain([llmRequest, 'access', _38 => _38.config, 'optionalAccess', _39 => _39.topP]) || 0,
             "adk.llm_request": JSON.stringify({
               model: this.model,
-              contents: _optionalChain([llmRequest, 'access', _39 => _39.contents, 'optionalAccess', _40 => _40.map, 'call', _41 => _41((content) => ({
+              contents: _optionalChain([llmRequest, 'access', _40 => _40.contents, 'optionalAccess', _41 => _41.map, 'call', _42 => _42((content) => ({
                 role: content.role,
-                parts: _optionalChain([content, 'access', _42 => _42.parts, 'optionalAccess', _43 => _43.map, 'call', _44 => _44((part) => ({
+                parts: _optionalChain([content, 'access', _43 => _43.parts, 'optionalAccess', _44 => _44.map, 'call', _45 => _45((part) => ({
                   text: typeof part.text === "string" ? part.text.substring(0, 200) + (part.text.length > 200 ? "..." : "") : "[non_text_content]"
                 }))])
               }))]),
@@ -1396,9 +1424,9 @@ var AiSdkLlm = (_class6 = class extends BaseLlm {
         messages,
         system: systemMessage,
         tools: Object.keys(tools).length > 0 ? tools : void 0,
-        maxTokens: _optionalChain([request, 'access', _45 => _45.config, 'optionalAccess', _46 => _46.maxOutputTokens]),
-        temperature: _optionalChain([request, 'access', _47 => _47.config, 'optionalAccess', _48 => _48.temperature]),
-        topP: _optionalChain([request, 'access', _49 => _49.config, 'optionalAccess', _50 => _50.topP])
+        maxTokens: _optionalChain([request, 'access', _46 => _46.config, 'optionalAccess', _47 => _47.maxOutputTokens]),
+        temperature: _optionalChain([request, 'access', _48 => _48.config, 'optionalAccess', _49 => _49.temperature]),
+        topP: _optionalChain([request, 'access', _50 => _50.config, 'optionalAccess', _51 => _51.topP])
       };
       if (stream) {
         const result = _ai.streamText.call(void 0, requestParams);
@@ -1538,7 +1566,7 @@ var AiSdkLlm = (_class6 = class extends BaseLlm {
    */
   convertToAiSdkTools(llmRequest) {
     const tools = {};
-    if (_optionalChain([llmRequest, 'access', _51 => _51.config, 'optionalAccess', _52 => _52.tools])) {
+    if (_optionalChain([llmRequest, 'access', _52 => _52.config, 'optionalAccess', _53 => _53.tools])) {
       for (const toolConfig of llmRequest.config.tools) {
         if ("functionDeclarations" in toolConfig) {
           for (const funcDecl of toolConfig.functionDeclarations) {
@@ -1572,7 +1600,7 @@ var AiSdkLlm = (_class6 = class extends BaseLlm {
       }
       return { role: "user", content: textContent };
     }
-    if (_optionalChain([content, 'access', _53 => _53.parts, 'optionalAccess', _54 => _54.some, 'call', _55 => _55((part) => part.functionCall)])) {
+    if (_optionalChain([content, 'access', _54 => _54.parts, 'optionalAccess', _55 => _55.some, 'call', _56 => _56((part) => part.functionCall)])) {
       const textParts = content.parts.filter((part) => part.text);
       const functionCalls = content.parts.filter((part) => part.functionCall);
       const contentParts2 = [];
@@ -1599,7 +1627,7 @@ var AiSdkLlm = (_class6 = class extends BaseLlm {
         content: contentParts2
       };
     }
-    if (_optionalChain([content, 'access', _56 => _56.parts, 'optionalAccess', _57 => _57.some, 'call', _58 => _58((part) => part.functionResponse)])) {
+    if (_optionalChain([content, 'access', _57 => _57.parts, 'optionalAccess', _58 => _58.some, 'call', _59 => _59((part) => part.functionResponse)])) {
       const functionResponses = content.parts.filter(
         (part) => part.functionResponse
       );
@@ -1704,7 +1732,7 @@ var AnthropicLlm = (_class7 = class extends BaseLlm {
       (content) => this.contentToAnthropicMessage(content)
     );
     let tools;
-    if (_optionalChain([llmRequest, 'access', _59 => _59.config, 'optionalAccess', _60 => _60.tools, 'optionalAccess', _61 => _61[0], 'optionalAccess', _62 => _62.functionDeclarations])) {
+    if (_optionalChain([llmRequest, 'access', _60 => _60.config, 'optionalAccess', _61 => _61.tools, 'optionalAccess', _62 => _62[0], 'optionalAccess', _63 => _63.functionDeclarations])) {
       tools = llmRequest.config.tools[0].functionDeclarations.map(
         (decl) => this.functionDeclarationToAnthropicTool(decl)
       );
@@ -1726,9 +1754,9 @@ var AnthropicLlm = (_class7 = class extends BaseLlm {
       messages: anthropicMessages,
       tools,
       tool_choice: tools ? { type: "auto" } : void 0,
-      max_tokens: _optionalChain([llmRequest, 'access', _63 => _63.config, 'optionalAccess', _64 => _64.maxOutputTokens]) || MAX_TOKENS,
-      temperature: _optionalChain([llmRequest, 'access', _65 => _65.config, 'optionalAccess', _66 => _66.temperature]),
-      top_p: _optionalChain([llmRequest, 'access', _67 => _67.config, 'optionalAccess', _68 => _68.topP])
+      max_tokens: _optionalChain([llmRequest, 'access', _64 => _64.config, 'optionalAccess', _65 => _65.maxOutputTokens]) || MAX_TOKENS,
+      temperature: _optionalChain([llmRequest, 'access', _66 => _66.config, 'optionalAccess', _67 => _67.temperature]),
+      top_p: _optionalChain([llmRequest, 'access', _68 => _68.config, 'optionalAccess', _69 => _69.topP])
     });
     yield this.anthropicMessageToLlmResponse(message);
   }
@@ -1789,7 +1817,7 @@ var AnthropicLlm = (_class7 = class extends BaseLlm {
     }
     if (part.function_response) {
       let content = "";
-      if (_optionalChain([part, 'access', _69 => _69.function_response, 'access', _70 => _70.response, 'optionalAccess', _71 => _71.result])) {
+      if (_optionalChain([part, 'access', _70 => _70.function_response, 'access', _71 => _71.response, 'optionalAccess', _72 => _72.result])) {
         content = String(part.function_response.response.result);
       }
       return {
@@ -1824,7 +1852,7 @@ var AnthropicLlm = (_class7 = class extends BaseLlm {
    */
   functionDeclarationToAnthropicTool(functionDeclaration) {
     const properties = {};
-    if (_optionalChain([functionDeclaration, 'access', _72 => _72.parameters, 'optionalAccess', _73 => _73.properties])) {
+    if (_optionalChain([functionDeclaration, 'access', _73 => _73.parameters, 'optionalAccess', _74 => _74.properties])) {
       for (const [key, value] of Object.entries(
         functionDeclaration.parameters.properties
       )) {
@@ -1915,7 +1943,7 @@ var GoogleLlm = class extends BaseLlm {
   /**
    * Constructor for Gemini
    */
-  constructor(model = "gemini-1.5-flash") {
+  constructor(model = "gemini-2.5-flash") {
     super(model);
   }
   /**
@@ -1952,7 +1980,7 @@ var GoogleLlm = class extends BaseLlm {
         response = resp;
         const llmResponse = LlmResponse.create(resp);
         usageMetadata = llmResponse.usageMetadata;
-        if (_optionalChain([llmResponse, 'access', _74 => _74.content, 'optionalAccess', _75 => _75.parts, 'optionalAccess', _76 => _76[0], 'optionalAccess', _77 => _77.text])) {
+        if (_optionalChain([llmResponse, 'access', _75 => _75.content, 'optionalAccess', _76 => _76.parts, 'optionalAccess', _77 => _77[0], 'optionalAccess', _78 => _78.text])) {
           const part0 = llmResponse.content.parts[0];
           if (part0.thought) {
             thoughtText += part0.text;
@@ -1980,7 +2008,7 @@ var GoogleLlm = class extends BaseLlm {
         }
         yield llmResponse;
       }
-      if ((text || thoughtText) && response && response.candidates && _optionalChain([response, 'access', _78 => _78.candidates, 'access', _79 => _79[0], 'optionalAccess', _80 => _80.finishReason]) === _genai.FinishReason.STOP) {
+      if ((text || thoughtText) && response && response.candidates && _optionalChain([response, 'access', _79 => _79.candidates, 'access', _80 => _80[0], 'optionalAccess', _81 => _81.finishReason]) === _genai.FinishReason.STOP) {
         const parts = [];
         if (thoughtText) {
           parts.push({ text: thoughtText, thought: true });
@@ -2004,7 +2032,7 @@ var GoogleLlm = class extends BaseLlm {
       });
       const llmResponse = LlmResponse.create(response);
       this.logger.debug(
-        `Google response: ${_optionalChain([llmResponse, 'access', _81 => _81.usageMetadata, 'optionalAccess', _82 => _82.candidatesTokenCount]) || 0} tokens`
+        `Google response: ${_optionalChain([llmResponse, 'access', _82 => _82.usageMetadata, 'optionalAccess', _83 => _83.candidatesTokenCount]) || 0} tokens`
       );
       yield llmResponse;
     }
@@ -2019,8 +2047,8 @@ var GoogleLlm = class extends BaseLlm {
    * Check if response has inline data
    */
   hasInlineData(response) {
-    const parts = _optionalChain([response, 'access', _83 => _83.candidates, 'optionalAccess', _84 => _84[0], 'optionalAccess', _85 => _85.content, 'optionalAccess', _86 => _86.parts]);
-    return _optionalChain([parts, 'optionalAccess', _87 => _87.some, 'call', _88 => _88((part) => _optionalChain([part, 'optionalAccess', _89 => _89.inlineData]))]) || false;
+    const parts = _optionalChain([response, 'access', _84 => _84.candidates, 'optionalAccess', _85 => _85[0], 'optionalAccess', _86 => _86.content, 'optionalAccess', _87 => _87.parts]);
+    return _optionalChain([parts, 'optionalAccess', _88 => _88.some, 'call', _89 => _89((part) => _optionalChain([part, 'optionalAccess', _90 => _90.inlineData]))]) || false;
   }
   /**
    * Convert LlmRequest contents to GoogleGenAI format
@@ -2054,7 +2082,7 @@ var GoogleLlm = class extends BaseLlm {
    * Sets display_name to null for the Gemini API (non-Vertex) backend.
    */
   removeDisplayNameIfPresent(dataObj) {
-    if (_optionalChain([dataObj, 'optionalAccess', _90 => _90.displayName])) {
+    if (_optionalChain([dataObj, 'optionalAccess', _91 => _91.displayName])) {
       dataObj.displayName = null;
     }
   }
@@ -2063,7 +2091,7 @@ var GoogleLlm = class extends BaseLlm {
    */
   buildFunctionDeclarationLog(funcDecl) {
     let paramStr = "{}";
-    if (_optionalChain([funcDecl, 'access', _91 => _91.parameters, 'optionalAccess', _92 => _92.properties])) {
+    if (_optionalChain([funcDecl, 'access', _92 => _92.parameters, 'optionalAccess', _93 => _93.properties])) {
       paramStr = JSON.stringify(funcDecl.parameters.properties);
     }
     return `${funcDecl.name}: ${paramStr}`;
@@ -2184,7 +2212,7 @@ var OpenAiLlm = class extends BaseLlm {
       (content) => this.contentToOpenAiMessage(content)
     );
     let tools;
-    if (_optionalChain([llmRequest, 'access', _93 => _93.config, 'optionalAccess', _94 => _94.tools, 'optionalAccess', _95 => _95[0], 'optionalAccess', _96 => _96.functionDeclarations])) {
+    if (_optionalChain([llmRequest, 'access', _94 => _94.config, 'optionalAccess', _95 => _95.tools, 'optionalAccess', _96 => _96[0], 'optionalAccess', _97 => _97.functionDeclarations])) {
       tools = llmRequest.config.tools[0].functionDeclarations.map(
         (funcDecl) => this.functionDeclarationToOpenAiTool(funcDecl)
       );
@@ -2202,9 +2230,9 @@ var OpenAiLlm = class extends BaseLlm {
       messages: openAiMessages,
       tools,
       tool_choice: tools ? "auto" : void 0,
-      max_tokens: _optionalChain([llmRequest, 'access', _97 => _97.config, 'optionalAccess', _98 => _98.maxOutputTokens]),
-      temperature: _optionalChain([llmRequest, 'access', _99 => _99.config, 'optionalAccess', _100 => _100.temperature]),
-      top_p: _optionalChain([llmRequest, 'access', _101 => _101.config, 'optionalAccess', _102 => _102.topP]),
+      max_tokens: _optionalChain([llmRequest, 'access', _98 => _98.config, 'optionalAccess', _99 => _99.maxOutputTokens]),
+      temperature: _optionalChain([llmRequest, 'access', _100 => _100.config, 'optionalAccess', _101 => _101.temperature]),
+      top_p: _optionalChain([llmRequest, 'access', _102 => _102.config, 'optionalAccess', _103 => _103.topP]),
       stream
     };
     if (stream) {
@@ -2224,7 +2252,7 @@ var OpenAiLlm = class extends BaseLlm {
         if (chunk.usage) {
           usageMetadata = chunk.usage;
         }
-        if (_optionalChain([llmResponse, 'access', _103 => _103.content, 'optionalAccess', _104 => _104.parts, 'optionalAccess', _105 => _105[0], 'optionalAccess', _106 => _106.text])) {
+        if (_optionalChain([llmResponse, 'access', _104 => _104.content, 'optionalAccess', _105 => _105.parts, 'optionalAccess', _106 => _106[0], 'optionalAccess', _107 => _107.text])) {
           const part0 = llmResponse.content.parts[0];
           if (part0.thought) {
             thoughtText += part0.text;
@@ -2265,10 +2293,10 @@ var OpenAiLlm = class extends BaseLlm {
                 function: { name: "", arguments: "" }
               };
             }
-            if (_optionalChain([toolCall, 'access', _107 => _107.function, 'optionalAccess', _108 => _108.name])) {
+            if (_optionalChain([toolCall, 'access', _108 => _108.function, 'optionalAccess', _109 => _109.name])) {
               accumulatedToolCalls[index].function.name += toolCall.function.name;
             }
-            if (_optionalChain([toolCall, 'access', _109 => _109.function, 'optionalAccess', _110 => _110.arguments])) {
+            if (_optionalChain([toolCall, 'access', _110 => _110.function, 'optionalAccess', _111 => _111.arguments])) {
               accumulatedToolCalls[index].function.arguments += toolCall.function.arguments;
             }
           }
@@ -2283,7 +2311,7 @@ var OpenAiLlm = class extends BaseLlm {
           }
           if (accumulatedToolCalls.length > 0) {
             for (const toolCall of accumulatedToolCalls) {
-              if (_optionalChain([toolCall, 'access', _111 => _111.function, 'optionalAccess', _112 => _112.name])) {
+              if (_optionalChain([toolCall, 'access', _112 => _112.function, 'optionalAccess', _113 => _113.name])) {
                 parts.push({
                   functionCall: {
                     id: toolCall.id,
@@ -2343,7 +2371,7 @@ var OpenAiLlm = class extends BaseLlm {
           response.usage
         );
         this.logger.debug(
-          `OpenAI response: ${_optionalChain([response, 'access', _113 => _113.usage, 'optionalAccess', _114 => _114.completion_tokens]) || 0} tokens`
+          `OpenAI response: ${_optionalChain([response, 'access', _114 => _114.usage, 'optionalAccess', _115 => _115.completion_tokens]) || 0} tokens`
         );
         yield llmResponse;
       }
@@ -2370,7 +2398,7 @@ var OpenAiLlm = class extends BaseLlm {
     }
     if (delta.tool_calls) {
       for (const toolCall of delta.tool_calls) {
-        if (toolCall.type === "function" && _optionalChain([toolCall, 'access', _115 => _115.function, 'optionalAccess', _116 => _116.name])) {
+        if (toolCall.type === "function" && _optionalChain([toolCall, 'access', _116 => _116.function, 'optionalAccess', _117 => _117.name])) {
           parts.push({
             functionCall: {
               id: toolCall.id || "",
@@ -2436,10 +2464,10 @@ var OpenAiLlm = class extends BaseLlm {
     if (role === "system") {
       return {
         role: "system",
-        content: _optionalChain([content, 'access', _117 => _117.parts, 'optionalAccess', _118 => _118[0], 'optionalAccess', _119 => _119.text]) || ""
+        content: _optionalChain([content, 'access', _118 => _118.parts, 'optionalAccess', _119 => _119[0], 'optionalAccess', _120 => _120.text]) || ""
       };
     }
-    if (_optionalChain([content, 'access', _120 => _120.parts, 'optionalAccess', _121 => _121.some, 'call', _122 => _122((part) => part.functionCall)])) {
+    if (_optionalChain([content, 'access', _121 => _121.parts, 'optionalAccess', _122 => _122.some, 'call', _123 => _123((part) => part.functionCall)])) {
       const functionCallPart = content.parts.find(
         (part) => part.functionCall
       );
@@ -2459,7 +2487,7 @@ var OpenAiLlm = class extends BaseLlm {
         ]
       };
     }
-    if (_optionalChain([content, 'access', _123 => _123.parts, 'optionalAccess', _124 => _124.some, 'call', _125 => _125((part) => part.functionResponse)])) {
+    if (_optionalChain([content, 'access', _124 => _124.parts, 'optionalAccess', _125 => _125.some, 'call', _126 => _126((part) => part.functionResponse)])) {
       const functionResponsePart = content.parts.find(
         (part) => part.functionResponse
       );
@@ -2471,7 +2499,7 @@ var OpenAiLlm = class extends BaseLlm {
         )
       };
     }
-    if (_optionalChain([content, 'access', _126 => _126.parts, 'optionalAccess', _127 => _127.length]) === 1 && content.parts[0].text) {
+    if (_optionalChain([content, 'access', _127 => _127.parts, 'optionalAccess', _128 => _128.length]) === 1 && content.parts[0].text) {
       return {
         role,
         content: content.parts[0].text
@@ -2494,7 +2522,7 @@ var OpenAiLlm = class extends BaseLlm {
         text: part.text
       };
     }
-    if (_optionalChain([part, 'access', _128 => _128.inline_data, 'optionalAccess', _129 => _129.mime_type]) && _optionalChain([part, 'access', _130 => _130.inline_data, 'optionalAccess', _131 => _131.data])) {
+    if (_optionalChain([part, 'access', _129 => _129.inline_data, 'optionalAccess', _130 => _130.mime_type]) && _optionalChain([part, 'access', _131 => _131.inline_data, 'optionalAccess', _132 => _132.data])) {
       return {
         type: "image_url",
         image_url: {
@@ -2622,8 +2650,8 @@ var OpenAiLlm = class extends BaseLlm {
    * Check if response has inline data (similar to Google LLM)
    */
   hasInlineData(response) {
-    const parts = _optionalChain([response, 'access', _132 => _132.content, 'optionalAccess', _133 => _133.parts]);
-    return _optionalChain([parts, 'optionalAccess', _134 => _134.some, 'call', _135 => _135((part) => part.inlineData)]) || false;
+    const parts = _optionalChain([response, 'access', _133 => _133.content, 'optionalAccess', _134 => _134.parts]);
+    return _optionalChain([parts, 'optionalAccess', _135 => _135.some, 'call', _136 => _136((part) => part.inlineData)]) || false;
   }
   /**
    * Gets the OpenAI client
@@ -2925,7 +2953,7 @@ var OAuth2Credential = class extends AuthCredential {
         "Cannot refresh token: no refresh token or refresh function"
       );
     }
-    const result = await _optionalChain([this, 'access', _136 => _136.refreshFunction, 'optionalCall', _137 => _137(this.refreshToken)]);
+    const result = await _optionalChain([this, 'access', _137 => _137.refreshFunction, 'optionalCall', _138 => _138(this.refreshToken)]);
     if (!result) {
       throw new Error("Failed to refresh token");
     }
@@ -2960,7 +2988,7 @@ var AuthHandler = class {
    * Gets the authentication token
    */
   getToken() {
-    return _optionalChain([this, 'access', _138 => _138.credential, 'optionalAccess', _139 => _139.getToken, 'call', _140 => _140()]);
+    return _optionalChain([this, 'access', _139 => _139.credential, 'optionalAccess', _140 => _140.getToken, 'call', _141 => _141()]);
   }
   /**
    * Gets headers for HTTP requests
@@ -2975,7 +3003,7 @@ var AuthHandler = class {
    * Refreshes the token if necessary
    */
   async refreshToken() {
-    if (_optionalChain([this, 'access', _141 => _141.credential, 'optionalAccess', _142 => _142.canRefresh, 'call', _143 => _143()])) {
+    if (_optionalChain([this, 'access', _142 => _142.credential, 'optionalAccess', _143 => _143.canRefresh, 'call', _144 => _144()])) {
       await this.credential.refresh();
     }
   }
@@ -3954,10 +3982,10 @@ var CreatedTool = class extends BaseTool {
   /**
    * Executes the tool function with validation
    */
-  async runAsync(args, context) {
+  async runAsync(args, context4) {
     try {
       const validatedArgs = this.schema.parse(args);
-      const result = await Promise.resolve(this.func(validatedArgs, context));
+      const result = await Promise.resolve(this.func(validatedArgs, context4));
       return _nullishCoalesce(result, () => ( {}));
     } catch (error) {
       if (error instanceof z.ZodError) {
@@ -4215,7 +4243,7 @@ var AgentTool = (_class15 = class extends BaseTool {
   /**
    * Execute the tool by running the agent with the provided input
    */
-  async runAsync(params, context) {
+  async runAsync(params, context4) {
     try {
       const input = params.input || Object.values(params)[0];
       if (!isLlmAgent(this.agent)) {
@@ -4223,7 +4251,7 @@ var AgentTool = (_class15 = class extends BaseTool {
           `Agent ${this.name} does not support running as a tool`
         );
       }
-      const parentInvocation = context._invocationContext;
+      const parentInvocation = context4._invocationContext;
       const childInvocationContext = new InvocationContext({
         invocationId: _uuid.v4.call(void 0, ),
         agent: this.agent,
@@ -4260,8 +4288,8 @@ var AgentTool = (_class15 = class extends BaseTool {
       } catch (e2) {
         toolResult = mergedText;
       }
-      if (this.outputKey && _optionalChain([context, 'optionalAccess', _144 => _144.state])) {
-        context.state[this.outputKey] = toolResult;
+      if (this.outputKey && _optionalChain([context4, 'optionalAccess', _145 => _145.state])) {
+        context4.state[this.outputKey] = toolResult;
       }
       return toolResult;
     } catch (error) {
@@ -4527,7 +4555,7 @@ var FileOperationsTool = class extends BaseTool {
       name: "file_operations",
       description: "Perform file system operations like reading, writing, and managing files"
     });
-    this.basePath = _optionalChain([options, 'optionalAccess', _145 => _145.basePath]) || process.cwd();
+    this.basePath = _optionalChain([options, 'optionalAccess', _146 => _146.basePath]) || process.cwd();
   }
   /**
    * Get the function declaration for the tool
@@ -4809,9 +4837,9 @@ var UserInteractionTool = class extends BaseTool {
   /**
    * Execute the user interaction
    */
-  async runAsync(args, context) {
+  async runAsync(args, context4) {
     try {
-      const actions = context.actions;
+      const actions = context4.actions;
       if (!actions || !actions.promptUser) {
         return {
           success: false,
@@ -4859,9 +4887,9 @@ var ExitLoopTool = (_class17 = class extends BaseTool {
   /**
    * Execute the exit loop action
    */
-  async runAsync(_args, context) {
+  async runAsync(_args, context4) {
     this.logger.debug("Executing exit loop tool");
-    context.actions.escalate = true;
+    context4.actions.escalate = true;
   }
 }, _class17);
 
@@ -4912,14 +4940,14 @@ var GetUserChoiceTool = (_class18 = class extends BaseTool {
    * This is a long running operation that will return null initially
    * and the actual choice will be provided asynchronously
    */
-  async runAsync(args, context) {
+  async runAsync(args, context4) {
     this.logger.debug(
       `Executing get_user_choice with options: ${args.options.join(", ")}`
     );
     if (args.question) {
       this.logger.debug(`Question: ${args.question}`);
     }
-    context.actions.skipSummarization = true;
+    context4.actions.skipSummarization = true;
     return null;
   }
 }, _class18);
@@ -4961,9 +4989,9 @@ var TransferToAgentTool = (_class19 = class extends BaseTool {
   /**
    * Execute the transfer to agent action
    */
-  async runAsync(args, context) {
+  async runAsync(args, context4) {
     this.logger.debug(`Executing transfer to agent: ${args.agent_name}`);
-    context.actions.transferToAgent = args.agent_name;
+    context4.actions.transferToAgent = args.agent_name;
   }
 }, _class19);
 
@@ -5004,13 +5032,13 @@ var LoadMemoryTool = (_class20 = class extends BaseTool {
   /**
    * Execute the memory loading action
    */
-  async runAsync(args, context) {
+  async runAsync(args, context4) {
     this.logger.debug(`Executing load_memory with query: ${args.query}`);
     try {
-      const searchResult = await context.searchMemory(args.query);
+      const searchResult = await context4.searchMemory(args.query);
       return {
         memories: searchResult.memories || [],
-        count: _optionalChain([searchResult, 'access', _146 => _146.memories, 'optionalAccess', _147 => _147.length]) || 0
+        count: _optionalChain([searchResult, 'access', _147 => _147.memories, 'optionalAccess', _148 => _148.length]) || 0
       };
     } catch (error) {
       console.error("Error searching memory:", error);
@@ -5057,7 +5085,7 @@ var LoadArtifactsTool = class extends BaseTool {
   /**
    * Execute the load artifacts operation
    */
-  async runAsync(args, context) {
+  async runAsync(args, context4) {
     const artifactNames = args.artifact_names || [];
     return { artifact_names: artifactNames };
   }
@@ -5567,7 +5595,7 @@ var McpClientService = (_class22 = class {
         },
         this,
         async (instance) => await instance.reinitialize(),
-        _optionalChain([this, 'access', _148 => _148.config, 'access', _149 => _149.retryOptions, 'optionalAccess', _150 => _150.maxRetries]) || 2
+        _optionalChain([this, 'access', _149 => _149.config, 'access', _150 => _150.retryOptions, 'optionalAccess', _151 => _151.maxRetries]) || 2
       );
       return await wrappedCall();
     } catch (error) {
@@ -5651,7 +5679,7 @@ var McpClientService = (_class22 = class {
     this.mcpSamplingHandler = null;
     if (this.client) {
       try {
-        _optionalChain([this, 'access', _151 => _151.client, 'access', _152 => _152.removeRequestHandler, 'optionalCall', _153 => _153("sampling/createMessage")]);
+        _optionalChain([this, 'access', _152 => _152.client, 'access', _153 => _153.removeRequestHandler, 'optionalCall', _154 => _154("sampling/createMessage")]);
       } catch (error) {
         this.logger.error("Failed to remove sampling handler:", error);
       }
@@ -6088,12 +6116,12 @@ var McpToolset = (_class24 = class {
    * Checks if a tool should be included based on the tool filter.
    * Similar to Python's _is_selected method.
    */
-  isSelected(tool, context) {
+  isSelected(tool, context4) {
     if (!this.toolFilter) {
       return true;
     }
     if (typeof this.toolFilter === "function") {
-      return this.toolFilter(tool, context);
+      return this.toolFilter(tool, context4);
     }
     if (Array.isArray(this.toolFilter)) {
       return this.toolFilter.includes(tool.name);
@@ -6146,7 +6174,7 @@ var McpToolset = (_class24 = class {
    * Retrieves tools from the MCP server and converts them to BaseTool instances.
    * Similar to Python's get_tools method.
    */
-  async getTools(context) {
+  async getTools(context4) {
     try {
       if (this.isClosing) {
         throw new McpError(
@@ -6154,7 +6182,7 @@ var McpToolset = (_class24 = class {
           "resource_closed_error" /* RESOURCE_CLOSED_ERROR */
         );
       }
-      if (this.tools.length > 0 && !_optionalChain([this, 'access', _154 => _154.config, 'access', _155 => _155.cacheConfig, 'optionalAccess', _156 => _156.enabled]) === false) {
+      if (this.tools.length > 0 && !_optionalChain([this, 'access', _155 => _155.config, 'access', _156 => _156.cacheConfig, 'optionalAccess', _157 => _157.enabled]) === false) {
         return this.tools;
       }
       if (!this.clientService) {
@@ -6168,7 +6196,7 @@ var McpToolset = (_class24 = class {
       }
       const tools = [];
       for (const mcpTool of toolsResponse.tools) {
-        if (this.isSelected(mcpTool, context)) {
+        if (this.isSelected(mcpTool, context4)) {
           try {
             const tool = await createTool2(mcpTool, client);
             tools.push(tool);
@@ -6180,7 +6208,7 @@ var McpToolset = (_class24 = class {
           }
         }
       }
-      if (_optionalChain([this, 'access', _157 => _157.config, 'access', _158 => _158.cacheConfig, 'optionalAccess', _159 => _159.enabled]) !== false) {
+      if (_optionalChain([this, 'access', _158 => _158.config, 'access', _159 => _159.cacheConfig, 'optionalAccess', _160 => _160.enabled]) !== false) {
         this.tools = tools;
       }
       return tools;
@@ -6205,9 +6233,9 @@ var McpToolset = (_class24 = class {
   /**
    * Refreshes the tool cache by clearing it and fetching tools again
    */
-  async refreshTools(context) {
+  async refreshTools(context4) {
     this.tools = [];
-    return this.getTools(context);
+    return this.getTools(context4);
   }
   /**
    * Closes the connection to the MCP server.
@@ -6251,6 +6279,7 @@ async function getMcpTools(config, toolFilter) {
 }
 
 // src/flows/llm-flows/functions.ts
+
 var AF_FUNCTION_CALL_ID_PREFIX = "adk-";
 var REQUEST_EUC_FUNCTION_CALL_NAME = "adk_request_credential";
 function generateClientFunctionCallId() {
@@ -6268,12 +6297,12 @@ function populateClientFunctionCallId(modelResponseEvent) {
   }
 }
 function removeClientFunctionCallId(content) {
-  if (_optionalChain([content, 'optionalAccess', _160 => _160.parts])) {
+  if (_optionalChain([content, 'optionalAccess', _161 => _161.parts])) {
     for (const part of content.parts) {
-      if (_optionalChain([part, 'access', _161 => _161.functionCall, 'optionalAccess', _162 => _162.id, 'optionalAccess', _163 => _163.startsWith, 'call', _164 => _164(AF_FUNCTION_CALL_ID_PREFIX)])) {
+      if (_optionalChain([part, 'access', _162 => _162.functionCall, 'optionalAccess', _163 => _163.id, 'optionalAccess', _164 => _164.startsWith, 'call', _165 => _165(AF_FUNCTION_CALL_ID_PREFIX)])) {
         part.functionCall.id = void 0;
       }
-      if (_optionalChain([part, 'access', _165 => _165.functionResponse, 'optionalAccess', _166 => _166.id, 'optionalAccess', _167 => _167.startsWith, 'call', _168 => _168(AF_FUNCTION_CALL_ID_PREFIX)])) {
+      if (_optionalChain([part, 'access', _166 => _166.functionResponse, 'optionalAccess', _167 => _167.id, 'optionalAccess', _168 => _168.startsWith, 'call', _169 => _169(AF_FUNCTION_CALL_ID_PREFIX)])) {
         part.functionResponse.id = void 0;
       }
     }
@@ -6340,23 +6369,40 @@ async function handleFunctionCallsAsync(invocationContext, functionCallEvent, to
       toolsDict
     );
     const functionArgs = functionCall.args || {};
-    const functionResponse = await callToolAsync(
-      tool,
-      functionArgs,
-      toolContext
-    );
-    if (tool.isLongRunning) {
+    const tracer2 = telemetryService.getTracer();
+    const span = tracer2.startSpan(`execute_tool ${tool.name}`);
+    const spanContext = _api.trace.setSpan(_api.context.active(), span);
+    try {
+      const functionResponse = await _api.context.with(spanContext, async () => {
+        const result = await callToolAsync(tool, functionArgs, toolContext);
+        if (tool.isLongRunning && !result) {
+          return null;
+        }
+        const functionResponseEvent = buildResponseEvent(
+          tool,
+          result,
+          toolContext,
+          invocationContext
+        );
+        telemetryService.traceToolCall(
+          tool,
+          functionArgs,
+          functionResponseEvent
+        );
+        return { result, event: functionResponseEvent };
+      });
       if (!functionResponse) {
         continue;
       }
+      functionResponseEvents.push(functionResponse.event);
+      span.setStatus({ code: 1 });
+    } catch (error) {
+      span.recordException(error);
+      span.setStatus({ code: 2, message: error.message });
+      throw error;
+    } finally {
+      span.end();
     }
-    const functionResponseEvent = buildResponseEvent(
-      tool,
-      functionResponse,
-      toolContext,
-      invocationContext
-    );
-    functionResponseEvents.push(functionResponseEvent);
   }
   if (!functionResponseEvents.length) {
     return null;
@@ -6418,7 +6464,7 @@ function mergeParallelFunctionResponseEvents(functionResponseEvents) {
   }
   const mergedParts = [];
   for (const event of functionResponseEvents) {
-    if (_optionalChain([event, 'access', _169 => _169.content, 'optionalAccess', _170 => _170.parts])) {
+    if (_optionalChain([event, 'access', _170 => _170.content, 'optionalAccess', _171 => _171.parts])) {
       for (const part of event.content.parts) {
         mergedParts.push(part);
       }
@@ -6456,7 +6502,7 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
   __init44() {this.responseProcessors = []}
   __init45() {this.logger = new Logger({ name: "BaseLlmFlow" })}
   async *runAsync(invocationContext) {
-    this.logger.info(`Agent '${invocationContext.agent.name}' started.`);
+    this.logger.debug(`Agent '${invocationContext.agent.name}' started.`);
     let stepCount = 0;
     while (true) {
       stepCount++;
@@ -6466,7 +6512,7 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
         yield event;
       }
       if (!lastEvent || lastEvent.isFinalResponse()) {
-        this.logger.info(
+        this.logger.debug(
           `Agent '${invocationContext.agent.name}' finished after ${stepCount} steps.`
         );
         break;
@@ -6496,7 +6542,7 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
       yield event;
     }
     if (invocationContext.endInvocation) {
-      this.logger.info("Invocation ended during preprocessing.");
+      this.logger.debug("Invocation ended during preprocessing.");
       return;
     }
     const modelResponseEvent = new Event({
@@ -6546,7 +6592,7 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
     if (tools.length > 0) {
       const toolsData = tools.map((tool) => ({
         Name: tool.name,
-        Description: _optionalChain([tool, 'access', _171 => _171.description, 'optionalAccess', _172 => _172.substring, 'call', _173 => _173(0, 50)]) + (_optionalChain([tool, 'access', _174 => _174.description, 'optionalAccess', _175 => _175.length]) > 50 ? "..." : ""),
+        Description: _optionalChain([tool, 'access', _172 => _172.description, 'optionalAccess', _173 => _173.substring, 'call', _174 => _174(0, 50)]) + (_optionalChain([tool, 'access', _175 => _175.description, 'optionalAccess', _176 => _176.length]) > 50 ? "..." : ""),
         "Long Running": tool.isLongRunning ? "Yes" : "No"
       }));
       this.logger.debugArray("\u{1F6E0}\uFE0F Available Tools", toolsData);
@@ -6609,14 +6655,14 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
       );
       if (functionResponseEvent) {
         yield functionResponseEvent;
-        const transferToAgent = _optionalChain([functionResponseEvent, 'access', _176 => _176.actions, 'optionalAccess', _177 => _177.transferToAgent]);
+        const transferToAgent = _optionalChain([functionResponseEvent, 'access', _177 => _177.actions, 'optionalAccess', _178 => _178.transferToAgent]);
         if (transferToAgent) {
-          this.logger.info(`\u{1F504} Live transfer to agent '${transferToAgent}'`);
+          this.logger.debug(`\u{1F504} Live transfer to agent '${transferToAgent}'`);
           const agentToRun = this._getAgentToRun(
             invocationContext,
             transferToAgent
           );
-          for await (const event of _optionalChain([agentToRun, 'access', _178 => _178.runLive, 'optionalCall', _179 => _179(invocationContext)]) || agentToRun.runAsync(invocationContext)) {
+          for await (const event of _optionalChain([agentToRun, 'access', _179 => _179.runLive, 'optionalCall', _180 => _180(invocationContext)]) || agentToRun.runAsync(invocationContext)) {
             yield event;
           }
         }
@@ -6648,9 +6694,9 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
         yield authEvent;
       }
       yield functionResponseEvent;
-      const transferToAgent = _optionalChain([functionResponseEvent, 'access', _180 => _180.actions, 'optionalAccess', _181 => _181.transferToAgent]);
+      const transferToAgent = _optionalChain([functionResponseEvent, 'access', _181 => _181.actions, 'optionalAccess', _182 => _182.transferToAgent]);
       if (transferToAgent) {
-        this.logger.info(`\u{1F504} Transferring to agent '${transferToAgent}'`);
+        this.logger.debug(`\u{1F504} Transferring to agent '${transferToAgent}'`);
         const agentToRun = this._getAgentToRun(
           invocationContext,
           transferToAgent
@@ -6694,27 +6740,27 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
     }
     invocationContext.incrementLlmCallCount();
     const isStreaming = invocationContext.runConfig.streamingMode === "sse" /* SSE */;
-    const tools = _optionalChain([llmRequest, 'access', _182 => _182.config, 'optionalAccess', _183 => _183.tools]) || [];
+    const tools = _optionalChain([llmRequest, 'access', _183 => _183.config, 'optionalAccess', _184 => _184.tools]) || [];
     const toolNames = tools.map((tool) => {
       if (tool.functionDeclarations && Array.isArray(tool.functionDeclarations)) {
         return tool.functionDeclarations.map((fn) => fn.name).join(", ");
       }
       if (tool.name) return tool.name;
-      if (_optionalChain([tool, 'access', _184 => _184.function, 'optionalAccess', _185 => _185.name])) return tool.function.name;
-      if (_optionalChain([tool, 'access', _186 => _186.function, 'optionalAccess', _187 => _187.function, 'optionalAccess', _188 => _188.name])) return tool.function.function.name;
+      if (_optionalChain([tool, 'access', _185 => _185.function, 'optionalAccess', _186 => _186.name])) return tool.function.name;
+      if (_optionalChain([tool, 'access', _187 => _187.function, 'optionalAccess', _188 => _188.function, 'optionalAccess', _189 => _189.name])) return tool.function.function.name;
       return "unknown";
     }).join(", ");
     const systemInstruction = llmRequest.getSystemInstructionText() || "";
     const truncatedSystemInstruction = systemInstruction.length > 100 ? `${systemInstruction.substring(0, 100)}...` : systemInstruction;
-    const contentPreview = _optionalChain([llmRequest, 'access', _189 => _189.contents, 'optionalAccess', _190 => _190.length]) > 0 ? LogFormatter.formatContentPreview(llmRequest.contents[0]) : "none";
+    const contentPreview = _optionalChain([llmRequest, 'access', _190 => _190.contents, 'optionalAccess', _191 => _191.length]) > 0 ? LogFormatter.formatContentPreview(llmRequest.contents[0]) : "none";
     this.logger.debugStructured("\u{1F4E4} LLM Request", {
       Model: llm.model,
       Agent: invocationContext.agent.name,
-      "Content Items": _optionalChain([llmRequest, 'access', _191 => _191.contents, 'optionalAccess', _192 => _192.length]) || 0,
+      "Content Items": _optionalChain([llmRequest, 'access', _192 => _192.contents, 'optionalAccess', _193 => _193.length]) || 0,
       "Content Preview": contentPreview,
       "System Instruction": truncatedSystemInstruction || "none",
       "Available Tools": toolNames || "none",
-      "Tool Count": _optionalChain([llmRequest, 'access', _193 => _193.config, 'optionalAccess', _194 => _194.tools, 'optionalAccess', _195 => _195.length]) || 0,
+      "Tool Count": _optionalChain([llmRequest, 'access', _194 => _194.config, 'optionalAccess', _195 => _195.tools, 'optionalAccess', _196 => _196.length]) || 0,
       Streaming: isStreaming ? "Yes" : "No"
     });
     let responseCount = 0;
@@ -6729,8 +6775,8 @@ var BaseLlmFlow = (_class25 = class {constructor() { _class25.prototype.__init43
         llmRequest,
         llmResponse
       );
-      const tokenCount = _optionalChain([llmResponse, 'access', _196 => _196.usageMetadata, 'optionalAccess', _197 => _197.totalTokenCount]) || "unknown";
-      const functionCalls = _optionalChain([llmResponse, 'access', _198 => _198.content, 'optionalAccess', _199 => _199.parts, 'optionalAccess', _200 => _200.filter, 'call', _201 => _201((part) => part.functionCall)]) || [];
+      const tokenCount = _optionalChain([llmResponse, 'access', _197 => _197.usageMetadata, 'optionalAccess', _198 => _198.totalTokenCount]) || "unknown";
+      const functionCalls = _optionalChain([llmResponse, 'access', _199 => _199.content, 'optionalAccess', _200 => _200.parts, 'optionalAccess', _201 => _201.filter, 'call', _202 => _202((part) => part.functionCall)]) || [];
       const functionCallsDisplay = LogFormatter.formatFunctionCalls(functionCalls);
       const responsePreview = LogFormatter.formatResponsePreview(llmResponse);
       this.logger.debugStructured("\u{1F4E5} LLM Response", {
@@ -6874,7 +6920,7 @@ var EnhancedAuthConfig = class {
    */
   generateCredentialKey() {
     const schemeKey = this.authScheme.type || "unknown";
-    const credentialKey = _optionalChain([this, 'access', _202 => _202.rawAuthCredential, 'optionalAccess', _203 => _203.type]) || "none";
+    const credentialKey = _optionalChain([this, 'access', _203 => _203.rawAuthCredential, 'optionalAccess', _204 => _204.type]) || "none";
     const timestamp = Date.now();
     return `adk_${schemeKey}_${credentialKey}_${timestamp}`;
   }
@@ -7031,7 +7077,7 @@ var AuthLlmRequestProcessor = class extends BaseLlmRequestProcessor {
    */
   parseAndStoreAuthResponse(authHandler, invocationContext) {
     try {
-      const credentialKey = _optionalChain([authHandler, 'access', _204 => _204.authConfig, 'access', _205 => _205.context, 'optionalAccess', _206 => _206.credentialKey]) || `temp:${Date.now()}`;
+      const credentialKey = _optionalChain([authHandler, 'access', _205 => _205.authConfig, 'access', _206 => _206.context, 'optionalAccess', _207 => _207.credentialKey]) || `temp:${Date.now()}`;
       const fullCredentialKey = credentialKey.startsWith("temp:") ? credentialKey : `temp:${credentialKey}`;
       invocationContext.session.state[fullCredentialKey] = authHandler.credential;
       if (authHandler.authConfig.authScheme.type === "oauth2" || authHandler.authConfig.authScheme.type === "openIdConnect") {
@@ -7074,8 +7120,6 @@ var BasicLlmRequestProcessor = class extends BaseLlmRequestProcessor {
     llmRequest.liveConnectConfig.realtimeInputConfig = runConfig.realtimeInputConfig;
     llmRequest.liveConnectConfig.enableAffectiveDialog = runConfig.enableAffectiveDialog;
     llmRequest.liveConnectConfig.proactivity = runConfig.proactivity;
-    const tools = await agent.canonicalTools();
-    llmRequest.appendTools(tools);
     for await (const _ of []) {
       yield _;
     }
@@ -7139,7 +7183,7 @@ var BuiltInCodeExecutor = class extends BaseCodeExecutor {
    * Pre-process the LLM request for Gemini 2.0+ models to use the code execution tool
    */
   processLlmRequest(llmRequest) {
-    if (!_optionalChain([llmRequest, 'access', _207 => _207.model, 'optionalAccess', _208 => _208.startsWith, 'call', _209 => _209("gemini-2")])) {
+    if (!_optionalChain([llmRequest, 'access', _208 => _208.model, 'optionalAccess', _209 => _209.startsWith, 'call', _210 => _210("gemini-2")])) {
       throw new Error(
         `Gemini code execution tool is not supported for model ${llmRequest.model}`
       );
@@ -7184,7 +7228,7 @@ var CodeExecutionUtils = class _CodeExecutionUtils {
    * Extracts the first code block from the content and truncates everything after it
    */
   static extractCodeAndTruncateContent(content, codeBlockDelimiters) {
-    if (!_optionalChain([content, 'optionalAccess', _210 => _210.parts, 'optionalAccess', _211 => _211.length])) {
+    if (!_optionalChain([content, 'optionalAccess', _211 => _211.parts, 'optionalAccess', _212 => _212.length])) {
       return null;
     }
     for (let idx = 0; idx < content.parts.length; idx++) {
@@ -7270,7 +7314,7 @@ ${fileNames}`);
    * Converts the code execution parts to text parts in a Content
    */
   static convertCodeExecutionParts(content, codeBlockDelimiter, executionResultDelimiters) {
-    if (!_optionalChain([content, 'access', _212 => _212.parts, 'optionalAccess', _213 => _213.length])) {
+    if (!_optionalChain([content, 'access', _213 => _213.parts, 'optionalAccess', _214 => _214.length])) {
       return;
     }
     const lastPart = content.parts[content.parts.length - 1];
@@ -7663,7 +7707,7 @@ async function* runPostProcessor(invocationContext, llmResponse) {
 function extractAndReplaceInlineFiles(codeExecutorContext, llmRequest) {
   const allInputFiles = codeExecutorContext.getInputFiles();
   const savedFileNames = new Set(allInputFiles.map((f) => f.name));
-  for (let i = 0; i < (_optionalChain([llmRequest, 'access', _214 => _214.contents, 'optionalAccess', _215 => _215.length]) || 0); i++) {
+  for (let i = 0; i < (_optionalChain([llmRequest, 'access', _215 => _215.contents, 'optionalAccess', _216 => _216.length]) || 0); i++) {
     const content = llmRequest.contents[i];
     if (content.role !== "user" || !content.parts) {
       continue;
@@ -7695,7 +7739,7 @@ Available file: \`${fileName}\`
 }
 function getOrSetExecutionId(invocationContext, codeExecutorContext) {
   const agent = invocationContext.agent;
-  if (!hasCodeExecutor(agent) || !_optionalChain([agent, 'access', _216 => _216.codeExecutor, 'optionalAccess', _217 => _217.stateful])) {
+  if (!hasCodeExecutor(agent) || !_optionalChain([agent, 'access', _217 => _217.codeExecutor, 'optionalAccess', _218 => _218.stateful])) {
     return void 0;
   }
   let executionId = codeExecutorContext.getExecutionId();
@@ -7926,7 +7970,7 @@ function rearrangeEventsForLatestFunctionResponse(events) {
       continue;
     }
     const functionResponses2 = event.getFunctionResponses();
-    if (_optionalChain([functionResponses2, 'optionalAccess', _218 => _218.some, 'call', _219 => _219((fr) => fr.id && functionResponsesIds.has(fr.id))])) {
+    if (_optionalChain([functionResponses2, 'optionalAccess', _219 => _219.some, 'call', _220 => _220((fr) => fr.id && functionResponsesIds.has(fr.id))])) {
       functionResponseEvents.push(event);
     }
   }
@@ -8025,7 +8069,7 @@ function mergeFunctionResponseEvents(functionResponseEvents) {
   const partIndicesInMergedEvent = {};
   for (let idx = 0; idx < partsInMergedEvent.length; idx++) {
     const part = partsInMergedEvent[idx];
-    if (_optionalChain([part, 'access', _220 => _220.functionResponse, 'optionalAccess', _221 => _221.id])) {
+    if (_optionalChain([part, 'access', _221 => _221.functionResponse, 'optionalAccess', _222 => _222.id])) {
       partIndicesInMergedEvent[part.functionResponse.id] = idx;
     }
   }
@@ -8034,7 +8078,7 @@ function mergeFunctionResponseEvents(functionResponseEvents) {
       throw new Error("There should be at least one function_response part.");
     }
     for (const part of event.content.parts) {
-      if (_optionalChain([part, 'access', _222 => _222.functionResponse, 'optionalAccess', _223 => _223.id])) {
+      if (_optionalChain([part, 'access', _223 => _223.functionResponse, 'optionalAccess', _224 => _224.id])) {
         const functionCallId = part.functionResponse.id;
         if (functionCallId in partIndicesInMergedEvent) {
           partsInMergedEvent[partIndicesInMergedEvent[functionCallId]] = part;
@@ -8303,7 +8347,7 @@ var PlanReActPlanner = class extends BasePlanner {
     let firstFcPartIndex = -1;
     for (let i = 0; i < responseParts.length; i++) {
       if (responseParts[i].functionCall) {
-        if (!_optionalChain([responseParts, 'access', _224 => _224[i], 'access', _225 => _225.functionCall, 'optionalAccess', _226 => _226.name])) {
+        if (!_optionalChain([responseParts, 'access', _225 => _225[i], 'access', _226 => _226.functionCall, 'optionalAccess', _227 => _227.name])) {
           continue;
         }
         preservedParts.push(responseParts[i]);
@@ -8342,7 +8386,7 @@ var PlanReActPlanner = class extends BasePlanner {
    * Handles non-function-call parts of the response
    */
   _handleNonFunctionCallParts(responsePart, preservedParts) {
-    if (_optionalChain([responsePart, 'access', _227 => _227.text, 'optionalAccess', _228 => _228.includes, 'call', _229 => _229(FINAL_ANSWER_TAG)])) {
+    if (_optionalChain([responsePart, 'access', _228 => _228.text, 'optionalAccess', _229 => _229.includes, 'call', _230 => _230(FINAL_ANSWER_TAG)])) {
       const [reasoningText, finalAnswerText] = this._splitByLastPattern(
         responsePart.text,
         FINAL_ANSWER_TAG
@@ -8591,7 +8635,7 @@ var SharedMemoryRequestProcessor = class extends BaseLlmRequestProcessor {
     const memoryService = invocationContext.memoryService;
     if (!memoryService) return;
     const lastUserEvent = invocationContext.session.events.findLast(
-      (e) => e.author === "user" && _optionalChain([e, 'access', _230 => _230.content, 'optionalAccess', _231 => _231.parts, 'optionalAccess', _232 => _232.length])
+      (e) => e.author === "user" && _optionalChain([e, 'access', _231 => _231.content, 'optionalAccess', _232 => _232.parts, 'optionalAccess', _233 => _233.length])
     );
     if (!lastUserEvent) return;
     const query = (_nullishCoalesce(lastUserEvent.content.parts, () => ( []))).map((p) => p.text || "").join(" ");
@@ -8602,7 +8646,7 @@ var SharedMemoryRequestProcessor = class extends BaseLlmRequestProcessor {
     });
     const sessionTexts = new Set(
       (llmRequest.contents || []).flatMap(
-        (c) => _optionalChain([c, 'access', _233 => _233.parts, 'optionalAccess', _234 => _234.map, 'call', _235 => _235((p) => p.text)]) || []
+        (c) => _optionalChain([c, 'access', _234 => _234.parts, 'optionalAccess', _235 => _235.map, 'call', _236 => _236((p) => p.text)]) || []
       )
     );
     for (const memory of results.memories) {
@@ -9025,7 +9069,7 @@ var LlmAgent = (_class27 = class _LlmAgent extends BaseAgent {
    * This matches the Python implementation's _llm_flow property
    */
   get llmFlow() {
-    if (this.disallowTransferToParent && this.disallowTransferToPeers && !_optionalChain([this, 'access', _236 => _236.subAgents, 'optionalAccess', _237 => _237.length])) {
+    if (this.disallowTransferToParent && this.disallowTransferToPeers && !_optionalChain([this, 'access', _237 => _237.subAgents, 'optionalAccess', _238 => _238.length])) {
       return new SingleFlow();
     }
     return new AutoFlow();
@@ -9041,7 +9085,7 @@ var LlmAgent = (_class27 = class _LlmAgent extends BaseAgent {
       );
       return;
     }
-    if (this.outputKey && event.isFinalResponse() && _optionalChain([event, 'access', _238 => _238.content, 'optionalAccess', _239 => _239.parts])) {
+    if (this.outputKey && event.isFinalResponse() && _optionalChain([event, 'access', _239 => _239.content, 'optionalAccess', _240 => _240.parts])) {
       let result = event.content.parts.map((part) => part.text || "").join("");
       if (this.outputSchema) {
         if (!result.trim()) {
@@ -9069,19 +9113,19 @@ var LlmAgent = (_class27 = class _LlmAgent extends BaseAgent {
    * Core logic to run this agent via text-based conversation
    * This matches the Python implementation's _run_async_impl
    */
-  async *runAsyncImpl(context) {
+  async *runAsyncImpl(context4) {
     this.logger.debug(`Starting LlmAgent execution for "${this.name}"`);
     try {
-      for await (const event of this.llmFlow.runAsync(context)) {
+      for await (const event of this.llmFlow.runAsync(context4)) {
         this.maybeSaveOutputToState(event);
         yield event;
       }
     } catch (error) {
       this.logger.error("Error in LlmAgent execution:", error);
       const errorEvent = new Event({
-        invocationId: context.invocationId,
+        invocationId: context4.invocationId,
         author: this.name,
-        branch: context.branch,
+        branch: context4.branch,
         content: {
           parts: [
             {
@@ -9269,7 +9313,7 @@ var LoopAgent = class extends BaseAgent {
       for (const subAgent of this.subAgents) {
         for await (const event of subAgent.runAsync(ctx)) {
           yield event;
-          if (_optionalChain([event, 'access', _240 => _240.actions, 'optionalAccess', _241 => _241.escalate])) {
+          if (_optionalChain([event, 'access', _241 => _241.actions, 'optionalAccess', _242 => _242.escalate])) {
             return;
           }
         }
@@ -9349,7 +9393,7 @@ var LangGraphAgent = (_class28 = class extends BaseAgent {
   /**
    * Gets the next nodes to execute based on the current node and its result
    */
-  async getNextNodes(currentNode, lastEvent, context) {
+  async getNextNodes(currentNode, lastEvent, context4) {
     if (!currentNode.targets || currentNode.targets.length === 0) {
       return [];
     }
@@ -9361,7 +9405,7 @@ var LangGraphAgent = (_class28 = class extends BaseAgent {
         continue;
       }
       if (targetNode.condition) {
-        const shouldExecute = await targetNode.condition(lastEvent, context);
+        const shouldExecute = await targetNode.condition(lastEvent, context4);
         if (!shouldExecute) {
           this.logger.debug(`Skipping node "${targetName}" due to condition`);
           continue;
@@ -9374,7 +9418,7 @@ var LangGraphAgent = (_class28 = class extends BaseAgent {
   /**
    * Core logic to run this agent via text-based conversation.
    */
-  async *runAsyncImpl(context) {
+  async *runAsyncImpl(context4) {
     this.logger.debug(
       `Starting graph execution from root node "${this.rootNode}"`
     );
@@ -9396,7 +9440,7 @@ var LangGraphAgent = (_class28 = class extends BaseAgent {
       return;
     }
     let stepCount = 0;
-    const nodesToExecute = [{ node: rootNode, context }];
+    const nodesToExecute = [{ node: rootNode, context: context4 }];
     const executedNodes = [];
     let lastEvent = null;
     while (nodesToExecute.length > 0 && stepCount < this.maxSteps) {
@@ -9404,7 +9448,7 @@ var LangGraphAgent = (_class28 = class extends BaseAgent {
       const { node } = nodesToExecute.shift();
       this.logger.debug(`Step ${stepCount}: Executing node "${node.name}"`);
       executedNodes.push(node.name);
-      const childContext = context.createChildContext(node.agent);
+      const childContext = context4.createChildContext(node.agent);
       try {
         const nodeEvents = [];
         for await (const event of node.agent.runAsync(childContext)) {
@@ -9417,7 +9461,7 @@ var LangGraphAgent = (_class28 = class extends BaseAgent {
           events: nodeEvents
         });
         if (lastEvent) {
-          const nextNodes = await this.getNextNodes(node, lastEvent, context);
+          const nextNodes = await this.getNextNodes(node, lastEvent, context4);
           for (const nextNode of nextNodes) {
             nodesToExecute.push({
               node: nextNode,
@@ -9460,8 +9504,8 @@ var LangGraphAgent = (_class28 = class extends BaseAgent {
    * Core logic to run this agent via video/audio-based conversation.
    * For LangGraph, this follows the same execution pattern as text-based.
    */
-  async *runLiveImpl(context) {
-    yield* this.runAsyncImpl(context);
+  async *runLiveImpl(context4) {
+    yield* this.runAsyncImpl(context4);
   }
   /**
    * Gets the execution results from the last run
@@ -9580,17 +9624,17 @@ var RunConfig = class {
    */
   
   constructor(config) {
-    this.speechConfig = _optionalChain([config, 'optionalAccess', _242 => _242.speechConfig]);
-    this.responseModalities = _optionalChain([config, 'optionalAccess', _243 => _243.responseModalities]);
-    this.saveInputBlobsAsArtifacts = _optionalChain([config, 'optionalAccess', _244 => _244.saveInputBlobsAsArtifacts]) || false;
-    this.supportCFC = _optionalChain([config, 'optionalAccess', _245 => _245.supportCFC]) || false;
-    this.streamingMode = _optionalChain([config, 'optionalAccess', _246 => _246.streamingMode]) || "NONE" /* NONE */;
-    this.outputAudioTranscription = _optionalChain([config, 'optionalAccess', _247 => _247.outputAudioTranscription]);
-    this.inputAudioTranscription = _optionalChain([config, 'optionalAccess', _248 => _248.inputAudioTranscription]);
-    this.realtimeInputConfig = _optionalChain([config, 'optionalAccess', _249 => _249.realtimeInputConfig]);
-    this.enableAffectiveDialog = _optionalChain([config, 'optionalAccess', _250 => _250.enableAffectiveDialog]);
-    this.proactivity = _optionalChain([config, 'optionalAccess', _251 => _251.proactivity]);
-    this.maxLlmCalls = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _252 => _252.maxLlmCalls]), () => ( 500));
+    this.speechConfig = _optionalChain([config, 'optionalAccess', _243 => _243.speechConfig]);
+    this.responseModalities = _optionalChain([config, 'optionalAccess', _244 => _244.responseModalities]);
+    this.saveInputBlobsAsArtifacts = _optionalChain([config, 'optionalAccess', _245 => _245.saveInputBlobsAsArtifacts]) || false;
+    this.supportCFC = _optionalChain([config, 'optionalAccess', _246 => _246.supportCFC]) || false;
+    this.streamingMode = _optionalChain([config, 'optionalAccess', _247 => _247.streamingMode]) || "NONE" /* NONE */;
+    this.outputAudioTranscription = _optionalChain([config, 'optionalAccess', _248 => _248.outputAudioTranscription]);
+    this.inputAudioTranscription = _optionalChain([config, 'optionalAccess', _249 => _249.inputAudioTranscription]);
+    this.realtimeInputConfig = _optionalChain([config, 'optionalAccess', _250 => _250.realtimeInputConfig]);
+    this.enableAffectiveDialog = _optionalChain([config, 'optionalAccess', _251 => _251.enableAffectiveDialog]);
+    this.proactivity = _optionalChain([config, 'optionalAccess', _252 => _252.proactivity]);
+    this.maxLlmCalls = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _253 => _253.maxLlmCalls]), () => ( 500));
     this.validateMaxLlmCalls();
   }
   /**
@@ -9734,7 +9778,7 @@ var InMemoryMemoryService = (_class30 = class {
     }
     const userSessions = this._sessionEvents.get(userKey);
     const filteredEvents = session.events.filter(
-      (event) => _optionalChain([event, 'access', _253 => _253.content, 'optionalAccess', _254 => _254.parts])
+      (event) => _optionalChain([event, 'access', _254 => _254.content, 'optionalAccess', _255 => _255.parts])
     );
     userSessions.set(session.id, filteredEvents);
   }
@@ -9873,7 +9917,7 @@ var InMemorySessionService = (_class31 = class extends BaseSessionService {const
     return this.createSessionImpl(appName, userId, state, sessionId);
   }
   createSessionImpl(appName, userId, state, sessionId) {
-    const finalSessionId = _optionalChain([sessionId, 'optionalAccess', _255 => _255.trim, 'call', _256 => _256()]) || _crypto.randomUUID.call(void 0, );
+    const finalSessionId = _optionalChain([sessionId, 'optionalAccess', _256 => _256.trim, 'call', _257 => _257()]) || _crypto.randomUUID.call(void 0, );
     const session = {
       appName,
       userId,
@@ -10030,7 +10074,7 @@ var InMemorySessionService = (_class31 = class extends BaseSessionService {const
       warning(`sessionId ${sessionId} not in sessions[appName][userId]`);
       return event;
     }
-    if (_optionalChain([event, 'access', _257 => _257.actions, 'optionalAccess', _258 => _258.stateDelta])) {
+    if (_optionalChain([event, 'access', _258 => _258.actions, 'optionalAccess', _259 => _259.stateDelta])) {
       for (const key in event.actions.stateDelta) {
         const value = event.actions.stateDelta[key];
         if (key.startsWith(State.APP_PREFIX)) {
@@ -10064,14 +10108,14 @@ function _findFunctionCallEventIfLastEventIsFunctionResponse(session) {
     return null;
   }
   const lastEvent = events[events.length - 1];
-  if (_optionalChain([lastEvent, 'access', _259 => _259.content, 'optionalAccess', _260 => _260.parts, 'optionalAccess', _261 => _261.some, 'call', _262 => _262((part) => part.functionResponse)])) {
-    const functionCallId = _optionalChain([lastEvent, 'access', _263 => _263.content, 'access', _264 => _264.parts, 'access', _265 => _265.find, 'call', _266 => _266(
+  if (_optionalChain([lastEvent, 'access', _260 => _260.content, 'optionalAccess', _261 => _261.parts, 'optionalAccess', _262 => _262.some, 'call', _263 => _263((part) => part.functionResponse)])) {
+    const functionCallId = _optionalChain([lastEvent, 'access', _264 => _264.content, 'access', _265 => _265.parts, 'access', _266 => _266.find, 'call', _267 => _267(
       (part) => part.functionResponse
-    ), 'optionalAccess', _267 => _267.functionResponse, 'optionalAccess', _268 => _268.id]);
+    ), 'optionalAccess', _268 => _268.functionResponse, 'optionalAccess', _269 => _269.id]);
     if (!functionCallId) return null;
     for (let i = events.length - 2; i >= 0; i--) {
       const event = events[i];
-      const functionCalls = _optionalChain([event, 'access', _269 => _269.getFunctionCalls, 'optionalCall', _270 => _270()]) || [];
+      const functionCalls = _optionalChain([event, 'access', _270 => _270.getFunctionCalls, 'optionalCall', _271 => _271()]) || [];
       for (const functionCall of functionCalls) {
         if (functionCall.id === functionCallId) {
           return event;
@@ -10174,11 +10218,11 @@ var Runner = (_class32 = class {
     runConfig = new RunConfig()
   }) {
     const span = tracer.startSpan("invocation");
+    const spanContext = _api.trace.setSpan(_api.context.active(), span);
     try {
-      const session = await this.sessionService.getSession(
-        this.appName,
-        userId,
-        sessionId
+      const session = await _api.context.with(
+        spanContext,
+        () => this.sessionService.getSession(this.appName, userId, sessionId)
       );
       if (!session) {
         throw new Error(`Session not found: ${sessionId}`);
@@ -10188,22 +10232,34 @@ var Runner = (_class32 = class {
         runConfig
       });
       if (newMessage) {
-        await this._appendNewMessageToSession(
-          session,
-          newMessage,
-          invocationContext,
-          runConfig.saveInputBlobsAsArtifacts || false
+        await _api.context.with(
+          spanContext,
+          () => this._appendNewMessageToSession(
+            session,
+            newMessage,
+            invocationContext,
+            runConfig.saveInputBlobsAsArtifacts || false
+          )
         );
       }
       invocationContext.agent = this._findAgentToRun(session, this.agent);
-      for await (const event of invocationContext.agent.runAsync(
-        invocationContext
-      )) {
+      const agentGenerator = invocationContext.agent.runAsync(invocationContext);
+      while (true) {
+        const result = await _api.context.with(
+          spanContext,
+          () => agentGenerator.next()
+        );
+        if (result.done) {
+          break;
+        }
+        const event = result.value;
         if (!event.partial) {
-          await this.sessionService.appendEvent(session, event);
-          if (this.memoryService) {
-            await this.memoryService.addSessionToMemory(session);
-          }
+          await _api.context.with(spanContext, async () => {
+            await this.sessionService.appendEvent(session, event);
+            if (this.memoryService) {
+              await this.memoryService.addSessionToMemory(session);
+            }
+          });
         }
         yield event;
       }
@@ -10262,15 +10318,15 @@ var Runner = (_class32 = class {
    */
   _findAgentToRun(session, rootAgent) {
     const event = _findFunctionCallEventIfLastEventIsFunctionResponse(session);
-    if (_optionalChain([event, 'optionalAccess', _271 => _271.author])) {
+    if (_optionalChain([event, 'optionalAccess', _272 => _272.author])) {
       return rootAgent.findAgent(event.author);
     }
-    const nonUserEvents = _optionalChain([session, 'access', _272 => _272.events, 'optionalAccess', _273 => _273.filter, 'call', _274 => _274((e) => e.author !== "user"), 'access', _275 => _275.reverse, 'call', _276 => _276()]) || [];
+    const nonUserEvents = _optionalChain([session, 'access', _273 => _273.events, 'optionalAccess', _274 => _274.filter, 'call', _275 => _275((e) => e.author !== "user"), 'access', _276 => _276.reverse, 'call', _277 => _277()]) || [];
     for (const event2 of nonUserEvents) {
       if (event2.author === rootAgent.name) {
         return rootAgent;
       }
-      const agent = _optionalChain([rootAgent, 'access', _277 => _277.findSubAgent, 'optionalCall', _278 => _278(event2.author)]);
+      const agent = _optionalChain([rootAgent, 'access', _278 => _278.findSubAgent, 'optionalCall', _279 => _279(event2.author)]);
       if (!agent) {
         this.logger.debug(
           `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
@@ -10729,7 +10785,7 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
       async ask(message) {
         const newMessage = typeof message === "string" ? { parts: [{ text: message }] } : typeof message === "object" && "contents" in message ? { parts: message.contents[message.contents.length - 1].parts } : message;
         let response = "";
-        if (!_optionalChain([sessionOptions, 'optionalAccess', _279 => _279.userId])) {
+        if (!_optionalChain([sessionOptions, 'optionalAccess', _280 => _280.userId])) {
           throw new Error("Session configuration is required");
         }
         for await (const event of baseRunner.runAsync({
@@ -10737,7 +10793,7 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
           sessionId: session.id,
           newMessage
         })) {
-          if (_optionalChain([event, 'access', _280 => _280.content, 'optionalAccess', _281 => _281.parts]) && Array.isArray(event.content.parts)) {
+          if (_optionalChain([event, 'access', _281 => _281.content, 'optionalAccess', _282 => _282.parts]) && Array.isArray(event.content.parts)) {
             const content = event.content.parts.map(
               (part) => (part && typeof part === "object" && "text" in part ? part.text : "") || ""
             ).join("");
@@ -10818,7 +10874,7 @@ var VertexAiSessionService = class extends BaseSessionService {
       path: `reasoningEngines/${reasoningEngineId}/sessions`,
       request_dict: sessionJsonDict
     });
-    console.info("Create Session response", apiResponse);
+    console.debug("Create Session response", apiResponse);
     const createdSessionId = apiResponse.name.split("/").slice(-3, -2)[0];
     const operationId = apiResponse.name.split("/").pop();
     let maxRetryAttempt = 5;
@@ -10829,7 +10885,7 @@ var VertexAiSessionService = class extends BaseSessionService {
         path: `operations/${operationId}`,
         request_dict: {}
       });
-      if (_optionalChain([lroResponse, 'optionalAccess', _282 => _282.done])) {
+      if (_optionalChain([lroResponse, 'optionalAccess', _283 => _283.done])) {
         break;
       }
       await new Promise((resolve) => setTimeout(resolve, 1e3));
@@ -11200,12 +11256,12 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
   }
   async createSession(appName, userId, state, sessionId) {
     await this.ensureInitialized();
-    const id = _optionalChain([sessionId, 'optionalAccess', _283 => _283.trim, 'call', _284 => _284()]) || this.generateSessionId();
+    const id = _optionalChain([sessionId, 'optionalAccess', _284 => _284.trim, 'call', _285 => _285()]) || this.generateSessionId();
     return await this.db.transaction().execute(async (trx) => {
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", appName).where("user_id", "=", userId).executeTakeFirst();
-      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _285 => _285.state]), {});
-      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _286 => _286.state]), {});
+      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _286 => _286.state]), {});
+      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _287 => _287.state]), {});
       if (!appState) {
         await trx.insertInto("app_states").values({
           app_name: appName,
@@ -11264,21 +11320,21 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
         return void 0;
       }
       let eventQuery = trx.selectFrom("events").selectAll().where("session_id", "=", sessionId).orderBy("timestamp", "desc");
-      if (_optionalChain([config, 'optionalAccess', _287 => _287.afterTimestamp])) {
+      if (_optionalChain([config, 'optionalAccess', _288 => _288.afterTimestamp])) {
         eventQuery = eventQuery.where(
           "timestamp",
           ">=",
           new Date(config.afterTimestamp * 1e3)
         );
       }
-      if (_optionalChain([config, 'optionalAccess', _288 => _288.numRecentEvents])) {
+      if (_optionalChain([config, 'optionalAccess', _289 => _289.numRecentEvents])) {
         eventQuery = eventQuery.limit(config.numRecentEvents);
       }
       const storageEvents = await eventQuery.execute();
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", appName).where("user_id", "=", userId).executeTakeFirst();
-      const currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _289 => _289.state]), {});
-      const currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _290 => _290.state]), {});
+      const currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _290 => _290.state]), {});
+      const currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _291 => _291.state]), {});
       const sessionState = this.parseJsonSafely(storageSession.state, {});
       const mergedState = this.mergeState(
         currentAppState,
@@ -11336,13 +11392,13 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
       }
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", session.appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", session.appName).where("user_id", "=", session.userId).executeTakeFirst();
-      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _291 => _291.state]), {});
-      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _292 => _292.state]), {});
+      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _292 => _292.state]), {});
+      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _293 => _293.state]), {});
       let sessionState = this.parseJsonSafely(storageSession.state, {});
       let appStateDelta = {};
       let userStateDelta = {};
       let sessionStateDelta = {};
-      if (_optionalChain([event, 'access', _293 => _293.actions, 'optionalAccess', _294 => _294.stateDelta])) {
+      if (_optionalChain([event, 'access', _294 => _294.actions, 'optionalAccess', _295 => _295.stateDelta])) {
         const deltas = this.extractStateDelta(event.actions.stateDelta);
         appStateDelta = deltas.appStateDelta;
         userStateDelta = deltas.userStateDelta;
@@ -11488,7 +11544,7 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
    * Overrides the base class method to work with plain object state.
    */
   updateSessionState(session, event) {
-    if (!_optionalChain([event, 'access', _295 => _295.actions, 'optionalAccess', _296 => _296.stateDelta])) {
+    if (!_optionalChain([event, 'access', _296 => _296.actions, 'optionalAccess', _297 => _297.stateDelta])) {
       return;
     }
     for (const [key, value] of Object.entries(event.actions.stateDelta)) {
@@ -11658,7 +11714,7 @@ var GcsArtifactService = class {
       };
       return part;
     } catch (error) {
-      if (_optionalChain([error, 'optionalAccess', _297 => _297.code]) === 404) {
+      if (_optionalChain([error, 'optionalAccess', _298 => _298.code]) === 404) {
         return null;
       }
       throw error;