github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-18 17:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

xai[patch]: support dedicated structured output feature (#29853)

Browse Source 
https://docs.x.ai/docs/guides/structured-outputs

Interface appears identical to OpenAI's.
```python
from langchain.chat_models import init_chat_model
from pydantic import BaseModel

class Joke(BaseModel):
    setup: str
    punchline: str

llm = init_chat_model("xai:grok-2").with_structured_output(
    Joke, method="json_schema"
)
llm.invoke("Tell me a joke about cats.")
```
This commit is contained in:
ccurme 2025-02-17 14:19:51 -05:00 committed by GitHub
parent 9d6fcd0bfb
commit 83dcef234d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -348,9 +348,14 @@ class ChatXAI(BaseChatOpenAI): # type: ignore[override]
self.client = openai.OpenAI( self.client = openai.OpenAI(
**client_params, **sync_specific **client_params, **sync_specific
).chat.completions ).chat.completions
self.root_client = openai.OpenAI(**client_params, **sync_specific)
if not (self.async_client or None): if not (self.async_client or None):
async_specific: dict = {"http_client": self.http_async_client} async_specific: dict = {"http_client": self.http_async_client}
self.async_client = openai.AsyncOpenAI( self.async_client = openai.AsyncOpenAI(
**client_params, **async_specific **client_params, **async_specific
).chat.completions ).chat.completions
self.root_async_client = openai.AsyncOpenAI(
**client_params,
**async_specific,
)
return self return self