Skip to content

Latest commit

 

History

History
103 lines (73 loc) · 2.91 KB

File metadata and controls

103 lines (73 loc) · 2.91 KB

CrewAI Agent with A2A Protocol

This sample demonstrates a simple image generation agent built with CrewAI and exposed through the A2A protocol.

How It Works

This agent utilizes CrewAI and the Google Gemini API to generate images based on text prompts. The A2A protocol enables standardized interaction with the agent, allowing clients to send requests and receive images as artifacts.

sequenceDiagram
    participant Client as A2A Client
    participant Server as A2A Server
    participant Agent as CrewAI Agent
    participant API as Gemini API

    Client->>Server: Send task with text prompt
    Server->>Agent: Forward prompt to image agent
    Note over Server,Agent: Optional: Simulated streaming updates
    Agent->>API: Generate image using Gemini
    API->>Agent: Return generated image
    Agent->>Server: Store image and return ID
    Server->>Client: Respond with image artifact
Loading

Key Components

  • CrewAI Agent: Image generation agent with specialized tools
  • A2A Server: Provides standardized protocol for interacting with the agent
  • Image Generation: Uses Gemini API to create images from text descriptions
  • Cache System: Stores generated images for retrieval (in-memory or file-based)

Prerequisites

  • Python 3.12 or higher
  • UV package manager (recommended)
  • Google API Key (for Gemini access)

Setup & Running

  1. Navigate to the samples directory:

    cd samples/python/agents/crewai
  2. Create an environment file with your API key (or Vertex AI credentials):

    echo "GOOGLE_API_KEY=your_api_key_here" > .env
  3. Set up the Python environment:

    uv python pin 3.12
    uv venv
    source .venv/bin/activate
  4. Run the agent with desired options:

    # Basic run
    uv run .
    
    # On custom host/port
    uv run . --host 0.0.0.0 --port 8080
  5. Run the A2A client:

    In a separate terminal:

    # Connect to the agent (specify the agent URL with correct port)
    cd samples/python/hosts/cli   
    uv run . --agent http://localhost:10001
    
    # If you changed the port when starting the agent, use that port instead
    # uv run . --agent http://localhost:YOUR_PORT

    Or run the demo app

Features & Improvements

Features:

  • Text-to-image generation using Google Gemini
  • Support for modifying existing images using references
  • Robust error handling with automatic retries
  • Optional file-based cache persistence
  • Improved artifact ID extraction from queries

Limitations:

  • No true streaming (CrewAI doesn't natively support it)
  • Limited agent interactions (no multi-turn conversations)

Learn More