resource_planner.ts

import s from 'dedent'
import { zodResponseFormat } from 'openai/helpers/zod'
import { z } from 'zod'

import { agent, AgentOptions } from '../agent.js'
import { handoff, request, response } from '../state.js'

const defaults: AgentOptions = {
  run: async (state, context, workflow) => {
    const res = await workflow.team[state.agent].provider.completions({
      messages: [
        {
          role: 'system',
          content: s`
            You are an agent selector that matches tasks to the most capable agent.
            Analyze the task requirements and each agent's capabilities to select the best match.
            
            Consider:
            1. Required tools and skills
            2. Agent's specialization
            3. Model capabilities
            4. Previous task context if available  
          `,
        },
        request(s`
          Here are the available agents:
          <agents>
            ${Object.entries(workflow.team).map(([name, agent]) =>
              agent.description ? `<agent name="${name}">${agent.description}</agent>` : ''
            )}
          </agents>`),
        response('What is the task?'),
        ...state.messages,
      ],
      temperature: 0.1,
      response_format: zodResponseFormat(
        z.object({
          agent: z.enum(Object.keys(workflow.team) as [string, ...string[]]),
          reasoning: z.string(),
        }),
        'agent_selection'
      ),
    })

    const message = res.choices[0].message.parsed
    if (!message) {
      throw new Error('No content in response')
    }

    return handoff(state, message.agent, state.messages)
  },
}

export const resourcePlanner = (options?: AgentOptions) =>
  agent({
    ...defaults,
    ...options,
  })