model-compose 0.2.8__tar.gz → 0.2.9__tar.gz

{model_compose-0.2.8/src/model_compose.egg-info → model_compose-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-compose
-Version: 0.2.8
+Version: 0.2.9
 Summary: model-compose: Declarative AI Model and Workflow Orchestrator
 Author-email: Hanyeol Cho <hanyeol.cho@gmail.com>
 License-Expression: MIT

{model_compose-0.2.8 → model_compose-0.2.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "model-compose"
-version = "0.2.8"
+version = "0.2.9"
 description = "model-compose: Declarative AI Model and Workflow Orchestrator"
 authors = [
     { name = "Hanyeol Cho", email = "hanyeol.cho@gmail.com" }

{model_compose-0.2.8 → model_compose-0.2.9}/src/mindor/core/controller/webui/webui.py

@@ -6,9 +6,8 @@ from mindor.core.controller.runner import ControllerRunner
 from mindor.core.workflow.schema import create_workflow_schema
 from mindor.core.services import AsyncService
 from .gradio import GradioWebUIBuilder
-
-from fastapi import FastAPI
 from gradio import mount_gradio_app
+from fastapi import FastAPI
 import uvicorn, gradio
 
 class ControllerWebUI(AsyncService):

{model_compose-0.2.8 → model_compose-0.2.9}/src/mindor/core/logger/services/console.py

@@ -1,6 +1,7 @@
 from typing import Type, Union, Literal, Optional, Dict, List, Tuple, Set, Annotated, Callable, Iterator, Any
 from mindor.dsl.schema.logger import ConsoleLoggerConfig
 from ..base import LoggerService, LoggerType, LoggingLevel, register_logger
+from uvicorn.logging import ColourizedFormatter
 import logging
 
 _level_map = {
@@ -17,7 +18,7 @@ class ConsoleLogger(LoggerService):
         super().__init__(id, config, daemon)
 
         self.logger: logging.Logger = logging.getLogger(id)
-        self.formatter: logging.Formatter = logging.Formatter("%(levelname)s: %(message)s")
+        self.formatter: logging.Formatter = ColourizedFormatter("%(levelprefix)s %(message)s")
         self.handler: logging.StreamHandler = None
 
         self._configure_logger()

{model_compose-0.2.8 → model_compose-0.2.9}/src/mindor/core/utils/time.py

@@ -2,6 +2,16 @@ from typing import Union, Optional
 from datetime import timedelta, datetime
 import zoneinfo
 
+class TimeTracker:
+    def __init__(self):
+        self._start = datetime.now()
+
+    def elapsed(self) -> float:
+        return (datetime.now() - self._start).total_seconds()
+
+    def reset(self) -> None:
+        self._start = datetime.now()
+
 def parse_duration(value: Union[str, float, int]) -> timedelta:
     if isinstance(value, (float, int)):
         return timedelta(seconds=value)

{model_compose-0.2.8 → model_compose-0.2.9}/src/mindor/core/workflow/job/impl/action.py

@@ -2,6 +2,7 @@ from typing import Type, Union, Literal, Optional, Dict, List, Tuple, Set, Annot
 from mindor.dsl.schema.job import ActionJobConfig, JobType
 from mindor.dsl.schema.component import ComponentConfig
 from mindor.core.component import ComponentService, ComponentGlobalConfigs, ComponentResolver, create_component
+from mindor.core.utils.time import TimeTracker
 from mindor.core.logger import logging
 from ..base import Job, JobType, WorkflowContext, register_job
 from datetime import datetime
@@ -24,14 +25,13 @@ class ActionJob(Job):
         async def _run_once():
             run_id: str = ulid.ulid()
 
-            started_at = datetime.now()
+            job_time_tracker = TimeTracker()
             logging.debug("[task-%s] Action 'run-%s' started for job '%s'", context.task_id, run_id, self.id)
 
             output = await component.run(await context.render_variable(self.config.action), run_id, input)
             context.register_source("output", output)
 
-            elapsed = (datetime.now() - started_at).total_seconds()
-            logging.debug("[task-%s] Action 'run-%s' completed in %.2f seconds.", context.task_id, run_id, elapsed)
+            logging.debug("[task-%s] Action 'run-%s' completed in %.2f seconds.", context.task_id, run_id, job_time_tracker.elapsed())
 
             output = (await context.render_variable(self.config.output, ignore_files=True)) if self.config.output else output
             outputs.append(output)

{model_compose-0.2.8 → model_compose-0.2.9}/src/mindor/core/workflow/job/impl/delay.py

@@ -1,7 +1,7 @@
 from typing import Type, Union, Literal, Optional, Dict, List, Tuple, Set, Annotated, Callable, Any
 from mindor.dsl.schema.job import DelayJobConfig, DelayJobMode
 from mindor.dsl.schema.component import ComponentConfig
-from mindor.core.utils.time import parse_duration, parse_datetime
+from mindor.core.utils.time import parse_duration, parse_datetime, TimeTracker
 from mindor.core.logger import logging
 from ..base import Job, JobType, WorkflowContext, register_job
 from datetime import datetime, timedelta
@@ -25,14 +25,13 @@ class DelayJob(Job):
         duration = parse_duration((await context.render_variable(self.config.duration)) or 0.0)
         duration = duration.total_seconds()
         
-        started_at = datetime.now()
+        job_time_tracker = TimeTracker()
         logging.debug("[task-%s] Delay started for time interval: %d seconds.", context.task_id, int(duration))
 
         if duration > 0.0:
             await asyncio.sleep(duration)
 
-        elapsed = (datetime.now() - started_at).total_seconds()
-        logging.debug("[task-%s] Delay completed in %.2f seconds.", context.task_id, elapsed)
+        logging.debug("[task-%s] Delay completed in %.2f seconds.", context.task_id, job_time_tracker.elapsed())
 
         return None
 
@@ -43,13 +42,12 @@ class DelayJob(Job):
         now = datetime.now(tz=time.tzinfo)
         duration = max((time - now).total_seconds(), 0.0)
 
-        started_at = datetime.now()
+        job_time_tracker = TimeTracker()
         logging.debug("[task-%s] Delay started, waiting until %s", context.task_id, time)
 
         if duration > 0.0:
             await asyncio.sleep(duration)
 
-        elapsed = (datetime.now() - started_at).total_seconds()
-        logging.debug("[task-%s] Delay completed in %.2f seconds.", context.task_id, elapsed)
+        logging.debug("[task-%s] Delay completed in %.2f seconds.", context.task_id, job_time_tracker.elapsed())
 
         return None

{model_compose-0.2.8 → model_compose-0.2.9}/src/mindor/core/workflow/workflow.py

@@ -2,6 +2,7 @@ from typing import Type, Union, Literal, Optional, Dict, List, Tuple, Set, Annot
 from mindor.dsl.schema.workflow import WorkflowConfig, JobConfig
 from mindor.dsl.schema.component import ComponentConfig
 from mindor.core.component import ComponentGlobalConfigs
+from mindor.core.utils.time import TimeTracker
 from mindor.core.logger import logging
 from .context import WorkflowContext
 from .job import Job, create_job
@@ -86,9 +87,10 @@ class WorkflowRunner:
         running_job_ids: Set[str] = set()
         completed_job_ids: Set[str] = set()
         scheduled_job_tasks: Dict[str, asyncio.Task] = {}
+        job_time_trackers: Dict[str, TimeTracker] = {}
         output: Any = None
 
-        started_at = datetime.now()
+        workflow_time_tracker = TimeTracker()
         logging.info("[task-%s] Workflow '%s' started.", context.task_id, self.id)
 
         while pending_jobs:
@@ -96,10 +98,12 @@ class WorkflowRunner:
 
             for job in runnable_jobs:
                 if job.id not in scheduled_job_tasks:
-                    logging.debug("[task-%s] Scheduling job: '%s'", context.task_id, job.id)
                     scheduled_job_tasks[job.id] = asyncio.create_task(job.run(context))
                     running_job_ids.add(job.id)
 
+                    job_time_trackers[job.id] = TimeTracker()
+                    logging.info("[task-%s] Job '%s' started.", context.task_id, job.id)
+
             if not scheduled_job_tasks:
                 raise RuntimeError("No runnable jobs but pending jobs remain.")
 
@@ -111,7 +115,7 @@ class WorkflowRunner:
                 completed_job_output = await completed_job_task
                 context.complete_job(completed_job_id, completed_job_output)
 
-                logging.info("[task-%s] Job '%s' completed.", context.task_id, completed_job_id)
+                logging.info("[task-%s] Job '%s' completed in %.2f seconds.", context.task_id, completed_job_id, job_time_trackers[completed_job_id].elapsed())
 
                 if self._is_terminal_job(completed_job_id):
                     if isinstance(output, dict) and isinstance(completed_job_output, dict):
@@ -124,8 +128,7 @@ class WorkflowRunner:
                 del pending_jobs[completed_job_id]
                 del scheduled_job_tasks[completed_job_id]
 
-        elapsed = (datetime.now() - started_at).total_seconds()
-        logging.info("[task-%s] Workflow '%s' completed in %.2f seconds.", context.task_id, self.id, elapsed)
+        logging.info("[task-%s] Workflow '%s' completed in %.2f seconds.", context.task_id, self.id, workflow_time_tracker.elapsed())
 
         return output
 

{model_compose-0.2.8 → model_compose-0.2.9/src/model_compose.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-compose
-Version: 0.2.8
+Version: 0.2.9
 Summary: model-compose: Declarative AI Model and Workflow Orchestrator
 Author-email: Hanyeol Cho <hanyeol.cho@gmail.com>
 License-Expression: MIT