tests(langchain-ibm): Fixed integration tests (#151)

Author: MateuszOssGit

libs/ibm/tests/integration_tests/test_embeddings.py

@@ -21,7 +21,7 @@
 WX_PROJECT_ID = os.environ.get("WATSONX_PROJECT_ID", "")
 
 URL = "https://us-south.ml.cloud.ibm.com"
-MODEL_ID = "ibm/granite-embedding-107m-multilingual"
+MODEL_ID = "ibm/granite-embedding-278m-multilingual"
 
 DOCUMENTS = ["What is a generative ai?", "What is a loan and how does it works?"]
 

libs/ibm/tests/integration_tests/test_embeddings_standard.py

@@ -10,7 +10,7 @@
 
 URL = "https://us-south.ml.cloud.ibm.com"
 
-MODEL_ID = "ibm/granite-embedding-107m-multilingual"
+MODEL_ID = "ibm/granite-embedding-278m-multilingual"
 
 
 class TestWatsonxEmbeddingsStandard(EmbeddingsIntegrationTests):

libs/ibm/tests/integration_tests/test_rerank.py

@@ -16,7 +16,7 @@
 
 URL = "https://us-south.ml.cloud.ibm.com"
 
-MODEL_ID = "ibm/slate-125m-english-rtrvr"
+MODEL_ID = "ibm/slate-125m-english-rtrvr-v2"
 
 CREATE_WATSONX_RERANK_INIT_PARAMETERS = [
     pytest.param(
@@ -82,7 +82,7 @@ def test_02_rerank_documents() -> None:
 
 
 def test_02_rerank_documents_with_params() -> None:
-    params = RerankParameters(truncate_input_tokens=1)
+    params = RerankParameters(truncate_input_tokens=2)
     test_documents = [
         Document(page_content="This is a test document."),
     ]