apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-09 07:42:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add try/except fallback in _extract_image for vision LLM failures

Browse source
This commit is contained in:
CREDO23 2026-04-09 15:11:24 +02:00
parent caaec2e0a7
commit d6c4fb8938
1 changed files with 16 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

25
surfsense_backend/app/etl_pipeline/etl_pipeline_service.py
View file

@ -57,16 +57,23 @@ class EtlPipelineService:
async def _extract_image(self, request: EtlRequest) -> EtlResult: async def _extract_image(self, request: EtlRequest) -> EtlResult:
if self._vision_llm: if self._vision_llm:
from app.etl_pipeline.parsers.vision_llm import parse_with_vision_llm try:
from app.etl_pipeline.parsers.vision_llm import parse_with_vision_llm
content = await parse_with_vision_llm( content = await parse_with_vision_llm(
request.file_path, request.filename, self._vision_llm request.file_path, request.filename, self._vision_llm
) )
return EtlResult( return EtlResult(
markdown_content=content, markdown_content=content,
etl_service="VISION_LLM", etl_service="VISION_LLM",
content_type="image", content_type="image",
) )
except Exception:
logging.warning(
"Vision LLM failed for %s, falling back to document parser",
request.filename,
exc_info=True,
)
logging.info( logging.info(
"No vision LLM provided, falling back to document parser for %s", "No vision LLM provided, falling back to document parser for %s",