How to add tools to CrewAI agent

pip install crewai openai

import os
from crewai import CrewAgent
from openai import OpenAI

# Initialize OpenAI client
client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])

# Define a simple tool function
# This tool returns the current weather for a given city (mocked for demo)
def get_weather(city: str) -> str:
    # In real use, call a weather API here
    return f"The weather in {city} is sunny with 75°F."

# Initialize CrewAI agent
agent = CrewAgent(openai_client=client)

# Register the tool with the agent
agent.add_tool(name="get_weather", func=get_weather, description="Get current weather for a city")

# Use the tool in an agent prompt
response = agent.run(
    prompt="What is the weather in New York?",
    tools=["get_weather"]
)

print(response)

import asyncio

# Async tool example
async def get_joke() -> str:
    # Simulate async API call
    await asyncio.sleep(0.1)
    return "Why did the AI cross the road? To optimize the other side!"

# Register async tool
agent.add_tool(name="get_joke", func=get_joke, description="Tell a joke asynchronously")

# Run async tool via agent
async def main():
    response = await agent.run_async(
        prompt="Tell me a joke.",
        tools=["get_joke"]
    )
    print(response)

asyncio.run(main())