From 56d248785cf79e34fff0bf0fc3aaf341c2ec1634 Mon Sep 17 00:00:00 2001 From: XuhuiZhou Date: Tue, 7 Jan 2025 17:16:31 -0500 Subject: [PATCH] run non streaming simulation --- sotopia/api/fastapi_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sotopia/api/fastapi_server.py b/sotopia/api/fastapi_server.py index 57ffa0f1..f1a3220c 100644 --- 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,