Class PiBlockHandler

apps/sim/executor/handlers/pi/pi-handler.ts:52–262

}

export class PiBlockHandler implements BlockHandler {
  canHandle(block: SerializedBlock): boolean {
    return block.metadata?.id === BlockType.PI
  }

  async execute(
    ctx: ExecutionContext,
    block: SerializedBlock,
    inputs: Record<string, any>
  ): Promise<BlockOutput | StreamingExecution> {
    const task = asOptString(inputs.task)
    if (!task) throw new Error('Task is required')
    const model = asOptString(inputs.model) ?? DEFAULT_MODEL

    // Validate the mode up front so an invalid value reports a mode error rather
    // than a misattributed credential error from key resolution below.
    if (inputs.mode !== 'cloud' && inputs.mode !== 'local') {
      throw new Error(`Invalid Pi mode: ${String(inputs.mode)}`)
    }
    const mode: 'cloud' | 'local' = inputs.mode

    const { providerId, apiKey, isBYOK } = await resolvePiModelKey({
      model,
      mode,
      workspaceId: ctx.workspaceId,
      userId: ctx.userId,
      apiKey: asRawString(inputs.apiKey),
      vertexCredential: asOptString(inputs.vertexCredential),
    })

    const skills = await resolvePiSkills(inputs.skills, ctx.workspaceId)
    const memoryConfig: PiMemoryConfig = {
      memoryType: asOptString(inputs.memoryType) as PiMemoryConfig['memoryType'],
      conversationId: asOptString(inputs.conversationId),
      slidingWindowSize: asOptString(inputs.slidingWindowSize),
      slidingWindowTokens: asOptString(inputs.slidingWindowTokens),
      model,
    }
    const initialMessages = await loadPiMemory(ctx, memoryConfig)

    const base = {
      model,
      providerId,
      apiKey,
      isBYOK,
      task,
      thinkingLevel: asOptString(inputs.thinkingLevel),
      skills,
      initialMessages,
    }

    if (mode === 'local') {
      const host = asOptString(inputs.host)
      const username = asOptString(inputs.username)
      const repoPath = asOptString(inputs.repoPath)
      if (!host || !username || !repoPath) {
        throw new Error('Local mode requires host, username, and repository path')
      }