import requests

def get_weather(city):
    # Call a weather API (example using wttr.in)
    url = f"https://wttr.in/{city}?format=%C+%t"
    response = requests.get(url)
    
    if response.status_code == 200:
        return response.text
    else:
        return "Couldn't fetch weather"

# Wire into your agent
@agent.tool
def weather(ctx, city):
    """Get current weather for a city"""
    return get_weather(city)

# 1. Install the Google client library
# pip install google-api-python-client google-auth

from googleapiclient.discovery import build
from google.oauth2.credentials import Credentials

# 2. Create a client with your OAuth token
def create_gmail_client(access_token):
    creds = Credentials(token=access_token)
    service = build('gmail', 'v1', credentials=creds)
    return service

# 3. Use it in your agent
@agent.tool
def read_emails(ctx, max_results=5):
    """Read recent unread emails"""
    service = create_gmail_client(ctx.user.google_token)
    
    results = service.users().messages().list(
        userId='me',
        q='is:unread',
        maxResults=max_results
    ).execute()
    
    messages = results.get('messages', [])
    return f"You have {len(messages)} unread emails"

# 1. Install Slack SDK
# pip install slack-sdk

import os
from slack_sdk import WebClient

# 2. Get bot token from Slack app settings
slack_token = os.getenv('SLACK_BOT_TOKEN')
client = WebClient(token=slack_token)

# 3. Use in agent
@agent.tool
def post_to_slack(ctx, channel, message):
    """Post a message to a Slack channel"""
    try:
        response = client.chat_postMessage(
            channel=channel,
            text=message
        )
        return "Message sent!"
    except Exception as e:
        return f"Error: {str(e)}"

# 1. Install Notion client
# pip install notion-client

from notion_client import Client

# 2. Initialize with integration tokennotion = Client(auth=os.getenv('NOTION_TOKEN'))

# 3. Use in agent
@agent.tool
def add_to_notion(ctx, database_id, title, content):
    """Add a page to a Notion database"""
    
    notion.pages.create(
        parent={"database_id": database_id},
        properties={
            "Name": {
                "title": [{"text": {"content": title}}]
            }
        },
        children=[{
            "object": "block",
            "type": "paragraph",
            "paragraph": {
                "rich_text": [{"text": {"content": content}}]
            }
        }]
    )
    
    return "Added to Notion!"

# Create a .env file (don't commit this!)
SLACK_BOT_TOKEN=xoxb-your-token-here
NOTION_TOKEN=secret_your_token_here
OPENAI_API_KEY=sk-your-key-here

# In your Python code
import os
from dotenv import load_dotenv

load_dotenv()  # Load variables from .env file

slack_token = os.getenv('SLACK_BOT_TOKEN')
notion_token = os.getenv('NOTION_TOKEN')

@agent.tool
def safe_api_call(ctx, service):
    """Example of robust error handling"""
    
    max_retries = 3
    
    for attempt in range(max_retries):
        try:
            result = call_external_api(service)
            if result:
                return result
                
        except requests.Timeout:
            if attempt == max_retries - 1:
                return "Service timed out. Please try again."
            time.sleep(2 ** attempt)  # Wait longer each retry
            
        except Exception as e:
            return f"Error: {str(e)}"
    
    return "Failed after retries"