add version (infiniflow#807)

### What problem does this PR solve? infiniflow#709 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
tiwater · May 16, 2024 · bba5314 · bba5314
1 parent 73c24ab
commit bba5314
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 73 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -15,6 +15,7 @@ ENV PYTHONPATH=/ragflow/
 ENV HF_ENDPOINT=https://hf-mirror.com
 
 ADD docker/entrypoint.sh ./entrypoint.sh
+ADD docker/.env ./
 RUN chmod +x ./entrypoint.sh
 
 ENTRYPOINT ["./entrypoint.sh"]
diff --git a/api/db/services/kb_service.py b/api/db/services/kb_service.py
diff --git a/api/versions.py b/api/versions.py
@@ -14,17 +14,15 @@
 #  limitations under the License.
 #
 import os
-
 import dotenv
 import typing
-
 from api.utils.file_utils import get_project_base_directory
 
 
 def get_versions() -> typing.Mapping[str, typing.Any]:
-    return dotenv.dotenv_values(
-        dotenv_path=os.path.join(get_project_base_directory(), "rag.env")
-    )
+    dotenv.load_dotenv(dotenv.find_dotenv())
+    return dotenv.dotenv_values()
+
 
 def get_rag_version() -> typing.Optional[str]:
-    return get_versions().get("RAG")
+    return get_versions().get("RAGFLOW_VERSION", "dev")