update chat models standard

libs/ibm/tests/integration_tests/test_chat_models_standard.py

@@ -31,11 +31,7 @@ def returns_usage_metadata(self) -> bool:
 
     @property
     def supports_image_inputs(self) -> bool:
-        return False  # Supported, but set False due to token limit in test account
-
-    @property
-    def supports_image_tool_message(self) -> bool:
-        return False  # Supported, but set False due to token limit in test account
+        return True
 
     @property
     def supported_usage_metadata_details(
@@ -66,6 +62,12 @@ def chat_model_params(self) -> dict:
             "project_id": WX_PROJECT_ID,
         }
 
+    @pytest.mark.xfail(reason="Supported for vision model.")
+    def test_image_inputs(self, model: BaseChatModel) -> None:
+        model.watsonx_model._inference.model_id = MODEL_ID_IMAGE  # type: ignore[attr-defined]
+        super().test_image_inputs(model)
+        model.watsonx_model._inference.model_id = MODEL_ID  # type: ignore[attr-defined]
+
     @pytest.mark.xfail(reason="Not implemented tool_choice as `any`.")
     def test_structured_few_shot_examples(self, model: BaseChatModel) -> None:
         super().test_structured_few_shot_examples(model)