mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-23 22:50:17 +08:00
e4c4fdabbd
### What problem does this PR solve? _Briefly describe what this PR aims to solve. Include background context that will help reviewers understand the purpose of the PR._ ### Type of change - [x] Refactoring --------- Signed-off-by: jinhai <haijin.chn@gmail.com>
51 lines
1.8 KiB
Python
51 lines
1.8 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_ragflow_version() -> typing.Optional[str]:
|
|
return RAGFLOW_VERSION_INFO
|
|
|
|
|
|
RAGFLOW_VERSION_INFO = "dev"
|
|
|
|
|
|
def get_closest_tag_and_count():
|
|
try:
|
|
# 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})"
|
|
except Exception as e:
|
|
return 'unknown'
|
|
|
|
|
|
if RAGFLOW_VERSION_INFO == 'dev':
|
|
RAGFLOW_VERSION_INFO = get_closest_tag_and_count()
|