from xpander_sdk import XpanderClient

# Initialize with API key
client = XpanderClient(api_key="your-xpander-api-key")

# List all agents
agents = client.list_agents()

# Filter agents by type
regular_agents = client.list_agents(type="regular")

# Get agent by ID
agent = client.get_agent("agent-id")

# Create a new agent
new_agent = client.create_agent(
    name="My New Agent",
    type="regular",
    description="This agent helps with customer service tasks"
)

# Update an existing agent
updated_agent = client.update_agent(
    "agent-id",
    name="Updated Agent Name",
    description="New description"
)

# Delete an agent
client.delete_agent("agent-id")

# List all operations for an agent
operations = client.list_operations(agent_id="agent-id")

# Get operation by ID
operation = client.get_operation("operation-id")

# Create a new operation
new_operation = client.create_operation(
    agent_id="agent-id",
    name="My Operation",
    type="function",
    prompt="You are a helpful assistant that...",
    input_schema={
        "type": "object",
        "properties": {
            "query": {"type": "string"}
        },
        "required": ["query"]
    },
    output_schema={
        "type": "object",
        "properties": {
            "answer": {"type": "string"}
        },
        "required": ["answer"]
    }
)

# Update an existing operation
updated_operation = client.update_operation(
    "operation-id",
    name="Updated Operation",
    prompt="New prompt content"
)

# Delete an operation
client.delete_operation("operation-id")

# Execute an operation
result = client.execute_operation(
    "operation-id",
    input={
        "query": "What is the capital of France?"
    }
)

# Stream operation results
for chunk in client.stream_operation("operation-id", input={"query": "Tell me a story"}):
    print(chunk.content, end="", flush=True)

try:
    result = client.execute_operation("operation-id", input={"query": "Hello"})
except xpander_sdk.APIError as e:
    if e.status_code == 401:
        print("Authentication error. Check your API key.")
    elif e.status_code == 404:
        print("Operation not found.")
    else:
        print(f"API error: {e}")