openhands-sdk 1.4.0__py3-none-any.whl → 1.4.1__py3-none-any.whl

openhands/sdk/conversation/conversation_stats.py

@@ -1,4 +1,6 @@
-from pydantic import BaseModel, Field, PrivateAttr
+from typing import Any
+
+from pydantic import BaseModel, Field, PrivateAttr, model_serializer
 
 from openhands.sdk.llm.llm_registry import RegistryEvent
 from openhands.sdk.llm.utils.metrics import Metrics
@@ -18,6 +20,41 @@ class ConversationStats(BaseModel):
 
     _restored_usage_ids: set[str] = PrivateAttr(default_factory=set)
 
+    @model_serializer(mode="wrap")
+    def _serialize_with_context(self, serializer: Any, info: Any) -> dict[str, Any]:
+        """Serialize metrics based on context.
+
+        By default, preserves full metrics history including costs,
+        response_latencies, and token_usages lists for persistence.
+
+        When context={'use_snapshot': True} is passed, converts Metrics to
+        MetricsSnapshot format to minimize payload size for network transmission.
+
+        Args:
+            serializer: Pydantic's default serializer
+            info: Serialization info containing context
+
+        Returns:
+            Dictionary with metrics serialized based on context
+        """
+        # Get the default serialization
+        data = serializer(self)
+
+        # Check if we should use snapshot serialization
+        context = info.context if info else None
+        use_snapshot = context.get("use_snapshot", False) if context else False
+
+        if use_snapshot and "usage_to_metrics" in data:
+            # Replace each Metrics with its snapshot
+            usage_to_snapshots = {}
+            for usage_id, metrics in self.usage_to_metrics.items():
+                snapshot = metrics.get_snapshot()
+                usage_to_snapshots[usage_id] = snapshot.model_dump()
+
+            data["usage_to_metrics"] = usage_to_snapshots
+
+        return data
+
     def get_combined_metrics(self) -> Metrics:
         total_metrics = Metrics()
         for metrics in self.usage_to_metrics.values():

openhands/sdk/event/conversation_state.py

@@ -49,6 +49,14 @@ class ConversationStateUpdateEvent(Event):
 
     @field_validator("value")
     def validate_value(cls, value, info):
+        # Prevent circular import
+        from openhands.sdk.conversation.conversation_stats import ConversationStats
+
+        # For ConversationStats, use snapshot serialization to avoid
+        # sending lengthy lists over WebSocket
+        if isinstance(value, ConversationStats):
+            return value.model_dump(mode="json", context={"use_snapshot": True})
+
         key = info.data.get("key")
         if key is None:
             # Allow value without key for flexibility

openhands/sdk/logger/logger.py

