Update agents.py

agents.py

@@ -6,16 +6,21 @@ from tools.image_generation_tools import ImageGenerationTools
 
 load_dotenv()
 secret_token = os.getenv("content")
-groq_api_key = os.getenv("groq_api_key")
+
 
 repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
 mistral = HuggingFaceEndpoint(repo_id=repo_id, huggingfacehub_api_token=secret_token, max_new_tokens=28000, temperature=0.3, repetition_penalty=1.1)
-# Assuming there's a Groq client you need to import and set up
-from groq import GroqClient
+
+from groq import Groq
+
+client = Groq(
+    api_key=os.environ.get("groq_api_key"),
+)
+
 
 class ContentAgents:
     def __init__(self):
-        self.llm = GroqClient(api_key=groq_api_key, model_name="Olamma")
+        self.llm = client(api_key=groq_api_key, model_name="llama3-8b-8192")
         self.image_generator = ImageGenerationTools()
 
     def generate_image_for_section(self, section_title, topic):