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="support_ops_env",
    max_concurrent_envs=4,
)


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()