add client CLI example

Author: mme

typescript-sdk/apps/cli-client/src/index.ts

@@ -0,0 +1,2 @@
+dist/
+mastra.db*

typescript-sdk/apps/client-cli/.gitignore

@@ -1,5 +1,5 @@
 {
-  "name": "cli-client",
+  "name": "client-cli",
   "version": "0.1.0",
   "private": true,
   "scripts": {
@@ -11,7 +11,15 @@
   "dependencies": {
     "@ag-ui/client": "workspace:*",
     "@ag-ui/core": "workspace:*",
-    "@ag-ui/mastra": "workspace:*"
+    "@ag-ui/mastra": "workspace:*",
+    "@ai-sdk/openai": "^1.3.22",
+    "@mastra/client-js": "^0.10.9",
+    "@mastra/core": "^0.10.10",
+    "@mastra/libsql": "^0.11.0",
+    "@mastra/loggers": "^0.10.3",
+    "@mastra/memory": "^0.11.1",
+    "open": "^10.1.2",
+    "zod": "^3.22.4"
   },
   "devDependencies": {
     "@types/node": "^20",

typescript-sdk/apps/cli-client/README.md renamed to typescript-sdk/apps/client-cli/README.md

@@ -0,0 +1,39 @@
+import { openai } from "@ai-sdk/openai";
+import { Agent } from "@mastra/core/agent";
+import { MastraAgent } from "@ag-ui/mastra";
+import { Memory } from "@mastra/memory";
+import { LibSQLStore } from "@mastra/libsql";
+import { weatherTool } from "./tools/weather.tool";
+import { browserTool } from "./tools/browser.tool";
+import { z } from "zod";
+
+export const agent = new MastraAgent({
+  // @ts-ignore
+  agent: new Agent({
+    name: "AG-UI Agent",
+    instructions: `
+        You are a helpful assistant that runs a CLI application.
+  
+        When helping users get weather details for specific locations, respond:
+        - Always ask for a location if none is provided.
+        - If the location name isn’t in English, please translate it
+        - If giving a location with multiple parts (e.g. "New York, NY"), use the most relevant part (e.g. "New York")
+        - Include relevant details like humidity, wind conditions, and precipitation
+        - Keep responses concise but informative
+  
+        Use the weatherTool to fetch current weather data.
+
+        When helping users browse the web, always use a full URL, for example: "https://www.google.com"
+        Use the browserTool to browse the web.
+
+  `,
+    model: openai("gpt-4o-mini"),
+    tools: { weatherTool, browserTool },
+    memory: new Memory({
+      storage: new LibSQLStore({
+        url: "file:./mastra.db",
+      }),
+    }),
+  }),
+  threadId: "1",
+});

typescript-sdk/apps/cli-client/package.json renamed to typescript-sdk/apps/client-cli/package.json

@@ -0,0 +1,81 @@
+import * as readline from "readline";
+import { agent } from "./agent";
+import { randomUUID } from "node:crypto";
+
+const rl = readline.createInterface({
+  input: process.stdin,
+  output: process.stdout,
+});
+
+async function chatLoop() {
+  console.log("🤖 AG-UI chat started! Type your messages and press Enter. Press Ctrl+D to quit.\n");
+
+  return new Promise<void>((resolve) => {
+    const promptUser = () => {
+      rl.question("> ", async (input) => {
+        if (input.trim() === "") {
+          promptUser();
+          return;
+        }
+        console.log("");
+
+        rl.pause();
+
+        agent.messages.push({
+          id: randomUUID(),
+          role: "user",
+          content: input.trim(),
+        });
+
+        try {
+          await agent.runAgent(
+            {},
+            {
+              onTextMessageStartEvent() {
+                process.stdout.write("🤖 AG-UI assistant: ");
+              },
+              onTextMessageContentEvent({ event }) {
+                process.stdout.write(event.delta);
+              },
+              onTextMessageEndEvent() {
+                console.log("\n");
+              },
+              onToolCallStartEvent({ event }) {
+                console.log("🔧 Tool call:", event.toolCallName);
+              },
+              onToolCallArgsEvent({ event }) {
+                process.stdout.write(event.delta);
+              },
+              onToolCallEndEvent() {
+                console.log("\n");
+              },
+              onToolCallResultEvent({ event }) {
+                if (event.content) {
+                  console.log("🔍 Tool call result:", event.content);
+                }
+              },
+            },
+          );
+        } catch (error) {
+          console.error("❌ Error running agent:", error);
+        }
+
+        rl.resume();
+        promptUser();
+      });
+    };
+
+    rl.on("close", () => {
+      console.log("\n👋 Goodbye!");
+      resolve();
+    });
+
+    promptUser();
+  });
+}
+
+async function main() {
+  await chatLoop();
+}
+
+main().catch(console.error);

typescript-sdk/apps/client-cli/src/agent.ts

@@ -0,0 +1,16 @@
+import { createTool } from "@mastra/core/tools";
+import { z } from "zod";
+import open from "open";
+
+export const browserTool = createTool({
+  id: "browser",
+  description: "Browse the web",
+  inputSchema: z.object({
+    url: z.string().describe("URL to browse"),
+  }),
+  outputSchema: z.string(),
+  execute: async ({ context }) => {
+    open(context.url);
+    return `Browsed ${context.url}`;
+  },
+});

typescript-sdk/apps/client-cli/src/index.ts

@@ -0,0 +1,102 @@
+import { createTool } from "@mastra/core/tools";
+import { z } from "zod";
+
+interface GeocodingResponse {
+  results: {
+    latitude: number;
+    longitude: number;
+    name: string;
+  }[];
+}
+interface WeatherResponse {
+  current: {
+    time: string;
+    temperature_2m: number;
+    apparent_temperature: number;
+    relative_humidity_2m: number;
+    wind_speed_10m: number;
+    wind_gusts_10m: number;
+    weather_code: number;
+  };
+}
+
+export const weatherTool = createTool({
+  id: "get-weather",
+  description: "Get current weather for a location",
+  inputSchema: z.object({
+    location: z.string().describe("City name"),
+  }),
+  outputSchema: z.object({
+    temperature: z.number(),
+    feelsLike: z.number(),
+    humidity: z.number(),
+    windSpeed: z.number(),
+    windGust: z.number(),
+    conditions: z.string(),
+    location: z.string(),
+  }),
+  execute: async ({ context }) => {
+    return await getWeather(context.location);
+  },
+});
+
+const getWeather = async (location: string) => {
+  const geocodingUrl = `https://geocoding-api.open-meteo.com/v1/search?name=${encodeURIComponent(location)}&count=1`;
+  const geocodingResponse = await fetch(geocodingUrl);
+  const geocodingData = (await geocodingResponse.json()) as GeocodingResponse;
+
+  if (!geocodingData.results?.[0]) {
+    throw new Error(`Location '${location}' not found`);
+  }
+
+  const { latitude, longitude, name } = geocodingData.results[0];
+
+  const weatherUrl = `https://api.open-meteo.com/v1/forecast?latitude=${latitude}&longitude=${longitude}&current=temperature_2m,apparent_temperature,relative_humidity_2m,wind_speed_10m,wind_gusts_10m,weather_code`;
+
+  const response = await fetch(weatherUrl);
+  const data = (await response.json()) as WeatherResponse;
+
+  return {
+    temperature: data.current.temperature_2m,
+    feelsLike: data.current.apparent_temperature,
+    humidity: data.current.relative_humidity_2m,
+    windSpeed: data.current.wind_speed_10m,
+    windGust: data.current.wind_gusts_10m,
+    conditions: getWeatherCondition(data.current.weather_code),
+    location: name,
+  };
+};
+
+function getWeatherCondition(code: number): string {
+  const conditions: Record<number, string> = {
+    0: "Clear sky",
+    1: "Mainly clear",
+    2: "Partly cloudy",
+    3: "Overcast",
+    45: "Foggy",
+    48: "Depositing rime fog",
+    51: "Light drizzle",
+    53: "Moderate drizzle",
+    55: "Dense drizzle",
+    56: "Light freezing drizzle",
+    57: "Dense freezing drizzle",
+    61: "Slight rain",
+    63: "Moderate rain",
+    65: "Heavy rain",
+    66: "Light freezing rain",
+    67: "Heavy freezing rain",
+    71: "Slight snow fall",
+    73: "Moderate snow fall",
+    75: "Heavy snow fall",
+    77: "Snow grains",
+    80: "Slight rain showers",
+    81: "Moderate rain showers",
+    82: "Violent rain showers",
+    85: "Slight snow showers",
+    86: "Heavy snow showers",
+    95: "Thunderstorm",
+    96: "Thunderstorm with slight hail",
+    99: "Thunderstorm with heavy hail",
+  };
+  return conditions[code] || "Unknown";
+}