@anvia/studio 0.1.0 → 0.1.1

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { Message, JsonObject, AgentTraceOptions, PromptResponse, AgentStreamEvent, JsonValue, Usage, AgentTraceInfo, Agent, AgentObserver, AgentRunStartArgs, AgentRunObserver } from '@anvia/core';
+import { Message, JsonObject, AgentTraceOptions, PromptResponse, AgentStreamEvent, JsonValue, Usage, AgentTraceInfo, Agent, MemoryStore, AgentObserver, AgentRunStartArgs, AgentRunObserver } from '@anvia/core';
 import { Hono } from 'hono';
 
 type StudioCapability = "agents" | "approvals" | "knowledge" | "observability" | "sessions" | "traces";
@@ -80,18 +80,21 @@ type StudioSessionListOptions = {
     agentId?: string;
     limit: number;
 };
-type StudioSessionAppendInput = {
+type StudioSessionRunStatus = "running" | "success" | "error";
+type StudioSessionRunTranscriptInput = {
     id: string;
+    runId: string;
     title?: string;
-    messages: Message[];
     transcript: StudioTranscriptEntry[];
+    status: StudioSessionRunStatus;
+    error?: JsonValue;
 };
-type StudioSessionStore = {
+type StudioSessionStore = MemoryStore & {
     readonly kind?: string;
     listSessions(options: StudioSessionListOptions): StudioSessionSummary[] | Promise<StudioSessionSummary[]>;
     createSession(input: StudioSessionCreateInput): StudioSessionSummary | Promise<StudioSessionSummary>;
     getSession(id: string): StudioSession | undefined | Promise<StudioSession | undefined>;
-    appendSessionRun(input: StudioSessionAppendInput): StudioSession | undefined | Promise<StudioSession | undefined>;
+    saveSessionRunTranscript(input: StudioSessionRunTranscriptInput): StudioSession | undefined | Promise<StudioSession | undefined>;
     deleteSession?(id: string): boolean | Promise<boolean>;
 };
 type StudioTraceStatus = "running" | "success" | "error";
@@ -341,4 +344,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 StudioSessionAppendInput, type StudioSessionCreateInput, type StudioSessionListOptions, 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 StudioTranscriptEntry, type StudioTranscriptReasoningEntry, type StudioTranscriptToolEntry, type StudioUiOptions, createSqliteSessionStore };

package/dist/index.js

@@ -1,7 +1,9 @@
 // src/runtime/studio.ts
 import {
   Agent,
-  createHook as createHook3
+  createHook as createHook3,
+  Message,
+  resolveMemoryOptions
 } from "@anvia/core";
 import { serve } from "@hono/node-server";
 import { Hono as HonoApp } from "hono";
@@ -544,15 +546,14 @@ var SqliteSessionStore = class {
     };
   }
   getSession(id) {
-    const db = this.database();
-    const row = db.prepare(
-      `SELECT id, agent_id, title, metadata_json, messages_json, transcript_json, created_at, updated_at
-         FROM runner_sessions
-         WHERE id = $id`
-    ).get({ $id: id });
-    return row === void 0 ? void 0 : toSession(row);
+    const row = this.getSessionRow(id);
+    return row === void 0 ? void 0 : toSession(row, this.listSessionRunRows(id));
   }
