fix: accept both 'action' and 'response' field names in /step endpoint

api.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.openapi.docs import get_swagger_ui_html
-from pydantic import BaseModel
+from pydantic import BaseModel, model_validator
 from typing import Optional
 
 from environment import CustomerSupportEnv, STEP_ORDER
@@ -33,7 +33,15 @@ class ResetRequest(BaseModel):
 
 
 class StepRequest(BaseModel):
-    action: str
+    action: Optional[str] = None    # accepts {"action": "..."}
+    response: Optional[str] = None  # accepts {"response": "..."}
+
+    @model_validator(mode="after")
+    def resolve_action(self):
+        self.action = self.action or self.response
+        if not self.action:
+            raise ValueError("Provide either 'action' or 'response' field with the agent reply.")
+        return self
 
 
 def _build_observation():