You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
613 B
18 lines
613 B
"""
|
|
Unit tests for ModelFactory with new default model (gemini-3.6-flash-high)
|
|
"""
|
|
import pytest
|
|
from unittest.mock import patch, Mock
|
|
from src.models.factory import ModelFactory
|
|
|
|
def test_model_factory_default():
|
|
factory = ModelFactory()
|
|
model = factory.get_model()
|
|
# Check that default model loaded is gemini-3.6-flash-high
|
|
assert model.id == "gemini-3.6-flash-high"
|
|
assert model.base_url == "http://ai.newhorizonco.uk/v1"
|
|
|
|
def test_model_factory_get_specific_model():
|
|
factory = ModelFactory()
|
|
model = factory.get_model("gemini_flash")
|
|
assert model.id == "gemini-3.6-flash-high"
|