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