mirror of
https://github.com/semantica-agi/semantica.git
synced 2026-09-06 04:00:19 +00:00
Zohaib HassnainandSameer Kadam
ea0dd17bff
* feat(llms): add Gemini, Ollama, DeepSeek, Novita provider wrappers * fix(llms): address Qodo review findings on provider wrappers PR * fix(llms): address review findings --------- Co-authored-by: Sameer Kadam <sskadam6305@gmail.com>
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
"""Tests for the Gemini LLM provider wrapper (semantica.llms.Gemini)."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from semantica.llms import Gemini
|
|
from semantica.utils.exceptions import ProcessingError
|
|
|
|
|
|
def test_construction_stores_model_and_api_key():
|
|
gemini = Gemini(model="gemini-pro", api_key="fake-key")
|
|
assert gemini.model == "gemini-pro"
|
|
assert gemini.api_key == "fake-key"
|
|
|
|
|
|
def test_is_available_false_with_no_key(monkeypatch):
|
|
monkeypatch.delenv("GEMINI_API_KEY", raising=False)
|
|
gemini = Gemini(api_key=None)
|
|
assert gemini.is_available() is False
|
|
|
|
|
|
def test_generate_raises_clear_error_when_unavailable(monkeypatch):
|
|
monkeypatch.delenv("GEMINI_API_KEY", raising=False)
|
|
gemini = Gemini(api_key=None)
|
|
with pytest.raises(ProcessingError, match="Gemini provider not available"):
|
|
gemini.generate("hello")
|
|
|
|
|
|
def test_generate_forwards_to_the_real_provider_when_available():
|
|
gemini = Gemini(api_key="fake-key")
|
|
gemini.provider = MagicMock()
|
|
gemini.provider.is_available.return_value = True
|
|
gemini.provider.generate.return_value = "a fake response"
|
|
|
|
result = gemini.generate("hello", temperature=0.5)
|
|
|
|
assert result == "a fake response"
|
|
gemini.provider.generate.assert_called_once_with("hello", temperature=0.5)
|
|
|
|
|
|
def test_generate_structured_forwards_to_the_real_provider():
|
|
gemini = Gemini(api_key="fake-key")
|
|
gemini.provider = MagicMock()
|
|
gemini.provider.is_available.return_value = True
|
|
gemini.provider.generate_structured.return_value = {"key": "value"}
|
|
|
|
result = gemini.generate_structured("hello")
|
|
|
|
assert result == {"key": "value"}
|
|
gemini.provider.generate_structured.assert_called_once_with("hello")
|
|
|
|
|
|
def test_generate_typed_forwards_schema_and_max_retries():
|
|
gemini = Gemini(api_key="fake-key")
|
|
gemini.provider = MagicMock()
|
|
gemini.provider.is_available.return_value = True
|
|
fake_schema = object()
|
|
gemini.provider.generate_typed.return_value = "typed result"
|
|
|
|
result = gemini.generate_typed("hello", fake_schema, max_retries=5)
|
|
|
|
assert result == "typed result"
|
|
gemini.provider.generate_typed.assert_called_once_with(
|
|
"hello", fake_schema, max_retries=5
|
|
)
|
|
|
|
|
|
def test_generate_structured_raises_clear_error_when_unavailable(monkeypatch):
|
|
monkeypatch.delenv("GEMINI_API_KEY", raising=False)
|
|
gemini = Gemini(api_key=None)
|
|
with pytest.raises(ProcessingError, match="Gemini provider not available"):
|
|
gemini.generate_structured("hello")
|
|
|
|
|
|
def test_generate_typed_raises_clear_error_when_unavailable(monkeypatch):
|
|
monkeypatch.delenv("GEMINI_API_KEY", raising=False)
|
|
gemini = Gemini(api_key=None)
|
|
with pytest.raises(ProcessingError, match="Gemini provider not available"):
|
|
gemini.generate_typed("hello", object())
|