@danceiny/gotry 0.0.1-rc.21 → 0.0.1-rc.24

package/dist/scripts/booking-copilot-dsh-core-proof-tests.js

@@ -1,9 +1,16 @@
+/**
+ * Real DeepSeek Harness subprocess proof for the embedded booking planner.
+ *
+ * A local OpenAI-compatible SSE fixture is the model transport; everything
+ * between it and BookingPlannerDecision is the published dsh core, its SDK
+ * stdio server, the GoTry plugin, tool execution, session events and adapter.
+ */
 import assert from 'node:assert/strict';
 import { createServer } from 'node:http';
 import { mkdtempSync, rmSync } from 'node:fs';
 import { tmpdir } from 'node:os';
 import { join } from 'node:path';
-import { createDshEmbeddedBookingPlanner, DSH_EMBEDDED_BOOKING_TOOL_NAMES } from '../src/booking-surface/dsh-planner.js';
+import { createDshEmbeddedBookingPlanner, DSH_EMBEDDED_BOOKING_TOOL_NAMES, } from '../src/booking-surface/dsh-planner.js';
 const action = {
     schemaVersion: 'booking.surface',
     kind: 'search.run',
@@ -12,147 +19,84 @@ const action = {
     expectedRevision: 0,
     reason: 'Run the authoritative workspace search.',
     factRefs: [],
-    input: {}
+    input: {},
 };
 function sse(res, chunks) {
     res.writeHead(200, {
         'content-type': 'text/event-stream; charset=utf-8',
-        'cache-control': 'no-cache'
+        'cache-control': 'no-cache',
     });
-    for (const chunk of chunks)res.write(`data: ${JSON.stringify(chunk)}\n\n`);
+    for (const chunk of chunks)
+        res.write(`data: ${JSON.stringify(chunk)}\n\n`);
     res.end('data: [DONE]\n\n');
 }
 function objectKeys(value) {
-    if (Array.isArray(value)) return value.flatMap(objectKeys);
-    if (typeof value !== 'object' || value === null) return [];
-    return Object.entries(value).flatMap(([key, child])=>[
-            key,
-            ...objectKeys(child)
-        ]);
+    if (Array.isArray(value))
+        return value.flatMap(objectKeys);
+    if (typeof value !== 'object' || value === null)
+        return [];
+    return Object.entries(value).flatMap(([key, child]) => [key, ...objectKeys(child)]);
 }
 const requests = [];
 let modelCall = 0;
 let continuationServed = false;
-const modelServer = createServer((req, res)=>{
+const modelServer = createServer((req, res) => {
     const parts = [];
-    req.on('data', (part)=>parts.push(part));
-    req.on('end', ()=>{
+    req.on('data', (part) => parts.push(part));
+    req.on('end', () => {
         const body = JSON.parse(Buffer.concat(parts).toString('utf8'));
-        requests.push({
-            headers: req.headers,
-            body
-        });
+        requests.push({ headers: req.headers, body });
         modelCall += 1;
+        // Script by request content, not call order: the prose-only retry budget
+        // issues additional runs whose scripted response must stay prose, not the
+        // continuation terminal.
         if (!continuationServed && JSON.stringify(body).includes('action.receipt.continuation')) {
             continuationServed = true;
-            sse(res, [
-                {
-                    id: 'chatcmpl-booking-continuation',
-                    object: 'chat.completion.chunk',
-                    created: 0,
-                    model: 'deepseek-v4-flash',
-                    choices: [
-                        {
-                            index: 0,
-                            delta: {
-                                role: 'assistant',
-                                tool_calls: [
-                                    {
-                                        index: 0,
-                                        id: 'call-booking-continuation',
-                                        type: 'function',
-                                        function: {
-                                            name: 'booking_search_hotels',
-                                            arguments: JSON.stringify({
-                                                decision: {
-                                                    kind: 'terminal',
-                                                    terminal: {
-                                                        status: 'stopped',
-                                                        summary: 'receipt continuation handled',
-                                                        factRefs: []
-                                                    }
-                                                }
-                                            })
-                                        }
-                                    }
-                                ]
-                            },
-                            finish_reason: 'tool_calls'
-                        }
-                    ]
-                }
-            ]);
+            sse(res, [{
+                    id: 'chatcmpl-booking-continuation', object: 'chat.completion.chunk', created: 0, model: 'deepseek-v4-flash',
+                    choices: [{ index: 0, delta: { role: 'assistant', tool_calls: [{ index: 0, id: 'call-booking-continuation', type: 'function', function: { name: 'booking_search_hotels', arguments: JSON.stringify({ decision: { kind: 'terminal', terminal: { status: 'stopped', summary: 'receipt continuation handled', factRefs: [] } } }) } }] }, finish_reason: 'tool_calls' }],
+                }]);
             return;
         }
         if (modelCall === 1) {
-            sse(res, [
-                {
+            sse(res, [{
                     id: 'chatcmpl-booking-1',
                     object: 'chat.completion.chunk',
                     created: 0,
                     model: 'deepseek-v4-flash',
-                    choices: [
-                        {
+                    choices: [{
                             index: 0,
                             delta: {
                                 role: 'assistant',
-                                tool_calls: [
-                                    {
+                                tool_calls: [{
                                         index: 0,
                                         id: 'call-booking-1',
                                         type: 'function',
                                         function: {
                                             name: 'booking_search_hotels',
-                                            arguments: JSON.stringify({
-                                                decision: {
-                                                    kind: 'operation',
-                                                    action
-                                                }
-                                            })
-                                        }
-                                    }
-                                ]
+                                            arguments: JSON.stringify({ decision: { kind: 'operation', action } }),
+                                        },
+                                    }],
                             },
-                            finish_reason: 'tool_calls'
-                        }
-                    ],
-                    usage: {
-                        prompt_tokens: 10,
-                        completion_tokens: 5,
-                        total_tokens: 15
-                    }
-                }
-            ]);
+                            finish_reason: 'tool_calls',
+                        }],
+                    usage: { prompt_tokens: 10, completion_tokens: 5, total_tokens: 15 },
+                }]);
             return;
         }
-        sse(res, [
-            {
+        sse(res, [{
                 id: 'chatcmpl-booking-2',
                 object: 'chat.completion.chunk',
                 created: 0,
                 model: 'deepseek-v4-flash',
-                choices: [
-                    {
-                        index: 0,
-                        delta: {
-                            role: 'assistant',
-                            content: 'Typed decision emitted.'
-                        },
-                        finish_reason: 'stop'
-                    }
-                ],
-                usage: {
-                    prompt_tokens: 20,
-                    completion_tokens: 4,
-                    total_tokens: 24
-                }
-            }
-        ]);
+                choices: [{ index: 0, delta: { role: 'assistant', content: 'Typed decision emitted.' }, finish_reason: 'stop' }],
+                usage: { prompt_tokens: 20, completion_tokens: 4, total_tokens: 24 },
+            }]);
     });
 });
-await new Promise((resolve, reject)=>{
+await new Promise((resolve, reject) => {
     modelServer.once('error', reject);
-    modelServer.listen(0, '127.0.0.1', ()=>resolve());
+    modelServer.listen(0, '127.0.0.1', () => resolve());
 });
 const address = modelServer.address();
 assert.ok(typeof address === 'object' && address);
@@ -165,18 +109,11 @@ const workspace = {
     locale: 'zh-CN',
     currency: 'AED',
     searchDraft: {},
-    results: {
-        status: 'idle'
-    },
+    results: { status: 'idle' },
     visibleHotels: [],
     loadedOffers: [],
     shortlistedOfferRefs: [],
-    capabilities: {
-        surface: 'tenant',
-        allowedActions: [
-            'search.run'
-        ]
-    }
+    capabilities: { surface: 'tenant', allowedActions: ['search.run'] },
 };
 const turn = {
     schemaVersion: 'booking.surface',
@@ -184,36 +121,23 @@ const turn = {
     taskId: 'task-real-dsh-1',
     turnId: 'real-dsh-turn-1',
     workspace,
-    request: {
-        text: '执行当前酒店搜索'
-    }
-};
-const availability = {
-    initialized: true,
-    recoveryStarted: false,
-    availabilityPhase: 'need_offers',
-    activeHotelOrdinal: 0,
-    hotelRefs: [],
-    hotels: {},
-    attempts: [],
-    queryReservations: []
+    request: { text: '执行当前酒店搜索' },
 };
+const availability = { initialized: true, recoveryStarted: false, availabilityPhase: 'need_offers', activeHotelOrdinal: 0, hotelRefs: [], hotels: {}, attempts: [], queryReservations: [] };
 const task = {
     schemaVersion: 'booking.surface',
     taskId: 'task-real-dsh-1',
     contextRef: action.contextRef,
     surface: 'tenant',
     revision: 0,
-    allowedActions: [
-        'search.run'
-    ],
+    allowedActions: ['search.run'],
     userTurnCount: 1,
     lastTurnId: 'real-dsh-turn-1',
     operationCount: 0,
     phase: 'planning',
     lastSequence: 0,
     availability,
-    workspaceSnapshot: workspace
+    workspaceSnapshot: workspace,
 };
 let planner;
 try {
@@ -224,88 +148,34 @@ try {
             DEEPSEEK_API_KEY: 'fixture-model-key',
             DEEPSEEK_BASE_URL: `http://127.0.0.1:${address.port}/v1`,
             GOTRY_BOOKING_COPILOT_API_KEY: 'must-not-enter-dsh',
-            HOTELBYTE_TOKEN: 'must-not-enter-dsh'
+            HOTELBYTE_TOKEN: 'must-not-enter-dsh',
         },
-        maxTokens: 512
+        maxTokens: 512,
     });
     const session = planner.plannerFactory(task);
-    const decisions = await session.next({
-        turn,
-        task
-    });
-    assert.deepEqual(decisions, [
-        {
-            kind: 'operation',
-            action
-        }
-    ]);
+    const decisions = await session.next({ turn, task });
+    assert.deepEqual(decisions, [{ kind: 'operation', action }]);
     assert.equal(requests.length, 2, 'real dsh loop executes the typed tool then reaches idle');
-    const toolNames = requests[0].body.tools.map((tool)=>tool.function.name).sort();
-    assert.deepEqual(toolNames, [
-        ...DSH_EMBEDDED_BOOKING_TOOL_NAMES
-    ].sort(), 'real model request exposes exactly the six embedded tools');
-    assert.ok(!toolNames.some((name)=>/gotry_book|payment|holder|guest/i.test(name)), 'real model request exposes no booking write or PII tool');
-    const executableKeys = requests[0].body.tools.flatMap((tool)=>objectKeys(tool.function.parameters));
-    assert.ok(!executableKeys.some((key)=>/^(book|payment|holder|guest|portalToken|supplierCost)$/i.test(key)), 'tool inputs expose no write or PII field');
+    const toolNames = requests[0].body.tools.map((tool) => tool.function.name).sort();
+    assert.deepEqual(toolNames, [...DSH_EMBEDDED_BOOKING_TOOL_NAMES].sort(), 'real model request exposes exactly the six embedded tools');
+    assert.ok(!toolNames.some((name) => /gotry_book|payment|holder|guest/i.test(name)), 'real model request exposes no booking write or PII tool');
+    const executableKeys = requests[0].body.tools.flatMap((tool) => objectKeys(tool.function.parameters));
+    assert.ok(!executableKeys.some((key) => /^(book|payment|holder|guest|portalToken|supplierCost)$/i.test(key)), 'tool inputs expose no write or PII field');
     assert.equal(requests[0].headers.authorization, 'Bearer fixture-model-key');
     assert.ok(!JSON.stringify(requests).includes('must-not-enter-dsh'), 'BFF and portal credentials never enter the dsh model transport');
-    const proseDecisions = await session.next({
-        turn,
-        task
-    });
+    const proseDecisions = await session.next({ turn, task });
     assert.equal(proseDecisions[0]?.kind, 'error', 'assistant prose without a tool call never becomes a decision');
     const continuation = {
-        schemaVersion: 'booking.surface',
-        kind: 'action.receipt.continuation',
-        taskId: task.taskId,
-        workspace,
-        receipt: {
-            schemaVersion: 'booking.surface',
-            kind: 'action.receipt',
-            actionId: action.actionId,
-            contextRef: workspace.contextRef,
-            status: 'applied',
-            revision: 1,
-            observation: {
-                kind: 'search.state',
-                resultCount: 1
-            },
-            resultContract: {
-                outcome: 'complete',
-                hardCriteriaMet: true,
-                factRefs: [],
-                gapCodes: [],
-                blockers: [],
-                relaxationsApplied: []
-            }
-        }
+        schemaVersion: 'booking.surface', kind: 'action.receipt.continuation', taskId: task.taskId, workspace,
+        receipt: { schemaVersion: 'booking.surface', kind: 'action.receipt', actionId: action.actionId, contextRef: workspace.contextRef, status: 'applied', revision: 1,
+            observation: { kind: 'search.state', resultCount: 1 }, resultContract: { outcome: 'complete', hardCriteriaMet: true, factRefs: [], gapCodes: [], blockers: [], relaxationsApplied: [] } },
     };
-    const continuationDecisions = await session.next({
-        turn: continuation,
-        task: {
-            ...task,
-            lastReceipt: continuation.kind === 'action.receipt.continuation' ? continuation.receipt : undefined
-        }
-    });
-    assert.deepEqual(continuationDecisions, [
-        {
-            kind: 'terminal',
-            terminal: {
-                status: 'stopped',
-                summary: 'receipt continuation handled',
-                factRefs: []
-            }
-        }
-    ], 'real dsh accepts action.receipt.continuation through the typed planner seam');
-} finally{
+    const continuationDecisions = await session.next({ turn: continuation, task: { ...task, lastReceipt: continuation.kind === 'action.receipt.continuation' ? continuation.receipt : undefined } });
+    assert.deepEqual(continuationDecisions, [{ kind: 'terminal', terminal: { status: 'stopped', summary: 'receipt continuation handled', factRefs: [] } }], 'real dsh accepts action.receipt.continuation through the typed planner seam');
+}
+finally {
     await planner?.close();
-    await new Promise((resolve, reject)=>modelServer.close((error)=>error ? reject(error) : resolve()));
-    rmSync(stateRoot, {
-        recursive: true,
-        force: true
-    });
+    await new Promise((resolve, reject) => modelServer.close((error) => error ? reject(error) : resolve()));
+    rmSync(stateRoot, { recursive: true, force: true });
 }
 console.log('BOOKING COPILOT DSH CORE PROOF: real subprocess/sdk/plugin/tool-call/session event/idle OK');
-
-
-//# sourceURL=ts/scripts/booking-copilot-dsh-core-proof-tests.ts