From acef961415e37dec1cd0ab8d7c0d6b35f4bc5f25 Mon Sep 17 00:00:00 2001
From: zhanghan2021 <zhanghan@kylinos.cn>
Date: Thu, 7 Nov 2024 15:53:47 +0800
Subject: [PATCH] add util tools for response and format log

---
 llmops/app_factory.py    |  5 +++++
 llmops/requirements.txt  |  2 ++
 llmops/utils/command.py  | 12 ++++++++++++
 llmops/utils/logger.py   | 11 +++++++++++
 llmops/utils/response.py | 18 ++++++++++++++++++
 5 files changed, 48 insertions(+)
 create mode 100644 llmops/requirements.txt
 create mode 100644 llmops/utils/command.py
 create mode 100644 llmops/utils/logger.py
 create mode 100644 llmops/utils/response.py

diff --git a/llmops/app_factory.py b/llmops/app_factory.py
index 9d8eeff5..b035671c 100644
--- a/llmops/app_factory.py
+++ b/llmops/app_factory.py
@@ -1,5 +1,6 @@
 from flask import Flask
 from config.config import init_config
+from utils.logger import setup_logger
 
 
 def create_app() -> Flask:
@@ -11,6 +12,10 @@ def create_app() -> Flask:
     app.config["PORT"] = config.app_conf.port
     app.config["DEBUG"] = config.app_conf.debug
 
+    # 设置日志
+    logger = setup_logger()
+    app.logger = logger
+
     return app
 
 
diff --git a/llmops/requirements.txt b/llmops/requirements.txt
new file mode 100644
index 00000000..23eaa329
--- /dev/null
+++ b/llmops/requirements.txt
@@ -0,0 +1,2 @@
+Flask==2.1.2
+PyYAML==6.0
\ No newline at end of file
diff --git a/llmops/utils/command.py b/llmops/utils/command.py
new file mode 100644
index 00000000..43d07e56
--- /dev/null
+++ b/llmops/utils/command.py
@@ -0,0 +1,12 @@
+import subprocess
+
+
+def ExecuteCommand(command):
+    try:
+        result = subprocess.run(
+            command, shell=True, text=True, capture_output=True, check=True
+        )
+        return (result.stdout, result.returncode)
+    except subprocess.CalledProcessError as e:
+        print(f"命令执行失败: {e}")
+        return (e.stderr, e.returncode)
diff --git a/llmops/utils/logger.py b/llmops/utils/logger.py
new file mode 100644
index 00000000..a9a5fb72
--- /dev/null
+++ b/llmops/utils/logger.py
@@ -0,0 +1,11 @@
+import logging
+
+
+def setup_logger():
+    logging.basicConfig(
+        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+        level=logging.INFO,
+    )
+    logger = logging.getLogger(__name__)
+    return logger
diff --git a/llmops/utils/response.py b/llmops/utils/response.py
new file mode 100644
index 00000000..8daf6241
--- /dev/null
+++ b/llmops/utils/response.py
@@ -0,0 +1,18 @@
+from flask import jsonify
+
+
+def success(data=None, msg=""):
+    return result(200, 200, data, msg)
+
+
+def fail(data=None, msg=""):
+    return result(200, 400, data, msg)
+
+
+def not_found(data=None, msg=""):
+    return result(200, 404, data, msg)
+
+
+def result(http_status, code, data=None, msg=""):
+    response = {"code": code, "data": data, "msg": msg}
+    return jsonify(response), http_status
-- 
Gitee