mirror of
https://github.com/bentoml/OpenLLM.git
synced 2026-01-20 13:29:35 -05:00
c7f4dc7bb2
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
40 lines
964 B
Python
Executable File
40 lines
964 B
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from markdown_it import MarkdownIt
|
|
|
|
|
|
md = MarkdownIt()
|
|
|
|
ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
with open(os.path.join(ROOT, "README.md"), "r") as f:
|
|
readme = md.parse(f.read())
|
|
|
|
# NOTE: Currently, we only have one table in README, which is the Model readme.
|
|
table = [r for r in readme if r.type == "html_block" and r.content.startswith("<td><a")]
|
|
|
|
prev = os.environ.pop("OPENLLMDEVDEBUG", None)
|
|
available = (
|
|
subprocess.check_output(
|
|
[sys.executable, "-m", "openllm", "models", "-o", "porcelain"],
|
|
)
|
|
.strip()
|
|
.decode("utf-8")
|
|
.count("\n")
|
|
+ 1
|
|
)
|
|
if prev:
|
|
os.environ["OPENLLMDEVDEBUG"] = prev
|
|
|
|
on_table = len(table) # NOTE: minus the header
|
|
|
|
if available - on_table != 0:
|
|
print("README.md is out of date! Make sure to run ./tools/update-readme.py")
|
|
raise SystemExit(1)
|
|
raise SystemExit(0)
|