rusticai-forge 0.2.2__tar.gz → 0.2.4__tar.gz

{rusticai_forge-0.2.2 → rusticai_forge-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rusticai-forge
-Version: 0.2.2
+Version: 0.2.4
 Summary: Python agent wrapper and execution engine for Forge
 Requires-Python: <3.14,>=3.13
 Description-Content-Type: text/markdown
@@ -8,7 +8,7 @@ Requires-Dist: pydantic<3.0.0,>=2.12.5
 Requires-Dist: redis<7.0.0,>=6.4.0
 Requires-Dist: nats-py<3.0.0,>=2.9.0
 Requires-Dist: pyzmq<27.0.0,>=26.0.0
-Requires-Dist: rusticai-core<2.0.0,>=1.1.0
+Requires-Dist: rusticai-core<2.0.0,>=1.2.0
 Requires-Dist: rusticai-redis<2.0.0,>=1.1.0
 Requires-Dist: psycopg[binary]<4.0.0,>=3.3.3
 Requires-Dist: httpx<1.0.0,>=0.28.1

{rusticai_forge-0.2.2 → rusticai_forge-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rusticai-forge"
-version = "0.2.2"
+version = "0.2.4"
 description = "Python agent wrapper and execution engine for Forge"
 readme = "README.md"
 requires-python = ">=3.13,<3.14"
@@ -10,7 +10,7 @@ dependencies = [
     "nats-py>=2.9.0,<3.0.0",
     "pyzmq>=26.0.0,<27.0.0",
     # DO NOT CHANGE THESE TO LOCAL FILE PATHS. Always use package references.
-    "rusticai-core>=1.1.0,<2.0.0",
+    "rusticai-core>=1.2.0,<2.0.0",
     "rusticai-redis>=1.1.0,<2.0.0",
     "psycopg[binary]>=3.3.3,<4.0.0",
     "httpx>=0.28.1,<1.0.0",

{rusticai_forge-0.2.2 → rusticai_forge-0.2.4}/src/rustic_ai/forge/agent_runner.py

@@ -204,10 +204,14 @@ def _load_backend_config(client_props: dict, client_type_str: str) -> dict:
         raw_backend_config = client_props["backend_config"]
     else:
         raw_backend_config = {
-            key: value for key, value in client_props.items() if key != "organization_id"
+            key: value
+            for key, value in client_props.items()
+            if key != "organization_id"
         }
     if not isinstance(raw_backend_config, dict):
-        raise ValueError("FORGE_CLIENT_PROPERTIES_JSON backend_config must be a JSON object.")
+        raise ValueError(
+            "FORGE_CLIENT_PROPERTIES_JSON backend_config must be a JSON object."
+        )
 
     backend_config = dict(raw_backend_config)
 
@@ -216,7 +220,9 @@ def _load_backend_config(client_props: dict, client_type_str: str) -> dict:
         if raw_supervisor_config:
             supervisor_config = json.loads(raw_supervisor_config)
             if not isinstance(supervisor_config, dict):
-                raise ValueError(f"{SUPERVISOR_ZMQ_CONFIG_ENV} must decode to a JSON object.")
+                raise ValueError(
+                    f"{SUPERVISOR_ZMQ_CONFIG_ENV} must decode to a JSON object."
+                )
             backend_config = supervisor_config | backend_config
 
         endpoint = os.getenv(SUPERVISOR_ZMQ_ENDPOINT_ENV, "")
@@ -248,7 +254,9 @@ def main():
         client_module_str = os.getenv("FORGE_CLIENT_MODULE", "")
         client_props = json.loads(os.getenv("FORGE_CLIENT_PROPERTIES_JSON", "{}"))
         if not isinstance(client_props, dict):
-            raise ValueError("FORGE_CLIENT_PROPERTIES_JSON must decode to a JSON object.")
+            raise ValueError(
+                "FORGE_CLIENT_PROPERTIES_JSON must decode to a JSON object."
+            )
 
         backend_config = _load_backend_config(client_props, client_type_str)
 

{rusticai_forge-0.2.2 → rusticai_forge-0.2.4}/src/rustic_ai/forge/messaging/backend.py

@@ -93,14 +93,18 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
         response = self._send_request("get_messages", {"topic": topic})
         return [_normalize_message(msg) for msg in response.get("messages", [])]
 
-    def get_messages_for_topic_since(self, topic: str, msg_id_since: int) -> List[Message]:
+    def get_messages_for_topic_since(
+        self, topic: str, msg_id_since: int
+    ) -> List[Message]:
         response = self._send_request(
             "get_since",
             {"topic": topic, "since_id": msg_id_since},
         )
         return [_normalize_message(msg) for msg in response.get("messages", [])]
 
-    def get_next_message_for_topic_since(self, topic: str, last_message_id: int) -> Optional[Message]:
+    def get_next_message_for_topic_since(
+        self, topic: str, last_message_id: int
+    ) -> Optional[Message]:
         response = self._send_request(
             "get_next",
             {"topic": topic, "since_id": last_message_id},
@@ -142,7 +146,9 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
                 timeout_ms=min(self.request_timeout_ms, 2_000),
             )
         except Exception:
-            logger.debug("Supervisor cleanup request failed during shutdown", exc_info=True)
+            logger.debug(
+                "Supervisor cleanup request failed during shutdown", exc_info=True
+            )
 
         self._shutdown_event.set()
         self._reconnect_event.set()
@@ -200,7 +206,9 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
 
     def _heartbeat_loop(self) -> None:
         interval = self.heartbeat_interval_ms / 1000.0
-        timeout_ms = max(1_000, min(self.request_timeout_ms, self.heartbeat_interval_ms))
+        timeout_ms = max(
+            1_000, min(self.request_timeout_ms, self.heartbeat_interval_ms)
+        )
 
         while not self._shutdown_event.wait(interval):
             try:
@@ -231,8 +239,9 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
                     self._drain_outbound(socket)
 
                     if self._reconnect_event.is_set():
-                        reconnect_cause = self._consume_transport_error() or RuntimeError(
-                            "supervisor ZeroMQ reconnect requested"
+                        reconnect_cause = (
+                            self._consume_transport_error()
+                            or RuntimeError("supervisor ZeroMQ reconnect requested")
                         )
                         socket, poller = self._reset_socket(socket, poller)
                         self._reconnect_event.clear()
@@ -257,7 +266,9 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
         finally:
             self._close_socket(socket, poller)
 
-    def _connect_with_retry(self, initial_cause: Exception) -> tuple[Optional[zmq.Socket], Optional[zmq.Poller]]:
+    def _connect_with_retry(
+        self, initial_cause: Exception
+    ) -> tuple[Optional[zmq.Socket], Optional[zmq.Poller]]:
         attempt = 0
         delay = self.retry_initial_delay
         start_time = time.monotonic()
@@ -340,9 +351,14 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
                 continue
 
             incoming = socket.recv_json()
-            if incoming.get("kind") == "response" and incoming.get("request_id") == request_id:
+            if (
+                incoming.get("kind") == "response"
+                and incoming.get("request_id") == request_id
+            ):
                 if not incoming.get("ok", False):
-                    raise RuntimeError(incoming.get("error", f"Supervisor {op} request failed"))
+                    raise RuntimeError(
+                        incoming.get("error", f"Supervisor {op} request failed")
+                    )
                 return incoming
 
             self._handle_envelope(incoming)
@@ -390,7 +406,9 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
             try:
                 handler(message)
             except Exception:
-                logger.exception("Supervisor ZeroMQ delivery handler failed for topic %s", topic)
+                logger.exception(
+                    "Supervisor ZeroMQ delivery handler failed for topic %s", topic
+                )
 
     def _send_request(
         self,
@@ -409,14 +427,20 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
         with self._pending_lock:
             self._pending[request_id] = queue
 
-        self._outbound.put({"kind": "request", "op": op, "request_id": request_id} | payload)
+        self._outbound.put(
+            {"kind": "request", "op": op, "request_id": request_id} | payload
+        )
 
         try:
-            response = queue.get(timeout=(timeout_ms or self.request_timeout_ms) / 1000.0)
+            response = queue.get(
+                timeout=(timeout_ms or self.request_timeout_ms) / 1000.0
+            )
         except Empty as exc:
             if allow_reconnect:
                 self._mark_transport_unhealthy(exc)
-            raise TimeoutError(f"Timed out waiting for supervisor response to {op}") from exc
+            raise TimeoutError(
+                f"Timed out waiting for supervisor response to {op}"
+            ) from exc
         finally:
             with self._pending_lock:
                 self._pending.pop(request_id, None)
@@ -451,7 +475,9 @@ class SupervisorZmqMessagingBackend(MessagingBackend):
         self._close_socket(socket, poller)
         return None, None
 
-    def _close_socket(self, socket: Optional[zmq.Socket], poller: Optional[zmq.Poller]) -> None:
+    def _close_socket(
+        self, socket: Optional[zmq.Socket], poller: Optional[zmq.Poller]
+    ) -> None:
         if socket is None:
             return
 

{rusticai_forge-0.2.2 → rusticai_forge-0.2.4}/src/rusticai_forge.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rusticai-forge
-Version: 0.2.2
+Version: 0.2.4
 Summary: Python agent wrapper and execution engine for Forge
 Requires-Python: <3.14,>=3.13
 Description-Content-Type: text/markdown
@@ -8,7 +8,7 @@ Requires-Dist: pydantic<3.0.0,>=2.12.5
 Requires-Dist: redis<7.0.0,>=6.4.0
 Requires-Dist: nats-py<3.0.0,>=2.9.0
 Requires-Dist: pyzmq<27.0.0,>=26.0.0
-Requires-Dist: rusticai-core<2.0.0,>=1.1.0
+Requires-Dist: rusticai-core<2.0.0,>=1.2.0
 Requires-Dist: rusticai-redis<2.0.0,>=1.1.0
 Requires-Dist: psycopg[binary]<4.0.0,>=3.3.3
 Requires-Dist: httpx<1.0.0,>=0.28.1

{rusticai_forge-0.2.2 → rusticai_forge-0.2.4}/src/rusticai_forge.egg-info/requires.txt

@@ -2,7 +2,7 @@ pydantic<3.0.0,>=2.12.5
 redis<7.0.0,>=6.4.0
 nats-py<3.0.0,>=2.9.0
 pyzmq<27.0.0,>=26.0.0
-rusticai-core<2.0.0,>=1.1.0
+rusticai-core<2.0.0,>=1.2.0
 rusticai-redis<2.0.0,>=1.1.0
 psycopg[binary]<4.0.0,>=3.3.3
 httpx<1.0.0,>=0.28.1