view mrjunejune/inference/litellm_config_test.py @ 265:056790c4fb0d

add role-aware Epi assistant prompts Add verified June knowledge, guest/member/admin Copilot profiles, profile-isolated session recovery, animated Epi greetings, and a single authoritative runtime config workflow for inference. Co-authored-by: Copilot <[email protected]>
author MrJuneJune <me@mrjunejune.com>
date Fri, 07 Aug 2026 10:50: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()