dsh-ptc-plus 0.2.1

package/internal/runtime-contexts.js

@@ -0,0 +1,31 @@
+import { latestRecoveryTip } from './recovery-tips.js'
+import { projectSessionLog } from './session-log-view.js'
+
+const REWRITE_FEEDBACK = 'tools:ptc-plus-rewrite-info'
+
+function continuationFeedback(view) {
+  const run = view.latestRun
+  const rewrites = run?.rewrites
+  if (rewrites === undefined || rewrites.length === 0) return undefined
+  const details = rewrites.map(rewrite => rewrite.description).join('; ')
+  if (run.journal === undefined) {
+    return `The preceding run_code cell had these source adjustments: ${details}. Its completion is unknown because no valid execution journal is available. Inspect its tool result and live bindings before continuing; do not assume it completed or failed, and do not replay it automatically.`
+  }
+  if (run.journal.completion?.kind !== 'return') {
+    return `The preceding run_code cell failed after a source adjustment: ${details}. Treat it as failed; inspect its tool result and live bindings before continuing, and do not replay it automatically.`
+  }
+  return undefined
+}
+
+/** Build all dynamic PTC contexts from one session-log projection. */
+export function sessionRuntimeContexts(agent, tipConfig) {
+  const view = projectSessionLog(agent)
+  const rewrite = continuationFeedback(view)
+  const tip = latestRecoveryTip(view, tipConfig)
+  return Object.freeze({
+    contexts: Object.freeze([
+      ...(rewrite === undefined ? [] : [{ name: REWRITE_FEEDBACK, text: rewrite }]),
+      ...(tip === undefined ? [] : [tip]),
+    ]),
+  })
+}

package/internal/sdk-projection.js

@@ -0,0 +1,88 @@
+/** Adapt the host program SDK without weakening the declared direct tool surface. */
+import { parse } from '@babel/parser'
+
+const PROJECTION_ERROR = 'ptc-plus: incompatible tools SDK projection: '
+const TYPESCRIPT_FENCE = '```ts\n'
+
+function incompatible(message, cause = undefined) {
+  return new Error(`${PROJECTION_ERROR}${message}`, cause === undefined ? undefined : { cause })
+}
+
+function lineStart(source, offset) {
+  return source.lastIndexOf('\n', offset - 1) + 1
+}
+
+function memberName(member) {
+  if (member.type !== 'TSPropertySignature') return undefined
+  if (member.key.type === 'Identifier') return member.key.name
+  if (member.key.type === 'StringLiteral') return member.key.value
+  return undefined
+}
+
+function supportedFence(source) {
+  const starts = []
+  for (let offset = 0; offset < source.length;) {
+    const start = source.indexOf(TYPESCRIPT_FENCE, offset)
+    if (start < 0) break
+    if (start === 0 || source[start - 1] === '\n') starts.push(start)
+    offset = start + TYPESCRIPT_FENCE.length
+  }
+  if (starts.length !== 1) {
+    throw incompatible(`expected exactly one ${JSON.stringify(TYPESCRIPT_FENCE.trim())} fence`)
+  }
+  const codeStart = starts[0] + TYPESCRIPT_FENCE.length
+  const codeEnd = source.indexOf('\n```', codeStart)
+  if (codeEnd < 0) throw incompatible('the TypeScript fence is not closed')
+  return { codeStart, codeEnd }
+}
+
+/** Remove one direct-only tool from the host-generated program SDK. */
+export function projectProgramSdk(nativeSdk, excludedName = 'edit_run_code') {
+  if (typeof nativeSdk !== 'string') return ''
+  if (!nativeSdk.includes(excludedName)) return nativeSdk
+
+  const { codeStart, codeEnd } = supportedFence(nativeSdk)
+  const code = nativeSdk.slice(codeStart, codeEnd)
+  let ast
+  try {
+    ast = parse(code, { sourceType: 'module', plugins: ['typescript'] })
+  } catch (error) {
+    throw incompatible('the TypeScript fence could not be parsed', error)
+  }
+
+  const removals = []
+  for (const interfaceName of ['ToolArgsMap', 'ToolOutputMap']) {
+    const declarations = ast.program.body.filter(statement => (
+      statement.type === 'TSInterfaceDeclaration' && statement.id.name === interfaceName
+    ))
+    if (declarations.length !== 1) {
+      throw incompatible(`expected exactly one ${interfaceName} interface`)
+    }
+    const declaration = declarations[0]
+    const members = declaration.body.body.filter(member => memberName(member) === excludedName)
+    if (members.length !== 1) {
+      throw incompatible(`expected exactly one ${excludedName} member in ${interfaceName}`)
+    }
+    const member = members[0]
+    if (declaration.body.body.length === 1) {
+      removals.push([declaration.body.start + 1, declaration.body.end - 1])
+      continue
+    }
+    const commentStart = member.leadingComments?.[0]?.start
+    const startOffset = commentStart ?? member.start
+    const start = lineStart(code, startOffset)
+    let end = member.end
+    if (code[end] === '\n') end += 1
+    removals.push([start, end])
+  }
+
+  let projectedCode = code
+  for (const [start, end] of removals.sort((left, right) => right[0] - left[0])) {
+    projectedCode = projectedCode.slice(0, start) + projectedCode.slice(end)
+  }
+  const projected = nativeSdk.slice(0, codeStart) + projectedCode + nativeSdk.slice(codeEnd)
+  if (projected.includes(excludedName)) {
+    throw incompatible(`residual ${excludedName} token remains after projection`)
+  }
+  return projected
+}