shigeki Ishida commited on
Commit
08d1517
·
1 Parent(s): 3cb0a6a

Add vllm version fetching from GitHub API

Browse files
Files changed (1) hide show
  1. src/envs.py +13 -1
src/envs.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
 
 
3
  from huggingface_hub import HfApi
4
 
5
  # Info to change for your repository
@@ -22,7 +23,18 @@ EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
22
  EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
23
  EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")
24
 
 
25
  # vllm version
26
- VLLM_CURRENT_VERSION = "v0.6.3"
 
 
 
 
 
 
 
 
 
 
27
 
28
  API = HfApi(token=TOKEN)
 
1
  import os
2
 
3
+ import requests
4
  from huggingface_hub import HfApi
5
 
6
  # Info to change for your repository
 
23
  EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
24
  EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")
25
 
26
+
27
  # vllm version
28
+ def get_latest_vllm_version() -> str:
29
+ """GitHubからvllmの最新バージョンを取得する"""
30
+ url = "https://api.github.com/repos/vllm-project/vllm/releases/latest"
31
+ response = requests.get(url)
32
+ if response.status_code == 200:
33
+ latest_release = response.json()
34
+ return latest_release["tag_name"]
35
+ return None # APIリクエストが失敗した場合はNoneを返す
36
+
37
+
38
+ VLLM_CURRENT_VERSION = get_latest_vllm_version()
39
 
40
  API = HfApi(token=TOKEN)