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

// Create tools for your agents with Nia API
export const browseEssays = tool({
  description: "Get all Paul Graham essays",
  inputSchema: z.object({}),
  execute: async () => {
    const res = await fetch(
      `${NIA_API}/data-sources/${SOURCE_ID}/tree`,
      { headers: { Authorization: `Bearer ${API_KEY}` }}
    );
    const data = await res.json();
    return { tree: data.tree_string, count: data.page_count };
  },
});