Update api_server.py
Browse files- api_server.py +2 -1
api_server.py
CHANGED
|
@@ -139,7 +139,6 @@ class RecordCreate(BaseModel):
|
|
| 139 |
@app.post("/api/predict")
|
| 140 |
def predict_completion(request: PredictionRequest, db: Session = Depends(get_db)):
|
| 141 |
|
| 142 |
-
# Debug
|
| 143 |
print("收到的 request:", request.dict())
|
| 144 |
|
| 145 |
if not request.prompt or request.prompt.strip() == "":
|
|
@@ -153,9 +152,11 @@ def predict_completion(request: PredictionRequest, db: Session = Depends(get_db)
|
|
| 153 |
except:
|
| 154 |
raise HTTPException(status_code=400, detail="patient_id 必須是整數")
|
| 155 |
|
|
|
|
| 156 |
result = run_ai_output(
|
| 157 |
input_text=request.prompt,
|
| 158 |
patient_id=patient_id,
|
|
|
|
| 159 |
)
|
| 160 |
|
| 161 |
return {"completions": [result]}
|
|
|
|
| 139 |
@app.post("/api/predict")
|
| 140 |
def predict_completion(request: PredictionRequest, db: Session = Depends(get_db)):
|
| 141 |
|
|
|
|
| 142 |
print("收到的 request:", request.dict())
|
| 143 |
|
| 144 |
if not request.prompt or request.prompt.strip() == "":
|
|
|
|
| 152 |
except:
|
| 153 |
raise HTTPException(status_code=400, detail="patient_id 必須是整數")
|
| 154 |
|
| 155 |
+
# ⭐ 正確傳入 db
|
| 156 |
result = run_ai_output(
|
| 157 |
input_text=request.prompt,
|
| 158 |
patient_id=patient_id,
|
| 159 |
+
db=db
|
| 160 |
)
|
| 161 |
|
| 162 |
return {"completions": [result]}
|