groundx 3.3.2__tar.gz → 3.3.4__tar.gz

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

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

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

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

{groundx-3.3.2 → groundx-3.3.4}/src/groundx/__init__.py

@@ -40,6 +40,7 @@ if typing.TYPE_CHECKING:
         IngestStatusProgressComplete,
         IngestStatusProgressErrors,
         IngestStatusProgressProcessing,
+        IngestStatusProgressQueued,
         MessageResponse,
         MeterDetail,
         ProcessLevel,
@@ -120,6 +121,7 @@ _dynamic_imports: typing.Dict[str, str] = {
     "IngestStatusProgressComplete": ".types",
     "IngestStatusProgressErrors": ".types",
     "IngestStatusProgressProcessing": ".types",
+    "IngestStatusProgressQueued": ".types",
     "MessageResponse": ".types",
     "MeterDetail": ".types",
     "ProcessLevel": ".types",
@@ -228,6 +230,7 @@ __all__ = [
     "IngestStatusProgressComplete",
     "IngestStatusProgressErrors",
     "IngestStatusProgressProcessing",
+    "IngestStatusProgressQueued",
     "MessageResponse",
     "MeterDetail",
     "ProcessLevel",

{groundx-3.3.2 → groundx-3.3.4}/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.4",
             "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.4",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-Key"] = self.api_key

{groundx-3.3.2 → groundx-3.3.4}/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,

{groundx-3.3.2 → groundx-3.3.4}/src/groundx/types/__init__.py

@@ -39,6 +39,7 @@ if typing.TYPE_CHECKING:
     from .ingest_status_progress_complete import IngestStatusProgressComplete
     from .ingest_status_progress_errors import IngestStatusProgressErrors
     from .ingest_status_progress_processing import IngestStatusProgressProcessing
+    from .ingest_status_progress_queued import IngestStatusProgressQueued
     from .message_response import MessageResponse
     from .meter_detail import MeterDetail
     from .process_level import ProcessLevel
@@ -107,6 +108,7 @@ _dynamic_imports: typing.Dict[str, str] = {
     "IngestStatusProgressComplete": ".ingest_status_progress_complete",
     "IngestStatusProgressErrors": ".ingest_status_progress_errors",
     "IngestStatusProgressProcessing": ".ingest_status_progress_processing",
+    "IngestStatusProgressQueued": ".ingest_status_progress_queued",
     "MessageResponse": ".message_response",
     "MeterDetail": ".meter_detail",
     "ProcessLevel": ".process_level",
@@ -199,6 +201,7 @@ __all__ = [
     "IngestStatusProgressComplete",
     "IngestStatusProgressErrors",
     "IngestStatusProgressProcessing",
+    "IngestStatusProgressQueued",
     "MessageResponse",
     "MeterDetail",
     "ProcessLevel",

{groundx-3.3.2 → groundx-3.3.4}/src/groundx/types/ingest_status_progress.py

@@ -8,6 +8,7 @@ from .ingest_status_progress_cancelled import IngestStatusProgressCancelled
 from .ingest_status_progress_complete import IngestStatusProgressComplete
 from .ingest_status_progress_errors import IngestStatusProgressErrors
 from .ingest_status_progress_processing import IngestStatusProgressProcessing
+from .ingest_status_progress_queued import IngestStatusProgressQueued
 
 
 class IngestStatusProgress(UniversalBaseModel):
@@ -15,6 +16,7 @@ class IngestStatusProgress(UniversalBaseModel):
     complete: typing.Optional[IngestStatusProgressComplete] = None
     errors: typing.Optional[IngestStatusProgressErrors] = None
     processing: typing.Optional[IngestStatusProgressProcessing] = None
+    queued: typing.Optional[IngestStatusProgressQueued] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

groundx-3.3.4/src/groundx/types/ingest_status_progress_queued.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .document_detail import DocumentDetail
+
+
+class IngestStatusProgressQueued(UniversalBaseModel):
+    documents: typing.Optional[typing.List[DocumentDetail]] = None
+    total: typing.Optional[int] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow