Changed to Groq

correctiveRag.py

@@ -29,21 +29,21 @@ PINECONE_API_KEY = {os.getenv("PINECONE_API_KEY")[:5]}
 """)
 
 # Prepare the LLM
-# from langchain_groq import ChatGroq
-# assert os.getenv('GROQ_MODEL') is not None, "GROQ_MODEL not set"
-# assert os.getenv('GROQ_API_KEY') is not None, "GROQ_API_KEY not set"
-# llm = ChatGroq(model_name=os.getenv('GROQ_MODEL'), temperature=0, verbose=True)
+from langchain_groq import ChatGroq
+assert os.getenv('GROQ_MODEL') is not None, "GROQ_MODEL not set"
+assert os.getenv('GROQ_API_KEY') is not None, "GROQ_API_KEY not set"
+llm = ChatGroq(model_name=os.getenv('GROQ_MODEL'), temperature=0, verbose=True)
 
 # For using Grok
-from langchain_openai import ChatOpenAI
-assert os.getenv('XAI_API_KEY') is not None, "XAI_API_KEY not set"
-assert os.getenv('XAI_MODEL') is not None, "XAI_MODEL not set"
-assert os.getenv('XAI_BASE_URL') is not None, "XAI_BASE_URL not set"
-llm = ChatOpenAI(
-    api_key=os.getenv("XAI_API_KEY"), 
-    base_url=os.getenv("XAI_BASE_URL"), 
-    model=os.getenv("XAI_MODEL"), 
-    temperature=0.1)
+# from langchain_openai import ChatOpenAI
+# assert os.getenv('XAI_API_KEY') is not None, "XAI_API_KEY not set"
+# assert os.getenv('XAI_MODEL') is not None, "XAI_MODEL not set"
+# assert os.getenv('XAI_BASE_URL') is not None, "XAI_BASE_URL not set"
+# llm = ChatOpenAI(
+#     api_key=os.getenv("XAI_API_KEY"), 
+#     base_url=os.getenv("XAI_BASE_URL"), 
+#     model=os.getenv("XAI_MODEL"), 
+#     temperature=0.1)
 
 
 # from langchain_openai import ChatOpenAI