UPDATE

src/components/ragAgent.py

@@ -45,7 +45,7 @@ class RAGAgent:
             retriever = vectorStore.as_retriever(search_kwargs = {"k": 5})
             llm = ChatCerebras(
                 model = config.get("RAGAGENT", "modelName"),
-                temperature = config.getint("RAGAGENT", "temperature"),
+                temperature = config.getfloat("RAGAGENT", "temperature"),
                 max_tokens = config.getint("RAGAGENT", "maxTokens")
             )
             chain = {"query": RunnablePassthrough(), "context": RunnablePassthrough() | retriever} | promptTemplate | llm | StrOutputParser()

src/components/reasoningAgent.py

@@ -17,7 +17,7 @@ class ReasoningAgent:
             promptTemplate = ChatPromptTemplate.from_template(prompts.get("reasoningTemplate"))
             llm = ChatCerebras(
                 model = config.get("REASONINGAGENT", "modelName"),
-                temperature = config.getint("REASONINGAGENT", "temperature")
+                temperature = config.getfloat("REASONINGAGENT", "temperature")
             )
             chain = {"query": RunnablePassthrough()} | promptTemplate | llm | StrOutputParser()
             self.chain = chain

src/components/sqlAgent.py

@@ -22,7 +22,7 @@ class PostgreSQLAgent:
             db = SQLDatabase(self.engine)
             llm = ChatCerebras(
                 model = self.config.get("SQLAGENT", "modelName"),
-                temperature = self.config.getint("SQLAGENT", "temperature"),
+                temperature = self.config.getfloat("SQLAGENT", "temperature"),
                 max_tokens = self.config.getint("SQLAGENT", "maxTokens")
             )
             self.toolkit = SQLDatabaseToolkit(db = db, llm = llm)

src/components/synthesizerAgent.py

@@ -17,7 +17,7 @@ class SynthesizerAgent:
             promptTemplate = ChatPromptTemplate.from_template(prompts.get("synthesizerTemplate"))
             llm = ChatCerebras(
                 model = config.get("SYNTHESIZERAGENT", "modelName"),
-                temperature = config.getint("SYNTHESIZERAGENT", "temperature"),
+                temperature = config.getfloat("SYNTHESIZERAGENT", "temperature"),
                 max_tokens = config.getint("SYNTHESIZERAGENT", "maxTokens")
             )
             chain = RunnablePassthrough() | promptTemplate | llm | StrOutputParser()