groundx 3.3.2__tar.gz → 3.3.3__tar.gz

{groundx-3.3.2 → groundx-3.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: groundx
-Version: 3.3.2
+Version: 3.3.3
 Summary: 
 License: MIT
 Requires-Python: >=3.8,<4.0

{groundx-3.3.2 → groundx-3.3.3}/pyproject.toml

@@ -4,7 +4,7 @@ dynamic = ["version"]
 
 [tool.poetry]
 name = "groundx"
-version = "3.3.2"
+version = "3.3.3"
 description = ""
 readme = "README.md"
 authors = []

{groundx-3.3.2 → groundx-3.3.3}/src/groundx/core/client_wrapper.py

@@ -24,12 +24,12 @@ class BaseClientWrapper:
         import platform
 
         headers: typing.Dict[str, str] = {
-            "User-Agent": "groundx/3.3.2",
+            "User-Agent": "groundx/3.3.3",
             "X-Fern-Language": "Python",
             "X-Fern-Runtime": f"python/{platform.python_version()}",
             "X-Fern-Platform": f"{platform.system().lower()}/{platform.release()}",
             "X-Fern-SDK-Name": "groundx",
-            "X-Fern-SDK-Version": "3.3.2",
+            "X-Fern-SDK-Version": "3.3.3",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-Key"] = self.api_key

{groundx-3.3.2 → groundx-3.3.3}/src/groundx/extract/services/status.py

@@ -1,10 +1,12 @@
-import typing
+import time, typing
 
 from fastapi import Response
 
 from .logger import Logger
 from ..settings.settings import ContainerSettings
 
+latency_to = 60
+
 
 class Status:
     def __init__(
@@ -85,6 +87,9 @@ class Status:
 
         return available, total
 
+    def key_api_latency(self, id: str) -> str:
+        return f"{id}:{time.time_ns()}:latency"
+
     def key_worker_available(self, id: str) -> str:
         return f"{self.config.service}:{id}:requests"
 
@@ -122,6 +127,16 @@ class Status:
     def refresh_worker_total(self, id: str, to: typing.Optional[int] = None) -> None:
         self.set_value(self.key_worker_total(id), self.config.workers, to)
 
+    def set_api_latency(
+        self, id: str, val: int, to: typing.Optional[int] = None
+    ) -> None:
+        if to is None:
+            global latency_to
+
+            to = latency_to
+
+        self.set_value(self.key_api_latency(id), val, to)
+
     def set_headers(
         self,
         response: Response,