Maverick

analyticsHub/components/codeGeneratorAgent.py

@@ -3,7 +3,7 @@ from langchain_core.prompts import PromptTemplate
 from langchain_core.runnables import RunnablePassthrough
 from ..utils.functions import readYaml, getConfig
 from ..utils.exceptions import CustomException
-from langchain_cerebras import ChatCerebras
+from langchain_groq import ChatGroq
 from ..utils.logger import logger
 import os
 
@@ -18,7 +18,7 @@ class CodeGenerator:
             logger.info("Constructing code generation chain.")
             promptTemplate = readYaml(self.yamlPath)["codeGeneratorAgentPrompt"]
             codeGeneratorPrompt = PromptTemplate.from_template(promptTemplate)
-            llm = ChatCerebras(
+            llm = ChatGroq(
                 model=self.config.get("CODEGENERATOR", "model"),
                 temperature=self.config.getfloat("CODEGENERATOR", "temperature")
             )

analyticsHub/components/failsafeAgent.py

@@ -3,7 +3,7 @@ from langchain_core.prompts import PromptTemplate
 from langchain_core.runnables import RunnablePassthrough
 from ..utils.functions import readYaml, getConfig
 from ..utils.exceptions import CustomException
-from langchain_cerebras import ChatCerebras
+from langchain_groq import ChatGroq
 from ..utils.logger import logger
 import os
 
@@ -18,7 +18,7 @@ class FailsafeCodeGenerator:
             logger.info("Constructing failsafe code generation chain.")
             promptTemplate = readYaml(self.yamlPath)["codeDebuggerAgentPrompt"]
             codeGeneratorPrompt = PromptTemplate.from_template(promptTemplate)
-            llm = ChatCerebras(
+            llm = ChatGroq(
                 model=self.config.get("FAILSAFECODEGENERATOR", "model"),
                 temperature=self.config.getfloat("FAILSAFECODEGENERATOR", "temperature")
             )

analyticsHub/components/metadataGenerator.py

@@ -3,7 +3,7 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.runnables import RunnableLambda
 from ..utils.functions import readYaml, getConfig
 from ..utils.exceptions import CustomException
-from langchain_cerebras import ChatCerebras
+from langchain_groq import ChatGroq
 from ..utils.logger import logger
 import os
 
@@ -18,7 +18,7 @@ class MetadataGenerator:
             logger.info("Constructing metadata generation chain.")
             promptTemplate = readYaml(self.yamlPath)["metadataGeneratorPrompt"]
             prompt = ChatPromptTemplate.from_template(promptTemplate)
-            llm = ChatCerebras(
+            llm = ChatGroq(
                 model=self.config.get("METADATAGENERATOR", "model"),
                 temperature=self.config.getfloat("METADATAGENERATOR", "temperature")
             )

analyticsHub/components/queryRephraserAgent.py

@@ -3,7 +3,7 @@ from langchain_core.prompts import PromptTemplate
 from ..utils.functions import readYaml, getConfig
 from ..utils.exceptions import CustomException
 from pydantic import Field, BaseModel
-from langchain_cerebras import ChatCerebras
+from langchain_groq import ChatGroq
 from ..utils.logger import logger
 import os
 
@@ -30,7 +30,7 @@ class QueryRephaser:
                 input_variables = ["metadata", "query"],
                 partial_variables = {"format_instructions": queryRephraseParser.get_format_instructions()}
             )
-            llm = ChatCerebras(
+            llm = ChatGroq(
                 model=self.config.get("QUERYREPHRASER", "model"),
                 temperature=self.config.getfloat("QUERYREPHRASER", "temperature"),
                 max_tokens=self.config.getint("QUERYREPHRASER", "maxTokens")

config.ini

@@ -1,18 +1,18 @@
 [QUERYREPHRASER]
-model = llama-3.3-70b
+model = meta-llama/llama-4-maverick-17b-128e-instruct
 temperature = 1
 maxTokens = 512
 
 [METADATAGENERATOR]
-model = llama-3.3-70b
+model = meta-llama/llama-4-maverick-17b-128e-instruct
 temperature = 1
 
 [CODEGENERATOR]
-model = llama-3.3-70b
+model = meta-llama/llama-4-maverick-17b-128e-instruct
 temperature = 1
 
 [FAILSAFECODEGENERATOR]
-model = llama-3.3-70b
+model = meta-llama/llama-4-maverick-17b-128e-instruct
 temperature = 1
 
 [SPEECHTOTEXT]