Update agents.py

agents.py

@@ -14,6 +14,12 @@ from smolagents import (
 )
 from tools import VisitWebpageTool
 
+###
+from crewai import Agent
+from langchain_openai import ChatOpenAI
+from tools import scrape_tool, search_tool, today_tool
+###
+
 MODEL_ID_1 = "gpt-4o-mini"
 MODEL_ID_2 = "gpt-4o"
 MODEL_ID_3 = "o4-mini"
@@ -96,4 +102,30 @@ def get_final_answer(question, answer):
     print(f"Answer: {answer}")
     print(f"Final answer: {final_answer}")
     
-    return final_answer
+    return final_answer
+
+###
+def get_researcher_agent(model, verbose):
+    return Agent(
+        role="Researcher",
+        goal="Research content on topic: {topic}.",
+        backstory="You're working on researching content on topic: {topic}. "
+                  "Your work is the basis for the Writer to write on this topic.",
+        llm=ChatOpenAI(model=model),
+        tools = [search_tool(), scrape_tool()],
+        allow_delegation=False,
+    	verbose=verbose
+    )
+
+def get_writer_agent(model, verbose):
+    return Agent(
+        role="Writer",
+        goal="Write an article on topic: {topic}.",
+        backstory="You're working on writing an article on topic: {topic}. "
+                  "You base your writing on the work of the Researcher, who provides context on this topic.",
+        llm=ChatOpenAI(model=model),
+        tools = [today_tool()],
+        allow_delegation=False,
+        verbose=verbose
+    )
+###