import asyncio
from amtp import AMTP, Message

# Create AMTP connection
amtp = AMTP("my-agent@company.com", "https://amtp.company.com")

# Handle messages
@amtp.on_message
async def handle_message(message):
    print(f"Received: {message.subject}")
    return {"status": "processed", "response": "Hello back!"}

# Send a message
message = Message(
    sender="my-agent@company.com",
    recipients=["user@partner.com"],
    subject="Hello from AI Agent",
    payload={"greeting": "Hello, World!"}
)

async def main():
    await amtp.send(message)
    await amtp.run()  # Run forever

asyncio.run(main())

import asyncio
from amtp import AMTP, Message

async def send_message():
    amtp = AMTP("sender@company.com", "https://amtp.company.com")
    await amtp.start()
    
    message = Message(
        sender="sender@company.com",
        recipients=["recipient@partner.com"],
        subject="Hello from AMTP",
        payload={"greeting": "Hello, World!"}
    )
    
    message_id = await amtp.send(message)
    print(f"Message sent: {message_id}")
    
    await amtp.stop()

asyncio.run(send_message())

from amtp import AMTP, Message, Schema

# Create your own schema
order_schema = Schema(
    id="myapp:commerce.order.v1",
    name="Order Schema",
    version="v1",
    schema_def={
        "type": "object",
        "properties": {
            "order_id": {"type": "string"},
            "customer": {"type": "object"},
            "items": {"type": "array"},
            "total": {"type": "number"}
        },
        "required": ["order_id", "customer", "items", "total"]
    }
)
order_schema.register()

# Use your schema in messages
message = Message(
    sender="sender@company.com",
    recipients=["recipient@partner.com"],
    subject="Order Processing",
    schema="myapp:commerce.order.v1",
    payload={
        "order_id": "ORD-12345",
        "customer": {"name": "John Doe", "email": "john@example.com"},
        "items": [{"sku": "WIDGET-001", "quantity": 2, "price": 29.99}],
        "total": 59.98
    }
)

# Create custom schema
custom_schema = Schema(
    id="myapp:notification.alert.v1",
    name="Alert Notification",
    version="v1",
    schema_def={
        "type": "object",
        "properties": {
            "alert_type": {"type": "string"},
            "severity": {"type": "string"},
            "message": {"type": "string"}
        },
        "required": ["alert_type", "severity", "message"]
    }
)
custom_schema.register()

from amtp import AMTP

# Basic configuration
amtp = AMTP(
    address="my-agent@company.com",
    gateway_url="https://amtp.company.com",
    delivery_mode="pull",  # or "push"
    tls_enabled=True,
    api_key="your-api-key"
)

amtp = AMTP(
    address="my-agent@company.com",
    gateway_url="https://amtp.company.com",
    delivery_mode="pull",
    tls_enabled=True,
    connect_timeout=30,
    read_timeout=60,
    max_retries=3,
    retry_delay=1.0,
    poll_interval=5.0,
    log_level="INFO"
)

from amtp import AMTP, Message

amtp = AMTP("my-agent@company.com", "https://amtp.company.com")

@amtp.on_message
async def handle_message(message: Message):
    if message.schema == "agntcy:commerce.order.v1":
        # Handle order
        return {"status": "order_processed", "order_id": message.payload["order_id"]}
    
    elif message.coordination_type == "parallel":
        # Handle workflow
        return {"status": "workflow_step_completed"}
    
    else:
        # Default handling
        return {"status": "received", "message": "Message processed"}

@amtp.on_error
async def handle_error(error):
    print(f"Error occurred: {error}")

from amtp import AMTP, Message, Error

async def robust_messaging():
    amtp = AMTP("my-agent@company.com", "https://amtp.company.com")
    
    try:
        await amtp.start()
        
        message = Message(
            sender="my-agent@company.com",
            recipients=["recipient@partner.com"],
            subject="Test Message",
            payload={"data": "test"}
        )
        
        message_id = await amtp.send(message)
        print(f"Message sent: {message_id}")
        
    except Error as e:
        print(f"AMTP Error: {e}")
        if e.details:
            print(f"Details: {e.details}")
    
    finally:
        await amtp.stop()

from amtp import Schema

# Create custom schema
notification_schema = Schema(
    id="myapp:notification.system.v1",
    name="System Notification",
    version="v1",
    description="System-wide notification schema",
    schema_def={
        "type": "object",
        "properties": {
            "event_type": {"type": "string"},
            "timestamp": {"type": "string"},
            "data": {"type": "object"}
        },
        "required": ["event_type", "timestamp"]
    }
)

# Register schema
notification_schema.register()

# Validate data
try:
    Schema.validate_payload("myapp:notification.system.v1", {
        "event_type": "user_login",
        "timestamp": "2025-01-01T12:00:00Z",
        "data": {"user_id": "123"}
    })
    print("Data is valid!")
except Error as e:
    print(f"Validation failed: {e}")