From 109a70fc64bdecac986f7d2725fbe99f33dc0db1 Mon Sep 17 00:00:00 2001 From: Leonid Ganeline Date: Tue, 18 Jun 2024 20:00:27 -0700 Subject: [PATCH] ibm: docstrings (#23149) Added missed docstrings. Format docstrings to the consistent format (used in the API Reference) --- libs/partners/ibm/langchain_ibm/embeddings.py | 2 ++ libs/partners/ibm/langchain_ibm/llms.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/partners/ibm/langchain_ibm/embeddings.py b/libs/partners/ibm/langchain_ibm/embeddings.py index 4abe9f542c..63549f3e24 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 962ff0cb11..039aa02680 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