Update api_server.py

api_server.py

@@ -4,7 +4,7 @@ from fastapi import FastAPI, HTTPException, Depends
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from backend.agent import run_agent
-from typing import Union
+from typing import List, Dict, Any
 
 
 app = FastAPI(title="Nursing Copilot API")
@@ -32,7 +32,7 @@ class PredictionRequest(BaseModel):
     model: str | None = "gpt2-nursing"
 
 class PredictionResponse(BaseModel):
-    completions: list[dict]
+    completions: List[Dict[str, Any]]
 
 # -----------------------------
 # API Endpoint
@@ -42,11 +42,10 @@ def predict_completion(request: PredictionRequest):
 
     input_text = request.prompt
 
-    # ⭐ 交給 agent（固定 or AI）
     result = run_agent(input_text)
 
-    # ⭐ agent 統一回傳 list
-    return {"completions": result}
+    # ⭐ 無論 fixed 或 AI，都包成 list
+    return {"completions": [result]}
 
 
 # -----------------------------