run non streaming simulation

sotopia-lab · Jan 7, 2025 · 56d2487 · 56d2487
1 parent 6b2db2a
commit 56d2487
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sotopia/api/fastapi_server.py b/sotopia/api/fastapi_server.py
@@ -229,7 +229,7 @@ async def send_error(
         )
 
 
-async def run_simulation(
+async def nonstreaming_simulation(
     episode_pk: str,
     simulation_request: SimulationRequest,
     simulation_status: NonStreamingSimulationStatus,
@@ -548,7 +548,7 @@ def simulate(simulation_request: SimulationRequest) -> Response:
                 simulation_status.save()
                 queue = rq.Queue("default", connection=get_redis_connection())
                 queue.enqueue(
-                    run_simulation,
+                    nonstreaming_simulation,
                     episode_pk=episode_pk,
                     simulation_request=simulation_request,
                     simulation_status=simulation_status,