Sae

A2A-Native Legal Agent for Contract Review

An autonomous legal services agent built on Google's A2A protocol, enabling AI agents to request contract clause review and risk analysis.

Features

• A2A Protocol Native: Full implementation of Google's Agent-to-Agent protocol
• Contract Clause Review: Extract and analyze contract clauses
• Risk Analysis: Identify legal risks with confidence scores
• Recommendations: Actionable suggestions for contract improvements
• Streaming Support: Real-time updates via Server-Sent Events

Quick Start

Prerequisites

• Python 3.12+
• UV package manager
• OpenAI API key
• Pinecone API key (optional, for RAG)

Installation

# Clone the repository
git clone https://github.com/your-org/sae-law-agent.git
cd sae-law-agent

# Install UV if not already installed
curl -LsSf https://astral.sh/uv/install.sh | sh

# Install dependencies
uv sync

# Copy environment template
cp .env.example .env

# Edit .env with your API keys

Running

# Start the server
uv run uvicorn sae.main:app --reload

# Server runs at http://localhost:8000

API Endpoints

Usage

A2A Protocol

Sae implements the A2A protocol for agent-to-agent communication.

1. Discover Agent Capabilities

curl http://localhost:8000/.well-known/agent.json

2. Submit a Contract for Review

curl -X POST http://localhost:8000/a2a \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "method": "tasks/send",
    "id": "req-001",
    "params": {
      "id": "task-001",
      "message": {
        "role": "user",
        "parts": [{
          "type": "text",
          "text": "Please review this NDA:\n\nCONFIDENTIALITY AGREEMENT\n\n1. Definition of Confidential Information..."
        }]
      }
    }
  }'

3. Check Task Status

curl -X POST http://localhost:8000/a2a \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "method": "tasks/get",
    "id": "req-002",
    "params": {
      "id": "task-001"
    }
  }'

4. Stream Updates (SSE)

curl -N http://localhost:8000/a2a/stream/task-001

Project Structure

sae-law-agent/
├── src/sae/
│   ├── main.py              # FastAPI application
│   ├── config.py            # Configuration management
│   ├── api/
│   │   ├── agent_card.py    # A2A Agent Card endpoint
│   │   ├── jsonrpc.py       # JSON-RPC handler
│   │   └── streaming.py     # SSE streaming
│   ├── agents/
│   │   ├── contract_review.py  # LangGraph workflow
│   │   ├── state.py         # Agent state definitions
│   │   └── nodes/           # Processing nodes
│   ├── models/
│   │   ├── a2a.py           # A2A protocol models
│   │   └── clauses.py       # Contract analysis models
│   └── services/
│       └── task_manager.py  # Task lifecycle management
├── tests/
├── Dockerfile
└── pyproject.toml

Docker

# Build
docker build -t sae-legal-agent .

# Run
docker run -p 8000:8000 \
  -e OPENAI_API_KEY=sk-... \
  -e PINECONE_API_KEY=... \
  sae-legal-agent

Environment Variables

License

Apache 2.0

Status

MVP Phase: Core contract review functionality implemented.

• A2A Protocol (Agent Card, JSON-RPC, SSE)
• LangGraph contract review workflow
• Clause extraction
• Risk analysis
• Recommendations
• RAG integration (Pinecone)
• Payment integration (x402)
• Production deployment