example_query_2agents.py

import os
from crewai import Agent, Task, Crew
from crewai_tools import BaseTool
from dotenv import load_dotenv
import wolframalpha

# Load environment variables from .env file
load_dotenv()

# Retrieve the API keys from the environment
OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
OPENAI_MODEL_NAME = os.getenv('OPENAI_MODEL_NAME')
WOLFRAM_APP_ID = os.getenv('APP_ID')

if not OPENAI_API_KEY or not WOLFRAM_APP_ID:
    raise ValueError("API keys not found in environment variables.")

# The question to query Wolfram Alpha
question = ("""
Mathematical calculations:
What is the square root of 12345678?
Unit conversions and comparisons:
Convert 100 miles per hour to meters per second
Compare the area of Texas to the area of France
Scientific data and constants:
What is the atomic mass of carbon?
Show me the electron configuration of sodium
Date and time calculations:
How many days between July 4, 1776 and today?
What day of the week was March 14, 1879?
Statistical data:
What is the population density of Tokyo?
Show me the GDP per capita of Canada over the last 10 years
Solving equations:
Solve x^2 + 3x - 4 = 0
Find the derivative of sin(x^2)
Astronomy and space:
What phase is the moon in today?
Distance between Earth and Mars
Nutrition information:
How many calories are in a medium apple?
Compare nutritional value of spinach and kale
Fun or unusual queries:
Convert 100 to Roman numerals
""")


# The Wolfram Alpha tool function
def wolfram_alpha_tool(query: str) -> str:
    try:
        # Initialize Wolfram Alpha client
        client = wolframalpha.Client(WOLFRAM_APP_ID)

        # Query Wolfram Alpha
        res = client.query(query)

        # Extract and return the result
        if not res['@success']:
            return "Query failed. Please check your query syntax."

        result = next(res.results).text if res.results else "No results found."
        return result

    except Exception as e:
        return f"An error occurred: {str(e)}"


# A custom tool using the wolfram_alpha_tool function
class WolframAlphaTool(BaseTool):
    name: str = "Wolfram Alpha Tool"
    description: str = "Queries Wolfram Alpha for information."

    def _run(self, query: str) -> str:
        result = wolfram_alpha_tool(query)
        if "error" in result.lower():
            raise ValueError(f"Wolfram Alpha Tool Error: {result}")
        return result


# Initialize the custom tool
wolfram_tool = WolframAlphaTool()

# The Researcher Agent
researcher = Agent(
    role='Researcher',
    goal=f'Fetch and analyze data from Wolfram Alpha about topic {question}',
    backstory='An expert in science and computational knowledge.',
    tools=[wolfram_tool],
    verbose=True
)

# The Writer Agent
writer = Agent(
    role='Writer',
    goal=f'Create a report based on the analysis about topic {question}.',
    backstory='A skilled writer with a knack for simplifying complex data.',
    verbose=True
)

# The Research Task
research_task = Task(
    description=f'Query Wolfram Alpha for information on the specified topic: {question}.',
    expected_output=f'Analysis report about topic {question}',
    agent=researcher,
    arguments={'query': question}
)

# The Writing Task
write_task = Task(
    description=f'Write a report based on the analysis from the Researcher about the topic: {question}.',
    expected_output='Formatted report',
    agent=writer
)

# The Crew
crew = Crew(
    agents=[researcher, writer],
    tasks=[research_task, write_task],
    verbose=True
)

# Kickoff the process and get the result
result = crew.kickoff()
print(result)