Skip to content

Commit 54be6e7

Browse files
committed
update doctests
Signed-off-by: Keval Mahajan <[email protected]>
1 parent e549836 commit 54be6e7

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

mcpgateway/services/mcp_client_chat_service.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1188,14 +1188,14 @@ class WatsonxProvider:
11881188
config: IBM watsonx.ai configuration object.
11891189
11901190
Examples:
1191-
>>> config = WatsonxConfig(
1191+
>>> config = WatsonxConfig( # doctest: +SKIP
11921192
... api_key="key",
11931193
... url="https://us-south.ml.cloud.ibm.com",
11941194
... project_id="project-id",
11951195
... model_id="ibm/granite-13b-chat-v2"
11961196
... )
1197-
>>> provider = WatsonxProvider(config)
1198-
>>> provider.get_model_name()
1197+
>>> provider = WatsonxProvider(config) # doctest: +SKIP
1198+
>>> provider.get_model_name() # doctest: +SKIP
11991199
'ibm/granite-13b-chat-v2'
12001200
12011201
Note:
@@ -1213,13 +1213,13 @@ def __init__(self, config: WatsonxConfig):
12131213
ImportError: If langchain-ibm is not installed.
12141214
12151215
Examples:
1216-
>>> config = WatsonxConfig(
1216+
>>> config = WatsonxConfig( # doctest: +SKIP
12171217
... api_key="key",
12181218
... url="https://us-south.ml.cloud.ibm.com",
12191219
... project_id="project-id",
12201220
... model_id="ibm/granite-13b-chat-v2"
12211221
... )
1222-
>>> provider = WatsonxProvider(config)
1222+
>>> provider = WatsonxProvider(config) # doctest: +SKIP
12231223
"""
12241224
if not _WATSONX_AVAILABLE:
12251225
raise ImportError("IBM watsonx.ai provider requires langchain-ibm package. Install it with: pip install langchain-ibm")
@@ -1241,13 +1241,13 @@ def get_llm(self) -> WatsonxLLM:
12411241
Exception: If LLM initialization fails (e.g., invalid credentials).
12421242
12431243
Examples:
1244-
>>> config = WatsonxConfig(
1244+
>>> config = WatsonxConfig( # doctest: +SKIP
12451245
... api_key="key",
12461246
... url="https://us-south.ml.cloud.ibm.com",
12471247
... project_id="project-id",
12481248
... model_id="ibm/granite-13b-chat-v2"
12491249
... )
1250-
>>> provider = WatsonxProvider(config)
1250+
>>> provider = WatsonxProvider(config) # doctest: +SKIP
12511251
>>> #llm = provider.get_llm() # Returns WatsonxLLM instance
12521252
"""
12531253
if self.llm is None:
@@ -1287,14 +1287,14 @@ def get_model_name(self) -> str:
12871287
str: The model ID configured for this provider.
12881288
12891289
Examples:
1290-
>>> config = WatsonxConfig(
1290+
>>> config = WatsonxConfig( # doctest: +SKIP
12911291
... api_key="key",
12921292
... url="https://us-south.ml.cloud.ibm.com",
12931293
... project_id="project-id",
12941294
... model_id="ibm/granite-13b-chat-v2"
12951295
... )
1296-
>>> provider = WatsonxProvider(config)
1297-
>>> provider.get_model_name()
1296+
>>> provider = WatsonxProvider(config) # doctest: +SKIP
1297+
>>> provider.get_model_name() # doctest: +SKIP
12981298
'ibm/granite-13b-chat-v2'
12991299
"""
13001300
return self.config.model_id

0 commit comments

Comments
 (0)