diff --git a/libs/partners/ibm/langchain_ibm/embeddings.py b/libs/partners/ibm/langchain_ibm/embeddings.py index 4abe9f542c4..63549f3e242 100644 --- a/libs/partners/ibm/langchain_ibm/embeddings.py +++ b/libs/partners/ibm/langchain_ibm/embeddings.py @@ -15,6 +15,8 @@ from langchain_core.utils import convert_to_secret_str, get_from_dict_or_env class WatsonxEmbeddings(BaseModel, LangChainEmbeddings): + """IBM WatsonX.ai embedding models.""" + model_id: str = "" """Type of model to use.""" diff --git a/libs/partners/ibm/langchain_ibm/llms.py b/libs/partners/ibm/langchain_ibm/llms.py index 962ff0cb113..039aa026805 100644 --- a/libs/partners/ibm/langchain_ibm/llms.py +++ b/libs/partners/ibm/langchain_ibm/llms.py @@ -15,7 +15,7 @@ logger = logging.getLogger(__name__) class WatsonxLLM(BaseLLM): """ - IBM watsonx.ai large language models. + IBM WatsonX.ai large language models. To use, you should have ``langchain_ibm`` python package installed, and the environment variable ``WATSONX_APIKEY`` set with your API key, or pass