Commit ·
900e1f4
1
Parent(s): a92b800
Updated structure and fixed module import issue
Browse files- README.md +1 -0
- openenv_interface.py +6 -0
- server.py +4 -2
README.md
CHANGED
|
@@ -3,6 +3,7 @@ title: ACRE - Autonomous Code Refactoring Environment
|
|
| 3 |
colorFrom: blue
|
| 4 |
colorTo: green
|
| 5 |
sdk: docker
|
|
|
|
| 6 |
app_port: 7860
|
| 7 |
pinned: false
|
| 8 |
license: mit
|
|
|
|
| 3 |
colorFrom: blue
|
| 4 |
colorTo: green
|
| 5 |
sdk: docker
|
| 6 |
+
app_file: server.py
|
| 7 |
app_port: 7860
|
| 8 |
pinned: false
|
| 9 |
license: mit
|
openenv_interface.py
CHANGED
|
@@ -1,7 +1,13 @@
|
|
| 1 |
from __future__ import annotations
|
| 2 |
|
|
|
|
|
|
|
| 3 |
from typing import Any, Dict, Optional, Tuple
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
try:
|
| 6 |
from openenv.env import Env as OpenEnvBase
|
| 7 |
except Exception: # pragma: no cover
|
|
|
|
| 1 |
from __future__ import annotations
|
| 2 |
|
| 3 |
+
import os
|
| 4 |
+
import sys
|
| 5 |
from typing import Any, Dict, Optional, Tuple
|
| 6 |
|
| 7 |
+
PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
|
| 8 |
+
if PROJECT_ROOT not in sys.path:
|
| 9 |
+
sys.path.insert(0, PROJECT_ROOT)
|
| 10 |
+
|
| 11 |
try:
|
| 12 |
from openenv.env import Env as OpenEnvBase
|
| 13 |
except Exception: # pragma: no cover
|
server.py
CHANGED
|
@@ -12,10 +12,10 @@ Endpoints (all required by OpenEnv spec):
|
|
| 12 |
from __future__ import annotations
|
| 13 |
|
| 14 |
import difflib
|
| 15 |
-
import sys
|
| 16 |
import os
|
| 17 |
import re
|
| 18 |
import json
|
|
|
|
| 19 |
from typing import Optional
|
| 20 |
|
| 21 |
import uvicorn
|
|
@@ -25,7 +25,9 @@ from fastapi.middleware.cors import CORSMiddleware
|
|
| 25 |
from fastapi.responses import HTMLResponse
|
| 26 |
from openai import OpenAI
|
| 27 |
|
| 28 |
-
|
|
|
|
|
|
|
| 29 |
|
| 30 |
try:
|
| 31 |
from stable_baselines3 import PPO
|
|
|
|
| 12 |
from __future__ import annotations
|
| 13 |
|
| 14 |
import difflib
|
|
|
|
| 15 |
import os
|
| 16 |
import re
|
| 17 |
import json
|
| 18 |
+
import sys
|
| 19 |
from typing import Optional
|
| 20 |
|
| 21 |
import uvicorn
|
|
|
|
| 25 |
from fastapi.responses import HTMLResponse
|
| 26 |
from openai import OpenAI
|
| 27 |
|
| 28 |
+
PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
|
| 29 |
+
if PROJECT_ROOT not in sys.path:
|
| 30 |
+
sys.path.insert(0, PROJECT_ROOT)
|
| 31 |
|
| 32 |
try:
|
| 33 |
from stable_baselines3 import PPO
|