@@ -105,7 +105,14 @@ def setup_logging(
     keep = ENV_BACKUP_COUNT if backup_count is None else backup_count
 
     root = logging.getLogger()
+    old_level = root.level
     root.setLevel(lvl)
+
+    # Set the level for any existing logger with the same intial level
+    for logger in logging.root.manager.loggerDict.values():
+        if isinstance(logger, logging.Logger) and logger.level == old_level:
+            logger.setLevel(lvl)
+
     # Do NOT clear existing handlers; Uvicorn installs these before importing the app.
     # Only add ours if there isn't already a comparable stream handler.
     has_stream = any(isinstance(h, logging.StreamHandler) for h in root.handlers)

{openhands_sdk-1.4.0.dist-info → openhands_sdk-1.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openhands-sdk
-Version: 1.4.0
+Version: 1.4.1
 Summary: OpenHands SDK - Core functionality for building AI agents
 Requires-Python: >=3.12
 Requires-Dist: deprecation>=2.1.0

{openhands_sdk-1.4.0.dist-info → openhands_sdk-1.4.1.dist-info}/RECORD

@@ -35,7 +35,7 @@ openhands/sdk/context/skills/types.py,sha256=qdakWgfs_88I_cnmVhO5bl3C3s11GvCydq7
 openhands/sdk/conversation/__init__.py,sha256=1-xh49S2KJhtAjkHDaDHU28UWhfQLl3CeFlo179gr00,1402
 openhands/sdk/conversation/base.py,sha256=3qUc4_PjrT1IztBX7EJKuypvRVWRIbMmIQ3Fc2lvspI,8308
 openhands/sdk/conversation/conversation.py,sha256=GsPK9MWTAWG47J6jQ1vcSD9ER31wbFkRF9zdFmCqz9I,5275
-openhands/sdk/conversation/conversation_stats.py,sha256=_-iKBOnT5PYhbDkm7kB3Jsw1RrOhViA0HdnfmCDwRn0,1643
+openhands/sdk/conversation/conversation_stats.py,sha256=ZlQ99kgG5YVCrZ4rqJlq63JaiInxX8jqv-q5lS7RN68,3038
 openhands/sdk/conversation/event_store.py,sha256=he-bwP823s5zAIdua_0ZgkkHQCJoAqbtV2SN0hibX30,5207
 openhands/sdk/conversation/events_list_base.py,sha256=n_YvgbhBPOPDbw4Kp68J0EKFM39vg95ng09GMfTz29s,505
 openhands/sdk/conversation/exceptions.py,sha256=C3pN3MJJIYdhcMHMqtOmVkR1BhVe3pfxBhxzQ6FVBjc,804
@@ -65,7 +65,7 @@ openhands/sdk/event/__init__.py,sha256=ir-jRVA0QjEbFuDzJOYRq2kXTgUHs8eJ7_skoCRNz
 openhands/sdk/event/base.py,sha256=QpQtYfYiZqaKWA-2xa4GEmk2qkUHKD6x-OIAk7SMnDs,5568
 openhands/sdk/event/condenser.py,sha256=sne9CxhhNq9UvJMKuKcw8tXuMgutuGh85TbbZHwYhPQ,2481
 openhands/sdk/event/conversation_error.py,sha256=gZMyliJx1xbyoJFYH0-AEHqznyiA8LZ7olPSidaNfuc,957
-openhands/sdk/event/conversation_state.py,sha256=nARmfgjj8PJ6luBjKlBjNIgdf1U0YxWrkIVAHCvbguY,3284
+openhands/sdk/event/conversation_state.py,sha256=V-ti5SLL5SL330sEOQgpy-U8tErVwZYG0iC3AqjTUwo,3650
 openhands/sdk/event/llm_completion_log.py,sha256=VCxJiZBsn1F6TRV6fwvsPs6W9DpjghfIFmJJlGKztXg,1232
 openhands/sdk/event/token.py,sha256=QlEbBrfZaHe9tp8-Ot4vTd0T-_Vj7lpUqKfVJcVHOeI,497
 openhands/sdk/event/types.py,sha256=3hyFDBNtYTGEwLV6muodJO7iDMoZTC4D0Dd4vYeF2dE,271
@@ -112,7 +112,7 @@ openhands/sdk/llm/utils/telemetry.py,sha256=BGzikbw1DAj96Bo60D7XIGh034zKIbO1zRx5
 openhands/sdk/llm/utils/unverified_models.py,sha256=SmYrX_WxXOJBanTviztqy1xPjOcLY4i3qvwNBEga_Dk,4797
 openhands/sdk/llm/utils/verified_models.py,sha256=add09fF6MHOCBS4Wp2BxLYYfijMtzkq1IFc3GNZgMWU,1357
 openhands/sdk/logger/__init__.py,sha256=vZvFDYfW01Y8Act3tveMs3XxTysJlt4HeT-n6X_ujYk,330
-openhands/sdk/logger/logger.py,sha256=Dqhl8EFVIprQmSQNKVnyIpybF6twZKzTJ21pHK7aW9k,6243
+openhands/sdk/logger/logger.py,sha256=kSpeol92dKesm24ZxVyYbtAxeYccmuFZJw3qNO9X-t4,6513
 openhands/sdk/logger/rolling.py,sha256=E6oy0asgmOhZHoWlSCw0QK1PKnS6kvtxjoWLAsqlGvs,3440
 openhands/sdk/mcp/__init__.py,sha256=-wQbZ405PjVRCBtSfirp4jsiRohd7IJAyAdicZ-M8Ok,588
 openhands/sdk/mcp/client.py,sha256=CLkFImydorlT_RBTE5UxvRNGMlc-uO3wvupoDzB5E20,2420
@@ -158,7 +158,7 @@ openhands/sdk/workspace/remote/__init__.py,sha256=eKkj6NOESMUBGDVC6_L2Wfuc4K6G-m
 openhands/sdk/workspace/remote/async_remote_workspace.py,sha256=ftv1Vdx4mmM3AjygJpemMJGvhaQel7ORxdQVk12z4ZE,5061
 openhands/sdk/workspace/remote/base.py,sha256=72C9MZV7ch5n6oHNvFMo6irW7b6Le8n4gk3yFuc0798,5605
 openhands/sdk/workspace/remote/remote_workspace_mixin.py,sha256=CzHfnLUIra5sgPkP9kcggb1vHGOPpYQzLsHvGO2rRt0,10963
-openhands_sdk-1.4.0.dist-info/METADATA,sha256=yVa8hBrqkKA5IVKx5WR_4CWNUtemIyGDKrxGvart6CM,545
-openhands_sdk-1.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-openhands_sdk-1.4.0.dist-info/top_level.txt,sha256=jHgVu9I0Blam8BXFgedoGKfglPF8XvW1TsJFIjcgP4E,10
-openhands_sdk-1.4.0.dist-info/RECORD,,
+openhands_sdk-1.4.1.dist-info/METADATA,sha256=9K8ieOX7Xf0snpoeuCY9KmXkd5GTHkg_3EpgTT6vGLo,545
+openhands_sdk-1.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+openhands_sdk-1.4.1.dist-info/top_level.txt,sha256=jHgVu9I0Blam8BXFgedoGKfglPF8XvW1TsJFIjcgP4E,10
+openhands_sdk-1.4.1.dist-info/RECORD,,