Intégrations

Vercel AI SDK

Encapsule Thunderbit comme Tool TypeScript pour les agents streamText / generateText

Le helper tool() du Vercel AI SDK transforme n'importe quelle fonction TypeScript en quelque chose qu'un LLM peut appeler. Encapsule /distill et le modèle peut lire des pages web en direct au milieu d'une conversation.

Installation

npm install ai @ai-sdk/openai zod

Définir le Tool

import { tool } from 'ai';
import { z } from 'zod';

const API = 'https://openapi.thunderbit.com/openapi/v1';
const H = { Authorization: `Bearer ${process.env.THUNDERBIT_API_KEY}` };

export const readUrl = tool({
  description:
    'Fetch a URL and return clean Markdown. Use for any web research task: docs, articles, product pages.',
  parameters: z.object({
    url: z.string().url(),
    renderMode: z.enum(['basic', 'advanced']).default('basic'),
  }),
  execute: async ({ url, renderMode }) => {
    const res = await fetch(`${API}/distill`, {
      method: 'POST',
      headers: { ...H, 'Content-Type': 'application/json' },
      body: JSON.stringify({ url, renderMode }),
    });
    const json = await res.json();
    return json.data.markdown;
  },
});

Utiliser avec streamText

import { openai } from '@ai-sdk/openai';
import { streamText } from 'ai';

const result = streamText({
  model: openai('gpt-4o'),
  tools: { readUrl },
  maxSteps: 5,
  prompt: 'Summarize the latest post on https://vercel.com/blog',
});

for await (const chunk of result.textStream) process.stdout.write(chunk);

Astuces

  • Limite le Markdown retourné à environ 8k tokens avant de retourner depuis execute — évite la saturation du contexte
  • Pour la parallélisation multi-URL, expose un second Tool readUrls adossé à /batch/distill
  • renderMode: 'advanced' uniquement quand la page est lourde en JS — basic est 3 à 5 fois moins coûteux

Liens connexes