ed
b5e512f483
Adds [C: caller] tags to functions/methods and [M: mutation] / [U: usage] tags to class variables based on cross-module call analysis.
17 lines
508 B
Python
17 lines
508 B
Python
from src import ai_client
|
|
|
|
def test_list_models_gemini_cli() -> None:
|
|
"""
|
|
|
|
|
|
Verifies that 'ai_client.list_models' correctly returns a list of models
|
|
for the 'gemini_cli' provider.
|
|
"""
|
|
models = ai_client.list_models("gemini_cli")
|
|
assert "gemini-3.1-pro-preview" in models
|
|
assert "gemini-3-flash-preview" in models
|
|
assert "gemini-2.5-pro" in models
|
|
assert "gemini-2.5-flash" in models
|
|
assert "gemini-2.0-flash" in models
|
|
assert "gemini-2.5-flash-lite" in models
|
|
assert len(models) == 6 |