File size: 1,020 Bytes
7547b3b
7d1e9a6
7547b3b
7d1e9a6
 
 
 
 
 
 
 
7547b3b
 
 
 
7d1e9a6
 
 
 
 
 
 
7547b3b
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
"""

app.py — Multi-task OpenEnv server with openenv-core bug fix

"""
import os, sys
sys.path.insert(0, os.path.dirname(__file__))

# ── Monkey-patch openenv-core bug (HTTP_422_UNPROCESSABLE_CONTENT missing) ──
import starlette.status as _status
if not hasattr(_status, "HTTP_422_UNPROCESSABLE_CONTENT"):
    _status.HTTP_422_UNPROCESSABLE_CONTENT = 422

from openenv.core import create_fastapi_app
from environment import PromptSensitivityEnvironment, SensitivityAction, SensitivityObservation
import uvicorn

TASK = os.environ.get("TASK", "1")

if TASK == "2":
    from environment2 import HallucinationDetectionEnvironment, HallucinationAction, HallucinationObservation
    app = create_fastapi_app(HallucinationDetectionEnvironment, HallucinationAction, HallucinationObservation)
else:
    app = create_fastapi_app(PromptSensitivityEnvironment, SensitivityAction, SensitivityObservation)

if __name__ == "__main__":
    uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=False)