File size: 424 Bytes
1b64cba
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from fastapi import FastAPI
from app.env import WorkflowEnv
from tasks.easy import create_easy_task

app = FastAPI()


@app.post("/reset")
def reset():
    state, _ = create_easy_task()
    env = WorkflowEnv(state)
    obs = env.reset()

    return {"status": "ok"}

@app.get("/")
def root():
    return {"message": "Workflow Env is running"}
    
@app.get("/")
def home():
    return {"message": "Workflow Env API running"}