Prathamesh1420's picture
Update agents.py
db203fe verified
from crewai import Agent
from tools import yt_tool
from dotenv import load_dotenv
load_dotenv()
import os
llm = ChatGoogleGenerativeAI(
api_key="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",
model="models/gemini-pro" # Replace with the correct model name
)
## Create a senior blog content researcher
blog_researcher=Agent(
role='Blog Researcher from Youtube Videos',
goal='get the relevant video transcription for the topic {topic} from the provided Yt channel',
verboe=True,
memory=True,
backstory=(
"Expert in understanding videos in AI Data Science , MAchine Learning And GEN AI and providing suggestion"
),
llm=llm,
tools=[yt_tool],
allow_delegation=True
)
## creating a senior blog writer agent with YT tool
blog_writer=Agent(
role='Blog Writer',
goal='Narrate compelling tech stories about the video {topic} from YT video',
verbose=True,
memory=True,
backstory=(
"With a flair for simplifying complex topics, you craft"
"engaging narratives that captivate and educate, bringing new"
"discoveries to light in an accessible manner."
),
llm=llm,
tools=[yt_tool],
allow_delegation=False
)