Temp server changes

lastmile-ai · Jan 4, 2024 · bbb06ae · bbb06ae
1 parent e926e4e
commit bbb06ae
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/src/aiconfig/editor/server/server_utils.py b/python/src/aiconfig/editor/server/server_utils.py
@@ -1,3 +1,4 @@
+from asyncio import Task, create_task
 import importlib
 import importlib.util
 import logging
@@ -75,6 +76,7 @@ def convert_to_mode(cls, value: Any) -> ServerMode:  # pylint: disable=no-self-a
 @dataclass
 class ServerState:
     aiconfig: AIConfigRuntime | None = None
+    tasks: dict[str, Task[Any]] = {}
 
 
 FlaskResponse = NewType("FlaskResponse", tuple[core_utils.JSONObject, int])