nucliadb-sdk 6.6.1.post4601__py3-none-any.whl → 6.6.1.post4617__py3-none-any.whl

nucliadb_sdk/v2/sdk.py

@@ -640,7 +640,12 @@ def _request_sync_builder(
     path_template = sdk_def.path_template
     path_params = sdk_def.path_params
 
-    def _func(self: NucliaDB, content: Optional[INPUT_TYPE] = None, **kwargs) -> OUTPUT_TYPE:
+    def _func(
+        self: NucliaDB,
+        content: Optional[INPUT_TYPE] = None,
+        headers: Optional[Dict[str, str]] = None,
+        **kwargs,
+    ) -> OUTPUT_TYPE:
         path, data, query_params = prepare_request(
             path_template=path_template,
             path_params=path_params,
@@ -648,7 +653,9 @@ def _request_sync_builder(
             content=content,
             **kwargs,
         )
-        resp = self._request(path, method, content=data, query_params=query_params)
+        resp = self._request(
+            path, method, content=data, query_params=query_params, extra_headers=headers
+        )
         if response_type is not None:
             if issubclass(response_type, SyncAskResponse):
                 return ask_response_parser(response_type, resp)  # type: ignore
@@ -726,7 +733,12 @@ def _request_async_builder(
     path_template = sdk_def.path_template
     path_params = sdk_def.path_params
 
-    async def _func(self: NucliaDBAsync, content: Optional[INPUT_TYPE] = None, **kwargs) -> OUTPUT_TYPE:
+    async def _func(
+        self: NucliaDBAsync,
+        content: Optional[INPUT_TYPE] = None,
+        headers: Optional[Dict[str, str]] = None,
+        **kwargs,
+    ) -> OUTPUT_TYPE:
         path, data, query_params = prepare_request(
             path_template=path_template,
             path_params=path_params,
@@ -734,7 +746,9 @@ def _request_async_builder(
             content=content,
             **kwargs,
         )
-        resp = await self._request(path, method, content=data, query_params=query_params)
+        resp = await self._request(
+            path, method, content=data, query_params=query_params, extra_headers=headers
+        )
         if response_type is not None:
             if isinstance(response_type, type) and issubclass(response_type, SyncAskResponse):
                 return ask_response_parser(response_type, resp)  # type: ignore
@@ -841,6 +855,7 @@ class _NucliaDBBase:
         method: str,
         query_params: Optional[Dict[str, str]] = None,
         content: Optional[RawRequestContent] = None,
+        extra_headers: Optional[Dict[str, str]] = None,
     ):
         raise NotImplementedError
 
@@ -939,6 +954,7 @@ class NucliaDB(_NucliaDBBase):
         method: str,
         query_params: Optional[Dict[str, str]] = None,
         content: Optional[RawRequestContent] = None,
+        extra_headers: Optional[Dict[str, str]] = None,
     ):
         url = f"{self.base_url}{path}"
         opts: Dict[str, Any] = {}
@@ -948,6 +964,8 @@ class NucliaDB(_NucliaDBBase):
             opts["content"] = content
         if query_params is not None:
             opts["params"] = query_params
+        if extra_headers is not None:
+            opts["headers"] = extra_headers
         response: httpx.Response = getattr(self.session, method.lower())(url, **opts)
         return self._check_response(response)
 
@@ -1137,6 +1155,7 @@ class NucliaDBAsync(_NucliaDBBase):
         method: str,
         query_params: Optional[Dict[str, str]] = None,
         content: Optional[RawRequestContent] = None,
+        extra_headers: Optional[Dict[str, str]] = None,
     ):
         url = f"{self.base_url}{path}"
         opts: Dict[str, Any] = {}
@@ -1146,6 +1165,8 @@ class NucliaDBAsync(_NucliaDBBase):
             opts["content"] = content
         if query_params is not None:
             opts["params"] = query_params
+        if extra_headers is not None:
+            opts["headers"] = extra_headers
         response: httpx.Response = await getattr(self.session, method.lower())(url, **opts)
         return self._check_response(response)
 

{nucliadb_sdk-6.6.1.post4601.dist-info → nucliadb_sdk-6.6.1.post4617.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_sdk
-Version: 6.6.1.post4601
+Version: 6.6.1.post4617
 Summary: NucliaDB SDK
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0
@@ -22,7 +22,7 @@ Requires-Dist: httpx
 Requires-Dist: orjson
 Requires-Dist: pydantic>=2.6
 Requires-Dist: nuclia-models>=0.43.0
-Requires-Dist: nucliadb-models>=6.6.1.post4601
+Requires-Dist: nucliadb-models>=6.6.1.post4617
 
 # NucliaDB SDK
 

{nucliadb_sdk-6.6.1.post4601.dist-info → nucliadb_sdk-6.6.1.post4617.dist-info}/RECORD

@@ -4,8 +4,8 @@ nucliadb_sdk/tests/__init__.py,sha256=74CI2MDrgAvfcsk1dRt543dUrqrfyWc6IFQdDdy2k_
 nucliadb_sdk/tests/fixtures.py,sha256=jeWHoLcJGOwX1-Dre-sqVFwwg5uWnR09ujLGPy7c3nQ,6731
 nucliadb_sdk/v2/__init__.py,sha256=XmxB8o1AUa3eWoxVHKMqXH88AESKM9IL8hFuI0KTR00,682
 nucliadb_sdk/v2/exceptions.py,sha256=fk4sEOUOgGpYe59DLxIkHflo4T6RcniwjY3VZGk4A70,1096
-nucliadb_sdk/v2/sdk.py,sha256=UWPVm9pHgkrEoycQasmM95VPDkUzugEZQ1DJNZiPIaw,48016
-nucliadb_sdk-6.6.1.post4601.dist-info/METADATA,sha256=5S486FJtLzw_lp08wc-lcEuT_xH2xM_Gs1pi1ttEP1U,6617
-nucliadb_sdk-6.6.1.post4601.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb_sdk-6.6.1.post4601.dist-info/top_level.txt,sha256=_dCwt_JnsZ3463lfvc5KcM2wUQJ9aSvKSsAAjGH8R0Y,13
-nucliadb_sdk-6.6.1.post4601.dist-info/RECORD,,
+nucliadb_sdk/v2/sdk.py,sha256=0NZ6GsWLKIjjuFeRA07dYIOxfhRzRWdUaywEMRjPMx4,48600
+nucliadb_sdk-6.6.1.post4617.dist-info/METADATA,sha256=tfck1NeBR4xPgmF8fLxDwID9kB0dMT-08wlyUqXem5Y,6617
+nucliadb_sdk-6.6.1.post4617.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb_sdk-6.6.1.post4617.dist-info/top_level.txt,sha256=_dCwt_JnsZ3463lfvc5KcM2wUQJ9aSvKSsAAjGH8R0Y,13
+nucliadb_sdk-6.6.1.post4617.dist-info/RECORD,,