引入日志模块

.gitignore

@@ -9,3 +9,4 @@ __pycache__/
 
 # OS
 .DS_Store" >> .gitignore
+*.log

app/__init__.py

@@ -1,12 +1,22 @@
+import logging
 from flask import Flask
+from flask_cors import CORS
 from app.api.v1.controllers import api_v1
 from app.api.v1.zhipu_controller import zhipu_controller
 from app.config import Config
 
 def create_app():
     app = Flask(__name__)
+    CORS(app)
+    
     app.config.from_object(Config)
 
+    # Set up logging using configuration from Config
+    logging.basicConfig(**Config.LOGGING_CONFIG)
+    logger = logging.getLogger(__name__)
+
+    logger.info("Application started")
+
     # Register Blueprints
     app.register_blueprint(api_v1, url_prefix='/api/v1')
 
@@ -15,4 +25,5 @@ def create_app():
     # Register Zhipu-related routes under /api/v1/zhipu
     app.register_blueprint(zhipu_controller, url_prefix='/api/v1')
 
+    logger.info("Application setup completed")
     return app

app/api/v1/zhipu_controller.py

@@ -1,9 +1,11 @@
+import logging
 from flask import Blueprint, request, Response
 from app.services.zhipu_service import ZhipuService
 from app.utils.prompt_repository import PromptRepository  # Add this import
 
 zhipu_controller = Blueprint('zhipu_controller', __name__)
 zhipu_service = ZhipuService()
+logger = logging.getLogger(__name__)
 
 @zhipu_controller.route('/zhipu/stream', methods=['POST'])
 def stream_sse():

app/config.py

@@ -1,6 +1,15 @@
 import os
+import logging
 
 class Config:
     SECRET_KEY = os.environ.get('SECRET_KEY') or 'supersecretkey'
     SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or 'sqlite:///app.db'
     SQLALCHEMY_TRACK_MODIFICATIONS = False
+
+    # Logging configuration
+    LOGGING_CONFIG = {
+        'level': logging.INFO,
+        'format': '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        'filename': 'app.log',
+        'filemode': 'a'
+    }

app/services/zhipu_service.py

@@ -1,6 +1,9 @@
 from zhipuai import ZhipuAI
+import logging
 
 
+logger = logging.getLogger(__name__)
+
 class ZhipuService:
     def __init__(self):
         self.model_name = "glm-4"

manage.py

@@ -1,6 +1,11 @@
+import logging
 from app import create_app
 
+logger = logging.getLogger(__name__)
+
 app = create_app()
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000)
+    logger.info("Starting the application")
+    app.run(host='0.0.0.0', port=5001)
+    logger.info("Application stopped")

requirements.txt

@@ -2,6 +2,7 @@ alembic==1.13.3
 blinker==1.8.2
 click==8.1.7
 Flask==3.0.3
+Flask_Cors==5.0.0
 Flask-Migrate==4.0.7
 Flask-SQLAlchemy==3.1.1
 importlib_metadata==8.5.0