mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-20 21:20:00 +08:00
201bbef7c0
### What problem does this PR solve? Print version when RAGFlow server startup ### Type of change - [x] New Feature (non-breaking change which adds functionality) --------- Signed-off-by: jinhai <haijin.chn@gmail.com> Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
#
|
|
# Copyright 2024 The InfiniFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
import dotenv
|
|
import typing
|
|
import subprocess
|
|
|
|
|
|
def get_versions() -> typing.Mapping[str, typing.Any]:
|
|
dotenv.load_dotenv(dotenv.find_dotenv())
|
|
return dotenv.dotenv_values()
|
|
|
|
|
|
def get_rag_version() -> typing.Optional[str]:
|
|
return get_versions().get("RAGFLOW_IMAGE", "infiniflow/ragflow:dev").split(":")[-1]
|
|
|
|
|
|
RAGFLOW_VERSION_INFO = "dev"
|
|
|
|
|
|
def get_closest_tag_and_count():
|
|
# Get the current commit hash
|
|
commit_id = subprocess.check_output(['git', 'rev-parse', '--short', 'HEAD']).strip().decode('utf-8')
|
|
# Get the closest tag
|
|
closest_tag = subprocess.check_output(['git', 'describe', '--tags', '--abbrev=0']).strip().decode('utf-8')
|
|
# Get the commit hash of the closest tag
|
|
closest_tag_commit = subprocess.check_output(['git', 'rev-list', '-n', '1', closest_tag]).strip().decode('utf-8')
|
|
# Get the commit count since the closest tag
|
|
process = subprocess.Popen(['git', 'rev-list', '--count', f'{closest_tag}..HEAD'], stdout=subprocess.PIPE)
|
|
commits_count, _ = process.communicate()
|
|
commits_count = int(commits_count.strip())
|
|
|
|
if commits_count == 0:
|
|
return closest_tag
|
|
else:
|
|
return f"{commit_id}({closest_tag}~{commits_count})"
|
|
|
|
|
|
if RAGFLOW_VERSION_INFO == 'dev':
|
|
RAGFLOW_VERSION_INFO = get_closest_tag_and_count()
|