from openai import OpenAI
from mcp_rank import Agent

agent = Agent(
    OpenAI(),
    api_key="sk_mcp_rank_...",
    max_tool_calls=10,  # Max iterations
)

response = agent.run("Send an email to the team about the project update")
print(response)

from openai import OpenAI, AsyncOpenAI
from mcp_rank import Agent

# Sync client
agent = Agent(OpenAI(), api_key="sk_mcp_rank_...")
response = agent.run("What's the weather?", model="gpt-4o")

# Async client
async_agent = Agent(AsyncOpenAI(), api_key="sk_mcp_rank_...")
response = await async_agent.run_async("What's the weather?")

from anthropic import Anthropic, AsyncAnthropic
from mcp_rank import Agent

agent = Agent(Anthropic(), api_key="sk_mcp_rank_...")
response = agent.run(
    "Summarize my recent emails",
    model="claude-3-5-sonnet-20241022"
)

# Behind the scenes:
# 1. Search: "send email" -> finds gmail-mcp-server
# 2. Discovery: gmail-mcp-server has send_email, list_emails tools
# 3. Format: Tools converted to OpenAI function calling format
# 4. Execute: Model calls send_email -> we invoke it on the server
# 5. Loop: Model gets result, generates final response

agent = Agent(
    OpenAI(),
    api_key="sk_mcp_rank_...",
    api_url="https://api.mcprank.com",  # API endpoint
    timeout=30.0,  # Request timeout
    max_tool_calls=10,  # Max tool iterations
)

try:
    response = agent.run("Send an email to bob@example.com")
except Exception as e:
    print(f"Agent failed: {e}")

# Good - specific task
response = agent.run("Send an email to bob@example.com with subject 'Meeting' and body 'See you at 3pm'")

# Less good - vague
response = agent.run("Help me with email")

response = agent.run(
    "Schedule a meeting with the team for next Tuesday at 2pm to discuss the Q1 roadmap",
    model="gpt-4o"  # Use a capable model for complex tasks
)

from mcp_rank import MCPRankClient
from mcp_rank.proxy import MCPServerProxy

async with MCPRankClient(api_key="sk_mcp_rank_...") as client:
    # Search for servers
    results = await client.search("email tools")
    
    # Get tools from a specific server
    proxy = MCPServerProxy()
    server = results["servers"][0]
    tools = await proxy.discover_tools(server["remote_url"])
    
    # Invoke a tool directly
    result, latency, error = await proxy.invoke_tool(
        remote_url=server["remote_url"],
        tool_name="send_email",
        arguments={"to": "bob@example.com", "subject": "Hello"}
    )