Vercel AI SDK + Jarvis SDK

// app/api/chat/route.ts
import { streamText, tool } from "ai";
import { openai } from "@ai-sdk/openai";
import { z } from "zod";

const JARVIS_API_KEY = process.env.JARVIS_API_KEY!;

async function callJarvis(module: string, action: string, params: Record<string, unknown> = {}) {
  const res = await fetch(`https://jarvissdk.com/api/v1/modules/${module}/execute`, {
    method: "POST",
    headers: { "x-api-key": JARVIS_API_KEY, "Content-Type": "application/json" },
    body: JSON.stringify({ action, params }),
  });
  const data = await res.json();
  return data.result ?? data;
}

export async function POST(req: Request) {
  const { messages } = await req.json();

  const result = streamText({
    model: openai("gpt-4o"),
    messages,
    tools: {
      executeModule: tool({
        description: "Execute a Jarvis SDK module (700+ tools available)",
        parameters: z.object({
          module: z.string().describe("Module name (e.g., text-toolkit, hash-toolkit)"),
          action: z.string().describe("Action name (e.g., summarize, sha256)"),
          params: z.record(z.unknown()).optional().describe("Action parameters"),
        }),
        execute: async ({ module, action, params }) => {
          return callJarvis(module, action, params ?? {});
        },
      }),
      searchTools: tool({
        description: "Search the Jarvis SDK marketplace for tools",
        parameters: z.object({
          query: z.string().describe("Search query"),
        }),
        execute: async ({ query }) => {
          const res = await fetch(
            `https://jarvissdk.com/api/v1/catalog/search?q=${encodeURIComponent(query)}`,
            { headers: { "x-api-key": JARVIS_API_KEY } },
          );
          return res.json();
        },
      }),
    },
  });

  return result.toDataStreamResponse();
}