unique-sdk 2026.34.0.dev3__tar.gz → 2026.34.0.dev5__tar.gz

{unique_sdk-2026.34.0.dev3 → unique_sdk-2026.34.0.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-sdk
-Version: 2026.34.0.dev3
+Version: 2026.34.0.dev5
 Summary: 
 Author: Martin Fadler, Konstantin Krauss, Andreas Hauri
 Author-email: Martin Fadler <martin.fadler@unique.ch>, Konstantin Krauss <konstantin@unique.ch>, Andreas Hauri <andreas@unique.ch>

{unique_sdk-2026.34.0.dev3 → unique_sdk-2026.34.0.dev5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_sdk"
-version = "2026.34.0.dev3"
+version = "2026.34.0.dev5"
 description = ""
 readme = "README.md"
 license = { text = "MIT" }

{unique_sdk-2026.34.0.dev3 → unique_sdk-2026.34.0.dev5}/unique_sdk/__init__.py

@@ -15,6 +15,10 @@ api_verify_mode: bool = True
 default_http_client: "HTTPClient | None" = None
 ingestion_upload_api_url_internal: str | None = None
 
+# Timeout (seconds) for the direct-to-blob requests in utils/file_io.py,
+# which bypass _http_client. Defaults to match _http_client's own default.
+blob_transfer_timeout: float = 600
+
 # Set to either 'debug' or 'info', controls console logging
 log: Literal["debug", "info"] | None = None
 

{unique_sdk-2026.34.0.dev3 → unique_sdk-2026.34.0.dev5}/unique_sdk/_api_resource.py

@@ -194,12 +194,16 @@ class APIResource(UniqueObject, Generic[T]):
         user_id: str | None = None,
         company_id: str | None = None,
         params=None,
+        supplied_headers: Mapping[str, str] | None = None,
     ):
         params = None if params is None else params.copy()
 
         requestor = APIRequestor(user_id=user_id, company_id=company_id)
 
-        response = requestor.request(method_, url_, params)
+        if supplied_headers is None:
+            response = requestor.request(method_, url_, params)
+        else:
+            response = requestor.request(method_, url_, params, supplied_headers)
         return convert_to_unique_object(response, user_id, company_id, params)
 
     # The `method_` and `url_` arguments are suffixed with an underscore to
@@ -213,10 +217,16 @@ class APIResource(UniqueObject, Generic[T]):
         user_id: str | None = None,
         company_id: str | None = None,
         params=None,
+        supplied_headers: Mapping[str, str] | None = None,
     ):
         params = None if params is None else params.copy()
 
         requestor = APIRequestor(user_id=user_id, company_id=company_id)
 
-        response = await requestor.request_async(method_, url_, params)
+        if supplied_headers is None:
+            response = await requestor.request_async(method_, url_, params)
+        else:
+            response = await requestor.request_async(
+                method_, url_, params, supplied_headers
+            )
         return convert_to_unique_object(response, user_id, company_id, params)

{unique_sdk-2026.34.0.dev3 → unique_sdk-2026.34.0.dev5}/unique_sdk/api_resources/_chat_completion.py

@@ -59,6 +59,7 @@ class ChatCompletion(APIResource["ChatCompletion"]):
         user_id: str | None = None,
         **params: Unpack["ChatCompletion.CreateParams"],
     ) -> "ChatCompletion":
+        supplied_headers = params.pop("headers", None)
         return cast(
             "ChatCompletion",
             cls._static_request(
@@ -67,6 +68,7 @@ class ChatCompletion(APIResource["ChatCompletion"]):
                 company_id=company_id,
                 user_id=user_id,
                 params=params,
+                supplied_headers=supplied_headers,
             ),
         )
 
@@ -77,6 +79,7 @@ class ChatCompletion(APIResource["ChatCompletion"]):
         user_id: str | None = None,
         **params: Unpack["ChatCompletion.CreateParams"],
     ) -> "ChatCompletion":
+        supplied_headers = params.pop("headers", None)
         return cast(
             "ChatCompletion",
             await cls._static_request_async(
@@ -85,5 +88,6 @@ class ChatCompletion(APIResource["ChatCompletion"]):
                 company_id=company_id,
                 user_id=user_id,
                 params=params,
+                supplied_headers=supplied_headers,
             ),
         )

{unique_sdk-2026.34.0.dev3 → unique_sdk-2026.34.0.dev5}/unique_sdk/utils/file_io.py

@@ -46,7 +46,7 @@ def download_file(url: str, filename: str):
     file_path = Path(random_dir) / filename
 
     # Download the file and save it to the random directory
-    response = requests.get(url)
+    response = requests.get(url, timeout=unique_sdk.blob_transfer_timeout)
     if response.status_code == 200:
         with open(file_path, "wb") as file:
             file.write(response.content)
@@ -100,6 +100,7 @@ def _put_preview_pdf(write_url: str, preview_pdf_path: str) -> None:
                 "X-Ms-Blob-Content-Type": _PREVIEW_PDF_MIME_TYPE,
                 "X-Ms-Blob-Type": "BlockBlob",
             },
+            timeout=unique_sdk.blob_transfer_timeout,
         )
     if response.status_code >= 400:
         raise RuntimeError(
@@ -219,13 +220,19 @@ def upload_file(
         raise ValueError("createdContent.writeUrl is None")
     uploadUrl = _apply_ingestion_upload_url_override(uploadUrl)
     with open(path_to_file, "rb") as file:
-        requests.put(
+        response = requests.put(
             uploadUrl,
             data=file,
             headers={
                 "X-Ms-Blob-Content-Type": mime_type,
                 "X-Ms-Blob-Type": "BlockBlob",
             },
+            timeout=unique_sdk.blob_transfer_timeout,
+        )
+    if response.status_code >= 400:
+        raise RuntimeError(
+            f"Content upload failed with status {response.status_code}: "
+            f"{response.text[:500]}"
         )
 
     # Step 3 — finalize upsert: ``byteSize`` + ``fileUrl`` flip the row
@@ -362,7 +369,9 @@ def download_content(
     # Issue the request before resolving the destination. A non-200
     # response should never leave a half-created directory or empty
     # file behind for callers who supplied ``target_path``.
-    response = requests.get(url, headers=headers)
+    response = requests.get(
+        url, headers=headers, timeout=unique_sdk.blob_transfer_timeout
+    )
     if response.status_code != 200:
         raise Exception(f"Error downloading file: Status code {response.status_code}")