Function runLocalPi

apps/sim/executor/handlers/pi/local-backend.ts:94–204

(params, context)

}

export const runLocalPi: PiBackendRun<PiLocalRunParams> = async (params, context) => {
  // Isolate Pi resource discovery: an empty cwd/agentDir keeps DefaultResourceLoader
  // from loading the Sim server's own .agents/skills, AGENTS.md, extensions, or settings.
  const isolatedDir = await mkdtemp(join(tmpdir(), 'sim-pi-'))
  // Clean up the scratch dir if the SSH connection fails — the try/finally below
  // is only entered once the session is open, so an early handshake failure would
  // otherwise orphan the directory.
  const session = await openSshSession(params.ssh).catch(async (error) => {
    await rm(isolatedDir, { recursive: true, force: true }).catch(() => {})
    throw error
  })

  try {
    const sdk = await loadPiSdk()

    const authStorage = sdk.AuthStorage.create()
    authStorage.setRuntimeApiKey(params.providerId, params.apiKey)

    const modelRegistry = sdk.ModelRegistry.create(authStorage)
    const thinkingLevel = mapThinkingLevel(params.thinkingLevel)
    // Parity with cloud: when the model isn't in Pi's bundled catalog under the
    // resolved provider, pass it through on that provider instead of failing.
    const model =
      modelRegistry.find(params.providerId, params.model) ??
      buildPiFallbackModel(modelRegistry, params.providerId, params.model, thinkingLevel)
    if (!model) {
      throw new Error(
        `Pi has no models for provider "${params.providerId}" (cannot run ${params.model})`
      )
    }

    const specs = [...buildSshToolSpecs(session, params.repoPath), ...params.tools]
    const customTools = specs.map((spec) => toPiTool(sdk, spec))

    const { session: agentSession } = await sdk.createAgentSession({
      cwd: isolatedDir,
      agentDir: isolatedDir,
      model,
      thinkingLevel,
      noTools: 'builtin',
      customTools,
      authStorage,
      modelRegistry,
      sessionManager: sdk.SessionManager.inMemory(isolatedDir),
    })

    const totals = createPiTotals()
    const unsubscribe = agentSession.subscribe((raw) => {
      const event = normalizePiEvent(raw)
      if (!event) return
      applyPiEvent(totals, event)
      context.onEvent(event)
    })

    const onAbort = () => {
      void agentSession.abort()
    }
    if (context.signal?.aborted) {