mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
|
"""Test WatsonxLLM API wrapper."""
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
from langchain_ibm import WatsonxEmbeddings
|
||
|
|
|
||
|
|
os.environ.pop("WATSONX_APIKEY", None)
|
||
|
|
os.environ.pop("WATSONX_PROJECT_ID", None)
|
||
|
|
|
||
|
|
MODEL_ID = "ibm/slate-125m-english-rtrvr"
|
||
|
|
|
||
|
|
|
||
|
|
def test_initialize_watsonx_embeddings_bad_path_without_url() -> None:
|
||
|
|
try:
|
||
|
|
WatsonxEmbeddings(
|
||
|
|
model_id=MODEL_ID,
|
||
|
|
)
|
||
|
|
except ValueError as e:
|
||
|
|
assert "WATSONX_URL" in e.__str__()
|
||
|
|
|
||
|
|
|
||
|
|
def test_initialize_watsonx_embeddings_cloud_bad_path() -> None:
|
||
|
|
try:
|
||
|
|
WatsonxEmbeddings(model_id=MODEL_ID, url="https://us-south.ml.cloud.ibm.com")
|
||
|
|
except ValueError as e:
|
||
|
|
assert "WATSONX_APIKEY" in e.__str__()
|
||
|
|
|
||
|
|
|
||
|
|
def test_initialize_watsonx_embeddings_cpd_bad_path_without_all() -> None:
|
||
|
|
try:
|
||
|
|
WatsonxEmbeddings(
|
||
|
|
model_id=MODEL_ID,
|
||
|
|
url="https://cpd-zen.apps.cpd48.cp.fyre.ibm.com",
|
||
|
|
)
|
||
|
|
except ValueError as e:
|
||
|
|
assert (
|
||
|
|
"WATSONX_APIKEY" in e.__str__()
|
||
|
|
and "WATSONX_PASSWORD" in e.__str__()
|
||
|
|
and "WATSONX_TOKEN" in e.__str__()
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_initialize_watsonx_embeddings_cpd_bad_path_password_without_username() -> None:
|
||
|
|
try:
|
||
|
|
WatsonxEmbeddings(
|
||
|
|
model_id=MODEL_ID,
|
||
|
|
url="https://cpd-zen.apps.cpd48.cp.fyre.ibm.com",
|
||
|
|
password="test_password",
|
||
|
|
)
|
||
|
|
except ValueError as e:
|
||
|
|
assert "WATSONX_USERNAME" in e.__str__()
|
||
|
|
|
||
|
|
|
||
|
|
def test_initialize_watsonx_embeddings_cpd_bad_path_apikey_without_username() -> None:
|
||
|
|
try:
|
||
|
|
WatsonxEmbeddings(
|
||
|
|
model_id=MODEL_ID,
|
||
|
|
url="https://cpd-zen.apps.cpd48.cp.fyre.ibm.com",
|
||
|
|
apikey="test_apikey",
|
||
|
|
)
|
||
|
|
except ValueError as e:
|
||
|
|
assert "WATSONX_USERNAME" in e.__str__()
|