Upload folder using huggingface_hub

server/app.py

@@ -1,41 +1,19 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import HTTPException
 from pydantic import BaseModel
-from typing import Dict, Any, List
-import asyncio
+from typing import Dict
 
+from openenv.core import create_app
+
+from models import Observation, Action
 from env.environment import SQLEnv
-from env.models import Observation, Action, Reward
 from env.tasks import TASKS
 
-app = FastAPI(title="SQL Query Optimizer OpenEnv")
-env = SQLEnv()
-
-class ResetRequest(BaseModel):
-    task_id: int
-
-@app.post("/reset", response_model=Observation)
-async def reset(req: ResetRequest):
-    try:
-        return env.reset(req.task_id)
-    except ValueError as e:
-        raise HTTPException(status_code=400, detail=str(e))
-
-@app.post("/step")
-async def step(action: Action):
-    try:
-        obs, reward, done, info = env.step(action)
-        return {
-            "observation": obs.model_dump(),
-            "reward": reward.model_dump(),
-            "done": done,
-            "info": info
-        }
-    except RuntimeError as e:
-        raise HTTPException(status_code=400, detail=str(e))
-
-@app.get("/state")
-async def state():
-    return env.state()
+app = create_app(
+    env=SQLEnv,
+    action_cls=Action,
+    observation_cls=Observation,
+    env_name="sql-query-optimizer"
+)
 
 @app.get("/tasks")
 async def get_tasks():
@@ -46,12 +24,6 @@ async def get_tasks():
         "action_schema": action_schema
     }
 
-@app.get("/grader")
-async def grader():
-    if not env.task:
-        raise HTTPException(status_code=400, detail="Environment not initialized.")
-    return {"grader_score": env.final_grader_score}
-
 class BaselineResponse(BaseModel):
     scores: Dict[int, float]
 
@@ -63,10 +35,3 @@ async def run_baseline():
         return BaselineResponse(scores=scores)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
-
-def main(host: str = "0.0.0.0", port: int = 8000):
-    import uvicorn
-    uvicorn.run(app, host=host, port=port)
-
-if __name__ == '__main__':
-    main()