#!/bin/bash
# Проверка доступных vision моделей в Ollama

set -e

CONTAINER_OLLAMA="ollama"

echo "=== Проверка vision моделей в Ollama ==="
echo ""

# Определяем команду docker
DOCKER_CMD="docker"
if ! docker ps >/dev/null 2>&1; then
    DOCKER_CMD="sudo docker"
fi

echo "1. Список установленных моделей:"
$DOCKER_CMD exec $CONTAINER_OLLAMA ollama list 2>/dev/null || sudo $DOCKER_CMD exec $CONTAINER_OLLAMA ollama list 2>/dev/null

echo ""
echo "2. Популярные vision модели для Ollama:"
echo ""
echo "   Рекомендуемые модели с поддержкой vision:"
echo "   - llava:latest (LLaVA 1.6) - 7B параметров, хорошо работает с изображениями"
echo "   - bakllava:latest (BakLLaVA) - 7B параметров, Mistral + LLaVA"
echo "   - llama3.2-vision:latest (Llama 3.2 Vision) - 11B параметров"
echo "   - gemma2:9b-it (Gemma 2) - может поддерживать vision"
echo ""
echo "3. Для установки vision модели выполните:"
echo ""
echo "   sudo docker exec ollama ollama pull llava:latest"
echo "   # или"
echo "   sudo docker exec ollama ollama pull bakllava:latest"
echo ""
echo "4. Проверка текущей модели gemma3n:e4b-it-fp16:"
echo ""
echo "   Модель gemma3n:e4b-it-fp16 может не поддерживать vision правильно."
echo "   Рекомендуется использовать специализированные vision модели:"
echo "   - llava:latest (лучший выбор для vision)"
echo "   - bakllava:latest (альтернатива)"
echo ""
echo "=== Готово ==="