nucliadb-sdk 6.3.5.post3980__py3-none-any.whl → 6.3.5.post3985__py3-none-any.whl

nucliadb_sdk/tests/fixtures.py

@@ -28,8 +28,8 @@ from typing import Optional
 from uuid import uuid4
 
 import docker  # type: ignore
+import httpx
 import pytest
-import requests
 from pytest_docker_fixtures import images  # type: ignore
 from pytest_docker_fixtures.containers._base import BaseImage  # type: ignore
 
@@ -71,7 +71,7 @@ class NucliaDB(BaseImage):
 
     def check(self):
         try:
-            response = requests.get(f"http://{self.host}:{self.get_port()}")
+            response = httpx.get(f"http://{self.host}:{self.get_port()}")
             return response.status_code == 200
         except Exception:
             return False
@@ -132,13 +132,13 @@ def nucliadb(pg):
             # Wait for service ready
             for _ in range(10):
                 try:
-                    if requests.get(f"http://localhost:8080").status_code == 200:
+                    if httpx.get(f"http://localhost:8080").status_code == 200:
                         break
                 except Exception:
                     pass
                 time.sleep(1)
 
-            if requests.get(f"http://localhost:8080").status_code != 200:
+            if httpx.get(f"http://localhost:8080").status_code != 200:
                 raise Exception("No NucliaDB Running")
 
         yield NucliaFixture(
@@ -204,7 +204,7 @@ async def init_fixture(
     kb_obj = sdk.create_knowledge_box(slug=slug)
     kbid = kb_obj.uuid
 
-    import_resp = requests.get(dataset_location)
+    import_resp = httpx.get(dataset_location)
     assert import_resp.status_code == 200, (
         f"Error pulling dataset {dataset_location}:{import_resp.status_code}"
     )

nucliadb_sdk/v2/sdk.py

@@ -19,6 +19,7 @@
 from __future__ import annotations
 
 import enum
+import importlib.metadata
 import inspect
 import io
 import warnings
@@ -119,6 +120,7 @@ OUTPUT_TYPE = TypeVar("OUTPUT_TYPE", bound=Union[BaseModel, None])
 RawRequestContent = Union[str, bytes, Iterable[bytes], AsyncIterable[bytes], dict[str, Any]]
 
 INPUT_TYPE = TypeVar("INPUT_TYPE", BaseModel, List[InputMessage], RawRequestContent, object, None)
+USER_AGENT = f"nucliadb-sdk/{importlib.metadata.version('nucliadb_sdk')}"
 
 
 class Region(enum.Enum):
@@ -821,7 +823,7 @@ class _NucliaDBBase:
             if api_key is not None:
                 headers["X-STF-SERVICEACCOUNT"] = f"Bearer {api_key}"
 
-        self.headers = headers
+        self.headers = {"User-Agent": USER_AGENT, **headers}
 
     def _request(
         self,

{nucliadb_sdk-6.3.5.post3980.dist-info → nucliadb_sdk-6.3.5.post3985.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_sdk
-Version: 6.3.5.post3980
+Version: 6.3.5.post3985
 Summary: NucliaDB SDK
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -20,11 +20,10 @@ Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: <4,>=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: httpx
-Requires-Dist: requests
 Requires-Dist: orjson
 Requires-Dist: pydantic>=2.6
 Requires-Dist: nuclia-models>=0.24.2
-Requires-Dist: nucliadb-models>=6.3.5.post3980
+Requires-Dist: nucliadb-models>=6.3.5.post3985
 
 # NucliaDB SDK
 

{nucliadb_sdk-6.3.5.post3980.dist-info → nucliadb_sdk-6.3.5.post3985.dist-info}/RECORD

@@ -1,11 +1,11 @@
 nucliadb_sdk/__init__.py,sha256=CkjYIXKAJkX7qnZiN-sYtvIDp0h4IEazc8HhvMuZiYc,990
 nucliadb_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nucliadb_sdk/tests/__init__.py,sha256=itSI7dtTwFP55YMX4iK7JzdMHS5CQVUiB1XzQu4UBh8,833
-nucliadb_sdk/tests/fixtures.py,sha256=Skx6UzLnxeTC17zE3HzEKWTHpYBsr_Pm2UyqDaxj23E,7052
+nucliadb_sdk/tests/fixtures.py,sha256=HrqlUoVHnF1PGn0MrXrBeT_Fjv8kUuNa9SINUXamPq4,7037
 nucliadb_sdk/v2/__init__.py,sha256=F9kJzOkphCL6fIGXBkQbXAAYS2ZzkOLzA4Nudwf1310,934
 nucliadb_sdk/v2/exceptions.py,sha256=cof-xdkqjWn1aGx1ul6Q4ydX97DbLie-LfwfX_PlhAQ,1348
-nucliadb_sdk/v2/sdk.py,sha256=5W_jBrEW6PUkHQyb-LUFBJn0LV4WvC_QGVeLdhSP6fM,46938
-nucliadb_sdk-6.3.5.post3980.dist-info/METADATA,sha256=_9Oa2Wmo5CVYeR1pClgSRUdivKHcew522UiKkP2x_aA,6719
-nucliadb_sdk-6.3.5.post3980.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nucliadb_sdk-6.3.5.post3980.dist-info/top_level.txt,sha256=_dCwt_JnsZ3463lfvc5KcM2wUQJ9aSvKSsAAjGH8R0Y,13
-nucliadb_sdk-6.3.5.post3980.dist-info/RECORD,,
+nucliadb_sdk/v2/sdk.py,sha256=sakK-_uwarkIL23P7AQXyu18-3vtRRAqx6syZyzBqXo,47068
+nucliadb_sdk-6.3.5.post3985.dist-info/METADATA,sha256=vdxFiQh_ePnnQ0hl6CGsLb8uDB374sVxn22opt10Bho,6695
+nucliadb_sdk-6.3.5.post3985.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nucliadb_sdk-6.3.5.post3985.dist-info/top_level.txt,sha256=_dCwt_JnsZ3463lfvc5KcM2wUQJ9aSvKSsAAjGH8R0Y,13
+nucliadb_sdk-6.3.5.post3985.dist-info/RECORD,,