litellm/tests/local_testing/test_model_alias_map.py

#### What this tests ####
#    This tests the model alias mapping - if user passes in an alias, and has set an alias, set it to the actual value

import os
import sys
import traceback

sys.path.insert(
    0, os.path.abspath("../..")
)  # Adds the parent directory to the system path
import pytest

import litellm
from litellm import completion, embedding

litellm.set_verbose = True

model_alias_map = {"good-model": "groq/llama-3.1-8b-instant"}


def test_model_alias_map(caplog):
    try:
        litellm.model_alias_map = model_alias_map
        response = completion(
            "good-model",
            messages=[{"role": "user", "content": "Hey, how's it going?"}],
            top_p=0.1,
            temperature=0.01,
            max_tokens=10,
        )
        print(response.model)

        captured_logs = [rec.levelname for rec in caplog.records]

        for log in captured_logs:
            assert "ERROR" not in log

        assert "llama-3.1-8b-instant" in response.model
    except litellm.ServiceUnavailableError:
        pass
    except Exception as e:
        pytest.fail(f"Error occurred: {e}")


# test_model_alias_map()