feat: return a list when the input is a list

modeling_jina_embeddings_v4.py

@@ -435,6 +435,8 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
             prefix=encode_kwargs.pop("prefix"),
         )
 
+        return_list = isinstance(texts, list)
+        
         if isinstance(texts, str):
             texts = [texts]
 
@@ -449,7 +451,7 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
             **encode_kwargs,
         )
 
-        return embeddings if len(texts) > 1 else embeddings[0]
+        return embeddings if return_list else embeddings[0] 
 
     def _load_images_if_needed(
         self, images: List[Union[str, Image.Image]]
@@ -497,6 +499,8 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
         encode_kwargs = self._validate_encoding_params(truncate_dim=truncate_dim)
         task = self._validate_task(task)
         
+        return_list = isinstance(images, list)
+
         # Convert single image to list
         if isinstance(images, (str, Image.Image)):
             images = [images]
@@ -516,7 +520,7 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
         if max_pixels:
             self.processor.image_processor.max_pixels = default_max_pixels
 
-        return embeddings if len(images) > 1 else embeddings[0]
+        return embeddings if return_list else embeddings[0]
 
     @classmethod
     def from_pretrained(