-  appendSessionRun(input) {
+  load(context) {
+    const session = this.getSession(context.sessionId);
+    return Promise.resolve(session?.messages ?? []);
+  }
+  append(input) {
     const db = this.database();
     try {
       db.exec("BEGIN IMMEDIATE");
@@ -560,39 +561,132 @@ var SqliteSessionStore = class {
         `SELECT id, agent_id, title, metadata_json, messages_json, transcript_json, created_at, updated_at
            FROM runner_sessions
            WHERE id = $id`
-      ).get({ $id: input.id });
+      ).get({ $id: input.context.sessionId });
       if (row === void 0) {
         db.exec("ROLLBACK");
-        return void 0;
+        return Promise.resolve();
       }
       const current = toSession(row);
       const messages = [...current.messages, ...input.messages];
-      const transcript = renumberTranscript([...current.transcript, ...input.transcript]);
-      const title = current.title ?? input.title;
       const updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+      db.prepare(
+        `UPDATE runner_sessions
+         SET messages_json = $messages,
+             updated_at = $updatedAt
+         WHERE id = $id`
+      ).run({
+        $id: input.context.sessionId,
+        $messages: JSON.stringify(messages),
+        $updatedAt: updatedAt
+      });
+      db.exec("COMMIT");
+      return Promise.resolve();
+    } catch (error) {
+      if (db.isTransaction) {
+        db.exec("ROLLBACK");
+      }
+      throw error;
+    }
+  }
+  clear(context) {
+    const db = this.database();
+    const updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+    try {
+      db.exec("BEGIN IMMEDIATE");
+      db.prepare(
+        `UPDATE runner_sessions
+         SET messages_json = '[]',
+             transcript_json = '[]',
+             updated_at = $updatedAt
+         WHERE id = $id`
+      ).run({
+        $id: context.sessionId,
+        $updatedAt: updatedAt
+      });
+      db.prepare("DELETE FROM runner_session_runs WHERE session_id = $id").run({
+        $id: context.sessionId
+      });
+      db.exec("COMMIT");
+      return Promise.resolve();
+    } catch (error) {
+      if (db.isTransaction) {
+        db.exec("ROLLBACK");
+      }
+      throw error;
+    }
+  }
+  async recordError(input) {
+    const runId = studioRunId(input.context) ?? input.runId;
+    const existing = this.getSessionRun(input.context.sessionId, runId);
+    const transcript = existing === void 0 || parseJsonArray(existing.transcript_json).length === 0 ? transcriptFromMessagesFallback(input.messages) : parseJsonArray(existing.transcript_json);
+    await this.saveSessionRunTranscript({
+      id: input.context.sessionId,
+      runId,
+      transcript,
+      status: "error",
+      error: serializeJsonError(input.error)
+    });
+  }
+  saveSessionRunTranscript(input) {
+    const db = this.database();
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    try {
+      db.exec("BEGIN IMMEDIATE");
+      const row = this.getSessionRow(input.id);
+      if (row === void 0) {
+        db.exec("ROLLBACK");
+        return void 0;
+      }
+      const current = toSession(row, this.listSessionRunRows(input.id));
+      const title = current.title ?? input.title;
+      db.prepare(
+        `INSERT INTO runner_session_runs (
+          run_id,
+          session_id,
+          status,
+          title,
+          transcript_json,
+          error_json,
+          created_at,
+          updated_at
+        ) VALUES (
+          $runId,
+          $sessionId,
+          $status,
+          $title,
+          $transcript,
+          $error,
+          $now,
+          $now
+        )
+        ON CONFLICT(run_id) DO UPDATE SET
+          status = excluded.status,
+          title = COALESCE(runner_session_runs.title, excluded.title),
+          transcript_json = excluded.transcript_json,
+          error_json = excluded.error_json,
+          updated_at = excluded.updated_at`
+      ).run({
+        $runId: input.runId,
+        $sessionId: input.id,
+        $status: input.status,
+        $title: input.title ?? null,
+        $transcript: JSON.stringify(renumberTranscript(input.transcript)),
+        $error: input.error === void 0 ? null : JSON.stringify(input.error),
+        $now: now
+      });
       db.prepare(
         `UPDATE runner_sessions
          SET title = $title,
-             messages_json = $messages,
-             transcript_json = $transcript,
              updated_at = $updatedAt
          WHERE id = $id`
       ).run({
         $id: input.id,
         $title: title ?? null,
-        $messages: JSON.stringify(messages),
-        $transcript: JSON.stringify(transcript),
-        $updatedAt: updatedAt
+        $updatedAt: now
       });
       db.exec("COMMIT");
-      return {
-        ...current,
-        ...title === void 0 ? {} : { title },
-        updatedAt,
-        messageCount: messages.length,
-        messages,
-        transcript
-      };
+      const updated = this.getSession(input.id);
+      return updated;
     } catch (error) {
       if (db.isTransaction) {
         db.exec("ROLLBACK");
@@ -605,6 +699,7 @@ var SqliteSessionStore = class {
     try {
       db.exec("BEGIN IMMEDIATE");
       db.prepare("DELETE FROM runner_traces WHERE session_id = $id").run({ $id: id });
+      db.prepare("DELETE FROM runner_session_runs WHERE session_id = $id").run({ $id: id });
       const result = db.prepare("DELETE FROM runner_sessions WHERE id = $id").run({ $id: id });
       db.exec("COMMIT");
       return Number(result.changes) > 0;
@@ -765,6 +860,19 @@ var SqliteSessionStore = class {
       ) STRICT;
       CREATE INDEX IF NOT EXISTS runner_sessions_agent_updated_idx
         ON runner_sessions(agent_id, updated_at DESC);
+      CREATE TABLE IF NOT EXISTS runner_session_runs (
+        run_id TEXT PRIMARY KEY,
+        session_id TEXT NOT NULL,
+        status TEXT NOT NULL,
+        title TEXT,
+        transcript_json TEXT NOT NULL,
+        error_json TEXT,
+        created_at TEXT NOT NULL,
+        updated_at TEXT NOT NULL,
+        FOREIGN KEY(session_id) REFERENCES runner_sessions(id) ON DELETE CASCADE
+      ) STRICT;
+      CREATE INDEX IF NOT EXISTS runner_session_runs_session_created_idx
+        ON runner_session_runs(session_id, created_at ASC);
       CREATE TABLE IF NOT EXISTS runner_traces (
         id TEXT PRIMARY KEY,
         session_id TEXT NOT NULL,
@@ -787,13 +895,39 @@ var SqliteSessionStore = class {
     this.db = db;
     return db;
   }
+  getSessionRow(id) {
+    return this.database().prepare(
+      `SELECT id, agent_id, title, metadata_json, messages_json, transcript_json, created_at, updated_at
+         FROM runner_sessions
+         WHERE id = $id`
+    ).get({ $id: id });
+  }
+  getSessionRun(sessionId, runId) {
+    return this.database().prepare(
+      `SELECT run_id, session_id, status, title, transcript_json, error_json, created_at, updated_at
+         FROM runner_session_runs
+         WHERE session_id = $sessionId AND run_id = $runId`
+    ).get({ $sessionId: sessionId, $runId: runId });
+  }
+  listSessionRunRows(sessionId) {
+    return this.database().prepare(
+      `SELECT run_id, session_id, status, title, transcript_json, error_json, created_at, updated_at
+         FROM runner_session_runs
+         WHERE session_id = $sessionId
+         ORDER BY created_at ASC`
+    ).all({ $sessionId: sessionId });
+  }
 };
-function toSession(row) {
+function toSession(row, runRows = []) {
   const summary = toSessionSummary(row);
+  const legacyTranscript = parseJsonArray(row.transcript_json);
+  const runTranscript = runRows.flatMap(
+    (runRow) => parseJsonArray(runRow.transcript_json)
+  );
   return {
     ...summary,
     messages: parseJsonArray(row.messages_json),
-    transcript: renumberTranscript(parseJsonArray(row.transcript_json))
+    transcript: renumberTranscript([...legacyTranscript, ...runTranscript])
   };
 }
 function toSessionSummary(row) {
@@ -852,6 +986,96 @@ function parseJsonValue(value) {
 function renumberTranscript(entries) {
   return entries.map((entry, entryId) => ({ ...entry, entryId }));
 }
+function studioRunId(context) {
+  const value = context.metadata?.studioRunId;
+  return typeof value === "string" && value.length > 0 ? value : void 0;
+}
+function serializeJsonError(error) {
+  if (error instanceof Error) {
+    return {
+      name: error.name,
+      message: error.message
+    };
+  }
+  if (error === null || typeof error === "string" || typeof error === "number" || typeof error === "boolean") {
+    return error;
+  }
+  return String(error);
+}
+function transcriptFromMessagesFallback(messages) {
+  const transcript = [];
+  for (const message of messages) {
+    if (message.role === "system") {
+      continue;
+    }
+    if (message.role === "user") {
+      for (const content of message.content) {
+        if (content.type === "text") {
+          transcript.push({
+            entryId: transcript.length,
+            kind: "message",
+            role: "user",
+            text: content.text
+          });
+        }
+      }
+      continue;
+    }
+    if (message.role === "tool") {
+      for (const content of message.content) {
+        transcript.push({
+          entryId: transcript.length,
+          kind: "tool",
+          toolName: "tool_result",
+          callId: content.callId ?? content.id,
+          result: content.content.map((item) => "text" in item ? item.text : "[image]").join("\n")
+        });
+      }
+      continue;
+    }
+    for (const content of message.content) {
+      if (content.type === "text") {
+        appendAssistantTranscriptText(transcript, content.text);
+      } else if (content.type === "reasoning") {
+        transcript.push({
+          entryId: transcript.length,
+          kind: "reasoning",
+          ...content.id === void 0 ? {} : { reasoningId: content.id },
+          text: content.text
+        });
+      } else if (content.type === "tool_call") {
+        transcript.push({
+          entryId: transcript.length,
+          kind: "tool",
+          toolName: content.function.name,
+          callId: content.callId ?? content.id,
+          args: formatJson(content.function.arguments)
+        });
+      }
+    }
+  }
+  return transcript;
+}
+function appendAssistantTranscriptText(transcript, text) {
+  const last = transcript.at(-1);
+  if (last?.kind === "message" && last.role === "assistant") {
+    last.text = `${last.text}${text}`;
+    return;
+  }
+  transcript.push({
+    entryId: transcript.length,
+    kind: "message",
+    role: "assistant",
+    text
+  });
+}
+function formatJson(value) {
+  try {
+    return JSON.stringify(value, null, 2);
+  } catch {
+    return String(value);
+  }
+}
 
 // src/runtime/shared.ts
 function resolveStores(options) {
@@ -1823,22 +2047,42 @@ function traceForRun(trace, agentId, session) {
     ...trace?.sessionId !== void 0 ? { sessionId: trace.sessionId } : session === void 0 ? {} : { sessionId: session.id }
   };
 }
-async function* persistStreamingSessionRun(props) {
+async function* persistStreamingSessionTranscript(props) {
   const transcript = [messageToTranscriptEntry(props.message, 0)];
-  for await (const event of props.stream) {
-    acceptTranscriptStreamEvent(transcript, event);
-    if (event.type === "final") {
-      const nextSession = await props.store.appendSessionRun({
+  const title = optionalTitle(props.message);
+  await props.store.saveSessionRunTranscript({
+    id: props.session.id,
+    runId: props.runId,
+    ...title,
+    transcript,
+    status: "running"
+  });
+  try {
+    for await (const event of props.stream) {
+      acceptTranscriptStreamEvent(transcript, event);
+      const nextSession = await props.store.saveSessionRunTranscript({
         id: props.session.id,
-        ...optionalTitle(props.message),
-        messages: event.messages,
-        transcript
+        runId: props.runId,
+        ...title,
+        transcript,
+        status: event.type === "final" ? "success" : event.type === "error" ? "error" : "running",
+        ...event.type === "error" ? { error: serializeError2(event.error) } : {}
       });
       if (nextSession === void 0) {
         throw new Error("Session not found");
       }
+      yield event;
     }
-    yield event;
+  } catch (error) {
+    await props.store.saveSessionRunTranscript({
+      id: props.session.id,
+      runId: props.runId,
+      ...title,
+      transcript,
+      status: "error",
+      error: serializeError2(error)
+    });
+    throw error;
   }
 }
 function acceptTranscriptStreamEvent(transcript, event) {
@@ -1854,7 +2098,7 @@ function acceptTranscriptStreamEvent(transcript, event) {
       kind: "tool",
       toolName: event.toolCall.function.name,
       callId: event.toolCall.callId ?? event.toolCall.id,
-      args: formatJson(event.toolCall.function.arguments)
+      args: formatJson2(event.toolCall.function.arguments)
     });
   }
   if (event.type === "tool_result") {
@@ -1987,7 +2231,7 @@ function transcriptFromMessages(messages) {
           kind: "tool",
           toolName: content.function.name,
           callId: content.callId ?? content.id,
-          args: formatJson(content.function.arguments)
+          args: formatJson2(content.function.arguments)
         });
       }
     }
@@ -2073,7 +2317,7 @@ function extractMessageText(message) {
       return [item.text];
     }
     if (item.type === "tool_call") {
-      return [`${item.function.name}(${formatJson(item.function.arguments)})`];
+      return [`${item.function.name}(${formatJson2(item.function.arguments)})`];
     }
     if (item.type === "tool_result") {
       return item.content.map((result) => "text" in result ? result.text : "[image]");
@@ -2081,7 +2325,7 @@ function extractMessageText(message) {
     return [];
   }).join("\n");
 }
-function formatJson(value) {
+function formatJson2(value) {
   try {
     return JSON.stringify(value, null, 2);
   } catch {
@@ -2408,9 +2652,7 @@ function agentMetadata(agent) {
 }
 function createStudioApp(options) {
   const stores = resolveStores(options);
-  const agents = normalizeAgents(options.agents).map(
-    (agent) => withStudioTraceObserver(agent, stores.traces)
-  );
+  const agents = normalizeAgents(options.agents).map((agent) => withStudioSessionMemory(agent, stores.sessions)).map((agent) => withStudioTraceObserver(agent, stores.traces));
   const agentMap = new Map(agents.map((agent) => [agent.id, agent]));
   const approvalRuntime = createApprovalRuntime();
   const questionRuntime = createQuestionRuntime();
@@ -2469,12 +2711,14 @@ function createStudioApp(options) {
       return errorResponse(c, 400, "bad_request", "Session belongs to another agent");
     }
     const runId = globalThis.crypto.randomUUID();
-    const request = agent.agent.prompt(body.message);
-    if (session !== void 0) {
-      request.withHistory(session.messages);
-    } else if (body.history !== void 0) {
-      request.withHistory(body.history);
-    }
+    const memoryMetadata = {
+      agentId,
+      ...body.metadata ?? {},
+      studioRunId: runId
+    };
+    const request = session !== void 0 ? agent.agent.session(session.id, { metadata: memoryMetadata }).prompt(body.message) : agent.agent.prompt(
+      body.history !== void 0 ? [...body.history, normalizePromptMessage(body.message)] : body.message
+    );
     if (body.maxTurns !== void 0) {
       request.maxTurns(body.maxTurns);
     }
@@ -2512,11 +2756,12 @@ function createStudioApp(options) {
         request.requestHook(effectiveHook);
       }
       const runStream = mergeRunAndApprovalEvents(request.stream(), runtimeEvents);
-      const stream = session === void 0 || stores.sessions === void 0 ? runStream : persistStreamingSessionRun({
+      const stream = session === void 0 || stores.sessions === void 0 ? runStream : persistStreamingSessionTranscript({
         stream: runStream,
         store: stores.sessions,
         session,
-        message: body.message
+        message: body.message,
+        runId
       });
       return streamAgentRunEvents(c, stream);
     }
@@ -2544,15 +2789,30 @@ function createStudioApp(options) {
       }
       const response = await request.send();
       if (session !== void 0 && stores.sessions !== void 0) {
-        await stores.sessions.appendSessionRun({
+        await stores.sessions.saveSessionRunTranscript({
           id: session.id,
+          runId,
           ...optionalTitle(body.message),
-          messages: response.messages,
-          transcript: transcriptFromMessages(response.messages)
+          transcript: transcriptFromMessages(response.messages),
+          status: "success"
         });
       }
       return c.json(response);
     } catch (error) {
+      if (session !== void 0 && stores.sessions !== void 0) {
+        const messages = await stores.sessions.load({
+          sessionId: session.id,
+          metadata: memoryMetadata
+        });
+        await stores.sessions.saveSessionRunTranscript({
+          id: session.id,
+          runId,
+          ...optionalTitle(body.message),
+          transcript: transcriptFromMessages(messages.slice(session.messageCount)),
+          status: "error",
+          error: serializeError2(error)
+        });
+      }
       return errorResponse(c, 500, "internal_error", "Agent run failed", serializeError2(error));
     }
   });
@@ -2584,36 +2844,60 @@ function createStudioApp(options) {
     ...stores.traces === void 0 ? {} : { traceStore: stores.traces }
   };
 }
+function normalizePromptMessage(message) {
+  return typeof message === "string" ? Message.user(message) : message;
+}
+function withStudioSessionMemory(studioAgent, sessionStore) {
+  if (sessionStore === void 0) {
+    return studioAgent;
+  }
+  return {
+    ...studioAgent,
+    agent: cloneAgent(studioAgent.agent, {
+      memory: {
+        store: sessionStore,
+        options: resolveMemoryOptions({ savePolicy: "message" })
+      }
+    })
+  };
+}
 function withStudioTraceObserver(studioAgent, traceStore) {
   if (traceStore === void 0 || hasStudioTraceObserver(studioAgent.agent)) {
     return studioAgent;
   }
   return {
     ...studioAgent,
-    agent: new Agent({
-      id: studioAgent.agent.id,
-      name: studioAgent.agent.name,
-      description: studioAgent.agent.description,
-      model: studioAgent.agent.model,
-      instructions: studioAgent.agent.instructions,
-      staticContext: studioAgent.agent.staticContext,
-      temperature: studioAgent.agent.temperature,
-      maxTokens: studioAgent.agent.maxTokens,
-      additionalParams: studioAgent.agent.additionalParams,
-      toolSet: studioAgent.agent.toolSet,
-      toolChoice: studioAgent.agent.toolChoice,
-      defaultMaxTurns: studioAgent.agent.defaultMaxTurns,
-      hook: studioAgent.agent.hook,
-      outputSchema: studioAgent.agent.outputSchema,
+    agent: cloneAgent(studioAgent.agent, {
       observers: [
         ...studioAgent.agent.observers,
         { observer: new StudioTraceObserver({ store: traceStore }) }
-      ],
-      dynamicContexts: studioAgent.agent.dynamicContexts,
-      dynamicTools: studioAgent.agent.dynamicTools
+      ]
     })
   };
 }
+function cloneAgent(agent, overrides = {}) {
+  return new Agent({
+    id: agent.id,
+    name: agent.name,
+    description: agent.description,
+    model: agent.model,
+    instructions: agent.instructions,
+    staticContext: agent.staticContext,
+    temperature: agent.temperature,
+    maxTokens: agent.maxTokens,
+    additionalParams: agent.additionalParams,
+    toolSet: agent.toolSet,
+    toolChoice: agent.toolChoice,
+    defaultMaxTurns: agent.defaultMaxTurns,
+    hook: agent.hook,
+    outputSchema: agent.outputSchema,
+    observers: agent.observers,
+    dynamicContexts: agent.dynamicContexts,
+    dynamicTools: agent.dynamicTools,
+    memory: agent.memory,
+    ...overrides
+  });
+}
 function hasStudioTraceObserver(agent) {
   return agent.observers.some(
     (registration) => registration.observer instanceof StudioTraceObserver