dragoneye-python 1.0.1__tar.gz → 1.0.2__tar.gz

{dragoneye_python-1.0.1 → dragoneye_python-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragoneye-python
-Version: 1.0.1
+Version: 1.0.2
 Summary: Official Python SDK for the Dragoneye computer vision API
 Author-email: Dragoneye <support@dragoneye.ai>
 License-Expression: MIT

{dragoneye_python-1.0.1 → dragoneye_python-1.0.2}/dragoneye_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragoneye-python
-Version: 1.0.1
+Version: 1.0.2
 Summary: Official Python SDK for the Dragoneye computer vision API
 Author-email: Dragoneye <support@dragoneye.ai>
 License-Expression: MIT

{dragoneye_python-1.0.1 → dragoneye_python-1.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dragoneye-python"
-version = "1.0.1"
+version = "1.0.2"
 description = "Official Python SDK for the Dragoneye computer vision API"
 readme = "README.md"
 requires-python = ">=3.8"

{dragoneye_python-1.0.1 → dragoneye_python-1.0.2}/src/dragoneye/classification.py

@@ -21,7 +21,6 @@ from .parquet_deserializer import (
     deserialize_video_predictions,
 )
 from .types.common import (
-    BASE_API_URL,
     PredictionTaskState,
     PredictionTaskUUID,
     PredictionType,
@@ -136,7 +135,7 @@ class Classification:
         Given a prediction task UUID, return
         """
         query = urlencode({"predictionTaskUuid": prediction_task_uuid})
-        url = f"{BASE_API_URL}/prediction-task/status?{query}"
+        url = f"{self._client.base_api_url}/prediction-task/status?{query}"
         headers = {"X-API-Key": self._client.api_key}
 
         @self._backoff_on_429
@@ -200,7 +199,7 @@ class Classification:
                 "response_version": "parquet",
             }
         )
-        url = f"{BASE_API_URL}/prediction-task/results?{query}"
+        url = f"{self._client.base_api_url}/prediction-task/results?{query}"
         headers = {"X-API-Key": self._client.api_key}
 
         @self._backoff_on_429
@@ -284,13 +283,14 @@ class Classification:
             mime_type=media.mime_type,
             frames_per_second=frames_per_second,
             file_name=media.name,
+            **kwargs,
         )
 
         await self._upload_media_to_prediction_task(
             media, prediction_task_begin_response.signed_urls[0]
         )
 
-        predict_url = f"{BASE_API_URL}/predict"
+        predict_url = f"{self._client.base_api_url}/predict"
         predict_data: dict[str, Any] = {
             "model_name": model_name,
             "prediction_task_uuid": prediction_task_begin_response.prediction_task_uuid,
@@ -385,8 +385,9 @@ class Classification:
         mime_type: str,
         frames_per_second: Optional[int],
         file_name: Optional[str],
+        **kwargs: Any,
     ) -> _PredictionTaskBeginResponse:
-        url = f"{BASE_API_URL}/prediction-task/begin"
+        url = f"{self._client.base_api_url}/prediction-task/begin"
 
         form_data = aiohttp.FormData()
         form_data.add_field("mimetype", mime_type)
@@ -397,6 +398,9 @@ class Classification:
         if frames_per_second is not None:
             form_data.add_field("frames_per_second", str(frames_per_second))
 
+        for key, value in kwargs.items():
+            form_data.add_field(key, str(value))
+
         headers = {
             "X-API-Key": self._client.api_key,
         }

{dragoneye_python-1.0.1 → dragoneye_python-1.0.2}/src/dragoneye/client.py

@@ -4,6 +4,7 @@ from typing import Callable, Optional
 import backoff
 
 from dragoneye.classification import Classification
+from dragoneye.types.common import BASE_API_URL
 
 
 class Dragoneye:
@@ -13,6 +14,7 @@ class Dragoneye:
         max_retries: int = 10,
         max_backoff_time: int = 120,
         backoff_jitter: Callable[[float], float] = backoff.full_jitter,
+        base_api_url: str = BASE_API_URL,
     ):
         if api_key is None:
             api_key = os.getenv("DRAGONEYE_API_KEY")
@@ -25,5 +27,6 @@ class Dragoneye:
         self.max_retries = max_retries
         self.max_backoff_time = max_backoff_time
         self.backoff_jitter = backoff_jitter
+        self.base_api_url = base_api_url
 
         self.classification = Classification(self)