add type check for run_async_server

sotopia-lab · Dec 4, 2024 · af63410 · af63410
1 parent 3e5beb6
commit af63410
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/sotopia/server.py b/sotopia/server.py
@@ -339,7 +339,13 @@ def get_agent_class(
         else [await i for i in episode_futures]
     )
 
-    return batch_results
+    if len(batch_results) > 0:
+        first_result = batch_results[0]
+        assert isinstance(
+            first_result, list
+        ), f"Unexpected result type: {type(first_result)}"
+
+    return batch_results  # type: ignore
 
 
 async def arun_one_script(