tool_use_env/server/app.py

from openenv.core.env_server.http_server import create_app

from tool_use_env.models import ToolUseAction, ToolUseObservation
from tool_use_env.server.tool_use_env_environment import ToolUseEnvironment


app = create_app(
    ToolUseEnvironment,
    ToolUseAction,
    ToolUseObservation,
    env_name="tool_use_env",
    max_concurrent_envs=1,
)


import uvicorn

def main(host: str = "0.0.0.0", port: int = 8000):
    uvicorn.run("tool_use_env.server.app:app", host=host, port=port)
    
@app.get("/")
def root():
    return {"status": "running"}

if __name__ == "__main__":
    main()