Merge pull request #1017 from winsonluk/x

Add xAI
assafelovic · Dec 10, 2024 · 27f4703 · 27f4703
2 parents bbbc900 + a36eed6
commit 27f4703
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/backend/server/server.py b/backend/server/server.py
@@ -39,6 +39,7 @@ class ConfigRequest(BaseModel):
     SERPAPI_API_KEY: str = ''
     SERPER_API_KEY: str = ''
     SEARX_URL: str = ''
+    XAI_API_KEY: str
 
 
 # App initialization

diff --git a/docs/docs/gpt-researcher/llms/llms.md b/docs/docs/gpt-researcher/llms/llms.md
@@ -240,6 +240,15 @@ STRATEGIC_LLM="bedrock:anthropic.claude-3-sonnet-20240229-v1:0"
 ```
 
 
+## xAI
+
+```bash
+FAST_LLM="xai:grok-beta"
+SMART_LLM="xai:grok-beta"
+STRATEGIC_LLM="xai:grok-beta"
+```
+
+
 ## Other Embedding Models
 
 ### Nomic

diff --git a/gpt_researcher/llm_provider/generic/base.py b/gpt_researcher/llm_provider/generic/base.py
@@ -17,6 +17,7 @@
     "huggingface",
     "groq",
     "bedrock",
+    "xai",
 }
 
 
@@ -102,6 +103,11 @@ def from_provider(cls, provider: str, **kwargs: Any):
                 model_id = kwargs.pop("model", None) or kwargs.pop("model_name", None)
                 kwargs = {"model_id": model_id, **kwargs}
             llm = ChatBedrock(**kwargs)
+        elif provider == "xai":
+            _check_pkg("langchain_xai")
+            from langchain_xai import ChatXAI
+
+            llm = ChatXAI(**kwargs)
         else:
             supported = ", ".join(_SUPPORTED_PROVIDERS)
             raise ValueError(