Spaces:
Sleeping
Sleeping
File size: 1,043 Bytes
656cdcb 79c94d2 5eb188c 485a57e 0d9a3e2 79c94d2 0d9a3e2 656cdcb 79c94d2 99cd67c 656cdcb 79c94d2 99cd67c 0d9a3e2 79c94d2 0d9a3e2 1f97a55 79c94d2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | from fastapi import FastAPI, Body
from pydantic import BaseModel
from typing import Any, Dict
app = FastAPI()
class Observation(BaseModel):
symptoms: str
severity: str
step_count: int
class StepResponse(BaseModel):
observation: Observation
reward: float
done: bool
info: Dict[str, Any]
@app.post("/reset")
async def reset(payload: dict = Body(default={})):
return StepResponse(
observation=Observation(
symptoms="Patient reports fever and sore throat",
severity="unknown",
step_count=0
),
reward=0.0,
done=False,
info={}
)
@app.post("/step")
async def step(payload: dict = Body(default={})):
return StepResponse(
observation=Observation(
symptoms="updated symptoms",
severity="low",
step_count=1
),
reward=0.3,
done=False,
info={}
)
@app.get("/state")
async def state():
return {
"status": "active",
"task": "easy"
}
|