Update agents.py

agents.py

@@ -11,16 +11,12 @@ secret_token = os.getenv("content")
 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)
 
-from groq import Groq
-
-client = Groq(
-    api_key=os.environ.get("groq_api_key"),
-)
+from langchain_groq import ChatGroq
 
 
 class ContentAgents:
     def __init__(self):
-        self.llm = client(api_key=os.environ.get("groq_api_key"), model_name="llama3-8b-8192")
+        self.llm = ChatGroq(temperature=0.6, groq_api_key=os.environ.get("groq_api_key"), model_name="llama3-8b-8192")
         self.image_generator = ImageGenerationTools()
 
     def generate_image_for_section(self, section_title, topic):