Benchmark a stock with Equity Strategist

import { Chaos, EQUITY_MODEL } from '@chaoslabs/ai-sdk';
 
const chaos = new Chaos({ apiKey: process.env.CHAOS_API_KEY! });
const tickers = ['NVDA', 'AAPL', 'MSFT', 'TSLA'];
 
const results = await Promise.all(
  tickers.map(async (t) => {
    const start = Date.now();
    const r = await chaos.chat.responses.create({
      model: EQUITY_MODEL,
      input: [
        {
          type: 'message',
          role: 'user',
          content: `Is ${t} fairly valued at the current price? Give a rationale.`,
        },
      ],
      metadata: {
        user_id: 'benchmark-runner',
        session_id: `benchmark-${t}-${start}`,
      },
    });
 
    // info blocks are emitted at runtime but aren't yet in the public Block helper union;
    // widen locally to read the discriminator + content.
    const blocks = (r.messages ?? [])
      .filter((m): m is Extract<typeof m, { type: 'block' }> => m.type === 'block')
      .map((m) => m.data.block as { type: string; content?: string });
    const rationale = blocks
      .filter((b) => b.type === 'info' && typeof b.content === 'string')
      .map((b) => b.content)
      .join('\n\n');
 
    return { ticker: t, rationale, latencyMs: Date.now() - start };
  })
);
 
console.table(results);