krons 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

krons/core/base/__init__.py

@@ -26,8 +26,6 @@ _LAZY_IMPORTS: dict[str, tuple[str, str]] = {
     "Edge": ("krons.core.base.graph", "Edge"),
     "EdgeCondition": ("krons.core.base.graph", "EdgeCondition"),
     "Graph": ("krons.core.base.graph", "Graph"),
-    # log
-    "DataLoggerConfig": ("krons.core.base.log", "DataLoggerConfig"),
     # node
     "NODE_REGISTRY": ("krons.core.base.node", "NODE_REGISTRY"),
     "PERSISTABLE_NODE_REGISTRY": ("krons.core.base.node", "PERSISTABLE_NODE_REGISTRY"),
@@ -79,7 +77,6 @@ if TYPE_CHECKING:
     from .eventbus import EventBus, Handler
     from .flow import Flow
     from .graph import Edge, EdgeCondition, Graph
-    from .log import DataLoggerConfig
     from .node import (
         NODE_REGISTRY,
         PERSISTABLE_NODE_REGISTRY,
@@ -100,7 +97,6 @@ __all__ = [
     "PERSISTABLE_NODE_REGISTRY",
     # classes
     "Broadcaster",
-    "DataLoggerConfig",
     "Edge",
     "EdgeCondition",
     "Element",

krons/core/base/flow.py

@@ -321,6 +321,13 @@ class Flow(Element, Generic[E, P]):
 
         return self.items.remove(uid)
 
+    @synchronized
+    def clear(self) -> None:
+        """Clear all items and progressions."""
+        self.items.clear()
+        self.progressions.clear()
+        self._progression_names.clear()
+
     def __repr__(self) -> str:
         name_str = f", name='{self.name}'" if self.name else ""
         return f"Flow(items={len(self.items)}, progressions={len(self.progressions)}{name_str})"

krons/session/session.py

@@ -14,11 +14,9 @@ import contextlib
 from collections.abc import AsyncGenerator, Iterable
 from pathlib import Path
 from typing import Any, Literal
-
-from krons.core.base.log import DataLoggerConfig
 from uuid import UUID
 
-from pydantic import Field, PrivateAttr, model_validator
+from pydantic import Field, PrivateAttr, field_serializer, model_validator
 
 from krons.core import Element, Flow, Pile, Progression
 from krons.core.types import HashableModel, Unset, UnsetType, not_sentinel
@@ -63,25 +61,41 @@ class SessionConfig(HashableModel):
     default_gen_model: str | None = None
     default_parse_model: str | None = None
     auto_create_default_branch: bool = True
-    log_config: DataLoggerConfig | None = Field(
+
+    # Logging configuration
+    log_persist_dir: str | Path | None = Field(
         default=None,
-        description="DataLoggerConfig for auto-logging (None = disabled)",
+        description="Directory for session dumps. None disables logging.",
+    )
+    log_auto_save_on_exit: bool = Field(
+        default=True,
+        description="Register atexit handler on Session creation.",
     )
 
+    @property
+    def logging_enabled(self) -> bool:
+        """True if logging is configured (log_persist_dir is set)."""
+        return self.log_persist_dir is not None
+
 
 class Session(Element):
     user: str | None = None
     communications: Flow[Message, Branch] = Field(
         default_factory=lambda: Flow(item_type=Message)
     )
-    resources: ResourceRegistry = Field(default_factory=ResourceRegistry)
-    operations: OperationRegistry = Field(default_factory=OperationRegistry)
+    resources: ResourceRegistry = Field(default_factory=ResourceRegistry, exclude=True)
+    operations: OperationRegistry = Field(default_factory=OperationRegistry, exclude=True)
     config: SessionConfig = Field(default_factory=SessionConfig)
     default_branch_id: UUID | None = None
 
     _registered_atexit: bool = PrivateAttr(default=False)
     _dump_count: int = PrivateAttr(default=0)
 
+    @field_serializer("communications")
+    def _serialize_communications(self, flow: Flow) -> dict:
+        """Use Flow's custom to_dict for proper nested serialization."""
+        return flow.to_dict(mode="json")
+
     @model_validator(mode="after")
     def _validate_default_branch(self) -> Session:
         """Auto-create default branch and register built-in operations."""
@@ -96,8 +110,8 @@ class Session(Element):
 
         # Register atexit handler if configured
         if (
-            self.config.log_config is not None
-            and self.config.log_config.auto_save_on_exit
+            self.config.logging_enabled
+            and self.config.log_auto_save_on_exit
             and not self._registered_atexit
         ):
             atexit.register(self._save_at_exit)
@@ -406,96 +420,87 @@ class Session(Element):
         async for result in handler(params, ctx):
             yield result
 
-    def dump_messages(self, clear: bool = False) -> Path | None:
-        """Sync dump all session messages to file.
+    def dump(self, clear: bool = False) -> Path | None:
+        """Sync dump entire session state for replay.
+
+        Serializes session (messages, branches, config) to JSON.
+        Resources and operations are excluded (re-register on restore).
+        To restore: Session.from_dict(data), then re-register resources.
 
         Args:
-            clear: Clear messages after dump (default False).
+            clear: Clear communications after dump (default False).
 
         Returns:
-            Path to written file, or None if no log_config or no messages.
+            Path to session file, or None if logging disabled or empty.
         """
-        from krons.utils import create_path, json_dumpb, json_lines_iter
+        from krons.utils import create_path, json_dumpb
         from krons.utils.concurrency import run_async
 
-        if self.config.log_config is None or len(self.messages) == 0:
+        if not self.config.logging_enabled or len(self.messages) == 0:
             return None
 
-        cfg = self.config.log_config
         self._dump_count += 1
 
         filepath = run_async(
             create_path(
-                directory=cfg.persist_dir,
+                directory=self.config.log_persist_dir,
                 filename=f"session_{str(self.id)[:8]}_{self._dump_count}",
-                extension=cfg.extension,
+                extension=".json",
                 timestamp=True,
                 file_exist_ok=True,
             )
         )
 
-        items = [msg.to_dict(mode="json") for msg in self.messages]
-
+        data = json_dumpb(self.to_dict(mode="json"), safe_fallback=True)
         std_path = Path(filepath)
-        if cfg.extension == ".jsonl":
-            with std_path.open("wb") as f:
-                for chunk in json_lines_iter(items, safe_fallback=True):
-                    f.write(chunk)
-        else:
-            data = json_dumpb(items, safe_fallback=True)
-            std_path.write_bytes(data)
+        std_path.write_bytes(data)
 
         if clear:
-            self.communications.items.clear()
+            self.communications.clear()
 
         return std_path
 
-    async def adump_messages(self, clear: bool = False) -> Path | None:
-        """Async dump all session messages to file with lock protection.
+    async def adump(self, clear: bool = False) -> Path | None:
+        """Async dump entire session state for replay.
+
+        Serializes the full session (messages, branches, config) to JSON.
+        To restore: Session.from_dict(data), then re-register resources.
 
         Args:
-            clear: Clear messages after dump (default False).
+            clear: Clear communications after dump (default False).
 
         Returns:
-            Path to written file, or None if no log_config or no messages.
+            Path to session file, or None if logging disabled or empty.
         """
-        from krons.utils import create_path, json_dumpb, json_lines_iter
+        from krons.utils import create_path, json_dumpb
 
-        if self.config.log_config is None or len(self.messages) == 0:
+        if not self.config.logging_enabled or len(self.messages) == 0:
             return None
 
-        cfg = self.config.log_config
-
         async with self.messages:
             self._dump_count += 1
 
             filepath = await create_path(
-                directory=cfg.persist_dir,
+                directory=self.config.log_persist_dir,
                 filename=f"session_{str(self.id)[:8]}_{self._dump_count}",
-                extension=cfg.extension,
+                extension=".json",
                 timestamp=True,
                 file_exist_ok=True,
             )
 
-            items = [msg.to_dict(mode="json") for msg in self.messages]
-
-            if cfg.extension == ".jsonl":
-                content = b"".join(json_lines_iter(items, safe_fallback=True))
-                await filepath.write_bytes(content)
-            else:
-                data = json_dumpb(items, safe_fallback=True)
-                await filepath.write_bytes(data)
+            data = json_dumpb(self.to_dict(mode="json"), safe_fallback=True)
+            await filepath.write_bytes(data)
 
             if clear:
-                self.communications.items.clear()
+                self.communications.clear()
 
         return Path(filepath)
 
     def _save_at_exit(self) -> None:
-        """atexit callback. Dumps messages synchronously. Errors are suppressed."""
+        """atexit callback. Dumps session synchronously. Errors are suppressed."""
         if len(self.messages) > 0:
             try:
-                self.dump_messages(clear=False)
+                self.dump(clear=False)
             except Exception:
                 pass  # Silent failure during interpreter shutdown
 

{krons-0.2.2.dist-info → krons-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: krons
-Version: 0.2.2
+Version: 0.2.3
 Summary: Spec-based composable framework for building type-safe systems
 Project-URL: Homepage, https://github.com/khive-ai/krons
 Project-URL: Repository, https://github.com/khive-ai/krons

{krons-0.2.2.dist-info → krons-0.2.3.dist-info}/RECORD

@@ -37,14 +37,13 @@ krons/agent/third_party/claude_code.py,sha256=NBYMsdPheb0PqGgTO2_aUAR2XoGDsPUwsB
 krons/agent/third_party/gemini_models.py,sha256=2dag2WbSTKLBd8t0F7t_TXCcl7aqsTFgDI6XLkhsZwM,17416
 krons/agent/third_party/openai_models.py,sha256=L-GkJ-bBW5zwPbgCWEYHMNqRiZfFTAUJE96AVZoDAkQ,8040
 krons/core/__init__.py,sha256=TKi3toVRYybyjUwUJlng--pfWzWihUWQBCNh0UTfTSQ,3611
-krons/core/base/__init__.py,sha256=emyypsuiFK0UzYsDeamD84ygXnO1TLx2A5mBxGTBidE,3755
+krons/core/base/__init__.py,sha256=Edl_HH2F1l0SMHMzuEPU_IFjkigRR1vtqVXcPm-mr1o,3614
 krons/core/base/broadcaster.py,sha256=Es__WfL-j2h5NS6aU18fgaFkZsJIL-zDe86Qvlea-io,4013
 krons/core/base/element.py,sha256=LqBbmPttXoDAFy56JzWhSJ4sBw_LW7hMGxZUog85N8g,7642
 krons/core/base/event.py,sha256=iZQ65Xg3a8eaCEl4Hca-pxPNtckhrHkydEqYyppbuo0,11524
 krons/core/base/eventbus.py,sha256=z45ORtGey5A1N9hdyEPOhlX1whaTTdqdk7g3XEDL-_8,3761
-krons/core/base/flow.py,sha256=eqsYm8QO7bD5Y9VYnlpHFWpaUCY0rCeKGrDrFUyBVA4,12364
+krons/core/base/flow.py,sha256=bFG5x7PEbWApsD24GTgpD5VsedLO6364scaAxvYw6LM,12561
 krons/core/base/graph.py,sha256=FiTMArxuz6At3oYPEoamES3FtBM-AM6QnA7EQ8G18mo,15972
-krons/core/base/log.py,sha256=SMJlruIG4YHk6Ls45jQ9SrPxFCcnypUmTGaWd90wiwM,923
 krons/core/base/node.py,sha256=0_-R6uNe6-LF3adQRZowSIICdy6ep3cQIQRdF0oosoQ,34699
 krons/core/base/pile.py,sha256=ZKvnX3BjMsrH989TLNZII1dDWv4yPhSRzLKXx2yzyKQ,21108
 krons/core/base/processor.py,sha256=GrdNZa5YD0tXle847ttnxuWe4Ego1MhULzWHmSz5RqQ,18768
@@ -87,7 +86,7 @@ krons/session/constraints.py,sha256=DvmLKyf5eYUX2nxCDcco1-uDipGjiIJwoUxWVtX2w84,
 krons/session/exchange.py,sha256=QChEqK5NFZJpHWxPrXd8LkGBCFg6LNONd1ZEy0MufpI,9165
 krons/session/message.py,sha256=mtZ-PYCCoyWLy7BSB5TL5a0hIt0aQX2zdckOCIl6NZ0,2007
 krons/session/registry.py,sha256=13pPKmDZJaj9q27oUOdtH6-jrHP_HDFF8lCdcNF--EE,999
-krons/session/session.py,sha256=_y3-NEQxgBa51mRd6LiD7zTqJFikKCP4avLqMai6O08,16966
+krons/session/session.py,sha256=2dykFkdK0DUu3kdv1v8RIevtuD3P50w_0vNIE3qs3D0,17204
 krons/utils/__init__.py,sha256=V-jTKULdofjJXxcEHygefKV_v6XE4H3poCdxPLjvf_4,1718
 krons/utils/_function_arg_parser.py,sha256=H5JVLBVY8W9ZNkZ4_YVtVVY1rFYvnIRhSrAhKEdj2Qc,3479
 krons/utils/_hash.py,sha256=W2Ma9v8-INPaGkur7GTtbF8KwuXSJNSwk8DCNPRvx8Q,6859
@@ -146,7 +145,7 @@ krons/work/rules/common/mapping.py,sha256=Loq54MNEtwpnHN0aypTjFOqwoOKLEysddHh-JE
 krons/work/rules/common/model.py,sha256=xmM6coEThf_fgIiqJiyDgvdfib_FpVeY6LgWPVcWSwU,3026
 krons/work/rules/common/number.py,sha256=cCukgMSpQu5RdYK5rXAUyop9qXgDRfLCioMvE8kIzHg,3162
 krons/work/rules/common/string.py,sha256=zHp_OLh0FL4PvmSlyDTEzb2I97-DBSEyI2zcMo10voA,5090
-krons-0.2.2.dist-info/METADATA,sha256=X1gYH9GGL-V2GhH3_iy96GzRZOCwO-HKt7NpNrzmgcg,2527
-krons-0.2.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-krons-0.2.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-krons-0.2.2.dist-info/RECORD,,
+krons-0.2.3.dist-info/METADATA,sha256=5VYw8_rlpQ5Q9W8EGwy8XZ19F--f_URzcmVAiG9WIqY,2527
+krons-0.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+krons-0.2.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+krons-0.2.3.dist-info/RECORD,,

krons/core/base/log.py

@@ -1,32 +0,0 @@
-# Copyright (c) 2025 - 2026, HaiyangLi <quantocean.li at gmail dot com>
-# SPDX-License-Identifier: Apache-2.0
-
-"""Logging configuration for Session message persistence.
-
-Provides DataLoggerConfig for configuring automatic message dumps.
-"""
-
-from __future__ import annotations
-
-from pathlib import Path
-from typing import Literal
-
-from pydantic import Field
-
-from krons.core.types import HashableModel
-
-__all__ = ("DataLoggerConfig",)
-
-
-class DataLoggerConfig(HashableModel):
-    """Configuration for Session message persistence.
-
-    Attributes:
-        persist_dir: Directory for dump files.
-        extension: Output format (.json array or .jsonl newline-delimited).
-        auto_save_on_exit: Register atexit handler on Session creation.
-    """
-
-    persist_dir: str | Path = Field(default="./logs")
-    extension: Literal[".json", ".jsonl"] = Field(default=".jsonl")
-    auto_save_on_exit: bool = Field(default=True)