Build an agent with Cloudflare Agents

import { tool } from "agents";
import { z } from "zod";
import { ATXPAccount, buildStreamableTransport, createMCPClient } from "@atxp/client";

export const generateImage = tool({
  description: "Generate an image from a text prompt using ATXP",
  parameters: z.object({
    prompt: z.string().describe("The text prompt to generate an image from"),
  }),
  execute: async ({ prompt }, { env, agent }) => {
    try {
      // Initialize ATXP account
      const account = new ATXPAccount(env.ATXP_CONNECTION_STRING);

      // Create transport for ATXP image server
      const transport = buildStreamableTransport({
        mcpServer: 'https://image.mcp.atxp.ai',
        account,
      });

      // Create MCP client
      const client = await createMCPClient({ transport });

      // Generate the image
      const result = await client.callTool('generate_image', {
        prompt: prompt,
        size: '1024x1024'
      });

      if (result.toolResult.content[0].type === 'text') {
        const response = JSON.parse(result.toolResult.content[0].text);

        if (response.image_url) {
          return `Image generated successfully! Here's your image: ![${prompt}](${response.image_url})`;
        } else {
          return `Image generation started. Task ID: ${response.task_id}`;
        }
      }

      return "Image generation request submitted successfully.";
    } catch (error) {
      console.error('Image generation error:', error);
      return `Failed to generate image: ${error.message}`;
    }
  },
});

import { generateImage } from "./tools/imageGeneration";

export const tools = {
  generateImage,
  // ... other tools
};

export const executions = {
  // Add any tools that require confirmation here
};

import { createAgent } from "agents";
import { tools, executions } from "./tools";

export default {
  async fetch(request: Request, env: Env): Promise<Response> {
    const agent = createAgent({
      model: openai("gpt-4o-mini", {
        apiKey: env.OPENAI_API_KEY,
      }),
      tools,
      executions,
      // ... other configuration
    });

    return agent.fetch(request, { env });
  },
};