Fix Issues

instructor/client_vertexai.py

@@ -39,18 +39,22 @@
 ) -> gm.Content:
     if isinstance(message["content"], str):
         return gm.Content(
-            role=message["role"], parts=[gm.Part.from_text(message["content"])]
+            role=message["role"],  # type:ignore
+            parts=[gm.Part.from_text(message["content"])],
         )
     elif isinstance(message["content"], list):
-        parts = []
+        parts: list[gm.Part] = []
         for item in message["content"]:
             if isinstance(item, str):
                 parts.append(gm.Part.from_text(item))
             elif isinstance(item, gm.Part):
                 parts.append(item)
             else:
                 raise ValueError(f"Unsupported content type in list: {type(item)}")
-        return gm.Content(role=message["role"], parts=parts)
+        return gm.Content(
+            role=message["role"],  # type:ignore
+            parts=parts,
+        )
     else:
         raise ValueError("Unsupported message content type")
 
@@ -82,7 +86,7 @@
 
 def vertexai_process_response(_kwargs: dict[str, Any], model: BaseModel):
     messages: list[dict[str, str]] = _kwargs.pop("messages")
     contents = _vertexai_message_list_parser(messages)
 
     tool = _create_vertexai_tool(model=model)
 
@@ -96,7 +100,7 @@
 
 def vertexai_process_json_response(_kwargs: dict[str, Any], model: BaseModel):
     messages: list[dict[str, str]] = _kwargs.pop("messages")
     contents = _vertexai_message_list_parser(messages)
 
     config: dict[str, Any] | None = _kwargs.pop("generation_config", None)
 

tests/llm/test_vertexai/test_message_parser.py

@@ -2,11 +2,8 @@
 import vertexai.generative_models as gm
 from instructor.client_vertexai import vertexai_message_parser
 
-from .util import models, modes
 
-
-@pytest.mark.parametrize("model, mode", [(models[0], modes[0])])
-def test_vertexai_message_parser_string_content(model, mode):
+def test_vertexai_message_parser_string_content():
     message = {"role": "user", "content": "Hello, world!"}
     result = vertexai_message_parser(message)
 
@@ -17,8 +14,7 @@ def test_vertexai_message_parser_string_content(model, mode):
     assert result.parts[0].text == "Hello, world!"
 
 
-@pytest.mark.parametrize("model, mode", [(models[0], modes[0])])
-def test_vertexai_message_parser_list_content(model, mode):
+def test_vertexai_message_parser_list_content():
     message = {
         "role": "user",
         "content": [
@@ -40,16 +36,14 @@ def test_vertexai_message_parser_list_content(model, mode):
     assert result.parts[2].text == " How are you?"
 
 
-@pytest.mark.parametrize("model, mode", [(models[0], modes[0])])
-def test_vertexai_message_parser_invalid_content(model, mode):
+def test_vertexai_message_parser_invalid_content():
     message = {"role": "user", "content": 123}  # Invalid content type
 
     with pytest.raises(ValueError, match="Unsupported message content type"):
         vertexai_message_parser(message)
 
 
-@pytest.mark.parametrize("model, mode", [(models[0], modes[0])])
-def test_vertexai_message_parser_invalid_list_item(model, mode):
+def test_vertexai_message_parser_invalid_list_item():
     message = {"role": "user", "content": ["Hello", 123, gm.Part.from_text("world!")]}
 
     with pytest.raises(ValueError, match="Unsupported content type in list"):