refactor: :recycle: from continuedev import run

2 files changed, 19 insertions, 15 deletions

diff --git a/continuedev/src/continuedev/__init__.py b/continuedev/src/continuedev/__init__.py
index e69de29b..1b4776a8 100644
--- a/continuedev/src/continuedev/__init__.py
+++ b/continuedev/src/continuedev/__init__.py
@@ -0,0 +1,19 @@
+import asyncio
+from typing import Union
+
+from .core.config import ContinueConfig
+from .core.main import Step
+from .headless import start_headless_session
+
+
+def run(step_or_config: Union[Step, ContinueConfig]):
+    if isinstance(step_or_config, ContinueConfig):
+        config = step_or_config
+    else:
+        config = ContinueConfig()
+        config.steps_on_startup = [step_or_config]
+
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(start_headless_session(config=config))
+    tasks = asyncio.all_tasks(loop)
+    loop.run_until_complete(asyncio.gather(*tasks))
diff --git a/continuedev/src/continuedev/headless/__init__.py b/continuedev/src/continuedev/headless/__init__.py
index fb2c9768..2ecdcce6 100644
--- a/continuedev/src/continuedev/headless/__init__.py
+++ b/continuedev/src/continuedev/headless/__init__.py
@@ -1,10 +1,8 @@
-import asyncio
 from typing import Optional, Union
 
 import typer
 
 from ..core.config import ContinueConfig
-from ..core.main import Step
 from ..server.session_manager import Session, session_manager
 from .headless_ide import LocalIdeProtocol
 
@@ -20,16 +18,3 @@ async def start_headless_session(
 
     ide = LocalIdeProtocol()
     return await session_manager.new_session(ide, config=config)
-
-
-def run(step_or_config: Union[Step, ContinueConfig]):
-    if isinstance(step_or_config, ContinueConfig):
-        config = step_or_config
-    else:
-        config = ContinueConfig()
-        config.steps_on_startup = [step_or_config]
-
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(start_headless_session(config=config))
-    tasks = asyncio.all_tasks(loop)
-    loop.run_until_complete(asyncio.gather(*tasks))