Mercurial
view mrjunejune/inference/litellm_config_test.py @ 266:efaf4c63cc94
fix clean production bundle staging
Recreate deployment staging before copying the Bazel bundle and verify required inference runtime paths before promoting it.
Co-authored-by: Copilot <[email protected]>
| author | MrJuneJune <me@mrjunejune.com> |
|---|---|
| date | Fri, 07 Aug 2026 12:52:30 -0700 |
| parents | 1f9877b637e9 |
| children |
line wrap: on
line source
import pathlib import unittest import yaml class LiteLlmConfigTest(unittest.TestCase): def test_config_uses_copilot_provider_without_embedded_secret(self) -> None: path = pathlib.Path(__file__).with_name("litellm_config.yaml") config = yaml.safe_load(path.read_text()) models = config["model_list"] self.assertEqual(len(models), 1) self.assertEqual(models[0]["model_name"], "jrpg-copilot") self.assertTrue( models[0]["litellm_params"]["model"].startswith( "github_copilot/", ) ) self.assertEqual( config["general_settings"]["master_key"], "os.environ/LITELLM_MASTER_KEY", ) if __name__ == "__main__": unittest.main()