Newer
Older
navi-1 / tests / unit / llm / test_ollama.py
@Eugene Sukhodolskiy Eugene Sukhodolskiy on 29 Apr 512 bytes Align Ollama HTTP timeout with LLM timeouts
"""Unit tests for Ollama backend helpers."""

import httpx

from navi.exceptions import LLMConnectionError
from navi.llm.ollama import _classify_error


def test_classify_read_timeout_as_connection_error():
    err = _classify_error(httpx.ReadTimeout("timed out"))

    assert isinstance(err, LLMConnectionError)


def test_classify_empty_timeout_message_as_connection_error():
    err = _classify_error(httpx.ReadTimeout(""))

    assert isinstance(err, LLMConnectionError)
    assert str(err) == "ReadTimeout"