The Agentic Scientist

from crewai import Agent, Task, Crew
from crewai_tools import BaseTool

class MoleculePriceTool(BaseTool):
    name: str = "Molecule Price Tool"
    description: str = "Gets the estimated price of a chemical molecule."
    def _run(self, molecule_name: str) -> str:
        if "benzene" in molecule_name.lower(): return "$50 per liter"
        return "Price not found"

researcher = Agent(
  role='Chemical Researcher',
  goal='Find the market price for specified chemicals',
  backstory='Expert in chemical procurement.',
  tools=[MoleculePriceTool()]
)
writer = Agent(
  role='Report Writer',
  goal='Write a concise report on the findings',
  backstory='Skilled in scientific documentation.'
)
find_price_task = Task(description='What is the price of Benzene?', expected_output='The market price.', agent=researcher)
write_report_task = Task(description='Summarize the findings.', expected_output='A one-paragraph summary.', agent=writer)

crew = Crew(agents=[researcher, writer], tasks=[find_price_task, write_report_task])
# result = crew.kickoff()
# print(result)
print("CrewAI workflow defined.")
                        

# This LangGraph example shows a similar multi-step process
# where state is passed explicitly between nodes.
from langgraph.graph import StateGraph, END
from typing import TypedDict

class ResearchState(TypedDict):
    chemical: str
    price: str
    report: str

def find_price(state):
    print("---FINDING PRICE---")
    # Simulates tool call
    if "benzene" in state["chemical"].lower():
        state["price"] = "$50 per liter"
    else:
        state["price"] = "Price not found"
    return state

def write_report(state):
    print("---WRITING REPORT---")
    state["report"] = f'The price for {state["chemical"]} is {state["price"]}.'
    return state

workflow = StateGraph(ResearchState)
workflow.add_node("find_price", find_price)
workflow.add_node("write_report", write_report)
workflow.set_entry_point("find_price")
workflow.add_edge("find_price", "write_report")
workflow.add_edge("write_report", END)
app = workflow.compile()

# To run: app.stream({"chemical": "Benzene"})
print("LangGraph workflow compiled.")