dromerosm commited on
Commit
f53cb24
1 Parent(s): 4f9016f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -52,14 +52,15 @@ def kickoff_crew(topic: str) -> dict:
52
  raise ValueError("API Key for Groq is not set in environment variables")
53
 
54
  # Initialize the Groq large language model
55
- groq_llm = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="llama3-70b-8192")
 
56
 
57
  # Define Agents with Groq LLM
58
  researcher = Agent(
59
  role='Researcher',
60
  goal='Collect detailed information on {topic}',
61
  tools=[search, search_results, web_scrapper],
62
- llm=groq_llm, # Assigning the Groq LLM here
63
  backstory=(
64
  "As a diligent researcher, you explore the depths of the internet to "
65
  "unearth crucial information and insights on the assigned topics. "
@@ -76,7 +77,7 @@ def kickoff_crew(topic: str) -> dict:
76
  editor = Agent(
77
  role='Editor',
78
  goal='Compile and refine the information into a comprehensive report on {topic}',
79
- llm=groq_llm, # Assigning the Groq LLM here
80
  backstory=(
81
  "With a keen eye for detail and a strong command of language, you transform "
82
  "raw data into polished, insightful reports that are both informative and engaging. "
 
52
  raise ValueError("API Key for Groq is not set in environment variables")
53
 
54
  # Initialize the Groq large language model
55
+ groq_llm_70b = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="llama3-70b-8192")
56
+ groq_llm_8b = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="llama3-8b-8192")
57
 
58
  # Define Agents with Groq LLM
59
  researcher = Agent(
60
  role='Researcher',
61
  goal='Collect detailed information on {topic}',
62
  tools=[search, search_results, web_scrapper],
63
+ llm=groq_llm_8b, # Assigning the Groq LLM here
64
  backstory=(
65
  "As a diligent researcher, you explore the depths of the internet to "
66
  "unearth crucial information and insights on the assigned topics. "
 
77
  editor = Agent(
78
  role='Editor',
79
  goal='Compile and refine the information into a comprehensive report on {topic}',
80
+ llm=groq_llm_70b, # Assigning the Groq LLM here
81
  backstory=(
82
  "With a keen eye for detail and a strong command of language, you transform "
83
  "raw data into polished, insightful reports that are both informative and engaging. "