diff options
Diffstat (limited to 'server/continuedev/__init__.py')
-rw-r--r-- | server/continuedev/__init__.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/server/continuedev/__init__.py b/server/continuedev/__init__.py new file mode 100644 index 00000000..1b4776a8 --- /dev/null +++ b/server/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)) |