Spaces:
Sleeping
Sleeping
Update agent/server.py
Browse files- agent/server.py +5 -3
agent/server.py
CHANGED
|
@@ -354,11 +354,13 @@ def recommend(req: RecommendRequest, response: Response):
|
|
| 354 |
|
| 355 |
response.headers["X-Cache"] = "MISS"
|
| 356 |
|
|
|
|
|
|
|
| 357 |
final_results, summary, debug_payload = _run_pipeline(
|
| 358 |
-
req.query, verbose=
|
| 359 |
)
|
| 360 |
resp = {"recommended_assessments": final_results}
|
| 361 |
-
if
|
| 362 |
resp["debug"] = _sanitize_debug(debug_payload)
|
| 363 |
resp["summary"] = summary
|
| 364 |
|
|
@@ -367,7 +369,7 @@ def recommend(req: RecommendRequest, response: Response):
|
|
| 367 |
|
| 368 |
@app.get("/health")
|
| 369 |
def health():
|
| 370 |
-
return {"status": "
|
| 371 |
|
| 372 |
|
| 373 |
@app.get("/")
|
|
|
|
| 354 |
|
| 355 |
response.headers["X-Cache"] = "MISS"
|
| 356 |
|
| 357 |
+
llm_model = req.llm_model or "Qwen/Qwen2.5-1.5B-Instruct"
|
| 358 |
+
verbose = bool(req.verbose) if req.verbose is not None else False
|
| 359 |
final_results, summary, debug_payload = _run_pipeline(
|
| 360 |
+
req.query, verbose=verbose, llm_model=llm_model
|
| 361 |
)
|
| 362 |
resp = {"recommended_assessments": final_results}
|
| 363 |
+
if verbose:
|
| 364 |
resp["debug"] = _sanitize_debug(debug_payload)
|
| 365 |
resp["summary"] = summary
|
| 366 |
|
|
|
|
| 369 |
|
| 370 |
@app.get("/health")
|
| 371 |
def health():
|
| 372 |
+
return {"status": "healthy"}
|
| 373 |
|
| 374 |
|
| 375 |
@app.get("/")
|