@anvia/core 0.14.0 → 0.16.0

package/dist/{chunk-BOJP4ARV.js → chunk-PJCJ45RJ.js}

@@ -4,7 +4,7 @@ import {
 import {
   parseToolArgs,
   toolResultContentToText
-} from "./chunk-2XD7Z7NB.js";
+} from "./chunk-4PMAWIGM.js";
 import {
   runControl,
   toolCallControl
@@ -12,15 +12,16 @@ import {
 import {
   CompletionRequestBuilder,
   isStreamingCompletionModel
-} from "./chunk-HHHNMP3V.js";
+} from "./chunk-K72JKDZ7.js";
 import {
   AssistantContent,
   Message,
   ToolContent,
   Usage,
   assertCompletionRequestSupported,
+  getAssistantGenerationMetadata,
   textFromAssistantContent
-} from "./chunk-HU55YP24.js";
+} from "./chunk-R6QERZAB.js";
 import {
   mapWithConcurrency
 } from "./chunk-OIMLU4SF.js";
@@ -293,6 +294,8 @@ var CompletionStreamAccumulator = class {
   reasoningByKey = /* @__PURE__ */ new Map();
   reasoningKeyById = /* @__PURE__ */ new Map();
   toolCalls = /* @__PURE__ */ new Map();
+  sources = /* @__PURE__ */ new Map();
+  providerToolCalls = /* @__PURE__ */ new Map();
   finalResponse;
   messageId;
   nextTextKey = 0;
@@ -313,7 +316,11 @@ var CompletionStreamAccumulator = class {
       if (event.name !== void 0 && event.name.length > 0) toolCall.name = event.name;
       if (event.signature !== void 0) toolCall.signature = event.signature;
       if (event.argumentsDelta !== void 0) {
-        toolCall.argumentsText += event.argumentsDelta;
+        if (event.argumentsMode === "replace") {
+          toolCall.argumentsText = event.argumentsDelta;
+        } else {
+          toolCall.argumentsText += event.argumentsDelta;
+        }
       }
       return void 0;
     }
@@ -321,6 +328,14 @@ var CompletionStreamAccumulator = class {
       this.upsertToolCall(event.toolCall);
       return { type: "tool_call", toolCall: event.toolCall };
     }
+    if (event.type === "source") {
+      this.sources.set(sourceKey(event.source), event.source);
+      return { type: "source", source: event.source };
+    }
+    if (event.type === "provider_tool_call") {
+      this.providerToolCalls.set(event.toolCall.id, event.toolCall);
+      return { type: "provider_tool_call", toolCall: event.toolCall };
+    }
     if (event.type === "message_id") {
       this.messageId = event.id;
       return void 0;
@@ -335,7 +350,7 @@ var CompletionStreamAccumulator = class {
     const accumulatedResponse = this.buildAccumulatedResponse();
     if (this.finalResponse !== void 0) {
       if (accumulatedResponse.choice.length === 0) {
-        return this.withMessageIdFallback(this.finalResponse, accumulatedResponse);
+        return this.withAccumulatedArtifacts(this.finalResponse, accumulatedResponse);
       }
       return this.mergeFinalResponse(accumulatedResponse, this.finalResponse);
     }
@@ -373,6 +388,14 @@ var CompletionStreamAccumulator = class {
     if (this.messageId !== void 0) {
       response.messageId = this.messageId;
     }
+    const sources = [...this.sources.values()];
+    if (sources.length > 0) {
+      response.sources = sources;
+    }
+    const providerToolCalls = [...this.providerToolCalls.values()];
+    if (providerToolCalls.length > 0) {
+      response.providerToolCalls = providerToolCalls;
+    }
     return response;
   }
   upsertToolCall(toolCall) {
@@ -405,7 +428,7 @@ var CompletionStreamAccumulator = class {
       if (finalResponse.messageId !== void 0) {
         mergedResponse.messageId = finalResponse.messageId;
       }
-      return this.withMessageIdFallback(mergedResponse, accumulatedResponse);
+      return this.withAccumulatedArtifacts(mergedResponse, accumulatedResponse);
     }
     const finalById = /* @__PURE__ */ new Map();
     const finalByCallId = /* @__PURE__ */ new Map();
@@ -438,7 +461,7 @@ var CompletionStreamAccumulator = class {
         choice.push(content);
       }
     }
-    return this.withMessageIdFallback({ ...finalResponse, choice }, accumulatedResponse);
+    return this.withAccumulatedArtifacts({ ...finalResponse, choice }, accumulatedResponse);
   }
   appendText(delta) {
     const lastPart = this.orderedParts.at(-1);
@@ -497,6 +520,19 @@ var CompletionStreamAccumulator = class {
     }
     return { ...response, messageId: accumulatedResponse.messageId };
   }
+  withAccumulatedArtifacts(response, accumulatedResponse) {
+    const withMessageId = this.withMessageIdFallback(response, accumulatedResponse);
+    const sources = mergeSources(accumulatedResponse.sources, response.sources);
+    const providerToolCalls = mergeProviderToolCalls(
+      accumulatedResponse.providerToolCalls,
+      response.providerToolCalls
+    );
+    return {
+      ...withMessageId,
+      ...sources.length === 0 ? {} : { sources },
+      ...providerToolCalls.length === 0 ? {} : { providerToolCalls }
+    };
+  }
   createTextKey() {
     this.nextTextKey += 1;
     return `text_${this.nextTextKey.toString()}`;
@@ -543,6 +579,23 @@ var CompletionStreamAccumulator = class {
     reasoning.content.push({ type: "redacted", data: event.delta });
   }
 };
+function sourceKey(source) {
+  return `${source.url}\0${source.startIndex ?? ""}\0${source.endIndex ?? ""}`;
+}
+function mergeSources(accumulated, final) {
+  const sources = /* @__PURE__ */ new Map();
+  for (const source of [...accumulated ?? [], ...final ?? []]) {
+    sources.set(sourceKey(source), source);
+  }
+  return [...sources.values()];
+}
+function mergeProviderToolCalls(accumulated, final) {
+  const toolCalls = /* @__PURE__ */ new Map();
+  for (const toolCall of [...accumulated ?? [], ...final ?? []]) {
+    toolCalls.set(toolCall.id, toolCall);
+  }
+  return [...toolCalls.values()];
+}
 function reasoningContent(reasoning) {
   const content = reasoning.content === void 0 ? { type: "reasoning", text: reasoning.text } : { type: "reasoning", text: reasoning.text, content: reasoning.content };
   return reasoning.id === void 0 ? content : { ...content, id: reasoning.id };
@@ -638,22 +691,42 @@ function addTurn(turn, event) {
     if (event.signature !== void 0) mapped.signature = event.signature;
     return mapped;
   }
+  if (event.type === "source") {
+    return { type: "source", turn, source: event.source };
+  }
+  if (event.type === "provider_tool_call") {
+    return { type: "provider_tool_call", turn, toolCall: event.toolCall };
+  }
   return { type: "tool_call", turn, toolCall: event.toolCall };
 }
+function addTurnToToolCallDelta(turn, event) {
+  const mapped = {
+    type: "tool_call_delta",
+    turn,
+    id: event.id
+  };
+  if (event.callId !== void 0) mapped.callId = event.callId;
+  if (event.name !== void 0) mapped.name = event.name;
+  if (event.argumentsDelta !== void 0) mapped.argumentsDelta = event.argumentsDelta;
+  if (event.argumentsMode !== void 0) mapped.argumentsMode = event.argumentsMode;
+  if (event.signature !== void 0) mapped.signature = event.signature;
+  return mapped;
+}
 function isGenerationDeltaEvent(type) {
-  return type === "text_delta" || type === "reasoning_delta" || type === "tool_call_delta" || type === "tool_call";
+  return type === "text_delta" || type === "reasoning_delta" || type === "tool_call_delta" || type === "tool_call" || type === "source" || type === "provider_tool_call";
 }
 
 // src/internal/prompt-runtime/tool-execution.ts
 var MCP_TOOL_METADATA_KEY = /* @__PURE__ */ Symbol.for("anvia.mcp.tool.metadata");
 var ToolCallExecutor = class {
-  constructor(agent, activeHook, approvals, runContext, concurrency, requestMiddlewares, cancel) {
+  constructor(agent, activeHook, approvals, runContext, concurrency, requestMiddlewares, includeToolCallDeltas, cancel) {
     this.agent = agent;
     this.activeHook = activeHook;
     this.approvals = approvals;
     this.runContext = runContext;
     this.concurrency = concurrency;
     this.requestMiddlewares = requestMiddlewares;
+    this.includeToolCallDeltas = includeToolCallDeltas;
     this.cancel = cancel;
   }
   agent;
@@ -662,6 +735,7 @@ var ToolCallExecutor = class {
   runContext;
   concurrency;
   requestMiddlewares;
+  includeToolCallDeltas;
   cancel;
   async execute(toolCalls, onResult, onStreamEvent, observation) {
     for (const toolCall of toolCalls) {
@@ -829,7 +903,7 @@ var ToolCallExecutor = class {
     }) : effectiveArgs;
     hookArgs.args = middlewareArgs;
     try {
-      return await this.agent.callTool(toolCall.function.name, middlewareArgs, {
+      const toolContext = {
         emitStreamEvent: async (event) => {
           const streamEventArgs = {
             turn: observation?.turn ?? 0,
@@ -848,7 +922,9 @@ var ToolCallExecutor = class {
             onStreamEvent?.(payload);
           }
         }
-      });
+      };
+      toolContext.includeToolCallDeltas = this.includeToolCallDeltas;
+      return await this.agent.callTool(toolCall.function.name, middlewareArgs, toolContext);
     } catch (error) {
       const errorAction = await this.activeHook?.onToolError?.({
         ...hookArgs,
@@ -1568,7 +1644,7 @@ var PromptRequest = class _PromptRequest {
         const ragText = extractRagText(prompt);
         const dynamicContext = await fetchDynamicContext(this.agent, ragText);
         const toolDefs = await fetchToolDefinitions(this.agent, ragText);
-        let request = new CompletionRequestBuilder(this.agent.model, prompt).instructions(this.agent.instructions).messages(historyForRequest).documents([...this.agent.staticContext, ...dynamicContext]).tools(toolDefs).temperature(this.agent.temperature).maxTokens(this.agent.maxTokens).additionalParams(this.agent.additionalParams).toolChoice(this.agent.toolChoice).outputSchema(this.agent.outputSchema).build();
+        let request = new CompletionRequestBuilder(this.agent.model, prompt).instructions(this.agent.instructions).messages(historyForRequest).documents([...this.agent.staticContext, ...dynamicContext]).tools([...toolDefs, ...this.agent.providerTools]).temperature(this.agent.temperature).maxTokens(this.agent.maxTokens).additionalParams(this.agent.additionalParams).toolChoice(this.agent.toolChoice).outputSchema(this.agent.outputSchema).build();
         request = await this.runCompletionRequestMiddlewares(
           request,
           currentTurns
@@ -1623,7 +1699,8 @@ var PromptRequest = class _PromptRequest {
             usage,
             messages: [...newMessages],
             trace: runObservers.trace,
-            guardrails: [...this.guardrailDecisions]
+            guardrails: [...this.guardrailDecisions],
+            ...generationArtifacts(newMessages)
           };
           await this.runRunEndHook(result, newMessages);
           await runObservers.end(result);
@@ -1674,11 +1751,12 @@ var PromptRequest = class _PromptRequest {
       throw finalError;
     }
   }
-  async *stream() {
+  async *stream(options = {}) {
     if (!this.agent.model.capabilities.streaming || !isStreamingCompletionModel(this.agent.model)) {
       throw new Error("This completion model does not support streaming");
     }
     this.startRun();
+    const includeToolCallDeltas = options.includeToolCallDeltas !== false;
     const runId = globalThis.crypto.randomUUID();
     const emit = async (event) => {
       await this.recordAgentEvent(runId, event);
@@ -1745,7 +1823,7 @@ var PromptRequest = class _PromptRequest {
         const ragText = extractRagText(prompt);
         const dynamicContext = await fetchDynamicContext(this.agent, ragText);
         const toolDefs = await fetchToolDefinitions(this.agent, ragText);
-        let request = new CompletionRequestBuilder(this.agent.model, prompt).instructions(this.agent.instructions).messages(historyForRequest).documents([...this.agent.staticContext, ...dynamicContext]).tools(toolDefs).temperature(this.agent.temperature).maxTokens(this.agent.maxTokens).additionalParams(this.agent.additionalParams).toolChoice(this.agent.toolChoice).outputSchema(this.agent.outputSchema).build();
+        let request = new CompletionRequestBuilder(this.agent.model, prompt).instructions(this.agent.instructions).messages(historyForRequest).documents([...this.agent.staticContext, ...dynamicContext]).tools([...toolDefs, ...this.agent.providerTools]).temperature(this.agent.temperature).maxTokens(this.agent.maxTokens).additionalParams(this.agent.additionalParams).toolChoice(this.agent.toolChoice).outputSchema(this.agent.outputSchema).build();
         request = await this.runCompletionRequestMiddlewares(
           request,
           currentTurns
@@ -1777,6 +1855,9 @@ var PromptRequest = class _PromptRequest {
                   firstDeltaMs = Date.now() - generationStartedAt;
                 }
                 const mapped = accumulator.accept(event);
+                if (includeToolCallDeltas && event.type === "tool_call_delta") {
+                  yield await emit(addTurnToToolCallDelta(currentTurns, event));
+                }
                 if (mapped !== void 0) {
                   await generationObservers.update?.({ turn: currentTurns, delta: mapped });
                   if (mapped.type === "tool_call") {
@@ -1870,7 +1951,11 @@ var PromptRequest = class _PromptRequest {
             pendingTurnMessages
           );
           if (!emittedTurnEnd && (bufferResponseEvents || bufferOutputDeltas)) {
-            for (const event of responseStreamEvents(currentTurns, response)) {
+            for (const event of responseStreamEvents(
+              currentTurns,
+              response,
+              bufferResponseEvents
+            )) {
               yield await emit(event);
             }
           }
@@ -1882,7 +1967,8 @@ var PromptRequest = class _PromptRequest {
             usage,
             messages: [...newMessages],
             trace: runObservers.trace,
-            guardrails: [...this.guardrailDecisions]
+            guardrails: [...this.guardrailDecisions],
+            ...generationArtifacts(newMessages)
           };
           await this.runRunEndHook(result, newMessages);
           await runObservers.end(result);
@@ -1900,7 +1986,9 @@ var PromptRequest = class _PromptRequest {
             usage: result.usage,
             messages: result.messages,
             trace: result.trace,
-            guardrails: result.guardrails
+            guardrails: result.guardrails,
+            sources: result.sources,
+            providerToolCalls: result.providerToolCalls
           });
           return;
         }
@@ -1936,7 +2024,8 @@ var PromptRequest = class _PromptRequest {
           {
             turn: currentTurns,
             runObservers,
-            toolDefinitions: request.tools
+            toolDefinitions: request.tools,
+            includeToolCallDeltas
           }
         );
         toolResultsPromise.then(
@@ -1973,8 +2062,8 @@ var PromptRequest = class _PromptRequest {
       throw finalError;
     }
   }
-  readableStream() {
-    return toReadableStream(this.stream());
+  readableStream(options = {}) {
+    return toReadableStream(this.stream(options));
   }
   async runCompletion(request, turn, runObservers) {
     assertCompletionRequestSupported(this.agent.model, request);
@@ -2016,7 +2105,9 @@ var PromptRequest = class _PromptRequest {
         generation: {
           provider: this.agent.model.provider,
           model: request.model ?? this.agent.model.defaultModel,
-          usage: { ...response.usage }
+          usage: { ...response.usage },
+          ...response.sources === void 0 ? {} : { sources: response.sources },
+          ...response.providerToolCalls === void 0 ? {} : { providerToolCalls: response.providerToolCalls }
         }
       }
     };
@@ -2092,6 +2183,7 @@ var PromptRequest = class _PromptRequest {
       },
       this.concurrency,
       this.requestMiddlewares,
+      observation?.includeToolCallDeltas ?? false,
       (reason) => this.cancelled(newMessages, reason)
     );
     return executor.execute(toolCalls, onResult, onStreamEvent, observation);
@@ -2359,7 +2451,7 @@ function normalizeSteeringInput(input) {
   }
   return Array.isArray(input) ? [...input] : [input];
 }
-function responseStreamEvents(turn, response) {
+function responseStreamEvents(turn, response, includeProviderArtifacts = true) {
   const events = [];
   for (const item of response.choice) {
     if (item.type === "text") {
@@ -2391,8 +2483,34 @@ function responseStreamEvents(turn, response) {
       events.push({ type: "tool_call", turn, toolCall: item });
     }
   }
+  if (includeProviderArtifacts) {
+    for (const source of response.sources ?? []) {
+      events.push({ type: "source", turn, source });
+    }
+    for (const toolCall of response.providerToolCalls ?? []) {
+      events.push({ type: "provider_tool_call", turn, toolCall });
+    }
+  }
   return events;
 }
+function generationArtifacts(messages) {
+  const sources = /* @__PURE__ */ new Map();
+  const providerToolCalls = /* @__PURE__ */ new Map();
+  for (const message of messages) {
+    const metadata = getAssistantGenerationMetadata(message);
+    for (const source of metadata?.sources ?? []) {
+      const key = `${source.url}\0${source.startIndex ?? ""}\0${source.endIndex ?? ""}`;
+      sources.set(key, source);
+    }
+    for (const toolCall of metadata?.providerToolCalls ?? []) {
+      providerToolCalls.set(toolCall.id, toolCall);
+    }
+  }
+  return {
+    ...sources.size === 0 ? {} : { sources: [...sources.values()] },
+    ...providerToolCalls.size === 0 ? {} : { providerToolCalls: [...providerToolCalls.values()] }
+  };
+}
 function reasoningDeltaEvent2(turn, delta, details = {}) {
   const event = {
     type: "reasoning_delta",
@@ -2450,4 +2568,4 @@ export {
   extractRagText,
   PromptRequest
 };
-//# sourceMappingURL=chunk-BOJP4ARV.js.map
+//# sourceMappingURL=chunk-PJCJ45RJ.js.map