mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 08:38:30 +00:00
Fix #2549: Improve error handling for Ollama connection errors
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
17
tests/test_ollama_connection.py
Normal file
17
tests/test_ollama_connection.py
Normal file
@@ -0,0 +1,17 @@
|
||||
import pytest
|
||||
from unittest.mock import patch, MagicMock
|
||||
from crewai.llm import LLM
|
||||
from crewai.utilities.exceptions.ollama_connection_exception import OllamaConnectionException
|
||||
|
||||
class TestOllamaConnection:
|
||||
def test_ollama_connection_error(self):
|
||||
with patch('litellm.completion') as mock_completion:
|
||||
mock_completion.side_effect = Exception("OllamaException - [Errno 111] Connection refused")
|
||||
|
||||
llm = LLM(model="ollama/llama3")
|
||||
|
||||
with pytest.raises(OllamaConnectionException) as exc_info:
|
||||
llm.call([{"role": "user", "content": "Hello"}])
|
||||
|
||||
assert "Failed to connect to Ollama" in str(exc_info.value)
|
||||
assert "Please make sure Ollama is installed and running" in str(exc_info.value)
|
||||
Reference in New Issue
Block a user