Update app.py

app.py

@@ -1,12 +1,16 @@
+import logging
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import subprocess
 
+# 配置日志
+logging.basicConfig(level=logging.INFO)
+
 app = FastAPI()
 
-# 健康检查端点
 @app.get("/health")
 def health_check():
+    logging.info("Health check endpoint was called")
     return {"status": "ok"}
 
 class PromptRequest(BaseModel):
@@ -15,12 +19,14 @@ class PromptRequest(BaseModel):
 @app.post("/generate")
 async def generate_response(request: PromptRequest):
     try:
-        # 调用 Ollama 模型生成响应
         result = subprocess.check_output(["ollama", "run", request.prompt], text=True)
+        logging.info(f"Model response: {result.strip()}")
         return {"response": result.strip()}
     except subprocess.CalledProcessError as e:
+        logging.error(f"Error during model generation: {e}")
         raise HTTPException(status_code=500, detail=f"Error: {e}")
 
 @app.get("/")
 def read_root():
+    logging.info("Root endpoint was called")
     return {"message": "Ollama API is running"}