agent-lab-sdk 0.1.43__py3-none-any.whl → 0.1.44__py3-none-any.whl

agent_lab_sdk/langgraph/checkpoint/agw_saver.py

@@ -550,6 +550,7 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         path: str,
         *,
         ok_statuses: Iterable[int] | None = None,
+        label_path: str | None = None,
         **kw,
     ) -> httpx.Response:
         if "json" in kw:
@@ -558,6 +559,7 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             logger.debug("AGW HTTP payload: %s", kw["data"].decode())
 
         ok_set = set(ok_statuses) if ok_statuses is not None else set()
+        metric_path = label_path or path
 
         attempt = 1
         while True:
@@ -576,7 +578,7 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             try:
                 resp = await client.request(method, path, **kw)
             except httpx.RequestError as exc:
-                AGW_HTTP_ERROR.labels(method, path).inc()
+                AGW_HTTP_ERROR.labels(method, metric_path).inc()
                 logger.warning(
                     "AGW request %s %s failed on attempt %d/%d: %s",
                     method,
@@ -586,7 +588,7 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
                     exc,
                 )
                 if attempt >= self.retry_max_attempts:
-                    AGW_HTTP_FINAL_ERROR.labels(method, path).inc()
+                    AGW_HTTP_FINAL_ERROR.labels(method, metric_path).inc()
                     if self._client is not None:
                         try:
                             await self._client.aclose()
@@ -619,16 +621,16 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
 
             status = resp.status_code
             if status < 400 or status in ok_set:
-                AGW_HTTP_SUCCESS.labels(method, path).inc()
+                AGW_HTTP_SUCCESS.labels(method, metric_path).inc()
                 return resp
 
-            AGW_HTTP_ERROR.labels(method, path).inc()
+            AGW_HTTP_ERROR.labels(method, metric_path).inc()
             if status in (404, 406):
-                AGW_HTTP_FINAL_ERROR.labels(method, path).inc()
+                AGW_HTTP_FINAL_ERROR.labels(method, metric_path).inc()
                 return resp
 
             if attempt >= self.retry_max_attempts:
-                AGW_HTTP_FINAL_ERROR.labels(method, path).inc()
+                AGW_HTTP_FINAL_ERROR.labels(method, metric_path).inc()
                 return resp
 
             try:
@@ -693,13 +695,15 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
         tid = api_cfg["threadId"]
 
         if cid:
+            path_template = "/checkpoint/{threadId}/{checkpointId}"
             path = f"/checkpoint/{tid}/{cid}"
             params = {"checkpointNs": api_cfg.get("checkpointNs", "")}
         else:
+            path_template = "/checkpoint/{threadId}"
             path = f"/checkpoint/{tid}"
             params = None
 
-        resp = await self._http("GET", path, params=params)
+        resp = await self._http("GET", path, params=params, label_path=path_template)
         logger.debug("AGW aget_tuple response: %s", resp.text)
 
         if not resp.text:
@@ -723,7 +727,12 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             "before": self._to_api_config(before) if before else None,
             "limit": limit,
         }
-        resp = await self._http("POST", "/checkpoint/list", json=payload)
+        resp = await self._http(
+            "POST",
+            "/checkpoint/list",
+            json=payload,
+            label_path="/checkpoint/list",
+        )
         logger.debug("AGW alist response: %s", resp.text)
         resp.raise_for_status()
         for item in resp.json():
@@ -742,7 +751,7 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             "metadata": self._enc_meta(get_checkpoint_metadata(cfg, metadata)),
             "newVersions": new_versions,
         }
-        resp = await self._http("POST", "/checkpoint", json=payload)
+        resp = await self._http("POST", "/checkpoint", json=payload, label_path="/checkpoint")
         logger.debug("AGW aput response: %s", resp.text)
         resp.raise_for_status()
         return resp.json()["config"]
@@ -761,12 +770,21 @@ class AsyncAGWCheckpointSaver(BaseCheckpointSaver):
             "taskId": task_id,
             "taskPath": task_path,
         }
-        resp = await self._http("POST", "/checkpoint/writes", json=payload)
+        resp = await self._http(
+            "POST",
+            "/checkpoint/writes",
+            json=payload,
+            label_path="/checkpoint/writes",
+        )
         logger.debug("AGW aput_writes response: %s", resp.text)
         resp.raise_for_status()
 
     async def adelete_thread(self, thread_id: str) -> None:
-        resp = await self._http("DELETE", f"/checkpoint/{thread_id}")
+        resp = await self._http(
+            "DELETE",
+            f"/checkpoint/{thread_id}",
+            label_path="/checkpoint/{threadId}",
+        )
         resp.raise_for_status()
 
     # =================================================================

{agent_lab_sdk-0.1.43.dist-info → agent_lab_sdk-0.1.44.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.43
+Version: 0.1.44
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved

{agent_lab_sdk-0.1.43.dist-info → agent_lab_sdk-0.1.44.dist-info}/RECORD

@@ -1,6 +1,6 @@
 agent_lab_sdk/__init__.py,sha256=1Dlmv-wuz1QuciymKtYtX7jXzr_fkeGTe7aENfEDl3E,108
 agent_lab_sdk/langgraph/checkpoint/__init__.py,sha256=DnKwR1LwbaQ3qhb124lE-tnojrUIVcCdNzHEHwgpL5M,86
-agent_lab_sdk/langgraph/checkpoint/agw_saver.py,sha256=N1S3xANf2cHiJa92nGeapcgq5PmcnK5dEqlqBZDsBkM,32160
+agent_lab_sdk/langgraph/checkpoint/agw_saver.py,sha256=21IkZ7aZrmYUpd3fmMuc4IhfQaDntoSWbRKnYlFjieY,32721
 agent_lab_sdk/langgraph/checkpoint/serde.py,sha256=UTSYbTbhBeL1CAr-XMbaH3SSIx9TeiC7ak22duXvqkw,5175
 agent_lab_sdk/llm/__init__.py,sha256=Yo9MbYdHS1iX05A9XiJGwWN1Hm4IARGav9mNFPrtDeA,376
 agent_lab_sdk/llm/agw_token_manager.py,sha256=_bPPI8muaEa6H01P8hHQOJHiiivaLd8N_d3OT9UT_80,4787
@@ -15,8 +15,8 @@ agent_lab_sdk/schema/log_message.py,sha256=nadi6lZGRuDSPmfbYs9QPpRJUT9Pfy8Y7pGCv
 agent_lab_sdk/storage/__init__.py,sha256=HAtUoqg3k0irqPMewayadVA9aXJOmYSxRr6a5J1scT0,174
 agent_lab_sdk/storage/storage.py,sha256=ELpt7GRwFD-aWa6ctinfA_QwcvzWLvKS0Wz8FlxVqAs,2075
 agent_lab_sdk/storage/storage_v2.py,sha256=ONseynX59xzWK17dfzxZvnii2rpz3Oo2Zo9Ck-lcGnw,1997
-agent_lab_sdk-0.1.43.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.43.dist-info/METADATA,sha256=i1vL5heUqbEJWzLygCLR1fXet6qHNMpqUyzn2UnywX0,20766
-agent_lab_sdk-0.1.43.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.43.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.43.dist-info/RECORD,,
+agent_lab_sdk-0.1.44.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.44.dist-info/METADATA,sha256=gbWuS14rM1KtS4W8f6Jze8orWGmjzcWgqJpdnfE7NzY,20766
+agent_lab_sdk-0.1.44.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.44.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.44.dist-info/RECORD,,