@raindrop-ai/deep-agents 0.1.3 → 0.1.4

package/README.md

@@ -56,6 +56,16 @@ await raindrop.shutdown();
 | `projectId` | `string` | - | Route events to a specific project (slug); omit for the default **Production** project |
 | `debug` | `boolean` | `false` | Enable verbose logging |
 | `traceChains` | `boolean` | `true` | Create spans for chain execution |
+| `tools` | `ToolDeclaration[]` | - | Overrides the tool catalog recorded as `ai.prompt.tools` on every model span. Use when the integration cannot see the tools the model was given. |
+
+### Tool catalog capture
+
+Every model span (`llm`) records the tools the agent's chat model was given for
+that call as `ai.prompt.tools` (one JSON document per tool: `name`, `description`,
+full JSON Schema as `inputSchema`). It is read from the callback's
+`invocation_params.tools` / `functions`, which ChatOpenAI, ChatAnthropic and similar
+models populate from `bindTools()`. Absent when the model reports no list; pass
+`tools` to record a catalog explicitly (`tools: []` records an empty one).
 
 ## Projects
 

package/dist/index.d.mts

@@ -509,6 +509,46 @@ declare global {
     var RAINDROP_ASYNC_LOCAL_STORAGE: (new <T>() => AsyncLocalStorageLike<T>) | undefined;
 }
 
