Vercel AI SDK

1
import { createOpenRouter } from '@openrouter/ai-sdk-provider';
2
import { streamText } from 'ai';
3
import { z } from 'zod';
4
5
export const getLasagnaRecipe = async (modelName: string) => {
6
  const openrouter = createOpenRouter({
7
    apiKey: '${API_KEY_REF}',
8
  });
9
10
  const response = streamText({
11
    model: openrouter(modelName),
12
    prompt: 'Write a vegetarian lasagna recipe for 4 people.',
13
  });
14
15
  await response.consumeStream();
16
  return response.text;
17
};
18
19
export const getWeather = async (modelName: string) => {
20
  const openrouter = createOpenRouter({
21
    apiKey: '${API_KEY_REF}',
22
  });
23
24
  const response = streamText({
25
    model: openrouter(modelName),
26
    prompt: 'What is the weather in San Francisco, CA in Fahrenheit?',
27
    tools: {
28
      getCurrentWeather: {
29
        description: 'Get the current weather in a given location',
30
        parameters: z.object({
31
          location: z
32
            .string()
33
            .describe('The city and state, e.g. San Francisco, CA'),
34
          unit: z.enum(['celsius', 'fahrenheit']).optional(),
35
        }),
36
        execute: async ({ location, unit = 'celsius' }) => {
37
          // Mock response for the weather
38
          const weatherData = {
39
            'Boston, MA': {
40
              celsius: '15°C',
41
              fahrenheit: '59°F',
42
            },
43
            'San Francisco, CA': {
44
              celsius: '18°C',
45
              fahrenheit: '64°F',
46
            },
47
          };
48
49
          const weather = weatherData[location];
50
          if (!weather) {
51
            return `Weather data for ${location} is not available.`;
52
          }
53
54
          return `The current weather in ${location} is ${weather[unit]}.`;
55
        },
56
      },
57
    },
58
  });
59
60
  await response.consumeStream();
61
  return response.text;
62
};