@google/adk 0.2.5 → 0.3.0

package/dist/esm/agents/functions.js

@@ -10,6 +10,11 @@ import { mergeEventActions } from "../events/event_actions.js";
 import { ToolContext } from "../tools/tool_context.js";
 import { randomUUID } from "../utils/env_aware_utils.js";
 import { logger } from "../utils/logger.js";
+import {
+  traceMergedToolCalls,
+  tracer,
+  traceToolCall
+} from "../telemetry/tracing.js";
 const AF_FUNCTION_CALL_ID_PREFIX = "adk-";
 const REQUEST_EUC_FUNCTION_CALL_NAME = "adk_request_credential";
 const REQUEST_CONFIRMATION_FUNCTION_CALL_NAME = "adk_request_confirmation";
@@ -127,8 +132,51 @@ function generateRequestConfirmationEvent({
   });
 }
 async function callToolAsync(tool, args, toolContext) {
-  logger.debug(`callToolAsync ${tool.name}`);
-  return await tool.runAsync({ args, toolContext });
+  return tracer.startActiveSpan(`execute_tool ${tool.name}`, async (span) => {
+    try {
+      logger.debug(`callToolAsync ${tool.name}`);
+      const result = await tool.runAsync({ args, toolContext });
+      traceToolCall({
+        tool,
+        args,
+        functionResponseEvent: buildResponseEvent(
+          tool,
+          result,
+          toolContext,
+          toolContext.invocationContext
+        )
+      });
+      return result;
+    } finally {
+      span.end();
+    }
+  });
+}
+function buildResponseEvent(tool, functionResult, toolContext, invocationContext) {
+  let responseResult;
+  if (typeof functionResult !== "object" || functionResult == null) {
+    responseResult = { result: functionResult };
+  } else {
+    responseResult = functionResult;
+  }
+  const partFunctionResponse = {
+    functionResponse: {
+      name: tool.name,
+      response: responseResult,
+      id: toolContext.functionCallId
+    }
+  };
+  const content = {
+    role: "user",
+    parts: [partFunctionResponse]
+  };
+  return createEvent({
+    invocationId: invocationContext.invocationId,
+    author: invocationContext.agent.name,
+    content,
+    actions: toolContext.actions,
+    branch: invocationContext.branch
+  });
 }
 async function handleFunctionCallsAsync({
   invocationContext,
@@ -169,14 +217,12 @@ async function handleFunctionCallList({
     if (toolConfirmationDict && functionCall.id) {
       toolConfirmation = toolConfirmationDict[functionCall.id];
     }
-    const { tool, toolContext } = getToolAndContext(
-      {
-        invocationContext,
-        functionCall,
-        toolsDict,
-        toolConfirmation
-      }
-    );
+    const { tool, toolContext } = getToolAndContext({
+      invocationContext,
+      functionCall,
+      toolsDict,
+      toolConfirmation
+    });
     logger.debug(`execute_tool ${tool.name}`);
     const functionArgs = (_a = functionCall.args) != null ? _a : {};
     let functionResponse = null;
@@ -200,21 +246,15 @@ async function handleFunctionCallList({
     }
     if (functionResponse == null) {
       try {
-        functionResponse = await callToolAsync(
-          tool,
-          functionArgs,
-          toolContext
-        );
+        functionResponse = await callToolAsync(tool, functionArgs, toolContext);
       } catch (e) {
         if (e instanceof Error) {
-          const onToolErrorResponse = await invocationContext.pluginManager.runOnToolErrorCallback(
-            {
-              tool,
-              toolArgs: functionArgs,
-              toolContext,
-              error: e
-            }
-          );
+          const onToolErrorResponse = await invocationContext.pluginManager.runOnToolErrorCallback({
+            tool,
+            toolArgs: functionArgs,
+            toolContext,
+            error: e
+          });
           if (onToolErrorResponse) {
             functionResponse = onToolErrorResponse;
           } else {
@@ -278,12 +318,24 @@ async function handleFunctionCallList({
   if (!functionResponseEvents.length) {
     return null;
   }
-  const mergedEvent = mergeParallelFunctionResponseEvents(functionResponseEvents);
+  const mergedEvent = mergeParallelFunctionResponseEvents(
+    functionResponseEvents
+  );
   if (functionResponseEvents.length > 1) {
-    logger.debug("execute_tool (merged)");
-    logger.debug("traceMergedToolCalls", {
-      responseEventId: mergedEvent.id,
-      functionResponseEvent: mergedEvent.id
+    tracer.startActiveSpan("execute_tool (merged)", (span) => {
+      try {
+        logger.debug("execute_tool (merged)");
+        logger.debug("traceMergedToolCalls", {
+          responseEventId: mergedEvent.id,
+          functionResponseEvent: mergedEvent.id
+        });
+        traceMergedToolCalls({
+          responseEventId: mergedEvent.id,
+          functionResponseEvent: mergedEvent
+        });
+      } finally {
+        span.end();
+      }
     });
   }
   return mergedEvent;
@@ -321,7 +373,9 @@ function mergeParallelFunctionResponseEvents(functionResponseEvents) {
     }
   }
   const baseEvent = functionResponseEvents[0];
-  const actionsList = functionResponseEvents.map((event) => event.actions || {});
+  const actionsList = functionResponseEvents.map(
+    (event) => event.actions || {}
+  );
   const mergedActions = mergeEventActions(actionsList);
   return createEvent({
     author: baseEvent.author,

package/dist/esm/agents/llm_agent.js

@@ -4,6 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 var _a, _b;
+import { context, trace } from "@opentelemetry/api";
 import { cloneDeep } from "lodash-es";
 import { z } from "zod";
 import {
@@ -38,6 +39,11 @@ import { ToolConfirmation } from "../tools/tool_confirmation.js";
 import { ToolContext } from "../tools/tool_context.js";
 import { base64Decode } from "../utils/env_aware_utils.js";
 import { logger } from "../utils/logger.js";
+import {
+  runAsyncGeneratorWithOtelContext,
+  traceCallLlm,
+  tracer
+} from "../telemetry/tracing.js";
 import { BaseAgent } from "./base_agent.js";
 import {
   BaseLlmRequestProcessor
@@ -60,13 +66,14 @@ import { injectSessionState } from "./instructions.js";
 import { ReadonlyContext } from "./readonly_context.js";
 import { StreamingMode } from "./run_config.js";
 const ADK_AGENT_NAME_LABEL_KEY = "adk_agent_name";
-async function convertToolUnionToTools(toolUnion, context) {
+async function convertToolUnionToTools(toolUnion, context2) {
   if (toolUnion instanceof BaseTool) {
     return [toolUnion];
   }
-  return await toolUnion.getTools(context);
+  return await toolUnion.getTools(context2);
 }
 class BasicLlmRequestProcessor extends BaseLlmRequestProcessor {
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     var _a2;
     const agent = invocationContext.agent;
@@ -91,6 +98,7 @@ class BasicLlmRequestProcessor extends BaseLlmRequestProcessor {
 }
 const BASIC_LLM_REQUEST_PROCESSOR = new BasicLlmRequestProcessor();
 class IdentityLlmRequestProcessor extends BaseLlmRequestProcessor {
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     const si = [`You are an agent. Your internal name is "${agent.name}".`];
@@ -105,6 +113,7 @@ class InstructionsLlmRequestProcessor extends BaseLlmRequestProcessor {
   /**
    * Handles instructions and global instructions for LLM flow.
    */
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     if (!(agent instanceof LlmAgent) || !(agent.rootAgent instanceof LlmAgent)) {
@@ -141,6 +150,7 @@ class InstructionsLlmRequestProcessor extends BaseLlmRequestProcessor {
 }
 const INSTRUCTIONS_LLM_REQUEST_PROCESSOR = new InstructionsLlmRequestProcessor();
 class ContentRequestProcessor {
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     if (!agent || !isLlmAgent(agent)) {
@@ -182,6 +192,7 @@ class AgentTransferLlmRequestProcessor extends BaseLlmRequestProcessor {
       }
     });
   }
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     if (!(invocationContext.agent instanceof LlmAgent)) {
       return;
@@ -250,7 +261,7 @@ to your parent agent.
 const AGENT_TRANSFER_LLM_REQUEST_PROCESSOR = new AgentTransferLlmRequestProcessor();
 class RequestConfirmationLlmRequestProcessor extends BaseLlmRequestProcessor {
   /** Handles tool confirmation information to build the LLM request. */
-  async *runAsync(invocationContext, llmRequest) {
+  async *runAsync(invocationContext) {
     const agent = invocationContext.agent;
     if (!isLlmAgent(agent)) {
       return;
@@ -383,7 +394,7 @@ class CodeExecutionRequestProcessor extends BaseLlmRequestProcessor {
     }
     for (const content of llmRequest.contents) {
       const delimeters = invocationContext.agent.codeExecutor.codeBlockDelimiters.length ? invocationContext.agent.codeExecutor.codeBlockDelimiters[0] : ["", ""];
-      const codeExecutionParts = convertCodeExecutionParts(
+      convertCodeExecutionParts(
         content,
         delimeters,
         invocationContext.agent.codeExecutor.executionResultDelimiters
@@ -593,7 +604,7 @@ async function* runPostProcessor(invocationContext, llmResponse) {
     codeExecutorContext,
     codeExecutionResult
   );
-  llmResponse.content = null;
+  llmResponse.content = void 0;
 }
 function extractAndReplaceInlineFiles(codeExecutorContext, llmRequest) {
   var _a2;
@@ -806,12 +817,12 @@ class LlmAgent extends (_b = BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
    * @param context The context to retrieve the session state.
    * @returns The resolved instruction field.
    */
-  async canonicalInstruction(context) {
+  async canonicalInstruction(context2) {
     if (typeof this.instruction === "string") {
       return { instruction: this.instruction, requireStateInjection: true };
     }
     return {
-      instruction: await this.instruction(context),
+      instruction: await this.instruction(context2),
       requireStateInjection: false
     };
   }
@@ -822,12 +833,15 @@ class LlmAgent extends (_b = BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
    * @param context The context to retrieve the session state.
    * @returns The resolved globalInstruction field.
    */
-  async canonicalGlobalInstruction(context) {
+  async canonicalGlobalInstruction(context2) {
     if (typeof this.globalInstruction === "string") {
-      return { instruction: this.globalInstruction, requireStateInjection: true };
+      return {
+        instruction: this.globalInstruction,
+        requireStateInjection: true
+      };
     }
     return {
-      instruction: await this.globalInstruction(context),
+      instruction: await this.globalInstruction(context2),
       requireStateInjection: false
     };
   }
@@ -836,10 +850,10 @@ class LlmAgent extends (_b = BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
    *
    * This method is only for use by Agent Development Kit.
    */
-  async canonicalTools(context) {
+  async canonicalTools(context2) {
     const resolvedTools = [];
     for (const toolUnion of this.tools) {
-      const tools = await convertToolUnionToTools(toolUnion, context);
+      const tools = await convertToolUnionToTools(toolUnion, context2);
       resolvedTools.push(...tools);
     }
     return resolvedTools;
@@ -943,10 +957,10 @@ class LlmAgent extends (_b = BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
     }
     event.actions.stateDelta[this.outputKey] = result;
   }
-  async *runAsyncImpl(context) {
+  async *runAsyncImpl(context2) {
     while (true) {
       let lastEvent = void 0;
-      for await (const event of this.runOneStepAsync(context)) {
+      for await (const event of this.runOneStepAsync(context2)) {
         lastEvent = event;
         this.maybeSaveOutputToState(event);
         yield event;
@@ -960,19 +974,20 @@ class LlmAgent extends (_b = BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
       }
     }
   }
-  async *runLiveImpl(context) {
-    for await (const event of this.runLiveFlow(context)) {
+  async *runLiveImpl(context2) {
+    for await (const event of this.runLiveFlow(context2)) {
       this.maybeSaveOutputToState(event);
       yield event;
     }
-    if (context.endInvocation) {
+    if (context2.endInvocation) {
       return;
     }
   }
   // --------------------------------------------------------------------------
   // #START LlmFlow Logic
   // --------------------------------------------------------------------------
-  async *runLiveFlow(invocationContext) {
+  // eslint-disable-next-line require-yield
+  async *runLiveFlow(_invocationContext) {
     await Promise.resolve();
     throw new Error("LlmAgent.runLiveFlow not implemented");
   }
@@ -1008,22 +1023,31 @@ class LlmAgent extends (_b = BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
       author: this.name,
       branch: invocationContext.branch
     });
-    for await (const llmResponse of this.callLlmAsync(
-      invocationContext,
-      llmRequest,
-      modelResponseEvent
-    )) {
-      for await (const event of this.postprocess(
-        invocationContext,
-        llmRequest,
-        llmResponse,
-        modelResponseEvent
-      )) {
-        modelResponseEvent.id = createNewEventId();
-        modelResponseEvent.timestamp = (/* @__PURE__ */ new Date()).getTime();
-        yield event;
+    const span = tracer.startSpan("call_llm");
+    const ctx = trace.setSpan(context.active(), span);
+    yield* runAsyncGeneratorWithOtelContext(
+      ctx,
+      this,
+      async function* () {
+        for await (const llmResponse of this.callLlmAsync(
+          invocationContext,
+          llmRequest,
+          modelResponseEvent
+        )) {
+          for await (const event of this.postprocess(
+            invocationContext,
+            llmRequest,
+            llmResponse,
+            modelResponseEvent
+          )) {
+            modelResponseEvent.id = createNewEventId();
+            modelResponseEvent.timestamp = (/* @__PURE__ */ new Date()).getTime();
+            yield event;
+          }
+        }
       }
-    }
+    );
+    span.end();
   }
   async *postprocess(invocationContext, llmRequest, llmResponse, modelResponseEvent) {
     var _a2;
@@ -1141,6 +1165,12 @@ class LlmAgent extends (_b = BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
         llmRequest,
         modelResponseEvent
       )) {
+        traceCallLlm({
+          invocationContext,
+          eventId: modelResponseEvent.id,
+          llmRequest,
+          llmResponse
+        });
         const alteredLlmResponse = await this.handleAfterModelCallback(
           invocationContext,
           llmResponse,

package/dist/esm/agents/loop_agent.js

@@ -38,7 +38,8 @@ class LoopAgent extends (_b = BaseAgent, _a = LOOP_AGENT_SIGNATURE_SYMBOL, _b) {
     }
     return;
   }
-  async *runLiveImpl(context) {
+  // eslint-disable-next-line require-yield
+  async *runLiveImpl(_context) {
     throw new Error("This is not supported yet for LoopAgent.");
   }
 }

package/dist/esm/agents/parallel_agent.js

@@ -20,15 +20,14 @@ class ParallelAgent extends (_b = BaseAgent, _a = PARALLEL_AGENT_SIGNATURE_SYMBO
   }
   async *runAsyncImpl(context) {
     const agentRuns = this.subAgents.map(
-      (subAgent) => subAgent.runAsync(
-        createBranchCtxForSubAgent(this, subAgent, context)
-      )
+      (subAgent) => subAgent.runAsync(createBranchCtxForSubAgent(this, subAgent, context))
     );
     for await (const event of mergeAgentRuns(agentRuns)) {
       yield event;
     }
   }
-  async *runLiveImpl(context) {
+  // eslint-disable-next-line require-yield
+  async *runLiveImpl(_context) {
     throw new Error("This is not supported yet for ParallelAgent.");
   }
 }

package/dist/esm/artifacts/gcs_artifact_service.js

@@ -12,10 +12,12 @@ class GcsArtifactService {
   async saveArtifact(request) {
     const versions = await this.listVersions(request);
     const version = versions.length > 0 ? Math.max(...versions) + 1 : 0;
-    const file = this.bucket.file(getFileName({
-      ...request,
-      version
-    }));
+    const file = this.bucket.file(
+      getFileName({
+        ...request,
+        version
+      })
+    );
     if (request.artifact.inlineData) {
       await file.save(JSON.stringify(request.artifact.inlineData.data), {
         contentType: request.artifact.inlineData.mimeType
@@ -39,11 +41,16 @@ class GcsArtifactService {
       }
       version = Math.max(...versions);
     }
-    const file = this.bucket.file(getFileName({
-      ...request,
-      version
-    }));
-    const [[metadata], [rawDataBuffer]] = await Promise.all([file.getMetadata(), file.download()]);
+    const file = this.bucket.file(
+      getFileName({
+        ...request,
+        version
+      })
+    );
+    const [[metadata], [rawDataBuffer]] = await Promise.all([
+      file.getMetadata(),
+      file.download()
+    ]);
     if (metadata.contentType === "text/plain") {
       return createPartFromText(rawDataBuffer.toString("utf-8"));
     }
@@ -56,10 +63,7 @@ class GcsArtifactService {
     const fileNames = [];
     const sessionPrefix = `${request.appName}/${request.userId}/${request.sessionId}/`;
     const usernamePrefix = `${request.appName}/${request.userId}/user/`;
-    const [
-      [sessionFiles],
-      [userSessionFiles]
-    ] = await Promise.all([
+    const [[sessionFiles], [userSessionFiles]] = await Promise.all([
       this.bucket.getFiles({ prefix: sessionPrefix }),
       this.bucket.getFiles({ prefix: usernamePrefix })
     ]);
@@ -73,13 +77,17 @@ class GcsArtifactService {
   }
   async deleteArtifact(request) {
     const versions = await this.listVersions(request);
-    await Promise.all(versions.map((version) => {
-      const file = this.bucket.file(getFileName({
-        ...request,
-        version
-      }));
-      return file.delete();
-    }));
+    await Promise.all(
+      versions.map((version) => {
+        const file = this.bucket.file(
+          getFileName({
+            ...request,
+            version
+          })
+        );
+        return file.delete();
+      })
+    );
     return;
   }
   async listVersions(request) {

package/dist/esm/artifacts/in_memory_artifact_service.js

@@ -39,7 +39,11 @@ class InMemoryArtifactService {
     }
     return Promise.resolve(versions[version]);
   }
-  listArtifactKeys({ appName, userId, sessionId }) {
+  listArtifactKeys({
+    appName,
+    userId,
+    sessionId
+  }) {
     const sessionPrefix = `${appName}/${userId}/${sessionId}/`;
     const usernamespacePrefix = `${appName}/${userId}/user/`;
     const filenames = [];
@@ -54,7 +58,12 @@ class InMemoryArtifactService {
     }
     return Promise.resolve(filenames.sort());
   }
-  deleteArtifact({ appName, userId, sessionId, filename }) {
+  deleteArtifact({
+    appName,
+    userId,
+    sessionId,
+    filename
+  }) {
     const path = artifactPath(appName, userId, sessionId, filename);
     if (!this.artifacts[path]) {
       return Promise.resolve();
@@ -62,13 +71,18 @@ class InMemoryArtifactService {
     delete this.artifacts[path];
     return Promise.resolve();
   }
-  listVersions({ appName, userId, sessionId, filename }) {
+  listVersions({
+    appName,
+    userId,
+    sessionId,
+    filename
+  }) {
     const path = artifactPath(appName, userId, sessionId, filename);
     const artifacts = this.artifacts[path];
     if (!artifacts) {
       return Promise.resolve([]);
     }
-    let versions = [];
+    const versions = [];
     for (let i = 0; i < artifacts.length; i++) {
       versions.push(i);
     }

package/dist/esm/auth/auth_handler.js

@@ -37,7 +37,9 @@ class AuthHandler {
       };
     }
     if (!this.authConfig.rawAuthCredential.oauth2.clientId || !this.authConfig.rawAuthCredential.oauth2.clientSecret) {
-      throw new Error(`Auth Scheme ${authSchemeType} requires both clientId and clientSecret in authCredential.oauth2.`);
+      throw new Error(
+        `Auth Scheme ${authSchemeType} requires both clientId and clientSecret in authCredential.oauth2.`
+      );
     }
     return {
       credentialKey: this.authConfig.credentialKey,

package/dist/esm/code_executors/base_code_executor.js

@@ -4,7 +4,9 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 var _a;
-const BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL = Symbol.for("google.adk.baseCodeExecutor");
+const BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL = Symbol.for(
+  "google.adk.baseCodeExecutor"
+);
 function isBaseCodeExecutor(obj) {
   return typeof obj === "object" && obj !== null && BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL in obj && obj[BASE_CODE_EXECUTOR_SIGNATURE_SYMBOL] === true;
 }

package/dist/esm/code_executors/built_in_code_executor.js

@@ -6,7 +6,9 @@
 var _a, _b;
 import { isGemini2OrAbove } from "../utils/model_name.js";
 import { BaseCodeExecutor } from "./base_code_executor.js";
-const BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL = Symbol.for("google.adk.builtInCodeExecutor");
+const BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL = Symbol.for(
+  "google.adk.builtInCodeExecutor"
+);
 function isBuiltInCodeExecutor(obj) {
   return typeof obj === "object" && obj !== null && BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL in obj && obj[BUILT_IN_CODE_EXECUTOR_SIGNATURE_SYMBOL] === true;
 }
@@ -16,7 +18,7 @@ class BuiltInCodeExecutor extends (_b = BaseCodeExecutor, _a = BUILT_IN_CODE_EXE
     /** A unique symbol to identify BuiltInCodeExecutor class. */
     this[_a] = true;
   }
-  executeCode(params) {
+  executeCode(_params) {
     return Promise.resolve({
       stdout: "",
       stderr: "",
@@ -30,7 +32,9 @@ class BuiltInCodeExecutor extends (_b = BaseCodeExecutor, _a = BUILT_IN_CODE_EXE
       llmRequest.config.tools.push({ codeExecution: {} });
       return;
     }
-    throw new Error(`Gemini code execution tool is not supported for model ${llmRequest.model}`);
+    throw new Error(
+      `Gemini code execution tool is not supported for model ${llmRequest.model}`
+    );
   }
 }
 export {

package/dist/esm/code_executors/code_executor_context.js

@@ -142,7 +142,9 @@ class CodeExecutorContext {
     if (!(CODE_EXECUTION_RESULTS_KEY in this.sessionState)) {
       this.sessionState.set(CODE_EXECUTION_RESULTS_KEY, {});
     }
-    const codeExecutionResults = this.sessionState.get(CODE_EXECUTION_RESULTS_KEY);
+    const codeExecutionResults = this.sessionState.get(
+      CODE_EXECUTION_RESULTS_KEY
+    );
     if (!(invocationId in codeExecutionResults)) {
       codeExecutionResults[invocationId] = [];
     }
@@ -154,12 +156,10 @@ class CodeExecutorContext {
     });
   }
   /**
-   * Gets the code executor context from the session state.
-   * @param invocationId The session state to get the code executor context
-   *     from.
+   * Gets the code execution context from the session state.
    * @return The code execution context for the given invocation ID.
    */
-  getCodeExecutionContext(invocationId) {
+  getCodeExecutionContext() {
     return this.sessionState.get(CONTEXT_KEY) || {};
   }
 }

package/dist/esm/common.js

@@ -63,7 +63,7 @@ import { GOOGLE_SEARCH, GoogleSearchTool } from "./tools/google_search_tool.js";
 import { LongRunningFunctionTool } from "./tools/long_running_tool.js";
 import { ToolConfirmation } from "./tools/tool_confirmation.js";
 import { ToolContext } from "./tools/tool_context.js";
-import { LogLevel, setLogLevel } from "./utils/logger.js";
+import { LogLevel, getLogger, setLogLevel, setLogger } from "./utils/logger.js";
 import { isGemini2OrAbove } from "./utils/model_name.js";
 import { zodObjectToSchema } from "./utils/simple_zod_to_json.js";
 import { GoogleLLMVariant } from "./utils/variant_utils.js";
@@ -124,6 +124,7 @@ export {
   getAskUserConfirmationFunctionCalls,
   getFunctionCalls,
   getFunctionResponses,
+  getLogger,
   hasTrailingCodeExecutionResult,
   isAgentTool,
   isBaseAgent,
@@ -138,6 +139,7 @@ export {
   isParallelAgent,
   isSequentialAgent,
   setLogLevel,
+  setLogger,
   stringifyContent,
   version,
   zodObjectToSchema

package/dist/esm/events/event.js

@@ -66,9 +66,7 @@ const ASCII_LETTERS_AND_NUMBERS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRS
 function createNewEventId() {
   let id = "";
   for (let i = 0; i < 8; i++) {
-    id += ASCII_LETTERS_AND_NUMBERS[Math.floor(
-      Math.random() * ASCII_LETTERS_AND_NUMBERS.length
-    )];
+    id += ASCII_LETTERS_AND_NUMBERS[Math.floor(Math.random() * ASCII_LETTERS_AND_NUMBERS.length)];
   }
   return id;
 }