@symerian/symi 3.0.18 → 3.0.19

package/extensions/learning-loop/src/capture/run-tracker.ts

@@ -1,181 +0,0 @@
-/**
- * In-memory run accumulator across hooks.
- *
- * Maintains a Map<RunId, PartialRun> that collects data from multiple
- * hooks (llm_input, after_tool_call, llm_output) within a single agent run.
- * A secondary Map<SessionKey, RunId> correlates hooks that don't carry runId.
- */
-
-import { hashText } from "../../../../src/memory/internal.js";
-import type { PartialRun, CompletedRun, ToolCallRecord, RunId, SessionKey } from "../types.js";
-
-export type RunTracker = ReturnType<typeof createRunTracker>;
-
-export function createRunTracker() {
-  const runs = new Map<RunId, PartialRun>();
-  const sessionToRun = new Map<SessionKey, RunId>();
-
-  function getOrCreate(runId: RunId): PartialRun {
-    let run = runs.get(runId);
-    if (!run) {
-      run = {
-        runId,
-        toolCalls: [],
-        startedAt: Date.now(),
-      };
-      runs.set(runId, run);
-    }
-    return run;
-  }
-
-  function bindSession(sessionKey: SessionKey, runId: RunId): void {
-    sessionToRun.set(sessionKey, runId);
-  }
-
-  function getRunIdForSession(sessionKey: SessionKey): RunId | undefined {
-    return sessionToRun.get(sessionKey);
-  }
-
-  /**
-   * Called from llm_input hook. Captures run start metadata.
-   */
-  function onLlmInput(params: {
-    runId: string;
-    sessionId: string;
-    provider: string;
-    model: string;
-    prompt: string;
-    sessionKey?: string;
-    agentId?: string;
-  }): void {
-    const run = getOrCreate(params.runId);
-    run.sessionId = params.sessionId;
-    run.provider = params.provider;
-    run.model = params.model;
-    run.promptHash = hashText(params.prompt);
-    run.promptLength = params.prompt.length;
-    run.agentId = params.agentId;
-    if (params.sessionKey) {
-      run.sessionKey = params.sessionKey;
-      bindSession(params.sessionKey, params.runId);
-    }
-  }
-
-  /**
-   * Called from llm_output hook. Captures usage and response data.
-   */
-  function onLlmOutput(params: {
-    runId: string;
-    sessionId: string;
-    assistantTexts: string[];
-    usage?: {
-      input?: number;
-      output?: number;
-      cacheRead?: number;
-      cacheWrite?: number;
-      total?: number;
-    };
-  }): void {
-    const run = getOrCreate(params.runId);
-    run.responseLength = params.assistantTexts.reduce((sum, t) => sum + t.length, 0);
-    run.usage = params.usage ?? {};
-  }
-
-  /**
-   * Called from after_tool_call hook. Captures tool call traces.
-   */
-  function onToolCall(params: {
-    toolName: string;
-    durationMs?: number;
-    success: boolean;
-    error?: string;
-    paramHash: string;
-    sessionKey?: string;
-  }): void {
-    // Find the run by session key
-    const runId = params.sessionKey ? sessionToRun.get(params.sessionKey) : undefined;
-    if (!runId) return;
-
-    const run = runs.get(runId);
-    if (!run) return;
-
-    const record: ToolCallRecord = {
-      toolName: params.toolName,
-      durationMs: params.durationMs ?? null,
-      success: params.success,
-      error: params.error ?? null,
-      paramHash: params.paramHash,
-    };
-    run.toolCalls.push(record);
-    run.responseToolCallCount = (run.responseToolCallCount ?? 0) + 1;
-  }
-
-  /**
-   * Called from agent_end hook. Finalizes the run and returns a CompletedRun
-   * if all required fields are present, otherwise null.
-   */
-  function finalize(params: {
-    sessionKey?: string;
-    agentId?: string;
-    sessionId?: string;
-    success: boolean;
-    error?: string;
-    durationMs?: number;
-  }): CompletedRun | null {
-    const runId = params.sessionKey ? sessionToRun.get(params.sessionKey) : undefined;
-    if (!runId) return null;
-
-    const run = runs.get(runId);
-    if (!run) return null;
-
-    // Must have captured llm_input at minimum
-    if (!run.provider || !run.model || !run.promptHash) return null;
-
-    const completed: CompletedRun = {
-      runId: run.runId,
-      sessionId: run.sessionId ?? params.sessionId ?? "",
-      sessionKey: run.sessionKey ?? params.sessionKey ?? "",
-      agentId: run.agentId ?? params.agentId ?? "",
-      provider: run.provider,
-      model: run.model,
-      promptHash: run.promptHash,
-      promptLength: run.promptLength ?? 0,
-      responseLength: run.responseLength ?? 0,
-      responseToolCallCount: run.responseToolCallCount ?? 0,
-      usage: run.usage ?? {},
-      toolCalls: run.toolCalls,
-      success: params.success,
-      error: params.error ?? null,
-      durationMs: params.durationMs ?? Date.now() - run.startedAt,
-      startedAt: run.startedAt,
-      completedAt: Date.now(),
-    };
-
-    // Clean up
-    runs.delete(runId);
-    if (params.sessionKey) {
-      sessionToRun.delete(params.sessionKey);
-    }
-
-    return completed;
-  }
-
-  function getActiveRunCount(): number {
-    return runs.size;
-  }
-
-  function clear(): void {
-    runs.clear();
-    sessionToRun.clear();
-  }
-
-  return {
-    onLlmInput,
-    onLlmOutput,
-    onToolCall,
-    finalize,
-    getActiveRunCount,
-    getRunIdForSession,
-    clear,
-  };
-}

