Update agents.py

agents.py

@@ -2,25 +2,25 @@ from crewai import Agent
 
 from tools import get_scrape_tool, get_search_tool
 
-def get_researcher_agent():
+def get_researcher_agent(model):
     return Agent(
         role="Researcher",
         goal="Research content on {topic}",
         backstory="You're working on researching content on topic: {topic}. "
                   "Your work is the basis for the Author to write a LinkedIn social media post on this topic.",
-        llm=ChatOpenAI(model="gpt-4o", temperature=0.7),
+        llm=ChatOpenAI(model=model, temperature=0.7),
         tools = [get_search_tool(), get_scrape_tool()],
         allow_delegation=False,
     	verbose=True
     )
 
-def get_author_agent():
+def get_author_agent(model):
     return Agent(
         role="Author",
         goal="Write a LinkedIn social media post on topic: {topic}",
         backstory="You're working on a writing a LinkedIn social media post on topic: {topic}. "
                   "You base your writing on the work of the Researcher, who provides context on this topic.",
-        llm=ChatOpenAI(model="gpt-4o", temperature=0.7),
+        llm=ChatOpenAI(model=model, temperature=0.7),
         allow_delegation=False,
         verbose=True
     )