From c79db3e17323633da81330073e68d59401ac4ac5 Mon Sep 17 00:00:00 2001 From: kdziedzic68 Date: Mon, 28 Oct 2024 15:25:21 +0100 Subject: [PATCH] fix after rebase issues --- .../ingestion/providers/unstructured/images.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/unstructured/images.py b/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/unstructured/images.py index 0e20beffe..b8edc9e8a 100644 --- a/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/unstructured/images.py +++ b/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/unstructured/images.py @@ -96,7 +96,12 @@ async def _to_image_element( img_bytes = crop_and_convert_to_bytes(image, top_x, top_y, bottom_x, bottom_y) prompt = _ImagePrompt(_ImagePromptInput(images=[img_bytes])) - image_description = await self.image_describer.get_image_description(prompt=prompt) + if self._llm: + image_description = await self._llm.generate(prompt=prompt) + elif self.image_describer: + image_description = await self.image_describer.get_image_description(prompt=prompt) + else: + image_description = "" return ImageElement( description=image_description, ocr_extracted_text=element.text,