fix: Update API responses to return JSON format and remove deprecated file references

README.md

@@ -88,7 +88,6 @@ SmartContractEnv/
 ├── .gitignore
 ├── demo.py
 ├── Dockerfile
-├── Docs.md
 ├── eval.py
 ├── inference.py
 ├── LICENSE.txt
@@ -308,8 +307,8 @@ The `inference.py` script runs an OpenAI-compatible model against all three task
 
 ```bash
 export OPENAI_API_KEY=your_key
-export API_BASE_URL=https://api.openai.com/v1   # or your custom endpoint
-export MODEL_NAME=gpt-4o-mini
+export API_BASE_URL=your custom endpoint
+export MODEL_NAME=your custom model
 
 python inference.py
 ```

inference.py

@@ -44,7 +44,7 @@ from dotenv import dotenv_values
 config = dotenv_values(".env")
 API_BASE_URL = config.get("API_BASE_URL", "https://api.openai.com/v1")
 MODEL_NAME   = config.get("MODEL_NAME",   "gpt-4o")
-HF_TOKEN     = config.get("HF_TOKEN",     "")
+HF_TOKEN     = config.get("HF_TOKEN")
 
 if not HF_TOKEN:
     print("[WARN] HF_TOKEN not set — API calls may fail.", file=sys.stderr)

server/app.py

@@ -166,7 +166,7 @@ def reset(
     env = _create_env(body.task_id)
     _sessions[session_id] = env
     result = env.reset(seed=body.seed)
-    return result.model_dump()
+    return JSONResponse(content=result.model_dump(), status_code=200)
 
 
 @app.post("/step")
@@ -193,7 +193,7 @@ def step(
         result = env.step(action)
     except RuntimeError as e:
         raise HTTPException(status_code=409, detail=str(e))
-    return result.model_dump()
+    return JSONResponse(content=result.model_dump(), status_code=200)
 
 
 @app.get("/state")
@@ -205,14 +205,14 @@ def state(session_id: str = Query(default=DEFAULT_SESSION)):
             status_code=400,
             detail=f"No active session '{session_id}'. Call /reset first.",
         )
-    return env.state().model_dump()
+    return JSONResponse(content=env.state().model_dump(), status_code=200)
 
 
 @app.get("/action_space")
 def action_space(task_id: str = "task1_vuln_detection"):
     """Describe the action space for a task."""
     if task_id == "task1_vuln_detection":
-        return {
+        return JSONResponse(content={
             "task_id": task_id,
             "actions": [
                 {"type": "list_functions",       "params": {},                                                    "reward": -0.05,                              "description": "List all function names"},
@@ -223,9 +223,9 @@ def action_space(task_id: str = "task1_vuln_detection"):
                 {"type": "get_call_graph",        "params": {},                                                   "reward": -0.08,                              "description": "Get function call graph"},
                 {"type": "submit",               "params": {"function_name": "str", "vulnerability_type": "str"},"reward": "+5.0 / +1.0 / -1.5",              "description": "Submit answer. Ends episode."},
             ],
-        }
+        }, status_code=200)
     if task_id == "task2_property_discovery":
-        return {
+        return JSONResponse(content={
             "task_id": task_id,
             "actions": [
                 {"type": "get_function_code",     "params": {},                         "reward": -0.06, "description": "Read full source of the target function"},
@@ -236,9 +236,9 @@ def action_space(task_id: str = "task1_vuln_detection"):
                 {"type": "get_similar_rule",      "params": {},                         "reward": -0.20, "description": "Get a similar property from another contract"},
                 {"type": "submit_property",       "params": {"property": "string"},     "reward": "0.0–5.0", "description": "Submit property. ONE attempt. Ends episode."},
             ],
-        }
+        }, status_code=200)
     if task_id == "task3_rule_checker":
-        return {
+        return JSONResponse(content={
             "task_id": task_id,
             "actions": [
                 {"type": "list_functions",          "params": {},                               "reward": -0.05, "description": "List all function names"},
@@ -249,14 +249,13 @@ def action_space(task_id: str = "task1_vuln_detection"):
                 {"type": "get_property_specification", "params": {},                            "reward": -0.03, "description": "Get formal pre/post-condition for the property"},
                 {"type": "submit_function",         "params": {"function_name": "string"},      "reward": "+5.0 / +1.5 / -1.5", "description": "Submit answer. ONE attempt. Ends episode."},
             ],
-        }
-    return {"error": f"No action space defined for task '{task_id}'"}
-
+        }, status_code=200)
+    raise HTTPException(status_code=400, detail=f"Unknown task_id '{task_id}'.")
 
 @app.get("/observation_space")
 def observation_space():
     """Describe the observation space (same for all tasks)."""
-    return {
+    return JSONResponse(content={
         "type": "object",
         "fields": {
             "task_id":              "string – active task identifier",
@@ -270,7 +269,7 @@ def observation_space():
             "done":                 "bool – True when episode is over",
             "extra":                "object – task-specific hints (target_function, hint, etc.)",
         },
-    }
+    }, status_code=200)
 
 
 # ─────────────────────────────────────────────────────────────────────────────