Function makeAdapter

packages/ai/tests/structured-output-middleware.test.ts:139–184

(opts: {
  agentIterations?: Array<Array<StreamChunk>>
  structuredValue?: Person
  structuredRunFinishedUsage?: RunFinishedEvent['usage']
  structuredOutputThrows?: Error
  noNativeStructuredOutputStream?: boolean
  recordStructuredCalls?: Recorder
})

}

function makeAdapter(opts: {
  agentIterations?: Array<Array<StreamChunk>>
  structuredValue?: Person
  structuredRunFinishedUsage?: RunFinishedEvent['usage']
  structuredOutputThrows?: Error
  noNativeStructuredOutputStream?: boolean
  recordStructuredCalls?: Recorder
}) {
  const value = opts.structuredValue ?? PERSON
  const recordIn = opts.recordStructuredCalls

  const { adapter } = createMockAdapter({
    iterations: opts.agentIterations,
    structuredOutput: async (o) => {
      if (recordIn) {
        recordIn.push({
          chatOptions: o.chatOptions,
          outputSchema: o.outputSchema,
        })
      }
      if (opts.structuredOutputThrows) throw opts.structuredOutputThrows
      return { data: value, rawText: JSON.stringify(value) }
    },
    ...(opts.noNativeStructuredOutputStream
      ? {}
      : {
          structuredOutputStream: (o) => {
            if (recordIn) {
              recordIn.push({
                chatOptions: o.chatOptions,
                outputSchema: o.outputSchema,
              })
            }
            const chunks = buildStructuredStream(
              value,
              opts.structuredRunFinishedUsage,
            )
            return (async function* () {
              for (const c of chunks) yield c
            })()
          },
        }),
  })

  return adapter
}

describe('chat({ outputSchema }) — Promise<T> path', () => {
  it('middleware observes chunks attributed to phase=structuredOutput', async () => {