mirror of
https://github.com/Memo-2023/mana-monorepo.git
synced 2026-05-15 16:39:39 +02:00
1495dbe476
Python/FastAPI service providing unified OpenAI-compatible API for Ollama and cloud LLM providers (OpenRouter, Groq, Together). Features: - Chat completions with streaming (SSE) - Vision/multimodal support - Embeddings generation - Multi-provider routing (provider/model format) - Prometheus metrics - Optional Redis caching
113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
"""Provider tests."""
|
|
|
|
import pytest
|
|
|
|
from src.models import ChatCompletionRequest, Message
|
|
from src.providers import OllamaProvider, OpenAICompatProvider, ProviderRouter
|
|
|
|
|
|
class TestProviderRouter:
|
|
"""Test provider routing logic."""
|
|
|
|
def test_parse_model_with_provider(self):
|
|
"""Test model parsing with provider prefix."""
|
|
router = ProviderRouter()
|
|
|
|
provider, model = router._parse_model("ollama/gemma3:4b")
|
|
assert provider == "ollama"
|
|
assert model == "gemma3:4b"
|
|
|
|
def test_parse_model_without_provider(self):
|
|
"""Test model parsing without provider prefix (defaults to ollama)."""
|
|
router = ProviderRouter()
|
|
|
|
provider, model = router._parse_model("gemma3:4b")
|
|
assert provider == "ollama"
|
|
assert model == "gemma3:4b"
|
|
|
|
def test_parse_model_openrouter(self):
|
|
"""Test model parsing for OpenRouter."""
|
|
router = ProviderRouter()
|
|
|
|
provider, model = router._parse_model("openrouter/meta-llama/llama-3.1-8b-instruct")
|
|
assert provider == "openrouter"
|
|
assert model == "meta-llama/llama-3.1-8b-instruct"
|
|
|
|
def test_get_invalid_provider(self):
|
|
"""Test getting invalid provider raises error."""
|
|
router = ProviderRouter()
|
|
|
|
with pytest.raises(ValueError, match="not available"):
|
|
router._get_provider("invalid_provider")
|
|
|
|
|
|
class TestOllamaProvider:
|
|
"""Test Ollama provider."""
|
|
|
|
def test_convert_simple_messages(self):
|
|
"""Test converting simple text messages."""
|
|
provider = OllamaProvider()
|
|
request = ChatCompletionRequest(
|
|
model="gemma3:4b",
|
|
messages=[
|
|
Message(role="user", content="Hello"),
|
|
],
|
|
)
|
|
|
|
messages = provider._convert_messages(request)
|
|
assert len(messages) == 1
|
|
assert messages[0]["role"] == "user"
|
|
assert messages[0]["content"] == "Hello"
|
|
|
|
def test_convert_multimodal_messages(self):
|
|
"""Test converting multimodal messages."""
|
|
provider = OllamaProvider()
|
|
request = ChatCompletionRequest(
|
|
model="llava:7b",
|
|
messages=[
|
|
Message(
|
|
role="user",
|
|
content=[
|
|
{"type": "text", "text": "What's in this image?"},
|
|
{
|
|
"type": "image_url",
|
|
"image_url": {"url": "data:image/png;base64,iVBORw0KGgo="},
|
|
},
|
|
],
|
|
),
|
|
],
|
|
)
|
|
|
|
messages = provider._convert_messages(request)
|
|
assert len(messages) == 1
|
|
assert messages[0]["role"] == "user"
|
|
assert messages[0]["content"] == "What's in this image?"
|
|
assert "images" in messages[0]
|
|
assert len(messages[0]["images"]) == 1
|
|
|
|
|
|
class TestOpenAICompatProvider:
|
|
"""Test OpenAI-compatible provider."""
|
|
|
|
def test_convert_simple_messages(self):
|
|
"""Test converting simple text messages."""
|
|
provider = OpenAICompatProvider(
|
|
name="test",
|
|
base_url="http://localhost",
|
|
api_key="test-key",
|
|
)
|
|
|
|
request = ChatCompletionRequest(
|
|
model="test-model",
|
|
messages=[
|
|
Message(role="system", content="You are helpful."),
|
|
Message(role="user", content="Hello"),
|
|
],
|
|
)
|
|
|
|
messages = provider._convert_messages(request)
|
|
assert len(messages) == 2
|
|
assert messages[0]["role"] == "system"
|
|
assert messages[0]["content"] == "You are helpful."
|
|
assert messages[1]["role"] == "user"
|
|
assert messages[1]["content"] == "Hello"
|