File size: 999 Bytes
3670b47
945b678
1d029cc
5d22cf7
4d658da
3670b47
5d83ebb
2bc4a9f
 
2f90406
4d658da
2bc4a9f
48a17ba
2f90406
3670b47
 
4d658da
3670b47
5d83ebb
2f90406
 
106a077
4d658da
48a17ba
2f90406
3670b47
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from crewai import Agent
from langchain_openai import ChatOpenAI
from tools import get_scrape_tool, get_search_tool

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 on this topic.",
        llm=ChatOpenAI(model=model, temperature=0.7),
        tools = [get_search_tool(), get_scrape_tool()],
        allow_delegation=False,
    	verbose=True
    )

def get_author_agent(model):
    return Agent(
        role="Author",
        goal="Write a LinkedIn post on topic: {topic}",
        backstory="You're working on a writing a LinkedIn post on topic: {topic}. "
                  "You base your writing on the work of the Researcher, who provides context on this topic.",
        llm=ChatOpenAI(model=model, temperature=0.7),
        allow_delegation=False,
        verbose=True
    )