feat: add langchain-groq dependency and update LLMEndpoint to use ChatGroq

core/pyproject.toml

@@ -24,9 +24,9 @@ dependencies = [
     "megaparse-sdk>=0.1.11",
     "langchain-mistralai>=0.2.3",
     "langchain-google-genai>=2.1.3",
-    "langchain-xai>=0.2.3",
     "fasttext-langdetect>=1.0.5",
     "langfuse>=2.57.0",
+    "langchain-groq>=0.3.2",
 ]
 readme = "README.md"
 requires-python = ">= 3.11"

core/quivr_core/llm/llm_endpoint.py

@@ -8,9 +8,9 @@
 from langchain_anthropic import ChatAnthropic
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_groq import ChatGroq
 from langchain_mistralai import ChatMistralAI
 from langchain_openai import AzureChatOpenAI, ChatOpenAI
-from langchain_xai import ChatXAI
 from pydantic import SecretStr
 
 from quivr_core.brain.info import LLMInfo
@@ -214,7 +214,7 @@ def from_config(cls, config: LLMEndpointConfig = LLMEndpointConfig()):
             ChatAnthropic,
             ChatMistralAI,
             ChatGoogleGenerativeAI,
-            ChatXAI,
+            ChatGroq,
         ]
         try:
             if config.supplier == DefaultModelSuppliers.AZURE:
@@ -275,7 +275,7 @@ def from_config(cls, config: LLMEndpointConfig = LLMEndpointConfig()):
                     temperature=config.temperature,
                 )
             elif config.supplier == DefaultModelSuppliers.GROQ:
-                _llm = ChatXAI(
+                _llm = ChatGroq(
                     model=config.model,
                     api_key=SecretStr(config.llm_api_key)
                     if config.llm_api_key