add timeout for sse_await

src/fg.py

@@ -69,9 +69,9 @@ class EditorAPIContext:
                 future = self.sse_futures[jdata["state"]]
                 future.set_result(jdata)
 
-    async def sse_await(self, state_id: str) -> None:
+    async def sse_await(self, state_id: str, timeout: float = 60.0) -> None:
         future = self.sse_futures[state_id]
-        jdata = await future
+        jdata = await asyncio.wait_for(future, timeout=timeout)
         if jdata["status"] != "ok":
             print("ERROR", jdata)
         assert jdata["status"] == "ok"