From 2f413e937b6fac5a1b196de08c248febce14b0e4 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Wed, 7 Jan 2026 11:07:22 +0200 Subject: [PATCH] Add VLM model to download_models() --- haiku_rag_slim/haiku/rag/client.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/haiku_rag_slim/haiku/rag/client.py b/haiku_rag_slim/haiku/rag/client.py index f7a7b32f..80fb1741 100644 --- a/haiku_rag_slim/haiku/rag/client.py +++ b/haiku_rag_slim/haiku/rag/client.py @@ -1672,6 +1672,9 @@ class HaikuRAG: and self._config.reranking.model.provider == "ollama" ): required_models.add(self._config.reranking.model.name) + pic_desc = self._config.processing.conversion_options.picture_description + if pic_desc.enabled and pic_desc.model.provider == "ollama": + required_models.add(pic_desc.model.name) if not required_models: return