@the-open-engine/zeroshot 6.38.1 → 6.39.1

package/src/agent/agent-context-sources.js

@@ -1,165 +1,112 @@
-const { isReplayableMessage } = require('./context-replay-policy');
-
+"use strict";
+const replayPolicy = require("./context-replay-policy");
+const { isReplayableMessage } = replayPolicy;
 function resolveSourceSince(source, cluster, lastTaskEndTime, lastAgentStartTime) {
-  const sinceValue = source.since;
-
-  if (sinceValue === 'cluster_start') {
-    return cluster.createdAt;
-  }
-
-  if (sinceValue === 'last_task_end') {
-    return lastTaskEndTime || cluster.createdAt;
-  }
-
-  if (sinceValue === 'last_agent_start') {
-    return lastAgentStartTime ? lastAgentStartTime + 1 : cluster.createdAt;
-  }
-
-  if (typeof sinceValue === 'string') {
-    const parsed = Date.parse(sinceValue);
-    if (Number.isNaN(parsed)) {
-      throw new Error(
-        `Unknown context source "since" value "${sinceValue}" for topic ${source.topic}. ` +
-          'Use cluster_start, last_task_end, last_agent_start, or an ISO timestamp.'
-      );
+    const sinceValue = source.since;
+    if (sinceValue === 'cluster_start') {
+        return cluster.createdAt;
     }
-    return parsed;
-  }
-
-  return sinceValue;
+    if (sinceValue === 'last_task_end') {
+        return lastTaskEndTime || cluster.createdAt;
+    }
+    if (sinceValue === 'last_agent_start') {
+        return lastAgentStartTime ? lastAgentStartTime + 1 : cluster.createdAt;
+    }
+    if (typeof sinceValue === 'string') {
+        const parsed = Date.parse(sinceValue);
+        if (Number.isNaN(parsed)) {
+            throw new Error(`Unknown context source "since" value "${sinceValue}" for topic ${source.topic}. ` +
+                'Use cluster_start, last_task_end, last_agent_start, or an ISO timestamp.');
+        }
+        return parsed;
+    }
+    return sinceValue;
 }
-
 function formatSourceMessagesSection(source, messages) {
-  let context = `\n## Messages from topic: ${source.topic}\n\n`;
-
-  for (const msg of messages) {
-    context += `[${new Date(msg.timestamp).toISOString()}] ${msg.sender}:\n`;
-    if (msg.content?.text) {
-      context += `${msg.content.text}\n`;
-    }
-    if (msg.content?.data) {
-      context += `Data: ${JSON.stringify(msg.content.data, null, 2)}\n`;
+    let context = `\n## Messages from topic: ${source.topic}\n\n`;
+    for (const msg of messages) {
+        context += `[${new Date(msg.timestamp).toISOString()}] ${msg.sender}:\n`;
+        if (msg.content?.text) {
+            context += `${msg.content.text}\n`;
+        }
+        if (msg.content?.data) {
+            context += `Data: ${JSON.stringify(msg.content.data, null, 2)}\n`;
+        }
+        context += '\n';
     }
-    context += '\n';
-  }
-
-  return context;
+    return context;
 }
-
 function resolveSourceSelection(source, { compact = false } = {}) {
-  const baseAmount = source.amount ?? source.limit;
-  const baseStrategy = source.strategy ?? (baseAmount !== undefined ? 'latest' : 'all');
-
-  if (!compact) {
-    return { amount: baseAmount, strategy: baseStrategy };
-  }
-
-  return {
-    amount: source.compactAmount ?? 1,
-    strategy: source.compactStrategy ?? (baseStrategy === 'all' ? 'latest' : baseStrategy),
-  };
+    const baseAmount = source.amount ?? source.limit;
+    const baseStrategy = source.strategy ?? (baseAmount !== undefined ? 'latest' : 'all');
+    if (!compact) {
+        return { amount: baseAmount, strategy: baseStrategy };
+    }
+    return {
+        amount: source.compactAmount ?? 1,
+        strategy: source.compactStrategy ?? (baseStrategy === 'all' ? 'latest' : baseStrategy),
+    };
 }
-
-function resolveSourceMessages({
-  source,
-  messageBus,
-  cluster,
-  lastTaskEndTime,
-  lastAgentStartTime,
-  afterId,
-  throughId,
-  triggeringMessageId,
-  compact = false,
-}) {
-  const sinceTimestamp = resolveSourceSince(
-    source,
-    cluster,
-    lastTaskEndTime,
-    lastAgentStartTime
-  );
-  const { amount, strategy } = resolveSourceSelection(source, { compact });
-  const messages = messageBus.query({
-    cluster_id: cluster.id,
-    topic: source.topic,
-    sender: source.sender,
-    since: sinceTimestamp,
-    afterId,
-    throughId,
-  });
-  const replayableMessages = messages.filter(
-    (message) => isReplayableMessage(message) && message.id !== triggeringMessageId
-  );
-
-  if (amount === undefined) {
-    return replayableMessages;
-  }
-
-  if (strategy === 'latest') {
-    return replayableMessages.slice(-amount);
-  }
-
-  return replayableMessages.slice(0, amount);
+function resolveSourceMessages({ source, messageBus, cluster, lastTaskEndTime, lastAgentStartTime, afterId, throughId, triggeringMessageId, compact = false, }) {
+    const sinceTimestamp = resolveSourceSince(source, cluster, lastTaskEndTime, lastAgentStartTime);
+    const { amount, strategy } = resolveSourceSelection(source, { compact });
+    const messages = messageBus.query({
+        cluster_id: cluster.id,
+        topic: source.topic,
+        sender: source.sender,
+        since: sinceTimestamp,
+        afterId,
+        throughId,
+    });
+    const replayableMessages = messages.filter((message) => isReplayableMessage(message) && message.id !== triggeringMessageId);
+    if (amount === undefined) {
+        return replayableMessages;
+    }
+    if (strategy === 'latest') {
+        return replayableMessages.slice(-amount);
+    }
+    return replayableMessages.slice(0, amount);
 }
-
 const REQUIRED_TOPICS = new Set(['STATE_SNAPSHOT', 'ISSUE_OPENED', 'PLAN_READY']);
 const HIGH_PRIORITY_TOPICS = new Set(['VALIDATION_RESULT', 'IMPLEMENTATION_READY']);
-
 function resolveSourcePriority(source) {
-  if (source.priority) {
-    return source.priority;
-  }
-
-  if (REQUIRED_TOPICS.has(source.topic)) {
-    return 'required';
-  }
-
-  if (HIGH_PRIORITY_TOPICS.has(source.topic)) {
-    return 'high';
-  }
-
-  return 'medium';
-}
-
-function buildSourcePack({
-  source,
-  index,
-  messageBus,
-  cluster,
-  lastTaskEndTime,
-  lastAgentStartTime,
-  afterId,
-  throughId,
-  triggeringMessageId,
-}) {
-  const render = (compact) => {
-    const messages = resolveSourceMessages({
-      source,
-      messageBus,
-      cluster,
-      lastTaskEndTime,
-      lastAgentStartTime,
-      afterId,
-      throughId,
-      triggeringMessageId,
-      compact,
-    });
-
-    if (messages.length === 0) {
-      return '';
+    if (source.priority) {
+        return source.priority;
     }
-
-    return formatSourceMessagesSection(source, messages);
-  };
-
-  return {
-    id: `source:${source.topic}:${index}`,
-    section: 'sources',
-    priority: resolveSourcePriority(source),
-    render: () => render(false),
-    compact: () => render(true),
-  };
+    if (REQUIRED_TOPICS.has(source.topic)) {
+        return 'required';
+    }
+    if (HIGH_PRIORITY_TOPICS.has(source.topic)) {
+        return 'high';
+    }
+    return 'medium';
+}
+function buildSourcePack({ source, index, messageBus, cluster, lastTaskEndTime, lastAgentStartTime, afterId, throughId, triggeringMessageId, }) {
+    const render = (compact) => {
+        const messages = resolveSourceMessages({
+            source,
+            messageBus,
+            cluster,
+            lastTaskEndTime,
+            lastAgentStartTime,
+            afterId,
+            throughId,
+            triggeringMessageId,
+            compact,
+        });
+        if (messages.length === 0) {
+            return '';
+        }
+        return formatSourceMessagesSection(source, messages);
+    };
+    return {
+        id: `source:${source.topic}:${index}`,
+        section: 'sources',
+        priority: resolveSourcePriority(source),
+        render: () => render(false),
+        compact: () => render(true),
+    };
 }
-
 module.exports = {
-  buildSourcePack,
+    buildSourcePack,
 };

package/src/agent/agent-context-sources.ts

@@ -0,0 +1,253 @@
+import replayPolicy = require('./context-replay-policy');
+
+type MessageTimestamp = string | number;
+
+interface ReplayFields {
+  readonly contextSafe?: unknown;
+  readonly replayPolicy?: unknown;
+  readonly [key: string]: unknown;
+}
+
+interface ContextMessage {
+  readonly id?: unknown;
+  readonly topic?: unknown;
+  readonly timestamp: MessageTimestamp;
+  readonly sender: string;
+  readonly metadata?: ReplayFields | null;
+  readonly content?: {
+    readonly text?: string | null;
+    readonly data?: ReplayFields | null;
+  } | null;
+}
+
+interface ContextSource {
+  topic: string;
+  sender?: unknown;
+  since?: unknown;
+  amount?: number;
+  limit?: number;
+  strategy?: string;
+  compactAmount?: number;
+  compactStrategy?: string;
+  priority?: string;
+}
+
+interface SourceCluster {
+  id: string;
+  createdAt: number;
+}
+
+interface MessageQuery {
+  cluster_id: string;
+  topic: string;
+  sender?: unknown;
+  since?: unknown;
+  afterId?: unknown;
+  throughId?: unknown;
+}
+
+interface SourceMessageBus {
+  query(criteria: MessageQuery): ContextMessage[];
+}
+
+interface SourceSelectionOptions {
+  compact?: boolean;
+}
+
+interface SourceSelection {
+  amount: number | undefined;
+  strategy: string;
+}
+
+interface SourceMessageParams {
+  source: ContextSource;
+  messageBus: SourceMessageBus;
+  cluster: SourceCluster;
+  lastTaskEndTime?: number | null | undefined;
+  lastAgentStartTime?: number | null | undefined;
+  afterId?: unknown;
+  throughId?: unknown;
+  triggeringMessageId?: unknown;
+  compact?: boolean;
+}
+
+interface SourcePackParams extends Omit<SourceMessageParams, 'compact'> {
+  index: number;
+}
+
+interface SourcePack {
+  id: string;
+  section: 'sources';
+  priority: string;
+  render(): string;
+  compact(): string;
+}
+
+const { isReplayableMessage } = replayPolicy;
+
+function resolveSourceSince(
+  source: ContextSource,
+  cluster: SourceCluster,
+  lastTaskEndTime: number | null | undefined,
+  lastAgentStartTime: number | null | undefined
+): unknown {
+  const sinceValue = source.since;
+
+  if (sinceValue === 'cluster_start') {
+    return cluster.createdAt;
+  }
+
+  if (sinceValue === 'last_task_end') {
+    return lastTaskEndTime || cluster.createdAt;
+  }
+
+  if (sinceValue === 'last_agent_start') {
+    return lastAgentStartTime ? lastAgentStartTime + 1 : cluster.createdAt;
+  }
+
+  if (typeof sinceValue === 'string') {
+    const parsed = Date.parse(sinceValue);
+    if (Number.isNaN(parsed)) {
+      throw new Error(
+        `Unknown context source "since" value "${sinceValue}" for topic ${source.topic}. ` +
+          'Use cluster_start, last_task_end, last_agent_start, or an ISO timestamp.'
+      );
+    }
+    return parsed;
+  }
+
+  return sinceValue;
+}
+
+function formatSourceMessagesSection(source: ContextSource, messages: ContextMessage[]): string {
+  let context = `\n## Messages from topic: ${source.topic}\n\n`;
+
+  for (const msg of messages) {
+    context += `[${new Date(msg.timestamp).toISOString()}] ${msg.sender}:\n`;
+    if (msg.content?.text) {
+      context += `${msg.content.text}\n`;
+    }
+    if (msg.content?.data) {
+      context += `Data: ${JSON.stringify(msg.content.data, null, 2)}\n`;
+    }
+    context += '\n';
+  }
+
+  return context;
+}
+
+function resolveSourceSelection(
+  source: ContextSource,
+  { compact = false }: SourceSelectionOptions = {}
+): SourceSelection {
+  const baseAmount = source.amount ?? source.limit;
+  const baseStrategy = source.strategy ?? (baseAmount !== undefined ? 'latest' : 'all');
+
+  if (!compact) {
+    return { amount: baseAmount, strategy: baseStrategy };
+  }
+
+  return {
+    amount: source.compactAmount ?? 1,
+    strategy: source.compactStrategy ?? (baseStrategy === 'all' ? 'latest' : baseStrategy),
+  };
+}
+
+function resolveSourceMessages({
+  source,
+  messageBus,
+  cluster,
+  lastTaskEndTime,
+  lastAgentStartTime,
+  afterId,
+  throughId,
+  triggeringMessageId,
+  compact = false,
+}: SourceMessageParams): ContextMessage[] {
+  const sinceTimestamp = resolveSourceSince(source, cluster, lastTaskEndTime, lastAgentStartTime);
+  const { amount, strategy } = resolveSourceSelection(source, { compact });
+  const messages = messageBus.query({
+    cluster_id: cluster.id,
+    topic: source.topic,
+    sender: source.sender,
+    since: sinceTimestamp,
+    afterId,
+    throughId,
+  });
+  const replayableMessages = messages.filter(
+    (message) => isReplayableMessage(message) && message.id !== triggeringMessageId
+  );
+
+  if (amount === undefined) {
+    return replayableMessages;
+  }
+
+  if (strategy === 'latest') {
+    return replayableMessages.slice(-amount);
+  }
+
+  return replayableMessages.slice(0, amount);
+}
+
+const REQUIRED_TOPICS = new Set(['STATE_SNAPSHOT', 'ISSUE_OPENED', 'PLAN_READY']);
+const HIGH_PRIORITY_TOPICS = new Set(['VALIDATION_RESULT', 'IMPLEMENTATION_READY']);
+
+function resolveSourcePriority(source: ContextSource): string {
+  if (source.priority) {
+    return source.priority;
+  }
+
+  if (REQUIRED_TOPICS.has(source.topic)) {
+    return 'required';
+  }
+
+  if (HIGH_PRIORITY_TOPICS.has(source.topic)) {
+    return 'high';
+  }
+
+  return 'medium';
+}
+
+function buildSourcePack({
+  source,
+  index,
+  messageBus,
+  cluster,
+  lastTaskEndTime,
+  lastAgentStartTime,
+  afterId,
+  throughId,
+  triggeringMessageId,
+}: SourcePackParams): SourcePack {
+  const render = (compact: boolean): string => {
+    const messages = resolveSourceMessages({
+      source,
+      messageBus,
+      cluster,
+      lastTaskEndTime,
+      lastAgentStartTime,
+      afterId,
+      throughId,
+      triggeringMessageId,
+      compact,
+    });
+
+    if (messages.length === 0) {
+      return '';
+    }
+
+    return formatSourceMessagesSection(source, messages);
+  };
+
+  return {
+    id: `source:${source.topic}:${index}`,
+    section: 'sources',
+    priority: resolveSourcePriority(source),
+    render: () => render(false),
+    compact: () => render(true),
+  };
+}
+
+export = {
+  buildSourcePack,
+};

package/src/agent/agent-context-types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/src/agent/agent-context-types.ts

@@ -0,0 +1,132 @@
+export interface JsonSchema {
+  readonly type?: string;
+  readonly description?: string;
+  readonly enum?: readonly unknown[];
+  readonly properties?: Readonly<Record<string, JsonSchema | null | undefined>>;
+}
+
+export interface LegacyOutputFormat {
+  readonly rules?: readonly string[];
+  readonly example?: unknown;
+}
+
+export interface PromptObject {
+  readonly system?: string | null;
+  readonly outputFormat?: LegacyOutputFormat | null;
+}
+
+export interface ContextSource {
+  topic: string;
+  sender?: unknown;
+  since?: unknown;
+  amount?: number;
+  limit?: number;
+  strategy?: string;
+  compactAmount?: number;
+  compactStrategy?: string;
+  priority?: string;
+}
+
+export interface ContextStrategy {
+  maxTokens?: number;
+  sources?: ContextSource[];
+  readonly [key: string]: unknown;
+}
+
+export interface AgentContextConfig {
+  readonly prompt?: string | PromptObject | null;
+  readonly promptConfig?: {
+    readonly type?: string | null;
+  } | null;
+  readonly cwd?: string | null;
+  readonly jsonSchema?: JsonSchema | null;
+  readonly outputFormat?: string | null;
+  readonly role?: string;
+  readonly requiredQualityGates?: unknown;
+  readonly commandProofs?: unknown;
+  readonly contextStrategy?: ContextStrategy | null;
+}
+
+export interface WorktreeContext {
+  readonly enabled?: boolean;
+  readonly path?: string | null;
+}
+
+export interface IsolationContext {
+  readonly enabled?: boolean;
+}
+
+export interface ValidationCriterion {
+  readonly status?: string;
+  readonly id?: string;
+  readonly reason?: string;
+}
+
+export interface ContextMessageData {
+  readonly criteriaResults?: readonly ValidationCriterion[] | null;
+  readonly contextSafe?: unknown;
+  readonly replayPolicy?: unknown;
+  readonly [key: string]: unknown;
+}
+
+export interface ContextMessage {
+  readonly id?: unknown;
+  readonly topic?: unknown;
+  readonly timestamp: string | number;
+  readonly sender: string;
+  readonly metadata?: ContextMessageData | null;
+  readonly content?: {
+    readonly text?: string | null;
+    readonly data?: ContextMessageData | null;
+  } | null;
+}
+
+export interface ContextQuery {
+  cluster_id: string;
+  topic: string;
+  sender?: unknown;
+  since?: unknown;
+  afterId?: unknown;
+  throughId?: unknown;
+  limit?: number;
+}
+
+export interface ContextMessageBus {
+  query(criteria: ContextQuery): ContextMessage[];
+  publish(message: unknown): unknown;
+}
+
+export interface ContextCluster {
+  id: string;
+  createdAt: number;
+}
+
+export interface TriggeringMessage {
+  id?: unknown;
+  topic: string;
+  sender: string;
+  content?: {
+    text?: string | null;
+  } | null;
+}
+
+export interface BuildContextParams {
+  id: string;
+  role: string;
+  iteration: number;
+  config: AgentContextConfig;
+  messageBus: ContextMessageBus;
+  cluster: ContextCluster;
+  triggeringMessage: TriggeringMessage;
+  lastTaskEndTime?: number | null | undefined;
+  lastAgentStartTime?: number | null | undefined;
+  selectedPrompt?: string | null | undefined;
+  queuedGuidance?: string | null | undefined;
+  worktree?: WorktreeContext | null | undefined;
+  isolation?: IsolationContext | null | undefined;
+  mode?: string | undefined;
+  continuationSequence?: unknown;
+  contextThroughId?: unknown;
+  previousPromptIdentity?: unknown;
+  currentPromptIdentity?: unknown;
+}