From bbb06ae01df69193d5d061b883e00ba29524b397 Mon Sep 17 00:00:00 2001
From: Sarmad Qadri <sarmad@lastmileai.dev>
Date: Thu, 4 Jan 2024 16:02:51 -0500
Subject: [PATCH] Temp server changes

---
 python/src/aiconfig/editor/server/server_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/src/aiconfig/editor/server/server_utils.py b/python/src/aiconfig/editor/server/server_utils.py
index f48704a27..b8502ff7e 100644
--- 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])