MarcoLeung052 commited on
Commit
fa0be41
·
verified ·
1 Parent(s): 2509709

Update api_server.py

Browse files
Files changed (1) hide show
  1. 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]}