diff --git a/litellm/proxy/types_utils/utils.py b/litellm/proxy/types_utils/utils.py index 676d7fb51b2..43fc87f8c58 100644 --- a/litellm/proxy/types_utils/utils.py +++ b/litellm/proxy/types_utils/utils.py @@ -1,3 +1,5 @@ +"""Utility helpers for loading proxy type-related instances and annotations.""" + import asyncio import importlib import importlib.util diff --git a/tests/test_litellm/proxy/test_proxy_utils.py b/tests/test_litellm/proxy/test_proxy_utils.py index 2605eadba7a..9ffe9e6fff5 100644 --- a/tests/test_litellm/proxy/test_proxy_utils.py +++ b/tests/test_litellm/proxy/test_proxy_utils.py @@ -1,7 +1,9 @@ +import ast import datetime as real_datetime import json import os import sys +from pathlib import Path import pytest from fastapi import HTTPException @@ -20,6 +22,12 @@ from unittest.mock import MagicMock from litellm.proxy.utils import get_custom_url, join_paths +def test_proxy_types_utils_has_module_docstring(): + path = Path("litellm/proxy/types_utils/utils.py") + module = ast.parse(path.read_text()) + assert ast.get_docstring(module) + + def test_get_custom_url(monkeypatch): monkeypatch.setenv("SERVER_ROOT_PATH", "/litellm") custom_url = get_custom_url(request_base_url="http://0.0.0.0:4000", route="ui/")