ibm: Add support for Chat Models (#22979)

langchain-ai · Jun 29, 2024 · a78ccb9 · a78ccb9
1 parent 16c5911
commit a78ccb9
Show file tree

Hide file tree

Showing 9 changed files with 1,965 additions and 239 deletions.
diff --git a/docs/docs/integrations/chat/ibm_watsonx.ipynb b/docs/docs/integrations/chat/ibm_watsonx.ipynb
diff --git a/docs/docs/integrations/providers/ibm.mdx b/docs/docs/integrations/providers/ibm.mdx
@@ -28,6 +28,16 @@ import os
 os.environ["WATSONX_APIKEY"] = "your IBM watsonx.ai api key"
 ```
 
+## Chat Model
+
+### ChatWatsonx
+
+See a [usage example](/docs/integrations/chat/ibm_watsonx).
+
+```python
+from langchain_ibm import ChatWatsonx
+```
+
 ## LLMs
 
 ### WatsonxLLM

diff --git a/libs/partners/ibm/langchain_ibm/__init__.py b/libs/partners/ibm/langchain_ibm/__init__.py
@@ -1,4 +1,5 @@
+from langchain_ibm.chat_models import ChatWatsonx
 from langchain_ibm.embeddings import WatsonxEmbeddings
 from langchain_ibm.llms import WatsonxLLM
 
-__all__ = ["WatsonxLLM", "WatsonxEmbeddings"]
+__all__ = ["WatsonxLLM", "WatsonxEmbeddings", "ChatWatsonx"]