Update src/components/summaryEngine.py

src/components/summaryEngine.py

@@ -21,15 +21,15 @@ class SummaryEngine:
             logger.info("Summarizing the details extracted from the images")
             allSummaries = "\n".join(texts)
             completion = litellm.completion(
-                model = self.config["SUMMARIZER"]["LLM"],
+                model = self.config.get("SUMMARIZER", "LLM"),
                 api_key = os.environ["GROQ_API_KEY"],
                 api_base = self.config["GROQ CONFIG"]["BASEURL"],
                 messages = [
                     {"role": "system", "content": self.prompts["summaryEnginePrompt"]},
                     {"role": "user", "content": f"AGGEREGATED SUMMARIES: {allSummaries}"}
                 ],
-                max_tokens = self.config["SUMMARIZER"]["MAXTOKENS"],
-                temperature = self.config["SUMMARIZER"]["TEMPERATURE"]
+                max_tokens = self.config.getint("SUMMARIZER", "MAXTOKENS"),
+                temperature = self.config.getfloat("SUMMARIZER", "TEMPERATURE")
             )
             response = completion["choices"][0]["message"]["content"]
             logger.info("Summary generated successfully")