DataOps-platform/tests/test_deepseek_client.py

from app.core.llm.deepseek_client import (
    get_llm_chat_completions_url,
    normalize_llm_base_url,
)


def test_normalize_llm_base_url_default():
    assert normalize_llm_base_url("") == "https://api.deepseek.com"
    assert normalize_llm_base_url("https://api.deepseek.com") == "https://api.deepseek.com"


def test_normalize_llm_base_url_strips_trailing_v1():
    assert normalize_llm_base_url("https://api.deepseek.com/v1") == "https://api.deepseek.com"
    assert normalize_llm_base_url("https://api.deepseek.com/v1/") == "https://api.deepseek.com"


def test_get_llm_chat_completions_url():
    assert (
        get_llm_chat_completions_url()
        == "https://api.deepseek.com/v1/chat/completions"
    )