picocode-core 0.9.195 → 0.9.196

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "picocode-core",
-  "version": "0.9.195",
+  "version": "0.9.196",
   "description": "The agent runtime behind pico: sessions, tools, subagents, MCP, memory, and model access",
   "type": "module",
   "license": "MIT",
@@ -25,7 +25,7 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.12.0",
-    "@prsm/ai": "^1.9.0",
+    "@prsm/ai": "^1.10.0",
     "diff": "^7.0.0",
     "globby": "^16.2.4",
     "proper-lockfile": "^4.1.2",

package/src/agent.js

@@ -58,10 +58,11 @@ export function compactionHistory(history, prompt) {
   return hydrateImages([...history.filter((m) => m.role !== 'system'), { role: 'user', content: prompt }])
 }
 
-export async function compactHistory({ history, modelName, auth, prompt, signal, onStream }) {
+export async function compactHistory({ history, modelName, speed, auth, prompt, signal, onStream }) {
   const out = await compose(
     model({
       model: modelName,
+      ...(speed && { speed }),
       ...(auth?.apiKey && { apiKey: auth.apiKey }),
       ...(auth?.headers && { headers: auth.headers }),
     }),
@@ -102,7 +103,7 @@ function estimateTokens(value) {
   return Math.ceil(JSON.stringify(value ?? '').length / 4)
 }
 
-export async function runTurn({ history, tools, recorder, modelName, effort, auth, system, signal, onStream, beforeRequest, stallMs = STALL_MS }) {
+export async function runTurn({ history, tools, recorder, modelName, effort, speed, auth, system, signal, onStream, beforeRequest, stallMs = STALL_MS }) {
   const collected = []
   let roundText = ''
   let usageSeen = null
@@ -201,6 +202,7 @@ export async function runTurn({ history, tools, recorder, modelName, effort, aut
             return { ...request, history: [...request.history, ...messages] }
           },
           ...(effort && { effort }),
+          ...(speed && { speed }),
           ...(auth?.apiKey && { apiKey: auth.apiKey }),
           ...(auth?.headers && { headers: auth.headers }),
         })({ ...ctx, history: hydratedHistory, abortSignal: internal.signal })

package/src/codex-models.js

@@ -18,6 +18,7 @@ export function mapCodexModels(entries) {
     desc: `${m.description || m.label || m.slug} · via ChatGPT plan`,
     price: null,
     effort: true,
+    speed: (m.service_tiers || []).some(tier => ['priority', 'fast'].includes(tier.id)) || (m.additional_speed_tiers || []).includes('fast'),
     context: m.context_window || null,
   }))
 }
@@ -27,7 +28,7 @@ export async function loadCodexModels(credentials, { force = false, fetcher = fe
   try {
     cached = JSON.parse(await readFile(cacheFile(), 'utf-8'))
   } catch {}
-  if (!force && cached && Date.now() - cached.at < TTL) return mapCodexModels(cached.models)
+  if (!force && cached && Date.now() - cached.at < TTL && cached.models.every(model => Array.isArray(model.service_tiers) || Array.isArray(model.additional_speed_tiers))) return mapCodexModels(cached.models)
   if (!credentials) return mapCodexModels(cached?.models || [])
 
   try {
@@ -42,6 +43,8 @@ export async function loadCodexModels(credentials, { force = false, fetcher = fe
     const data = await response.json()
     const models = (data.models || []).map((m) => ({
       slug: m.slug,
+      service_tiers: m.service_tiers || [],
+      additional_speed_tiers: m.additional_speed_tiers || [],
       description: m.description,
       label: m.label || m.display_name,
       context_window: m.context_window || m.max_context_window || null,

package/src/controller.js

@@ -36,6 +36,11 @@ import { MAX_DELIBERATION_ROUNDS } from './deliberation.js'
 import { buildUserContent, finalizeUserContent, inputTextFromContent, mediaTypeFor, stashImages } from './attachments.js'
 import { settlePending } from './pending.js'
 
+export const SPEED_LEVELS = [
+  { key: 'standard', usage: '1x usage' },
+  { key: 'fast', usage: '2.5x usage' },
+]
+
 export const EFFORT_LEVELS = [
   { key: null, desc: 'let the provider decide how much to think' },
   { key: 'low', desc: 'quick answers, minimal thinking' },
@@ -114,6 +119,8 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
     defaultModel: boot.initialModel,
     effort: boot.initialEffort,
     defaultEffort: boot.initialEffort,
+    speedDefaults: { ...boot.speedDefaults },
+    speed: boot.speedDefaults?.[boot.initialModel.name] === 'fast' ? 'fast' : 'standard',
     busy: false,
     compacting: false,
     compactOutcome: null,
@@ -293,6 +300,9 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
       // by file needs to hear about it
       emit('session', state.session)
     }
+    if (!state.events.some(event => event.type === 'speed')) {
+      state.events.push(makeEvent('speed', { to: state.speed }))
+    }
     while (state.persisted < state.events.length) {
       state.session.append(state.events[state.persisted])
       state.persisted++
@@ -350,6 +360,7 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
         recorder,
         modelName: worker.name,
         effort: worker.effort ? 'low' : null,
+        ...(worker.provider === 'codex' && { speed: 'standard' }),
         auth,
         system: context.system,
         signal,
@@ -421,6 +432,7 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
           recorder: toolset.recorder,
           modelName: roleWorkers[role].name,
           effort: roleWorkers[role].effort ? 'low' : null,
+          ...(roleWorkers[role].provider === 'codex' && { speed: 'standard' }),
           auth: roleAuths[role],
           system: context.system,
           signal: workerSignal,
@@ -606,6 +618,7 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
       const raw = await compactHistory({
         history: current.providerHistory,
         modelName: state.model.name,
+        ...(state.model.provider === 'codex' && { speed: speedApplies() ? state.speed : 'standard' }),
         auth,
         prompt: compactionPrompt(instructions),
         signal: controller.signal,
@@ -828,6 +841,7 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
         recorder,
         modelName: state.model.name,
         effort: effortApplies() ? state.effort ?? 'auto' : null,
+        ...(state.model.provider === 'codex' && { speed: speedApplies() ? state.speed : 'standard' }),
         auth,
         system: context.system,
         signal: controller.signal,
@@ -946,7 +960,7 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
     set({ queued: [...state.expedited, ...state.queued], expedited: [] })
   }
 
-  function resetConversation({ model, effort } = {}) {
+  function resetConversation({ model, effort, speed } = {}) {
     state.session = null
     state.events = []
     state.persisted = 0
@@ -958,6 +972,7 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
     state.peerPaused = false
     state.model = model ?? state.defaultModel
     state.effort = effort === undefined ? state.defaultEffort : effort
+    state.speed = speed ?? defaultSpeed(state.model)
   }
 
   async function newSession() {
@@ -1192,6 +1207,7 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
       reDerive()
       restoreModelFromSession()
       state.effort = state.derived.effort === undefined ? state.defaultEffort : state.derived.effort
+      state.speed = state.derived.speed ?? 'standard'
       state.sent = userEntries(state.derived).map((e) => ({ text: recallText(e), at: header.createdAt }))
       changed()
       emit('session', state.session)
@@ -1319,6 +1335,8 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
     }
     persist(makeEvent('model_switch', { from: state.model.name, to: pick.name }))
     state.model = pick
+    state.speed = defaultSpeed(pick)
+    persist(makeEvent('speed', { to: state.speed }))
     if (asDefault) {
       state.defaultModel = pick
       writeConfig({ defaultModel: pick.name }).catch(() => {})
@@ -1335,6 +1353,23 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
     switchModel(pick, { note: adhoc ? ' (not in catalog, pricing unknown)' : '' })
   }
 
+  const defaultSpeed = model => state.speedDefaults[model.name] === 'fast' ? 'fast' : 'standard'
+  const speedApplies = () => state.model.provider === 'codex' && !!state.model.speed
+
+  function applySpeed(next, { asDefault = false } = {}) {
+    if (!speedApplies()) return flash(`${state.model.name} does not support Fast mode`)
+    if (!SPEED_LEVELS.some(level => level.key === next)) return flash('usage: /speed <standard|fast>')
+    persist(makeEvent('speed', { to: next }))
+    state.speed = next
+    if (asDefault) {
+      state.speedDefaults[state.model.name] = next
+      writeConfig({ speedDefaults: { [state.model.name]: next } }).catch(error => flash(`speed default failed: ${errorText(error, 80)}`))
+    }
+    changed()
+    flash(`speed: ${next}`)
+  }
+
+  const speedCommand = arg => applySpeed(arg.toLowerCase())
   const effortApplies = () => !!state.model.effort
 
   function setEffort(next, { asDefault = false } = {}) {
@@ -1714,6 +1749,9 @@ export function createController({ boot, run = runTurn, peerDirectory } = {}) {
     resolveModel,
     switchModel,
     switchModelByName,
+    speedApplies,
+    applySpeed,
+    speedCommand,
     effortApplies,
     setEffort,
     setEffortByName,

package/src/derive.js

@@ -196,6 +196,7 @@ export function deriveState(events) {
     historyEventIds: [],
     model: null,
     effort: undefined,
+    speed: undefined,
     usage: spentUsage,
     usageByModel: spentUsageByModel,
     usageActive: emptyUsage(),
@@ -310,6 +311,9 @@ export function deriveState(events) {
       case 'model_switch':
         state.model = event.data.to
         break
+      case 'speed':
+        state.speed = event.data.to
+        break
       case 'effort':
         state.effort = event.data.to
         break