Add main module header comment
Co-authored-by: ishaan-berri <ishaan-berri@users.noreply.github.com>
This commit is contained in:
parent
454ce5073f
commit
062b5b31fb
BIN
.evidence/main_header_repro.png
Normal file
BIN
.evidence/main_header_repro.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 60 KiB |
@ -1,3 +1,5 @@
|
||||
# LiteLLM main module: public completion, embedding, streaming, and moderation entrypoints.
|
||||
#
|
||||
# +-----------------------------------------------+
|
||||
# | |
|
||||
# | Give Feedback / Get Help |
|
||||
|
||||
13
tests/test_litellm/test_main_module_header.py
Normal file
13
tests/test_litellm/test_main_module_header.py
Normal file
@ -0,0 +1,13 @@
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def test_main_py_starts_with_brief_file_description():
|
||||
repo_root = Path(__file__).resolve().parents[2]
|
||||
main_py = repo_root / "litellm" / "main.py"
|
||||
|
||||
first_two_lines = main_py.read_text(encoding="utf-8").splitlines()[:2]
|
||||
|
||||
assert any(
|
||||
"LiteLLM main module" in line and "entrypoints" in line
|
||||
for line in first_two_lines
|
||||
)
|
||||
Loading…
Reference in New Issue
Block a user