Skip to content

datarobot_genai.core.utils.urls

urls

get_api_base 

get_api_base(api_base: str, deployment_id: str | None) -> str

Construct the LiteLLM API base URL for a deployment.

Parameters

api_base : str Base URL for the LiteLLM API. deployment_id : str | None Deployment identifier. When provided, a chat/completions URL is produced.

Returns

str Normalized URL for the given deployment. Ensures a trailing slash unless the path ends with "chat/completions" or already has a meaningful path component.

Source code in datarobot_genai/core/utils/urls.py 
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
def get_api_base(api_base: str, deployment_id: str | None) -> str:
    """
    Construct the LiteLLM API base URL for a deployment.

    Parameters
    ----------
    api_base : str
        Base URL for the LiteLLM API.
    deployment_id : str | None
        Deployment identifier. When provided, a chat/completions URL is produced.

    Returns
    -------
    str
        Normalized URL for the given deployment. Ensures a trailing slash unless the path
        ends with "chat/completions" or already has a meaningful path component.
    """
    # Normalize the URL and drop a trailing /api/v2 if present
    parsed = urlparse(api_base)
    path = re.sub(r"/api/v2/?$", "", parsed.path)
    base_url = urlunparse(
        (
            parsed.scheme,
            parsed.netloc,
            path,
            parsed.params,
            parsed.query,
            parsed.fragment,
        )
    )
    base_url = base_url.rstrip("/")

    # If the base_url already ends with chat/completions, return it.
    if base_url.endswith("chat/completions"):
        return base_url

    # If the path contains deployments or genai, it's already a complete API path preserve it.
    if path and ("deployments" in path or "genai" in path):
        return f"{base_url}/" if not base_url.endswith("/") else base_url

    # For all other cases (including custom base paths), apply deployment logic if needed.
    if deployment_id:
        return f"{base_url}/api/v2/deployments/{deployment_id}/chat/completions"
    # Otherwise, just return the base URL with a trailing slash for normalization.
    return f"{base_url}/"