Use default contract on first message

Author: MaxLinCode

apps/web/src/lib/server/turn-router.test.ts

@@ -17,8 +17,10 @@ vi.mock("./slot-extractor", () => ({
 }));
 
 import { classifyTurn } from "./llm-classifier";
+import { extractSlots } from "./slot-extractor";
 
 const mockClassifyTurn = vi.mocked(classifyTurn);
+const mockExtractSlots = vi.mocked(extractSlots);
 
 function mockClassification(output: Partial<TurnClassifierOutput>) {
   const full: TurnClassifierOutput = {
@@ -36,6 +38,11 @@ describe("turn router", () => {
       turnType: "planning_request",
       confidence: 0.95
     });
+    mockExtractSlots.mockResolvedValueOnce({
+      extractedValues: { day: "tomorrow", time: "18:00" },
+      confidence: { day: 0.95, time: 0.95 },
+      unresolvable: []
+    });
 
     const result = await routeMessageTurn({
       rawText: "Schedule gym tomorrow at 6pm for 1 hour",

apps/web/src/lib/server/turn-router.ts

@@ -27,7 +27,7 @@ export type TurnRouterInput = TurnRoutingInput;
 export type TurnRouterResult = RoutedTurn;
 
 const DEFAULT_CONTRACT: WriteContract = {
-  requiredSlots: [],
+  requiredSlots: ["day", "time"],
   intentKind: "plan"
 };
 
@@ -46,13 +46,10 @@ export async function routeMessageTurn(input: TurnRouterInput): Promise<TurnRout
   const activeContract = discourseState.pending_write_contract ?? DEFAULT_CONTRACT;
   let slotExtraction = null;
 
-  if (
-    SLOT_COMMITTING_TURN_TYPES.has(classification.turnType) &&
-    discourseState.pending_write_contract
-  ) {
+  if (SLOT_COMMITTING_TURN_TYPES.has(classification.turnType)) {
     slotExtraction = await extractSlots({
       currentTurnText: input.normalizedText,
-      pendingSlots: derivePendingSlots(discourseState),
+      pendingSlots: derivePendingSlots(activeContract, discourseState.resolved_slots ?? {}),
       priorResolvedSlots: discourseState.resolved_slots ?? {},
       conversationContext: deriveConversationContext(input.recentTurns)
     });
@@ -83,11 +80,8 @@ export async function routeMessageTurn(input: TurnRouterInput): Promise<TurnRout
   });
 }
 
-function derivePendingSlots(discourseState: DiscourseState): SlotKey[] {
-  const contract = discourseState.pending_write_contract;
-  if (!contract) return [];
-  const resolved = discourseState.resolved_slots ?? {};
-  return contract.requiredSlots.filter((slot) => resolved[slot] === undefined);
+function derivePendingSlots(contract: WriteContract, resolvedSlots: Record<string, unknown>): SlotKey[] {
+  return contract.requiredSlots.filter((slot) => resolvedSlots[slot] === undefined);
 }
 
 function deriveConversationContext(recentTurns: ConversationTurn[]): string {