Function registerSlidingWindowFunction

src/functions/sliding-window.ts:116–267

(
  sdk: ISdk,
  kv: StateKV,
  provider: MemoryProvider,
)

}

export function registerSlidingWindowFunction(
  sdk: ISdk,
  kv: StateKV,
  provider: MemoryProvider,
): void {
  sdk.registerFunction("mem::enrich-window", 
    async (data: {
      observationId: string;
      sessionId: string;
      lookback?: number;
      lookahead?: number;
    }) => {
      if (
        !data ||
        typeof data.sessionId !== "string" ||
        !data.sessionId.trim() ||
        typeof data.observationId !== "string" ||
        !data.observationId.trim()
      ) {
        return { success: false, error: "sessionId and observationId are required" };
      }
      const sessionId = data.sessionId.trim();
      const observationId = data.observationId.trim();
      const hprev = data.lookback ?? 3;
      const hnext = data.lookahead ?? 2;

      const allObs = await kv.list<CompressedObservation>(
        KV.observations(sessionId),
      );
      allObs.sort(
        (a, b) =>
          new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime(),
      );

      const primaryIdx = allObs.findIndex((o) => o.id === observationId);
      if (primaryIdx === -1) {
        return { success: false, error: "Observation not found" };
      }

      const primary = allObs[primaryIdx];
      const before = allObs.slice(Math.max(0, primaryIdx - hprev), primaryIdx);
      const after = allObs.slice(primaryIdx + 1, primaryIdx + 1 + hnext);

      if (before.length === 0 && after.length === 0) {
        return {
          success: true,
          enriched: null,
          reason: "No adjacent context available",
        };
      }

      try {
        const prompt = buildWindowPrompt(primary, before, after);
        const response = await provider.compress(
          SLIDING_WINDOW_SYSTEM,
          prompt,
        );
        const parsed = parseEnrichedXml(response);