github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-07 13:40:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

update xai

Browse Source
This commit is contained in:
Chester Curme 2025-02-21 21:13:30 -05:00
parent 230f2e030a
commit be587d1640
1 changed files with 3 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
libs/partners/xai/langchain_xai/chat_models.py
View File

@ -7,7 +7,6 @@ from typing import (
Optional, Optional,
) )
import openai
from langchain_core.language_models.chat_models import LangSmithParams from langchain_core.language_models.chat_models import LangSmithParams
from langchain_core.utils import secret_from_env from langchain_core.utils import secret_from_env
from langchain_openai.chat_models.base import BaseChatOpenAI from langchain_openai.chat_models.base import BaseChatOpenAI
@ -325,7 +324,7 @@ class ChatXAI(BaseChatOpenAI): # type: ignore[override]
if self.n is not None and self.n > 1 and self.streaming: if self.n is not None and self.n > 1 and self.streaming:
raise ValueError("n must be 1 when streaming.") raise ValueError("n must be 1 when streaming.")
client_params: dict = { self._client_params: dict = {
"api_key": ( "api_key": (
self.xai_api_key.get_secret_value() if self.xai_api_key else None self.xai_api_key.get_secret_value() if self.xai_api_key else None
), ),
@ -335,27 +334,12 @@ class ChatXAI(BaseChatOpenAI): # type: ignore[override]
"default_query": self.default_query, "default_query": self.default_query,
} }
if self.max_retries is not None: if self.max_retries is not None:
client_params["max_retries"] = self.max_retries self._client_params["max_retries"] = self.max_retries
if client_params["api_key"] is None: if self._client_params["api_key"] is None:
raise ValueError( raise ValueError(
"xAI API key is not set. Please set it in the `xai_api_key` field or " "xAI API key is not set. Please set it in the `xai_api_key` field or "
"in the `XAI_API_KEY` environment variable." "in the `XAI_API_KEY` environment variable."
) )
if not (self.client or None):
sync_specific: dict = {"http_client": self.http_client}
self.client = openai.OpenAI(
**client_params, **sync_specific
).chat.completions
self.root_client = openai.OpenAI(**client_params, **sync_specific)
if not (self.async_client or None):
async_specific: dict = {"http_client": self.http_async_client}
self.async_client = openai.AsyncOpenAI(
**client_params, **async_specific
).chat.completions
self.root_async_client = openai.AsyncOpenAI(
**client_params,
**async_specific,
)
return self return self