mirror of
https://github.com/bentoml/OpenLLM.git
synced 2026-01-22 14:31:26 -05:00
26 lines
869 B
Python
Executable File
26 lines
869 B
Python
Executable File
#!/usr/bin/env python3
|
|
from __future__ import annotations
|
|
import os, 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())
|
|
sys.path.insert(0, os.path.join(ROOT, 'openllm-python', 'src'))
|
|
import openllm
|
|
|
|
# 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 = len(openllm.CONFIG_MAPPING.keys())
|
|
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)
|