@opengeni/db 0.14.3 → 0.16.2

package/src/session-queue-commands.ts

@@ -1,9 +1,12 @@
 import {
   metadataWithTurnExecutionPolicyV1,
   mergeResourceRefs,
+  ResourceRef,
+  resourceMountPath,
+  stableJson,
   turnExecutionPolicyAuditMetadata,
+  type LatencyMode,
   type ReasoningEffort,
-  type ResourceRef,
   type TurnExecutionPolicyV1,
 } from "@opengeni/contracts";
 import { and, asc, eq, inArray, sql } from "drizzle-orm";
@@ -367,6 +370,27 @@ function draftIsNonEmpty(draft: ComposerDraftRow): boolean {
   return draft.text.length > 0 || draft.resources.length > 0 || draft.sourceTurnId !== null;
 }
 
+function withCanonicalResourceMountPaths(resources: readonly unknown[]): unknown[] {
+  const seen = new Set<string>();
+  const canonical: unknown[] = [];
+  for (const value of resources) {
+    const parsed = ResourceRef.safeParse(value);
+    if (!parsed.success) {
+      canonical.push(value);
+      continue;
+    }
+    const normalized = {
+      ...(value as Record<string, unknown>),
+      mountPath: resourceMountPath(parsed.data),
+    };
+    const key = stableJson(normalized);
+    if (seen.has(key)) continue;
+    seen.add(key);
+    canonical.push(normalized);
+  }
+  return canonical;
+}
+
 export async function getComposerDraftInTransaction(
   db: Database,
   input: {
@@ -403,6 +427,7 @@ export async function saveComposerDraftInTransaction(
     resources: ResourceRef[];
     model: string;
     reasoningEffort: ReasoningEffort;
+    latencyMode?: LatencyMode;
   },
 ): Promise<ComposerDraftRow> {
   await lockWorkspaceInferenceControl(db, input.workspaceId, "share");
@@ -430,11 +455,12 @@ export async function saveComposerDraftInTransaction(
     subjectId: input.subjectId,
     revision,
     text: input.text,
-    resources: input.resources,
+    resources: withCanonicalResourceMountPaths(input.resources),
     tools: [],
     toolsProvided: false,
     model: input.model,
     reasoningEffort: input.reasoningEffort,
+    latencyMode: input.latencyMode ?? "standard",
     // A queue edit is still the same accepted work item. Preserve its frozen
     // initiator through arbitrary draft saves; only a genuinely new compose or
     // Steer captures the submitting actor.
@@ -802,11 +828,12 @@ export async function editQueuedTurnInTransaction(
     subjectId: input.subjectId,
     revision: nextDraftRevision,
     text: turn.prompt,
-    resources: turn.resources,
+    resources: withCanonicalResourceMountPaths(turn.resources),
     tools: [],
     toolsProvided: false,
     model: turn.model,
     reasoningEffort: turn.reasoningEffort,
+    latencyMode: turn.latencyMode,
     sourceTurnId: turn.id,
     sourceTurnVersion: turn.version,
     updatedAt: new Date(),
@@ -1122,6 +1149,7 @@ export async function submitHumanPromptInTransaction(
     resources: ResourceRef[];
     model?: string | null;
     reasoningEffort?: ReasoningEffort | null;
+    latencyMode?: LatencyMode | null;
     reasoningEffortFallback: ReasoningEffort;
     /** Trusted API/core admission snapshot. Omitted only by legacy low-level callers. */
     turnExecutionPolicy?: TurnExecutionPolicyV1;
@@ -1149,9 +1177,10 @@ export async function submitHumanPromptInTransaction(
     expectedDraftRevision: input.expectedDraftRevision ?? null,
     text: input.text,
     turnInstructions: input.turnInstructions ?? null,
-    resources: input.resources,
+    resources: withCanonicalResourceMountPaths(input.resources),
     model: input.model ?? null,
     reasoningEffort: input.reasoningEffort ?? null,
+    latencyMode: input.latencyMode ?? null,
     source: input.source,
     mcpCredentialUpdates: input.mcpCredentialUpdates ?? [],
     ...(input.actor.type === "service"
@@ -1256,15 +1285,17 @@ export async function submitHumanPromptInTransaction(
       draft &&
       canonicalSessionCommandHash({
         text: draft.text,
-        resources: draft.resources,
+        resources: withCanonicalResourceMountPaths(draft.resources),
         model: draft.model,
         reasoningEffort: draft.reasoningEffort,
+        latencyMode: draft.latencyMode,
       }) !==
         canonicalSessionCommandHash({
           text: input.text,
-          resources: input.resources,
+          resources: withCanonicalResourceMountPaths(input.resources),
           model: input.model ?? session.model,
           reasoningEffort: input.reasoningEffort ?? input.reasoningEffortFallback,
+          latencyMode: input.turnExecutionPolicy?.latencyMode ?? input.latencyMode ?? "standard",
         })
     ) {
       throw new QueueCommandConflictError(
@@ -1384,6 +1415,7 @@ export async function submitHumanPromptInTransaction(
         ...(input.resources.length ? { resources: input.resources } : {}),
         ...(input.model ? { model: input.model } : {}),
         ...(input.reasoningEffort ? { reasoningEffort: input.reasoningEffort } : {}),
+        ...(input.latencyMode ? { latencyMode: input.latencyMode } : {}),
         delivery: input.delivery,
         initiator: frozenInitiator.initiator,
       }),
@@ -1413,6 +1445,7 @@ export async function submitHumanPromptInTransaction(
       toolsProvided: false,
       model: input.model ?? session.model,
       reasoningEffort: input.reasoningEffort ?? input.reasoningEffortFallback,
+      latencyMode: input.turnExecutionPolicy?.latencyMode ?? input.latencyMode ?? "standard",
       sandboxBackend: session.sandboxBackend,
       metadata: input.turnExecutionPolicy
         ? metadataWithTurnExecutionPolicyV1({}, input.turnExecutionPolicy)