package/extensions/learning-loop/src/capture/serializer.ts

@@ -1,74 +0,0 @@
-/**
- * Data normalization utilities for captured run data.
- */
-
-import type { CompletedRun, RunRow, ToolCallRow, UsageData } from "../types.js";
-
-/**
- * Normalize usage data, clamping negative values to 0.
- */
-export function normalizeUsage(usage: UsageData | undefined): UsageData {
-  if (!usage) return { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 };
-  return {
-    input: Math.max(0, usage.input ?? 0),
-    output: Math.max(0, usage.output ?? 0),
-    cacheRead: Math.max(0, usage.cacheRead ?? 0),
-    cacheWrite: Math.max(0, usage.cacheWrite ?? 0),
-    total: Math.max(0, usage.total ?? 0),
-  };
-}
-
-/**
- * Reconstruct a CompletedRun from DB rows.
- */
-export function reconstructCompletedRun(row: RunRow, toolCalls: ToolCallRow[]): CompletedRun {
-  return {
-    runId: row.run_id,
-    sessionId: row.session_id,
-    sessionKey: row.session_key,
-    agentId: row.agent_id,
-    provider: row.provider,
-    model: row.model,
-    promptHash: row.prompt_hash,
-    promptLength: row.prompt_length,
-    responseLength: row.response_length,
-    responseToolCallCount: row.response_tool_call_count,
-    usage: {
-      input: row.usage_input,
-      output: row.usage_output,
-      cacheRead: row.usage_cache_read,
-      cacheWrite: row.usage_cache_write,
-      total: row.usage_total,
-    },
-    toolCalls: toolCalls.map((tc) => ({
-      toolName: tc.tool_name,
-      durationMs: tc.duration_ms,
-      success: Boolean(tc.success),
-      error: tc.error,
-      paramHash: tc.param_hash,
-    })),
-    success: Boolean(row.success),
-    error: row.error,
-    durationMs: row.duration_ms,
-    startedAt: row.started_at,
-    completedAt: row.completed_at,
-  };
-}
-
-/**
- * Normalize a completed run, ensuring all numeric fields are non-negative
- * and strings are trimmed.
- */
-export function normalizeCompletedRun(run: CompletedRun): CompletedRun {
-  return {
-    ...run,
-    provider: run.provider.trim(),
-    model: run.model.trim(),
-    promptLength: Math.max(0, run.promptLength),
-    responseLength: Math.max(0, run.responseLength),
-    responseToolCallCount: Math.max(0, run.responseToolCallCount),
-    durationMs: Math.max(0, run.durationMs),
-    usage: normalizeUsage(run.usage),
-    error: run.error?.trim() || null,
-  };
-}