File size: 433 Bytes
49ded5c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from fastapi import FastAPI
from pydantic import BaseModel
from guess_env.server.guess_environment import GuessEnvironment
from guess_env.models import GuessAction
app = FastAPI()
env = GuessEnvironment(max_guesses=5)
class StepRequest(BaseModel):
action: GuessAction
@app.post("/reset")
def reset():
return env.reset()
@app.post("/step")
def step(request: StepRequest):
obs = env.step(request.action)
return obs |