fix: fix vllm test

Author: vishnurk6247

flo_ai/tests/unit-tests/test_openai_vllm.py

@@ -379,8 +379,9 @@ def test_openai_vllm_format_image_in_message(self):
         result = llm.format_image_in_message(image)
 
         assert result is not None
-        assert result['type'] == 'input_image'
-        assert result['image']['url'] == 'https://example.com/image.jpg'
+        assert len(result) == 1
+        assert result[0]['type'] == 'image_url'
+        assert result[0]['image_url']['url'] == 'https://example.com/image.jpg'
 
     @pytest.mark.asyncio
     @patch('flo_ai.llm.openai_llm.AsyncOpenAI')