From 8360aa54492bc5b5e07fcd07b0b85284910f1a14 Mon Sep 17 00:00:00 2001 From: peets <100425207+ElHachem02@users.noreply.github.com> Date: Tue, 4 Nov 2025 08:39:15 +0100 Subject: [PATCH] fix: extract response from api_image_request in picture description (#2571) Signed-off-by: Peter El Hachem Co-authored-by: Peter El Hachem --- docling/models/picture_description_api_model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docling/models/picture_description_api_model.py b/docling/models/picture_description_api_model.py index a3c0c2ee..771608e2 100644 --- a/docling/models/picture_description_api_model.py +++ b/docling/models/picture_description_api_model.py @@ -51,7 +51,7 @@ class PictureDescriptionApiModel(PictureDescriptionBaseModel): # Note: technically we could make a batch request here, # but not all APIs will allow for it. For example, vllm won't allow more than 1. def _api_request(image): - return api_image_request( + response, _ = api_image_request( image=image, prompt=self.options.prompt, url=self.options.url, @@ -60,5 +60,7 @@ class PictureDescriptionApiModel(PictureDescriptionBaseModel): **self.options.params, ) + return response + with ThreadPoolExecutor(max_workers=self.concurrency) as executor: yield from executor.map(_api_request, images)