+/**
+ * `ai.prompt.tools`: the tool catalog a model call was given, one JSON
+ * document per tool, in the shape the Vercel AI SDK records on its
+ * `doGenerate` / `doStream` spans. Every integration emits this attribute on
+ * its model-call spans so downstream consumers (agent replay, world builder)
+ * can read what the model could call instead of guessing from the calls it
+ * happened to make.
+ *
+ * An absent attribute means the integration could not observe the tool list;
+ * an empty array means the model had no tools. Callers who know better than
+ * the integration pass a `tools` override, which replaces the inferred list.
+ */
+declare const PROMPT_TOOLS_ATTRIBUTE = "ai.prompt.tools";
+type JsonSchema = Record<string, unknown>;
+type FunctionToolDeclaration = {
+    type: "function";
+    name: string;
+    description?: string;
+    inputSchema?: JsonSchema;
+};
+type ProviderDefinedToolDeclaration = {
+    type: "provider-defined";
+    id?: string;
+    name: string;
+    args?: Record<string, unknown>;
+};
+type ToolDeclaration = FunctionToolDeclaration | ProviderDefinedToolDeclaration;
+/**
+ * What callers hand to a `tools` override. The canonical `ToolDeclaration`
+ * shape is accepted as-is; so is any framework shape `normalizeToolDeclarations`
+ * recognizes (OpenAI, Anthropic, Bedrock, LangChain, Pi, Mastra, AI SDK).
+ */
+type ToolDeclarationInput = ToolDeclaration | {
+    name: string;
+    description?: string;
+    inputSchema?: JsonSchema;
+    parameters?: JsonSchema;
+};
+type PromptToolsOverride = ReadonlyArray<ToolDeclarationInput | unknown>;
+
 interface RaindropDeepAgentsHandlerOptions {
     eventShipper: EventShipper;
     traceShipper: TraceShipper;
@@ -526,6 +566,12 @@ interface RaindropDeepAgentsHandlerOptions {
      */
     maxTextFieldChars?: number;
     eventMaxTextFieldChars?: number;
+    /**
+     * Overrides the tool catalog recorded as `ai.prompt.tools` on every model
+     * span. Use when the integration cannot see the tools the model was given.
+     * Replaces the inferred list entirely; `[]` records an empty catalog.
+     */
+    tools?: PromptToolsOverride;
 }
 declare class RaindropDeepAgentsHandler extends BaseCallbackHandler {
     name: string;
@@ -538,6 +584,7 @@ declare class RaindropDeepAgentsHandler extends BaseCallbackHandler {
     private traceChains;
     private readonly configuredSpanMaxChars;
     private readonly eventMaxChars;
+    private readonly promptTools;
     private spans;
     private rootRunIds;
     private eventIds;
@@ -550,14 +597,15 @@ declare class RaindropDeepAgentsHandler extends BaseCallbackHandler {
      */
     _lastRootEventId: string | undefined;
     constructor(opts: RaindropDeepAgentsHandlerOptions);
+    private promptToolsAttribute;
     private get spanMaxChars();
     private getEventId;
     private getParent;
     private cleanup;
     private finalizeEventIfRoot;
     private rootErrorPatch;
-    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
-    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
     handleLLMEnd(output: LLMResult, runId: string, _parentRunId?: string): Promise<void>;
     handleLLMError(err: unknown, runId: string): Promise<void>;
     handleChainStart(chain: Serialized, _inputs: ChainValues, runId: string, parentRunId?: string, _tags?: string[], _metadata?: Record<string, unknown>, _runType?: string, runName?: string): Promise<void>;
@@ -610,6 +658,11 @@ interface DeepAgentsOptions {
     maxTextFieldChars?: number;
     /** Application Git identity. Auto-detects commit SHA by default; pass `false` to disable. */
     appGit?: AppGitOptions | false;
+    /**
+     * Overrides the tool catalog recorded as `ai.prompt.tools` on every model
+     * span. Use when the integration cannot see the tools the model was given.
+     */
+    tools?: PromptToolsOverride;
 }
 type RaindropDeepAgentsClient = {
     /**
@@ -697,4 +750,4 @@ type RaindropDeepAgentsClient = {
  */
 declare function createRaindropDeepAgents(opts: DeepAgentsOptions): RaindropDeepAgentsClient;
 
-export { type DeepAgentsOptions, type RaindropDeepAgentsClient, RaindropDeepAgentsHandler, createRaindropDeepAgents };
+export { type DeepAgentsOptions, PROMPT_TOOLS_ATTRIBUTE, type PromptToolsOverride, type RaindropDeepAgentsClient, RaindropDeepAgentsHandler, type ToolDeclaration, createRaindropDeepAgents };

package/dist/index.d.ts

@@ -509,6 +509,46 @@ declare global {
     var RAINDROP_ASYNC_LOCAL_STORAGE: (new <T>() => AsyncLocalStorageLike<T>) | undefined;
 }
 
+/**
+ * `ai.prompt.tools`: the tool catalog a model call was given, one JSON
+ * document per tool, in the shape the Vercel AI SDK records on its
+ * `doGenerate` / `doStream` spans. Every integration emits this attribute on
+ * its model-call spans so downstream consumers (agent replay, world builder)
+ * can read what the model could call instead of guessing from the calls it
+ * happened to make.
+ *
+ * An absent attribute means the integration could not observe the tool list;
+ * an empty array means the model had no tools. Callers who know better than
+ * the integration pass a `tools` override, which replaces the inferred list.
+ */
+declare const PROMPT_TOOLS_ATTRIBUTE = "ai.prompt.tools";
+type JsonSchema = Record<string, unknown>;
+type FunctionToolDeclaration = {
+    type: "function";
+    name: string;
+    description?: string;
+    inputSchema?: JsonSchema;
+};
+type ProviderDefinedToolDeclaration = {
+    type: "provider-defined";
+    id?: string;
+    name: string;
+    args?: Record<string, unknown>;
+};
+type ToolDeclaration = FunctionToolDeclaration | ProviderDefinedToolDeclaration;
+/**
+ * What callers hand to a `tools` override. The canonical `ToolDeclaration`
+ * shape is accepted as-is; so is any framework shape `normalizeToolDeclarations`
+ * recognizes (OpenAI, Anthropic, Bedrock, LangChain, Pi, Mastra, AI SDK).
+ */
+type ToolDeclarationInput = ToolDeclaration | {
+    name: string;
+    description?: string;
+    inputSchema?: JsonSchema;
+    parameters?: JsonSchema;
+};
+type PromptToolsOverride = ReadonlyArray<ToolDeclarationInput | unknown>;
+
 interface RaindropDeepAgentsHandlerOptions {
     eventShipper: EventShipper;
     traceShipper: TraceShipper;
@@ -526,6 +566,12 @@ interface RaindropDeepAgentsHandlerOptions {
      */
     maxTextFieldChars?: number;
     eventMaxTextFieldChars?: number;
+    /**
+     * Overrides the tool catalog recorded as `ai.prompt.tools` on every model
+     * span. Use when the integration cannot see the tools the model was given.
+     * Replaces the inferred list entirely; `[]` records an empty catalog.
+     */
+    tools?: PromptToolsOverride;
 }
 declare class RaindropDeepAgentsHandler extends BaseCallbackHandler {
     name: string;
@@ -538,6 +584,7 @@ declare class RaindropDeepAgentsHandler extends BaseCallbackHandler {
     private traceChains;
     private readonly configuredSpanMaxChars;
     private readonly eventMaxChars;
+    private readonly promptTools;
     private spans;
     private rootRunIds;
     private eventIds;
@@ -550,14 +597,15 @@ declare class RaindropDeepAgentsHandler extends BaseCallbackHandler {
      */
     _lastRootEventId: string | undefined;
     constructor(opts: RaindropDeepAgentsHandlerOptions);
+    private promptToolsAttribute;
     private get spanMaxChars();
     private getEventId;
     private getParent;
     private cleanup;
     private finalizeEventIfRoot;
     private rootErrorPatch;
-    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
-    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, _extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, runName?: string): Promise<void>;
     handleLLMEnd(output: LLMResult, runId: string, _parentRunId?: string): Promise<void>;
     handleLLMError(err: unknown, runId: string): Promise<void>;
     handleChainStart(chain: Serialized, _inputs: ChainValues, runId: string, parentRunId?: string, _tags?: string[], _metadata?: Record<string, unknown>, _runType?: string, runName?: string): Promise<void>;
@@ -610,6 +658,11 @@ interface DeepAgentsOptions {
     maxTextFieldChars?: number;
     /** Application Git identity. Auto-detects commit SHA by default; pass `false` to disable. */
     appGit?: AppGitOptions | false;
+    /**
+     * Overrides the tool catalog recorded as `ai.prompt.tools` on every model
+     * span. Use when the integration cannot see the tools the model was given.
+     */
+    tools?: PromptToolsOverride;
 }
 type RaindropDeepAgentsClient = {
     /**
@@ -697,4 +750,4 @@ type RaindropDeepAgentsClient = {
  */
 declare function createRaindropDeepAgents(opts: DeepAgentsOptions): RaindropDeepAgentsClient;
 
-export { type DeepAgentsOptions, type RaindropDeepAgentsClient, RaindropDeepAgentsHandler, createRaindropDeepAgents };
+export { type DeepAgentsOptions, PROMPT_TOOLS_ATTRIBUTE, type PromptToolsOverride, type RaindropDeepAgentsClient, RaindropDeepAgentsHandler, type ToolDeclaration, createRaindropDeepAgents };

package/dist/index.js

@@ -20,6 +20,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  PROMPT_TOOLS_ATTRIBUTE: () => PROMPT_TOOLS_ATTRIBUTE,
   RaindropDeepAgentsHandler: () => RaindropDeepAgentsHandler,
   createRaindropDeepAgents: () => createRaindropDeepAgents
 });
@@ -90,7 +91,7 @@ function evalScopeAttributes() {
   return { [EVAL_CORRELATION_ID_ATTRIBUTE]: scope.correlationId };
 }
 
-// ../core/dist/chunk-WJOR6M7H.js
+// ../core/dist/chunk-I6TVJQEC.js
 var STATE_KEY = /* @__PURE__ */ Symbol.for("raindrop.tracing.replayTraceRoutingState");
 function state() {
   const holder = globalThis;
@@ -2077,6 +2078,158 @@ var TraceShipper = class {
     }
   }
 };
+var PROMPT_TOOLS_ATTRIBUTE = "ai.prompt.tools";
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+var JSON_SCHEMA_KEYS = [
+  "type",
+  "properties",
+  "$schema",
+  "$ref",
+  "anyOf",
+  "oneOf",
+  "allOf",
+  "enum",
+  "const",
+  "items",
+  "additionalProperties",
+  "required"
+];
+function isJsonSchema(value) {
+  if (!isRecord(value)) return false;
+  if (typeof value["~standard"] === "object") return false;
+  if (isRecord(value._def) || isRecord(value._zod)) return false;
+  if (Object.keys(value).length === 0) return true;
+  return JSON_SCHEMA_KEYS.some((key) => key in value);
+}
+function toJsonSchema(value, options) {
+  if (value === void 0 || value === null) return void 0;
+  if (isJsonSchema(value)) return value;
+  if (isRecord(value) && isJsonSchema(value.jsonSchema)) return value.jsonSchema;
+  if (isRecord(value) && isJsonSchema(value.json)) return value.json;
+  if (options == null ? void 0 : options.schema) {
+    try {
+      return options.schema(value);
+    } catch (e) {
+      return void 0;
+    }
+  }
+  return void 0;
+}
+function pickString(value) {
+  return typeof value === "string" && value.length > 0 ? value : void 0;
+}
+function functionTool(name, description, schema, options) {
+  const tool = { type: "function", name };
+  const desc = pickString(description);
+  if (desc !== void 0) tool.description = desc;
+  const inputSchema = toJsonSchema(schema, options);
+  if (inputSchema !== void 0) tool.inputSchema = inputSchema;
+  return tool;
+}
+function normalizeToolDeclaration(tool, options, fallbackName) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+  if (!isRecord(tool)) return void 0;
+  if (tool.type === "provider-defined" || tool.type === "provider") {
+    const name2 = (_b = (_a = pickString(tool.name)) != null ? _a : pickString(tool.id)) != null ? _b : fallbackName;
+    if (name2 === void 0) return void 0;
+    const declaration = { type: "provider-defined", name: name2 };
+    const id = pickString(tool.id);
+    if (id !== void 0) declaration.id = id;
+    if (isRecord(tool.args)) declaration.args = tool.args;
+    return declaration;
+  }
+  if (isRecord(tool.toolSpec)) {
+    const spec = tool.toolSpec;
+    const name2 = (_c = pickString(spec.name)) != null ? _c : fallbackName;
+    if (name2 === void 0) return void 0;
+    return functionTool(name2, spec.description, spec.inputSchema, options);
+  }
+  if (isRecord(tool.function)) {
+    const fn = tool.function;
+    const name2 = (_d = pickString(fn.name)) != null ? _d : fallbackName;
+    if (name2 === void 0) return void 0;
+    return functionTool(name2, fn.description, (_e = fn.parameters) != null ? _e : fn.input_schema, options);
+  }
+  const name = (_g = (_f = pickString(tool.name)) != null ? _f : pickString(tool.id)) != null ? _g : fallbackName;
+  if (name === void 0) return void 0;
+  const type = pickString(tool.type);
+  if (type !== void 0 && type !== "function" && type !== "dynamic" && type !== "custom") {
+    if (!("input_schema" in tool) && !("inputSchema" in tool) && !("parameters" in tool)) {
+      return { type: "provider-defined", id: type, name };
+    }
+  }
+  return functionTool(
+    name,
+    tool.description,
+    (_j = (_i = (_h = tool.inputSchema) != null ? _h : tool.input_schema) != null ? _i : tool.parameters) != null ? _j : tool.schema,
+    options
+  );
+}
+var NOT_A_GEMINI_GROUP_KEYS = ["name", "id", "type", "function", "toolSpec", "cachePoint"];
+function isGeminiToolGroup(tool) {
+  if (Array.isArray(tool.functionDeclarations)) return true;
+  const keys = Object.keys(tool);
+  if (keys.length === 0 || NOT_A_GEMINI_GROUP_KEYS.some((key) => key in tool)) return false;
+  return keys.every((key) => isRecord(tool[key]));
+}
+function expandGeminiTool(tool, options) {
+  const declarations = [];
+  for (const [key, value] of Object.entries(tool)) {
+    if (key === "functionDeclarations" && Array.isArray(value)) {
+      for (const declaration of value) {
+        const normalized = normalizeToolDeclaration(declaration, options);
+        if (normalized) declarations.push(normalized);
+      }
+    } else if (isRecord(value)) {
+      declarations.push({ type: "provider-defined", id: key, name: key, args: value });
+    }
+  }
+  return declarations;
+}
+function normalizeToolDeclarations(tools, options) {
+  if (tools === void 0 || tools === null) return void 0;
+  if (Array.isArray(tools)) {
+    return tools.flatMap((tool) => {
+      if (isRecord(tool) && isGeminiToolGroup(tool)) {
+        return expandGeminiTool(tool, options);
+      }
+      const declaration = normalizeToolDeclaration(tool, options);
+      return declaration ? [declaration] : [];
+    });
+  }
+  if (!isRecord(tools)) return void 0;
+  return Object.entries(tools).flatMap(([key, tool]) => {
+    const declaration = normalizeToolDeclaration(tool, options, key);
+    return declaration ? [declaration] : [];
+  });
+}
+function resolvePromptTools(args) {
+  var _a;
+  if (args.override !== void 0) {
+    return (_a = normalizeToolDeclarations(args.override, args.options)) != null ? _a : [];
+  }
+  return normalizeToolDeclarations(args.inferred, args.options);
+}
+function promptToolsAttribute(tools) {
+  const values = promptToolsAttributeValue(tools);
+  if (values === void 0) return void 0;
+  return {
+    key: PROMPT_TOOLS_ATTRIBUTE,
+    value: { arrayValue: { values: values.map((stringValue) => ({ stringValue })) } }
+  };
+}
+function promptToolsAttributeValue(tools) {
+  if (tools === void 0) return void 0;
+  return tools.map((tool) => {
+    try {
+      return JSON.stringify(tool);
+    } catch (e) {
+      return JSON.stringify({ type: tool.type, name: tool.name });
+    }
+  });
+}
 
 // ../core/dist/index.node.js
 var import_async_hooks = require("async_hooks");
@@ -2391,7 +2544,7 @@ var CAPTURE_TRUNCATED_ROLE = "raindrop";
 var MESSAGE_OVERHEAD = 32;
 var PART_OVERHEAD = 24;
 var MAX_DEPTH2 = 12;
-function isRecord(value) {
+function isRecord2(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 function stringField(record, key) {
@@ -2445,7 +2598,7 @@ function takeValue(value, budget, depth) {
   }
   budget.remaining -= 2;
   const out = {};
-  if (!isRecord(value)) return out;
+  if (!isRecord2(value)) return out;
   for (const key in value) {
     if (!Object.prototype.hasOwnProperty.call(value, key)) continue;
     if (budget.remaining <= 0) {
@@ -2473,7 +2626,7 @@ function joinedText(content, budget) {
       break;
     }
     budget.remaining -= 1;
-    if (!isRecord(block)) continue;
+    if (!isRecord2(block)) continue;
     const text = stringField(block, "text");
     if (text !== void 0) texts.push(takeText(text, budget));
   }
@@ -2481,13 +2634,13 @@ function joinedText(content, budget) {
 }
 function contentPart(block, budget) {
   budget.remaining -= PART_OVERHEAD;
-  if (!isRecord(block)) return takeValue(block, budget, 0);
+  if (!isRecord2(block)) return takeValue(block, budget, 0);
   const type = stringField(block, "type");
   const text = stringField(block, "text");
   if (type === "text" && text !== void 0) return textPart(text, budget);
   if (type === "image_url") {
     const imageUrl = block["image_url"];
-    const url = typeof imageUrl === "string" ? imageUrl : isRecord(imageUrl) ? stringField(imageUrl, "url") : void 0;
+    const url = typeof imageUrl === "string" ? imageUrl : isRecord2(imageUrl) ? stringField(imageUrl, "url") : void 0;
     if (url !== void 0) return { type: "image", image: takeText(url, budget) };
   }
   if (type === "thinking") {
@@ -2499,8 +2652,8 @@ function contentPart(block, budget) {
     const sourceType = stringField(block, "source_type");
     const source = block["source"];
     if (sourceType === "text" && text !== void 0) return textPart(text, budget);
-    const data = sourceType === "url" ? stringField(block, "url") : sourceType === "base64" ? stringField(block, "data") : isRecord(source) ? stringField(source, "url") ?? stringField(source, "data") : void 0;
-    const sourceMediaType = mediaType ?? (isRecord(source) ? stringField(source, "media_type") : void 0);
+    const data = sourceType === "url" ? stringField(block, "url") : sourceType === "base64" ? stringField(block, "data") : isRecord2(source) ? stringField(source, "url") ?? stringField(source, "data") : void 0;
+    const sourceMediaType = mediaType ?? (isRecord2(source) ? stringField(source, "media_type") : void 0);
     if (data !== void 0) {
       const filename = stringField(block, "filename");
       return type === "image" ? { type: "image", image: takeText(data, budget), mediaType: reserve(sourceMediaType, budget) } : {
@@ -2521,7 +2674,7 @@ function contentParts(content, budget, skipToolUseIds) {
       parts.push({ type: "text", text: TRUNCATION_MARKER2 });
       break;
     }
-    if (isRecord(block) && (block["type"] === "tool_use" || block["type"] === "tool_call")) {
+    if (isRecord2(block) && (block["type"] === "tool_use" || block["type"] === "tool_call")) {
       const id = stringField(block, "id");
       if (id !== void 0 && skipToolUseIds.has(id)) continue;
     }
@@ -2664,12 +2817,12 @@ function capturePrompt(prompt, limit) {
   }
 }
 function captureToolCalls(message, limit) {
-  if (!isRecord(message) || !Array.isArray(message["tool_calls"])) return void 0;
+  if (!isRecord2(message) || !Array.isArray(message["tool_calls"])) return void 0;
   const calls = [];
   let structural = 0;
   for (const call of message["tool_calls"]) {
     structural += PART_OVERHEAD;
-    if (isRecord(call)) {
+    if (isRecord2(call)) {
       structural += (stringField(call, "id")?.length ?? 0) + (stringField(call, "name")?.length ?? 0);
       calls.push(call);
     }
@@ -2692,7 +2845,7 @@ function captureToolCalls(message, limit) {
 function parseToolCallInput(input) {
   try {
     const parsed = JSON.parse(input);
-    if (isRecord(parsed) && parsed["type"] === "tool_call" && "args" in parsed) {
+    if (isRecord2(parsed) && parsed["type"] === "tool_call" && "args" in parsed) {
       return { id: stringField(parsed, "id"), args: JSON.stringify(parsed["args"]) };
     }
   } catch {
@@ -2710,6 +2863,12 @@ function unwrapToolOutput(output) {
 }
 
 // src/callback-handler.ts
+function inferInvocationTools(extraParams) {
+  const params = extraParams?.invocation_params;
+  if (typeof params !== "object" || params === null) return void 0;
+  const record = { ...params };
+  return record.tools ?? record.functions;
+}
 function resolveEventId(source) {
   if (typeof source !== "function") return randomUUID();
   try {
@@ -2778,6 +2937,12 @@ var RaindropDeepAgentsHandler = class extends import_base.BaseCallbackHandler {
     this.traceChains = opts.traceChains ?? true;
     this.configuredSpanMaxChars = opts.maxTextFieldChars ?? eventMaxTextFieldChars();
     this.eventMaxChars = opts.eventMaxTextFieldChars ?? eventMaxTextFieldChars();
+    this.promptTools = opts.tools;
+  }
+  promptToolsAttribute(extraParams) {
+    return promptToolsAttribute(
+      resolvePromptTools({ override: this.promptTools, inferred: inferInvocationTools(extraParams) })
+    );
   }
   get spanMaxChars() {
     return spanAttributeLimit(this.configuredSpanMaxChars);
@@ -2841,7 +3006,7 @@ var RaindropDeepAgentsHandler = class extends import_base.BaseCallbackHandler {
   // ---------------------------------------------------------------------------
   // LLM events
   // ---------------------------------------------------------------------------
-  async handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, tags, metadata, runName) {
+  async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, runName) {
     try {
       if (this.spans.has(runId)) return;
       const eventId = this.getEventId(runId, parentRunId);
@@ -2860,7 +3025,8 @@ var RaindropDeepAgentsHandler = class extends import_base.BaseCallbackHandler {
         attributes: [
           attrString("ai.model.provider", modelId),
           attrString("ai.operationId", "ai.generateText"),
-          ...promptAttributes(capturePrompt(prompt, this.spanMaxChars))
+          ...promptAttributes(capturePrompt(prompt, this.spanMaxChars)),
+          this.promptToolsAttribute(extraParams)
         ]
       });
       this.spans.set(runId, span);
@@ -2875,7 +3041,7 @@ var RaindropDeepAgentsHandler = class extends import_base.BaseCallbackHandler {
     } catch {
     }
   }
-  async handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, tags, metadata, runName) {
+  async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, runName) {
     try {
       if (this.spans.has(runId)) return;
       const eventId = this.getEventId(runId, parentRunId);
@@ -2894,7 +3060,8 @@ var RaindropDeepAgentsHandler = class extends import_base.BaseCallbackHandler {
         attributes: [
           attrString("ai.model.provider", modelId),
           attrString("ai.operationId", "ai.generateText"),
-          ...promptAttributes(captureConversation(conversation, this.spanMaxChars))
+          ...promptAttributes(captureConversation(conversation, this.spanMaxChars)),
+          this.promptToolsAttribute(extraParams)
         ]
       });
       this.spans.set(runId, span);
@@ -3210,7 +3377,7 @@ function asNumber(val) {
 // package.json
 var package_default = {
   name: "@raindrop-ai/deep-agents",
-  version: "0.1.3",
+  version: "0.1.4",
   description: "Raindrop integration for LangChain Deep Agents",
   main: "dist/index.js",
   module: "dist/index.mjs",
@@ -3324,7 +3491,8 @@ function createRaindropDeepAgents(opts) {
     eventId: opts.eventId,
     traceChains: opts.traceChains,
     maxTextFieldChars: maxChars,
-    eventMaxTextFieldChars: maxChars
+    eventMaxTextFieldChars: maxChars,
+    tools: opts.tools
   });
   return {
     handler,
@@ -3372,6 +3540,7 @@ function createRaindropDeepAgents(opts) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  PROMPT_TOOLS_ATTRIBUTE,
   RaindropDeepAgentsHandler,
   createRaindropDeepAgents
 });

package/dist/index.mjs

@@ -63,7 +63,7 @@ function evalScopeAttributes() {
   return { [EVAL_CORRELATION_ID_ATTRIBUTE]: scope.correlationId };
 }
 
-// ../core/dist/chunk-WJOR6M7H.js
+// ../core/dist/chunk-I6TVJQEC.js
 var STATE_KEY = /* @__PURE__ */ Symbol.for("raindrop.tracing.replayTraceRoutingState");
 function state() {
   const holder = globalThis;
@@ -2050,6 +2050,158 @@ var TraceShipper = class {
     }
   }
 };
+var PROMPT_TOOLS_ATTRIBUTE = "ai.prompt.tools";
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+var JSON_SCHEMA_KEYS = [
+  "type",
+  "properties",
+  "$schema",
+  "$ref",
+  "anyOf",
+  "oneOf",
+  "allOf",
+  "enum",
+  "const",
+  "items",
+  "additionalProperties",
+  "required"
+];
+function isJsonSchema(value) {
+  if (!isRecord(value)) return false;
+  if (typeof value["~standard"] === "object") return false;
+  if (isRecord(value._def) || isRecord(value._zod)) return false;
+  if (Object.keys(value).length === 0) return true;
+  return JSON_SCHEMA_KEYS.some((key) => key in value);
+}
+function toJsonSchema(value, options) {
+  if (value === void 0 || value === null) return void 0;
+  if (isJsonSchema(value)) return value;
+  if (isRecord(value) && isJsonSchema(value.jsonSchema)) return value.jsonSchema;
+  if (isRecord(value) && isJsonSchema(value.json)) return value.json;
+  if (options == null ? void 0 : options.schema) {
+    try {
+      return options.schema(value);
+    } catch (e) {
+      return void 0;
+    }
+  }
+  return void 0;
+}
+function pickString(value) {
+  return typeof value === "string" && value.length > 0 ? value : void 0;
+}
+function functionTool(name, description, schema, options) {
+  const tool = { type: "function", name };
+  const desc = pickString(description);
+  if (desc !== void 0) tool.description = desc;
+  const inputSchema = toJsonSchema(schema, options);
+  if (inputSchema !== void 0) tool.inputSchema = inputSchema;
+  return tool;
+}
+function normalizeToolDeclaration(tool, options, fallbackName) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+  if (!isRecord(tool)) return void 0;
+  if (tool.type === "provider-defined" || tool.type === "provider") {
+    const name2 = (_b = (_a = pickString(tool.name)) != null ? _a : pickString(tool.id)) != null ? _b : fallbackName;
+    if (name2 === void 0) return void 0;
+    const declaration = { type: "provider-defined", name: name2 };
+    const id = pickString(tool.id);
+    if (id !== void 0) declaration.id = id;
+    if (isRecord(tool.args)) declaration.args = tool.args;
+    return declaration;
+  }
+  if (isRecord(tool.toolSpec)) {
+    const spec = tool.toolSpec;
+    const name2 = (_c = pickString(spec.name)) != null ? _c : fallbackName;
+    if (name2 === void 0) return void 0;
+    return functionTool(name2, spec.description, spec.inputSchema, options);
+  }
+  if (isRecord(tool.function)) {
+    const fn = tool.function;
+    const name2 = (_d = pickString(fn.name)) != null ? _d : fallbackName;
+    if (name2 === void 0) return void 0;
+    return functionTool(name2, fn.description, (_e = fn.parameters) != null ? _e : fn.input_schema, options);
+  }
+  const name = (_g = (_f = pickString(tool.name)) != null ? _f : pickString(tool.id)) != null ? _g : fallbackName;
+  if (name === void 0) return void 0;
+  const type = pickString(tool.type);
+  if (type !== void 0 && type !== "function" && type !== "dynamic" && type !== "custom") {
+    if (!("input_schema" in tool) && !("inputSchema" in tool) && !("parameters" in tool)) {
+      return { type: "provider-defined", id: type, name };
+    }
+  }
+  return functionTool(
+    name,
+    tool.description,
+    (_j = (_i = (_h = tool.inputSchema) != null ? _h : tool.input_schema) != null ? _i : tool.parameters) != null ? _j : tool.schema,
+    options
+  );
+}
+var NOT_A_GEMINI_GROUP_KEYS = ["name", "id", "type", "function", "toolSpec", "cachePoint"];
+function isGeminiToolGroup(tool) {
+  if (Array.isArray(tool.functionDeclarations)) return true;
+  const keys = Object.keys(tool);
+  if (keys.length === 0 || NOT_A_GEMINI_GROUP_KEYS.some((key) => key in tool)) return false;
+  return keys.every((key) => isRecord(tool[key]));
+}
+function expandGeminiTool(tool, options) {
+  const declarations = [];
+  for (const [key, value] of Object.entries(tool)) {
+    if (key === "functionDeclarations" && Array.isArray(value)) {
+      for (const declaration of value) {
+        const normalized = normalizeToolDeclaration(declaration, options);
+        if (normalized) declarations.push(normalized);
+      }
+    } else if (isRecord(value)) {
+      declarations.push({ type: "provider-defined", id: key, name: key, args: value });
+    }
+  }
+  return declarations;
+}
+function normalizeToolDeclarations(tools, options) {
+  if (tools === void 0 || tools === null) return void 0;
+  if (Array.isArray(tools)) {
+    return tools.flatMap((tool) => {
+      if (isRecord(tool) && isGeminiToolGroup(tool)) {
+        return expandGeminiTool(tool, options);
+      }
+      const declaration = normalizeToolDeclaration(tool, options);
+      return declaration ? [declaration] : [];
+    });
+  }
+  if (!isRecord(tools)) return void 0;
+  return Object.entries(tools).flatMap(([key, tool]) => {
+    const declaration = normalizeToolDeclaration(tool, options, key);
+    return declaration ? [declaration] : [];
+  });
+}
+function resolvePromptTools(args) {
+  var _a;
+  if (args.override !== void 0) {
+    return (_a = normalizeToolDeclarations(args.override, args.options)) != null ? _a : [];
+  }
+  return normalizeToolDeclarations(args.inferred, args.options);
+}
+function promptToolsAttribute(tools) {
+  const values = promptToolsAttributeValue(tools);
+  if (values === void 0) return void 0;
+  return {
+    key: PROMPT_TOOLS_ATTRIBUTE,
+    value: { arrayValue: { values: values.map((stringValue) => ({ stringValue })) } }
+  };
+}
+function promptToolsAttributeValue(tools) {
+  if (tools === void 0) return void 0;
+  return tools.map((tool) => {
+    try {
+      return JSON.stringify(tool);
+    } catch (e) {
+      return JSON.stringify({ type: tool.type, name: tool.name });
+    }
+  });
+}
 
 // ../core/dist/index.node.js
 import { AsyncLocalStorage } from "async_hooks";
@@ -2369,7 +2521,7 @@ var CAPTURE_TRUNCATED_ROLE = "raindrop";
 var MESSAGE_OVERHEAD = 32;
 var PART_OVERHEAD = 24;
 var MAX_DEPTH2 = 12;
-function isRecord(value) {
+function isRecord2(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 function stringField(record, key) {
@@ -2423,7 +2575,7 @@ function takeValue(value, budget, depth) {
   }
   budget.remaining -= 2;
   const out = {};
-  if (!isRecord(value)) return out;
+  if (!isRecord2(value)) return out;
   for (const key in value) {
     if (!Object.prototype.hasOwnProperty.call(value, key)) continue;
     if (budget.remaining <= 0) {
@@ -2451,7 +2603,7 @@ function joinedText(content, budget) {
       break;
     }
     budget.remaining -= 1;
-    if (!isRecord(block)) continue;
+    if (!isRecord2(block)) continue;
     const text = stringField(block, "text");
     if (text !== void 0) texts.push(takeText(text, budget));
   }
@@ -2459,13 +2611,13 @@ function joinedText(content, budget) {
 }
 function contentPart(block, budget) {
   budget.remaining -= PART_OVERHEAD;
-  if (!isRecord(block)) return takeValue(block, budget, 0);
+  if (!isRecord2(block)) return takeValue(block, budget, 0);
   const type = stringField(block, "type");
   const text = stringField(block, "text");
   if (type === "text" && text !== void 0) return textPart(text, budget);
   if (type === "image_url") {
     const imageUrl = block["image_url"];
-    const url = typeof imageUrl === "string" ? imageUrl : isRecord(imageUrl) ? stringField(imageUrl, "url") : void 0;
+    const url = typeof imageUrl === "string" ? imageUrl : isRecord2(imageUrl) ? stringField(imageUrl, "url") : void 0;
     if (url !== void 0) return { type: "image", image: takeText(url, budget) };
   }
   if (type === "thinking") {
@@ -2477,8 +2629,8 @@ function contentPart(block, budget) {
     const sourceType = stringField(block, "source_type");
     const source = block["source"];
     if (sourceType === "text" && text !== void 0) return textPart(text, budget);
-    const data = sourceType === "url" ? stringField(block, "url") : sourceType === "base64" ? stringField(block, "data") : isRecord(source) ? stringField(source, "url") ?? stringField(source, "data") : void 0;
-    const sourceMediaType = mediaType ?? (isRecord(source) ? stringField(source, "media_type") : void 0);
+    const data = sourceType === "url" ? stringField(block, "url") : sourceType === "base64" ? stringField(block, "data") : isRecord2(source) ? stringField(source, "url") ?? stringField(source, "data") : void 0;
+    const sourceMediaType = mediaType ?? (isRecord2(source) ? stringField(source, "media_type") : void 0);
     if (data !== void 0) {
       const filename = stringField(block, "filename");
       return type === "image" ? { type: "image", image: takeText(data, budget), mediaType: reserve(sourceMediaType, budget) } : {
@@ -2499,7 +2651,7 @@ function contentParts(content, budget, skipToolUseIds) {
       parts.push({ type: "text", text: TRUNCATION_MARKER2 });
       break;
     }
-    if (isRecord(block) && (block["type"] === "tool_use" || block["type"] === "tool_call")) {
+    if (isRecord2(block) && (block["type"] === "tool_use" || block["type"] === "tool_call")) {
       const id = stringField(block, "id");
       if (id !== void 0 && skipToolUseIds.has(id)) continue;
     }
@@ -2642,12 +2794,12 @@ function capturePrompt(prompt, limit) {
   }
 }
 function captureToolCalls(message, limit) {
-  if (!isRecord(message) || !Array.isArray(message["tool_calls"])) return void 0;
+  if (!isRecord2(message) || !Array.isArray(message["tool_calls"])) return void 0;
   const calls = [];
   let structural = 0;
   for (const call of message["tool_calls"]) {
     structural += PART_OVERHEAD;
-    if (isRecord(call)) {
+    if (isRecord2(call)) {
       structural += (stringField(call, "id")?.length ?? 0) + (stringField(call, "name")?.length ?? 0);
       calls.push(call);
     }
@@ -2670,7 +2822,7 @@ function captureToolCalls(message, limit) {
 function parseToolCallInput(input) {
   try {
     const parsed = JSON.parse(input);
-    if (isRecord(parsed) && parsed["type"] === "tool_call" && "args" in parsed) {
+    if (isRecord2(parsed) && parsed["type"] === "tool_call" && "args" in parsed) {
       return { id: stringField(parsed, "id"), args: JSON.stringify(parsed["args"]) };
     }
   } catch {
@@ -2688,6 +2840,12 @@ function unwrapToolOutput(output) {
 }
 
 // src/callback-handler.ts
+function inferInvocationTools(extraParams) {
+  const params = extraParams?.invocation_params;
+  if (typeof params !== "object" || params === null) return void 0;
+  const record = { ...params };
+  return record.tools ?? record.functions;
+}
 function resolveEventId(source) {
   if (typeof source !== "function") return randomUUID();
   try {
@@ -2756,6 +2914,12 @@ var RaindropDeepAgentsHandler = class extends BaseCallbackHandler {
     this.traceChains = opts.traceChains ?? true;
     this.configuredSpanMaxChars = opts.maxTextFieldChars ?? eventMaxTextFieldChars();
     this.eventMaxChars = opts.eventMaxTextFieldChars ?? eventMaxTextFieldChars();
+    this.promptTools = opts.tools;
+  }
+  promptToolsAttribute(extraParams) {
+    return promptToolsAttribute(
+      resolvePromptTools({ override: this.promptTools, inferred: inferInvocationTools(extraParams) })
+    );
   }
   get spanMaxChars() {
     return spanAttributeLimit(this.configuredSpanMaxChars);
@@ -2819,7 +2983,7 @@ var RaindropDeepAgentsHandler = class extends BaseCallbackHandler {
   // ---------------------------------------------------------------------------
   // LLM events
   // ---------------------------------------------------------------------------
-  async handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, tags, metadata, runName) {
+  async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, runName) {
     try {
       if (this.spans.has(runId)) return;
       const eventId = this.getEventId(runId, parentRunId);
@@ -2838,7 +3002,8 @@ var RaindropDeepAgentsHandler = class extends BaseCallbackHandler {
         attributes: [
           attrString("ai.model.provider", modelId),
           attrString("ai.operationId", "ai.generateText"),
-          ...promptAttributes(capturePrompt(prompt, this.spanMaxChars))
+          ...promptAttributes(capturePrompt(prompt, this.spanMaxChars)),
+          this.promptToolsAttribute(extraParams)
         ]
       });
       this.spans.set(runId, span);
@@ -2853,7 +3018,7 @@ var RaindropDeepAgentsHandler = class extends BaseCallbackHandler {
     } catch {
     }
   }
-  async handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, tags, metadata, runName) {
+  async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, runName) {
     try {
       if (this.spans.has(runId)) return;
       const eventId = this.getEventId(runId, parentRunId);
@@ -2872,7 +3037,8 @@ var RaindropDeepAgentsHandler = class extends BaseCallbackHandler {
         attributes: [
           attrString("ai.model.provider", modelId),
           attrString("ai.operationId", "ai.generateText"),
-          ...promptAttributes(captureConversation(conversation, this.spanMaxChars))
+          ...promptAttributes(captureConversation(conversation, this.spanMaxChars)),
+          this.promptToolsAttribute(extraParams)
         ]
       });
       this.spans.set(runId, span);
@@ -3188,7 +3354,7 @@ function asNumber(val) {
 // package.json
 var package_default = {
   name: "@raindrop-ai/deep-agents",
-  version: "0.1.3",
+  version: "0.1.4",
   description: "Raindrop integration for LangChain Deep Agents",
   main: "dist/index.js",
   module: "dist/index.mjs",
@@ -3302,7 +3468,8 @@ function createRaindropDeepAgents(opts) {
     eventId: opts.eventId,
     traceChains: opts.traceChains,
     maxTextFieldChars: maxChars,
-    eventMaxTextFieldChars: maxChars
+    eventMaxTextFieldChars: maxChars,
+    tools: opts.tools
   });
   return {
     handler,
@@ -3349,6 +3516,7 @@ function createRaindropDeepAgents(opts) {
   };
 }
 export {
+  PROMPT_TOOLS_ATTRIBUTE,
   RaindropDeepAgentsHandler,
   createRaindropDeepAgents
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@raindrop-ai/deep-agents",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Raindrop integration for LangChain Deep Agents",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -36,7 +36,7 @@
     "typescript": "^5.3.3",
     "vitest": "^2.1.9",
     "zod": "^3.23.0",
-    "@raindrop-ai/core": "0.3.0"
+    "@raindrop-ai/core": "0.3.1"
   },
   "tsup": {
     "entry": [