Initial commit

templates/tools-basic.ts

@@ -0,0 +1,75 @@
+// Basic tool calling example
+// AI SDK Core - Tool calling with generateText()
+
+import { generateText, tool } from 'ai';
+import { openai } from '@ai-sdk/openai';
+import { z } from 'zod';
+
+async function main() {
+  const result = await generateText({
+    model: openai('gpt-4'),
+    tools: {
+      weather: tool({
+        description: 'Get the current weather for a location',
+        inputSchema: z.object({
+          location: z.string().describe('City name, e.g. "San Francisco"'),
+          unit: z.enum(['celsius', 'fahrenheit']).optional().describe('Temperature unit'),
+        }),
+        execute: async ({ location, unit = 'fahrenheit' }) => {
+          // Simulate API call to weather service
+          console.log(`[Tool] Fetching weather for ${location}...`);
+
+          // In production, call real weather API here
+          const mockWeather = {
+            location,
+            temperature: unit === 'celsius' ? 22 : 72,
+            condition: 'sunny',
+            humidity: 65,
+            unit,
+          };
+
+          return mockWeather;
+        },
+      }),
+      convertTemperature: tool({
+        description: 'Convert temperature between Celsius and Fahrenheit',
+        inputSchema: z.object({
+          value: z.number(),
+          from: z.enum(['celsius', 'fahrenheit']),
+          to: z.enum(['celsius', 'fahrenheit']),
+        }),
+        execute: async ({ value, from, to }) => {
+          console.log(`[Tool] Converting ${value}°${from} to ${to}...`);
+
+          if (from === to) return { value, unit: to };
+
+          let result: number;
+          if (from === 'celsius' && to === 'fahrenheit') {
+            result = (value * 9 / 5) + 32;
+          } else {
+            result = (value - 32) * 5 / 9;
+          }
+
+          return { value: Math.round(result * 10) / 10, unit: to };
+        },
+      }),
+    },
+    prompt: 'What is the weather in Tokyo? Please tell me in Celsius.',
+    maxOutputTokens: 200,
+  });
+
+  console.log('\n--- AI Response ---');
+  console.log(result.text);
+
+  console.log('\n--- Tool Calls ---');
+  console.log('Number of tool calls:', result.toolCalls?.length || 0);
+  if (result.toolCalls) {
+    result.toolCalls.forEach((call, i) => {
+      console.log(`\n${i + 1}. ${call.toolName}`);
+      console.log('   Input:', JSON.stringify(call.input));
+      console.log('   Output:', JSON.stringify(call.output));
+    });
+  }
+}
+
+main().catch(console.error);