penguiflow 2.2.2__tar.gz → 2.2.3__tar.gz

{penguiflow-2.2.2 → penguiflow-2.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: penguiflow
-Version: 2.2.2
+Version: 2.2.3
 Summary: Async agent orchestration primitives.
 Author: PenguiFlow Team
 License: MIT License

{penguiflow-2.2.2 → penguiflow-2.2.3}/penguiflow/__init__.py

@@ -105,4 +105,4 @@ __all__ = [
     "TrajectoryStep",
 ]
 
-__version__ = "2.2.2"
+__version__ = "2.2.3"

{penguiflow-2.2.2 → penguiflow-2.2.3}/penguiflow/core.py

@@ -14,6 +14,7 @@ from collections import deque
 from collections.abc import Awaitable, Callable, Mapping, Sequence
 from contextlib import suppress
 from dataclasses import dataclass
+from types import TracebackType
 from typing import Any, cast
 
 from .bus import BusEnvelope, MessageBus
@@ -27,6 +28,14 @@ from .types import WM, FinalAnswer, Message, StreamChunk
 
 logger = logging.getLogger("penguiflow.core")
 
+ExcInfo = tuple[type[BaseException], BaseException, TracebackType | None]
+
+
+def _capture_exc_info(exc: BaseException | None) -> ExcInfo | None:
+    if exc is None:
+        return None
+    return (type(exc), exc, exc.__traceback__)
+
 BUDGET_EXCEEDED_TEXT = "Hop budget exhausted"
 DEADLINE_EXCEEDED_TEXT = "Deadline exceeded"
 TOKEN_BUDGET_EXCEEDED_TEXT = "Token budget exhausted"
@@ -750,6 +759,7 @@ class PenguiFlow:
                 raise
             except TimeoutError as exc:
                 latency = (time.perf_counter() - start) * 1000
+                exc_info = _capture_exc_info(exc)
                 await self._emit_event(
                     event="node_timeout",
                     node=node,
@@ -759,6 +769,7 @@ class PenguiFlow:
                     latency_ms=latency,
                     level=logging.WARNING,
                     extra={"exception": repr(exc)},
+                    exc_info=exc_info,
                 )
                 if attempt >= node.policy.max_retries:
                     timeout_message: str | None = None
@@ -783,6 +794,7 @@ class PenguiFlow:
                         flow_error=flow_error,
                         latency=latency,
                         attempt=attempt,
+                        exc_info=exc_info,
                     )
                     return
                 attempt += 1
@@ -801,6 +813,7 @@ class PenguiFlow:
                 continue
             except Exception as exc:  # noqa: BLE001
                 latency = (time.perf_counter() - start) * 1000
+                exc_info = _capture_exc_info(exc)
                 await self._emit_event(
                     event="node_error",
                     node=node,
@@ -810,6 +823,7 @@ class PenguiFlow:
                     latency_ms=latency,
                     level=logging.ERROR,
                     extra={"exception": repr(exc)},
+                    exc_info=exc_info,
                 )
                 if attempt >= node.policy.max_retries:
                     flow_error = self._create_flow_error(
@@ -830,6 +844,7 @@ class PenguiFlow:
                         flow_error=flow_error,
                         latency=latency,
                         attempt=attempt,
+                        exc_info=exc_info,
                     )
                     return
                 attempt += 1
@@ -890,6 +905,7 @@ class PenguiFlow:
         flow_error: FlowError,
         latency: float | None,
         attempt: int,
+        exc_info: ExcInfo | None,
     ) -> None:
         original = flow_error.unwrap()
         exception_repr = repr(original) if original is not None else flow_error.message
@@ -906,6 +922,7 @@ class PenguiFlow:
             latency_ms=latency,
             level=logging.ERROR,
             extra=extra,
+            exc_info=exc_info,
         )
         if self._emit_errors_to_rookery and flow_error.trace_id is not None:
             await self._emit_to_rookery(flow_error, source=context.owner)
@@ -1365,6 +1382,7 @@ class PenguiFlow:
         latency_ms: float | None,
         level: int,
         extra: dict[str, Any] | None = None,
+        exc_info: ExcInfo | None = None,
     ) -> None:
         node_name = getattr(node, "name", None)
         node_id = getattr(node, "node_id", node_name)
@@ -1398,7 +1416,12 @@ class PenguiFlow:
             extra=extra or {},
         )
 
-        logger.log(level, event, extra=event_obj.to_payload())
+        payload = event_obj.to_payload()
+        log_kwargs: dict[str, Any] = {"extra": payload}
+        if exc_info is not None:
+            log_kwargs["exc_info"] = exc_info
+
+        logger.log(level, event, **log_kwargs)
 
         if self._state_store is not None:
             stored_event = StoredEvent.from_flow_event(event_obj)

{penguiflow-2.2.2 → penguiflow-2.2.3}/penguiflow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: penguiflow
-Version: 2.2.2
+Version: 2.2.3
 Summary: Async agent orchestration primitives.
 Author: PenguiFlow Team
 License: MIT License

{penguiflow-2.2.2 → penguiflow-2.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "penguiflow"
-version = "2.2.2"
+version = "2.2.3"
 description = "Async agent orchestration primitives."
 readme = "README.md"
 requires-python = ">=3.11"

{penguiflow-2.2.2 → penguiflow-2.2.3}/tests/test_errors.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import asyncio
+import logging
 
 import pytest
 
@@ -104,3 +105,39 @@ async def test_flow_error_metadata_for_timeouts() -> None:
     assert result.metadata["attempt"] == 0
 
     await flow.stop()
+
+
+@pytest.mark.asyncio
+async def test_node_error_logs_exc_info(caplog: pytest.LogCaptureFixture) -> None:
+    async def broken(_message: str, _ctx) -> None:
+        raise RuntimeError("dependency missing")
+
+    node = Node(
+        broken,
+        name="broken",
+        policy=NodePolicy(validate="none", max_retries=0),
+    )
+    flow = create(node.to(), emit_errors_to_rookery=True)
+    flow.run()
+
+    msg = Message(payload="payload", headers=Headers(tenant="demo"))
+
+    with caplog.at_level(logging.ERROR, logger="penguiflow.core"):
+        await flow.emit(msg)
+        await asyncio.wait_for(flow.fetch(), timeout=0.5)
+
+    node_error_records = [
+        record for record in caplog.records if record.message == "node_error"
+    ]
+    assert node_error_records, "expected node_error log entry"
+    assert node_error_records[0].exc_info is not None
+    assert isinstance(node_error_records[0].exc_info[1], RuntimeError)
+
+    node_failed_records = [
+        record for record in caplog.records if record.message == "node_failed"
+    ]
+    assert node_failed_records, "expected node_failed log entry"
+    assert node_failed_records[0].exc_info is not None
+    assert isinstance(node_failed_records[0].exc_info[1], RuntimeError)
+
+    await flow.stop()