Commit ·
9c6cff5
1
Parent(s): e7eb0fa
Final multi-mode OpenEnv fix
Browse files- pyproject.toml +1 -1
- server/app.py +16 -0
pyproject.toml
CHANGED
|
@@ -19,7 +19,7 @@ dependencies = [
|
|
| 19 |
]
|
| 20 |
|
| 21 |
[project.scripts]
|
| 22 |
-
server = "server.app:
|
| 23 |
|
| 24 |
[tool.setuptools.packages.find]
|
| 25 |
where = ["."]
|
|
|
|
| 19 |
]
|
| 20 |
|
| 21 |
[project.scripts]
|
| 22 |
+
server = "server.app:main"
|
| 23 |
|
| 24 |
[tool.setuptools.packages.find]
|
| 25 |
where = ["."]
|
server/app.py
CHANGED
|
@@ -19,6 +19,7 @@ from fastapi import FastAPI, HTTPException
|
|
| 19 |
from fastapi.middleware.cors import CORSMiddleware
|
| 20 |
from fastapi.responses import HTMLResponse, JSONResponse
|
| 21 |
from openai import OpenAI
|
|
|
|
| 22 |
|
| 23 |
# Ensure project root is importable when executed in Spaces/Docker.
|
| 24 |
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
@@ -394,3 +395,18 @@ def optimize(req: OptimizeRequest) -> OptimizeResponse:
|
|
| 394 |
task_score=task_score,
|
| 395 |
)
|
| 396 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
from fastapi.middleware.cors import CORSMiddleware
|
| 20 |
from fastapi.responses import HTMLResponse, JSONResponse
|
| 21 |
from openai import OpenAI
|
| 22 |
+
import uvicorn
|
| 23 |
|
| 24 |
# Ensure project root is importable when executed in Spaces/Docker.
|
| 25 |
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
| 395 |
task_score=task_score,
|
| 396 |
)
|
| 397 |
|
| 398 |
+
|
| 399 |
+
def main() -> None:
|
| 400 |
+
"""
|
| 401 |
+
Entry point for OpenEnv multi-mode deployment.
|
| 402 |
+
|
| 403 |
+
- API mode: OpenEnv imports `server.app:app`
|
| 404 |
+
- CLI mode: OpenEnv / HF can run `server` script -> `server.app:main`
|
| 405 |
+
"""
|
| 406 |
+
port = int(os.getenv("PORT", "7860"))
|
| 407 |
+
uvicorn.run("server.app:app", host="0.0.0.0", port=port)
|
| 408 |
+
|
| 409 |
+
|
| 410 |
+
if __name__ == "__main__":
|
| 411 |
+
main()
|
| 412 |
+
|