diff --git a/llmops/app_factory.py b/llmops/app_factory.py index 9d8eeff59ac61023564888dedaf7c5639503e9c6..b035671c5682a2ae1c94497ae49fdff2df0cc35f 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 0000000000000000000000000000000000000000..23eaa329a1a64b66e9f0be5346495ce24a15e1a4 --- /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 0000000000000000000000000000000000000000..43d07e5680290af63a6eb0948be4fd7551a27f6a --- /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 0000000000000000000000000000000000000000..a9a5fb72dc6cf6f3eb33a66b5986b186188e604e --- /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 0000000000000000000000000000000000000000..8daf6241caae934026094e07d045a34ad1b7623a --- /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