@anvia/core 0.15.0 → 0.17.0

package/dist/{chunk-CNANX5OL.js → chunk-OPAEENGD.js}

@@ -4,26 +4,27 @@ import {
 import {
   parseToolArgs,
   toolResultContentToText
-} from "./chunk-BIYXGG33.js";
+} from "./chunk-SWUIBS4N.js";
 import {
   runControl,
   toolCallControl
 } from "./chunk-2ODTMRHP.js";
+import {
+  mapWithConcurrency
+} from "./chunk-OIMLU4SF.js";
 import {
   CompletionRequestBuilder,
   isStreamingCompletionModel
-} from "./chunk-QTZHDPQX.js";
+} from "./chunk-GCED5ZNM.js";
 import {
   AssistantContent,
   Message,
   ToolContent,
   Usage,
   assertCompletionRequestSupported,
+  getAssistantGenerationMetadata,
   textFromAssistantContent
-} from "./chunk-TIV3WFSR.js";
-import {
-  mapWithConcurrency
-} from "./chunk-OIMLU4SF.js";
+} from "./chunk-PJDL5UT2.js";
 import {
   appendGuardrailPolicies,
   hasEnforcedOutputGuardrails,
@@ -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;
@@ -325,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;
@@ -339,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);
     }
@@ -377,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) {
@@ -409,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();
@@ -442,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);
@@ -501,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()}`;
@@ -547,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 };
@@ -642,6 +691,12 @@ 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) {
@@ -658,7 +713,7 @@ function addTurnToToolCallDelta(turn, event) {
   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
@@ -1589,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
@@ -1644,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);
@@ -1767,18 +1823,27 @@ 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
         );
         assertCompletionRequestSupported(this.agent.model, request, { streaming: true });
         const providerRequest = this.providerTraceRequest(request, { stream: true });
-        const generationObservers = await runObservers.startGeneration(
-          this.generationStartArgs(currentTurns, request, providerRequest)
+        const generationStartArgs = this.generationStartArgs(
+          currentTurns,
+          request,
+          providerRequest
         );
-        const accumulator = new CompletionStreamAccumulator();
+        const generationObservers = await runObservers.startGeneration(generationStartArgs);
         const generationStartedAt = Date.now();
+        yield await emit({
+          type: "generation_start",
+          turn: currentTurns,
+          request,
+          modelInfo: generationStartArgs.modelInfo
+        });
+        const accumulator = new CompletionStreamAccumulator();
         let firstDeltaMs;
         const bufferResponseEvents = this.shouldBufferStreamResponseEvents();
         const emittedToolCallIds = /* @__PURE__ */ new Set();
@@ -1860,7 +1925,12 @@ var PromptRequest = class _PromptRequest {
                 yield await emit(event);
               }
             }
-            yield await emit({ type: "turn_end", turn: currentTurns, response });
+            yield await emit({
+              type: "turn_end",
+              turn: currentTurns,
+              response,
+              firstDeltaMs
+            });
             emittedTurnEnd = true;
           }
           if (this.steeringMessages.length > 0) {
@@ -1895,19 +1965,29 @@ 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);
             }
           }
           if (!emittedTurnEnd) {
-            yield await emit({ type: "turn_end", turn: currentTurns, response });
+            yield await emit({
+              type: "turn_end",
+              turn: currentTurns,
+              response,
+              firstDeltaMs
+            });
           }
           const result = {
             output: guardedOutput.output,
             usage,
             messages: [...newMessages],
             trace: runObservers.trace,
-            guardrails: [...this.guardrailDecisions]
+            guardrails: [...this.guardrailDecisions],
+            ...generationArtifacts(newMessages)
           };
           await this.runRunEndHook(result, newMessages);
           await runObservers.end(result);
@@ -1925,7 +2005,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;
         }
@@ -1946,7 +2028,12 @@ var PromptRequest = class _PromptRequest {
           [assistantMessage],
           pendingTurnMessages
         );
-        yield await emit({ type: "turn_end", turn: currentTurns, response });
+        yield await emit({
+          type: "turn_end",
+          turn: currentTurns,
+          response,
+          firstDeltaMs
+        });
         const toolResultEvents = createAsyncQueue();
         const toolResultsPromise = this.executeToolCalls(
           runId,
@@ -2042,7 +2129,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 }
         }
       }
     };
@@ -2177,6 +2266,7 @@ var PromptRequest = class _PromptRequest {
         agentDescription: this.agent.description,
         instructions: this.agent.instructions,
         trace: this.traceOptions,
+        promptRef: this.traceOptions?.promptRef,
         prompt: this.promptMessage,
         history: this.chatHistory,
         maxTurns: this.maxTurnCount
@@ -2386,7 +2476,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") {
@@ -2418,8 +2508,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",
@@ -2477,4 +2593,4 @@ export {
   extractRagText,
   PromptRequest
 };
-//# sourceMappingURL=chunk-CNANX5OL.js.map
+//# sourceMappingURL=chunk-OPAEENGD.js.map