@anvia/studio 0.1.1 → 0.1.2

package/dist/index.d.ts

@@ -53,9 +53,30 @@ type StudioTranscriptToolEntry = {
     callId?: string;
     args?: string;
     result?: string;
+    childEvents?: StudioTranscriptChildAgentEvent[];
     approval?: StudioToolApprovalTranscript;
     question?: StudioToolQuestionTranscript;
 };
+type StudioTranscriptChildAgentEvent = {
+    kind: "message";
+    agentId: string;
+    agentName?: string;
+    text: string;
+} | {
+    kind: "reasoning";
+    agentId: string;
+    agentName?: string;
+    reasoningId?: string;
+    text: string;
+} | {
+    kind: "tool";
+    agentId: string;
+    agentName?: string;
+    toolName: string;
+    callId?: string;
+    args?: string;
+    result?: string;
+};
 type StudioTranscriptEntry = StudioTranscriptChatEntry | StudioTranscriptReasoningEntry | StudioTranscriptToolEntry;
 type StudioSessionSummary = {
     id: string;
@@ -98,9 +119,10 @@ type StudioSessionStore = MemoryStore & {
     deleteSession?(id: string): boolean | Promise<boolean>;
 };
 type StudioTraceStatus = "running" | "success" | "error";
-type StudioTraceObservationKind = "generation" | "tool";
+type StudioTraceObservationKind = "agent" | "generation" | "tool";
 type StudioTraceObservation = {
     id: string;
+    parentObservationId?: string;
     kind: StudioTraceObservationKind;
     name: string;
     status: StudioTraceStatus;
@@ -344,4 +366,4 @@ type SqliteSessionStoreOptions = {
 };
 declare function createSqliteSessionStore(options?: SqliteSessionStoreOptions): StudioSessionStore & StudioTraceStore;
 
-export { type AgentRunRequest, type AgentRunResponse, type AgentRunStreamEvent, type AnviaStudio, type SqliteSessionStoreOptions, Studio, type StudioAgent, type StudioAgentConfig, type StudioAgentKnowledgeConfig, type StudioCapability, type StudioCapabilityConfig, type StudioConfig, type StudioErrorCode, type StudioErrorResponse, type StudioKnowledgeEvidence, type StudioKnowledgeEvidenceDocument, type StudioKnowledgeSourceKind, type StudioKnowledgeSourceSummary, type StudioKnowledgeSummary, type StudioOptions, type StudioServeOptions, type StudioSession, type StudioSessionCreateInput, type StudioSessionListOptions, type StudioSessionRunStatus, type StudioSessionRunTranscriptInput, type StudioSessionStore, type StudioSessionSummary, type StudioSessionTraceListOptions, type StudioStaticKnowledgeDocument, type StudioStores, type StudioToolApproval, type StudioToolApprovalDecision, type StudioToolApprovalRequestEvent, type StudioToolApprovalResultEvent, type StudioToolApprovalStatus, type StudioToolApprovalTranscript, type StudioToolQuestion, type StudioToolQuestionAnswer, type StudioToolQuestionChoice, type StudioToolQuestionPrompt, type StudioToolQuestionRequestEvent, type StudioToolQuestionResultEvent, type StudioToolQuestionStatus, type StudioToolQuestionTranscript, type StudioTrace, type StudioTraceListOptions, type StudioTraceObservation, type StudioTraceObservationKind, StudioTraceObserver, type StudioTraceObserverOptions, type StudioTraceStatus, type StudioTraceStore, type StudioTraceSummary, type StudioTranscriptChatEntry, type StudioTranscriptEntry, type StudioTranscriptReasoningEntry, type StudioTranscriptToolEntry, type StudioUiOptions, createSqliteSessionStore };
+export { type AgentRunRequest, type AgentRunResponse, type AgentRunStreamEvent, type AnviaStudio, type SqliteSessionStoreOptions, Studio, type StudioAgent, type StudioAgentConfig, type StudioAgentKnowledgeConfig, type StudioCapability, type StudioCapabilityConfig, type StudioConfig, type StudioErrorCode, type StudioErrorResponse, type StudioKnowledgeEvidence, type StudioKnowledgeEvidenceDocument, type StudioKnowledgeSourceKind, type StudioKnowledgeSourceSummary, type StudioKnowledgeSummary, type StudioOptions, type StudioServeOptions, type StudioSession, type StudioSessionCreateInput, type StudioSessionListOptions, type StudioSessionRunStatus, type StudioSessionRunTranscriptInput, type StudioSessionStore, type StudioSessionSummary, type StudioSessionTraceListOptions, type StudioStaticKnowledgeDocument, type StudioStores, type StudioToolApproval, type StudioToolApprovalDecision, type StudioToolApprovalRequestEvent, type StudioToolApprovalResultEvent, type StudioToolApprovalStatus, type StudioToolApprovalTranscript, type StudioToolQuestion, type StudioToolQuestionAnswer, type StudioToolQuestionChoice, type StudioToolQuestionPrompt, type StudioToolQuestionRequestEvent, type StudioToolQuestionResultEvent, type StudioToolQuestionStatus, type StudioToolQuestionTranscript, type StudioTrace, type StudioTraceListOptions, type StudioTraceObservation, type StudioTraceObservationKind, StudioTraceObserver, type StudioTraceObserverOptions, type StudioTraceStatus, type StudioTraceStore, type StudioTraceSummary, type StudioTranscriptChatEntry, type StudioTranscriptChildAgentEvent, type StudioTranscriptEntry, type StudioTranscriptReasoningEntry, type StudioTranscriptToolEntry, type StudioUiOptions, createSqliteSessionStore };

package/dist/index.js

@@ -79,34 +79,38 @@ var StudioRunTraceObserver = class {
   }
   startTool(args) {
     const startedAt = /* @__PURE__ */ new Date();
+    const childTrace = new ChildAgentToolTraceAccumulator(args);
     return {
+      streamEvent: (streamArgs) => {
+        childTrace.accept(streamArgs);
+      },
       end: (endArgs) => {
-        this.observations.push(
-          traceObservation({
-            kind: "tool",
-            name: args.toolName,
-            status: "success",
-            turn: args.turn,
-            startedAt,
-            input: parseOrString(args.args),
-            output: parseOrString(endArgs.result),
-            metadata: toolMetadata(args, endArgs.skipped)
-          })
-        );
+        const parentObservation = traceObservation({
+          kind: "tool",
+          name: args.toolName,
+          status: "success",
+          turn: args.turn,
+          startedAt,
+          input: parseOrString(args.args),
+          output: parseOrString(endArgs.result),
+          metadata: toolMetadata(args, endArgs.skipped)
+        });
+        this.observations.push(parentObservation);
+        this.observations.push(...childTrace.observations(parentObservation.id));
       },
       error: (errorArgs) => {
-        this.observations.push(
-          traceObservation({
-            kind: "tool",
-            name: args.toolName,
-            status: "error",
-            turn: args.turn,
-            startedAt,
-            input: parseOrString(args.args),
-            error: serializeError(errorArgs.error),
-            metadata: toolMetadata(args, false)
-          })
-        );
+        const parentObservation = traceObservation({
+          kind: "tool",
+          name: args.toolName,
+          status: "error",
+          turn: args.turn,
+          startedAt,
+          input: parseOrString(args.args),
+          error: serializeError(errorArgs.error),
+          metadata: toolMetadata(args, false)
+        });
+        this.observations.push(parentObservation);
+        this.observations.push(...childTrace.observations(parentObservation.id));
       }
     };
   }
@@ -157,10 +161,191 @@ var StudioRunTraceObserver = class {
     await store.saveTrace(trace);
   }
 };
+var ChildAgentToolTraceAccumulator = class {
+  constructor(parent) {
+    this.parent = parent;
+  }
+  parent;
+  agentStarts = /* @__PURE__ */ new Map();
+  generationStarts = /* @__PURE__ */ new Map();
+  toolStarts = [];
+  completedObservations = [];
+  accept(args) {
+    const wrapper = args.event;
+    const child = isRecord(wrapper.event) ? wrapper.event : void 0;
+    if (child === void 0) {
+      return;
+    }
+    const agentId = wrapper.agentId;
+    const agentName = wrapper.agentName;
+    const childTurn = typeof child.turn === "number" ? child.turn : this.parent.turn;
+    if (!this.agentStarts.has(agentId)) {
+      this.agentStarts.set(agentId, {
+        startedAt: /* @__PURE__ */ new Date(),
+        agentId,
+        ...agentName === void 0 ? {} : { agentName }
+      });
+    }
+    if (child.type === "turn_start") {
+      this.generationStarts.set(generationKey(agentId, childTurn), {
+        startedAt: /* @__PURE__ */ new Date(),
+        input: toJsonValue({
+          prompt: child.prompt,
+          history: child.history
+        }),
+        agentId,
+        ...agentName === void 0 ? {} : { agentName },
+        childTurn
+      });
+      return;
+    }
+    if (child.type === "turn_end") {
+      const key = generationKey(agentId, childTurn);
+      const start = this.generationStarts.get(key);
+      this.generationStarts.delete(key);
+      this.completedObservations.push(
+        traceObservation({
+          kind: "generation",
+          name: `${agentLabel(agentId, agentName)}.model.turn.${childTurn}`,
+          status: "success",
+          turn: this.parent.turn,
+          startedAt: start?.startedAt ?? /* @__PURE__ */ new Date(),
+          ...start?.input === void 0 ? {} : { input: start.input },
+          output: toJsonValue(child.response),
+          metadata: this.childMetadata(agentId, agentName, childTurn)
+        })
+      );
+      return;
+    }
+    if (child.type === "tool_call" && isRecord(child.toolCall)) {
+      const toolCall = child.toolCall;
+      const toolCallFunction = isRecord(toolCall.function) ? toolCall.function : void 0;
+      const toolName = typeof toolCallFunction?.name === "string" ? toolCallFunction.name : "tool";
+      const callId = typeof toolCall.callId === "string" ? toolCall.callId : typeof toolCall.id === "string" ? toolCall.id : void 0;
+      this.toolStarts.push({
+        startedAt: /* @__PURE__ */ new Date(),
+        agentId,
+        ...agentName === void 0 ? {} : { agentName },
+        childTurn,
+        toolName,
+        ...callId === void 0 ? {} : { toolCallId: callId },
+        input: toJsonValue(toolCallFunction?.arguments ?? {}),
+        completed: false
+      });
+      return;
+    }
+    if (child.type === "tool_result") {
+      const toolName = typeof child.toolName === "string" ? child.toolName : "tool";
+      const toolCallId = typeof child.toolCallId === "string" ? child.toolCallId : void 0;
+      const internalCallId = typeof child.internalCallId === "string" ? child.internalCallId : void 0;
+      const start = this.findToolStart(agentId, toolName, toolCallId);
+      const input = start?.input ?? (typeof child.args === "string" ? parseOrString(child.args) : void 0);
+      if (start !== void 0) {
+        start.completed = true;
+      }
+      this.completedObservations.push(
+        traceObservation({
+          kind: "tool",
+          name: `${agentLabel(agentId, agentName)}.${toolName}`,
+          status: "success",
+          turn: this.parent.turn,
+          startedAt: start?.startedAt ?? /* @__PURE__ */ new Date(),
+          ...input === void 0 ? {} : { input },
+          ...typeof child.result === "string" ? { output: parseOrString(child.result) } : {},
+          metadata: {
+            ...this.childMetadata(agentId, agentName, childTurn),
+            ...toolCallId === void 0 ? {} : { toolCallId },
+            ...internalCallId === void 0 ? {} : { internalCallId }
+          }
+        })
+      );
+      return;
+    }
+    if (child.type === "error") {
+      this.completedObservations.push(
+        traceObservation({
+          kind: "tool",
+          name: `${agentLabel(agentId, agentName)}.error`,
+          status: "error",
+          turn: this.parent.turn,
+          startedAt: /* @__PURE__ */ new Date(),
+          error: serializeError(child.error),
+          metadata: this.childMetadata(agentId, agentName, childTurn)
+        })
+      );
+    }
+  }
+  observations(parentObservationId) {
+    const observations = [];
+    const agentObservationIds = /* @__PURE__ */ new Map();
+    for (const agentStart of this.agentStarts.values()) {
+      const agentChildren = this.completedObservations.filter(
+        (observation) => isRecord(observation.metadata) && observation.metadata.childAgentId === agentStart.agentId
+      );
+      const childStartTimes = agentChildren.map((observation) => Date.parse(observation.startedAt));
+      const childEndTimes = agentChildren.map(
+        (observation) => Date.parse(observation.endedAt ?? observation.startedAt)
+      );
+      const startedAt = childStartTimes.length === 0 ? agentStart.startedAt : new Date(Math.min(agentStart.startedAt.getTime(), ...childStartTimes));
+      const endedAt = childEndTimes.length === 0 ? /* @__PURE__ */ new Date() : new Date(Math.max(...childEndTimes));
+      const agentObservation = traceObservation({
+        parentObservationId,
+        kind: "agent",
+        name: `${agentLabel(agentStart.agentId, agentStart.agentName)}.run`,
+        status: agentChildren.some((observation) => observation.status === "error") ? "error" : "success",
+        turn: this.parent.turn,
+        startedAt,
+        endedAt,
+        metadata: this.childMetadata(agentStart.agentId, agentStart.agentName, this.parent.turn)
+      });
+      observations.push(agentObservation);
+      agentObservationIds.set(agentStart.agentId, agentObservation.id);
+    }
+    for (const observation of this.completedObservations) {
+      const childAgentId = isRecord(observation.metadata) ? stringValue(observation.metadata.childAgentId) : void 0;
+      const childAgentObservationId = childAgentId === void 0 ? void 0 : agentObservationIds.get(childAgentId);
+      observations.push({
+        ...observation,
+        parentObservationId: childAgentObservationId ?? parentObservationId
+      });
+    }
+    return observations;
+  }
+  findToolStart(agentId, toolName, toolCallId) {
+    for (let index = this.toolStarts.length - 1; index >= 0; index -= 1) {
+      const start = this.toolStarts[index];
+      if (start === void 0 || start.completed || start.agentId !== agentId || start.toolName !== toolName) {
+        continue;
+      }
+      if (toolCallId === void 0 || start.toolCallId === toolCallId) {
+        return start;
+      }
+    }
+    return void 0;
+  }
+  childMetadata(agentId, agentName, childTurn) {
+    return compactJsonObject({
+      source: "agent_tool_event",
+      childAgentId: agentId,
+      childAgentName: agentName,
+      childTurn,
+      parentToolName: this.parent.toolName,
+      parentInternalCallId: this.parent.internalCallId,
+      parentToolCallId: this.parent.toolCallId
+    });
+  }
+};
+function generationKey(agentId, turn) {
+  return `${agentId}:${turn}`;
+}
+function agentLabel(agentId, agentName) {
+  return (agentName ?? agentId).replaceAll(/\s+/g, "_");
+}
 function traceObservation(props) {
-  const endedAt = /* @__PURE__ */ new Date();
+  const endedAt = props.endedAt ?? /* @__PURE__ */ new Date();
   return {
     id: globalThis.crypto.randomUUID(),
+    ...props.parentObservationId === void 0 ? {} : { parentObservationId: props.parentObservationId },
     kind: props.kind,
     name: props.name,
     status: props.status,
@@ -209,6 +394,12 @@ function parseOrString(value) {
     return value;
   }
 }
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function stringValue(value) {
+  return typeof value === "string" ? value : void 0;
+}
 function serializeError(error) {
   if (error instanceof Error) {
     return compactJsonObject({
@@ -1593,7 +1784,7 @@ async function recentKnowledgeEvidence(traceStore, limit) {
 }
 function evidenceFromTrace(trace) {
   return trace.observations.flatMap((observation) => {
-    if (observation.kind !== "generation" || !isRecord(observation.input)) {
+    if (observation.kind !== "generation" || !isRecord2(observation.input)) {
       return [];
     }
     const documents = Array.isArray(observation.input.documents) ? observation.input.documents.flatMap((document) => evidenceDocument(document)) : [];
@@ -1646,7 +1837,7 @@ function messageText(value) {
   if (typeof value === "string") {
     return value.trim();
   }
-  if (!isRecord(value)) {
+  if (!isRecord2(value)) {
     return "";
   }
   if (typeof value.text === "string") {
@@ -1664,7 +1855,7 @@ function contentText(value) {
   if (typeof value === "string") {
     return value.trim();
   }
-  if (!isRecord(value)) {
+  if (!isRecord2(value)) {
     return "";
   }
   if (typeof value.text === "string") {
@@ -1673,12 +1864,12 @@ function contentText(value) {
   return "";
 }
 function evidenceDocument(value) {
-  if (!isRecord(value)) {
+  if (!isRecord2(value)) {
     return [];
   }
   const id = typeof value.id === "string" ? value.id : void 0;
   const text = typeof value.text === "string" ? value.text : void 0;
-  const additionalProps = isRecord(value.additionalProps) ? jsonObjectFromRecord(value.additionalProps) : void 0;
+  const additionalProps = isRecord2(value.additionalProps) ? jsonObjectFromRecord(value.additionalProps) : void 0;
   if (id === void 0 && text === void 0 && additionalProps === void 0) {
     return [];
   }
@@ -1691,7 +1882,7 @@ function evidenceDocument(value) {
   ];
 }
 function evidenceToolName(value) {
-  if (!isRecord(value) || typeof value.name !== "string") {
+  if (!isRecord2(value) || typeof value.name !== "string") {
     return [];
   }
   return [value.name];
@@ -1699,7 +1890,7 @@ function evidenceToolName(value) {
 function jsonObjectFromRecord(value) {
   return compactJsonObject2(value);
 }
-function isRecord(value) {
+function isRecord2(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 
@@ -2117,6 +2308,22 @@ function acceptTranscriptStreamEvent(transcript, event) {
     matched.args = matched.args ?? event.args;
     matched.result = event.result;
   }
+  if (event.type === "agent_tool_event") {
+    const matched = findTranscriptToolEntry(transcript, event.toolName, event.toolCallId);
+    if (matched === void 0) {
+      transcript.push({
+        entryId: transcript.length,
+        kind: "tool",
+        toolName: event.toolName,
+        ...event.toolCallId === void 0 ? {} : { callId: event.toolCallId },
+        childEvents: [childAgentTranscriptEvent(event)].filter(
+          (childEvent) => childEvent !== void 0
+        )
+      });
+      return;
+    }
+    appendChildAgentTranscriptEvent(matched, event);
+  }
   if (event.type === "tool_approval_request") {
     const matched = findTranscriptToolEntry(
       transcript,
@@ -2189,6 +2396,112 @@ function approvalCallId(approval) {
 function questionCallId(question) {
   return question.callId ?? question.toolCallId;
 }
+function appendChildAgentTranscriptEvent(entry, event) {
+  const childEvent = childAgentTranscriptEvent(event);
+  if (childEvent === void 0) {
+    return;
+  }
+  const childEvents = entry.childEvents ?? [];
+  if (childEvent.kind === "message") {
+    const last = childEvents.at(-1);
+    if (last?.kind === "message" && last.agentId === childEvent.agentId) {
+      last.text = `${last.text}${childEvent.text}`;
+    } else {
+      childEvents.push(childEvent);
+    }
+  } else if (childEvent.kind === "reasoning") {
+    const last = childEvents.at(-1);
+    if (last?.kind === "reasoning" && last.agentId === childEvent.agentId && (last.reasoningId ?? "") === (childEvent.reasoningId ?? "")) {
+      last.text = `${last.text}${childEvent.text}`;
+    } else {
+      childEvents.push(childEvent);
+    }
+  } else {
+    const matched = findChildAgentToolEvent(childEvents, childEvent);
+    if (matched === void 0) {
+      childEvents.push(childEvent);
+    } else {
+      if (matched.args === void 0 && childEvent.args !== void 0) {
+        matched.args = childEvent.args;
+      }
+      if (childEvent.result !== void 0) {
+        matched.result = childEvent.result;
+      }
+    }
+  }
+  entry.childEvents = childEvents;
+}
+function childAgentTranscriptEvent(event) {
+  const child = event.event;
+  if (child.type === "text_delta") {
+    return {
+      kind: "message",
+      agentId: event.agentId,
+      ...event.agentName === void 0 ? {} : { agentName: event.agentName },
+      text: child.delta
+    };
+  }
+  if (child.type === "reasoning_delta") {
+    return {
+      kind: "reasoning",
+      agentId: event.agentId,
+      ...event.agentName === void 0 ? {} : { agentName: event.agentName },
+      ...child.id === void 0 ? {} : { reasoningId: child.id },
+      text: child.delta
+    };
+  }
+  if (child.type === "tool_call") {
+    return {
+      kind: "tool",
+      agentId: event.agentId,
+      ...event.agentName === void 0 ? {} : { agentName: event.agentName },
+      toolName: child.toolCall.function.name,
+      ...child.toolCall.callId === void 0 && child.toolCall.id === void 0 ? {} : { callId: child.toolCall.callId ?? child.toolCall.id },
+      args: formatJson2(child.toolCall.function.arguments)
+    };
+  }
+  if (child.type === "tool_result") {
+    return {
+      kind: "tool",
+      agentId: event.agentId,
+      ...event.agentName === void 0 ? {} : { agentName: event.agentName },
+      toolName: child.toolName,
+      ...child.toolCallId === void 0 ? {} : { callId: child.toolCallId },
+      args: child.args,
+      result: child.result
+    };
+  }
+  if (child.type === "error") {
+    return {
+      kind: "message",
+      agentId: event.agentId,
+      ...event.agentName === void 0 ? {} : { agentName: event.agentName },
+      text: `Error: ${errorText(child.error)}`
+    };
+  }
+  return void 0;
+}
+function errorText(error) {
+  if (error instanceof Error) {
+    return error.message;
+  }
+  if (typeof error === "string") {
+    return error;
+  }
+  return JSON.stringify(serializeError2(error));
+}
+function findChildAgentToolEvent(childEvents, event) {
+  for (let index = childEvents.length - 1; index >= 0; index -= 1) {
+    const childEvent = childEvents[index];
+    if (childEvent?.kind !== "tool" || childEvent.agentId !== event.agentId || childEvent.toolName !== event.toolName || childEvent.result !== void 0) {
+      continue;
+    }
+    if (event.callId === void 0 || childEvent.callId === event.callId) {
+      return childEvent;
+    }
+  }
+  return void 0;
+}
 function transcriptFromMessages(messages) {
   const transcript = [];
   for (const message of messages) {