fix after rebase issues

deepsense-ai · Oct 28, 2024 · c79db3e · c79db3e
1 parent b98d5f1
commit c79db3e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...ts-document-search/src/ragbits/document_search/ingestion/providers/unstructured/images.py b/...ts-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,