Bundle neptune-api 0.26.0 inside neptune-query (#175)

* Copy code from neptune-api 0.26.0 to src/neptune_query/generated/neptune_api/

* Update nq code to use the bundled neptune_api rather than one from PyPI

* formatting fix

Author: gabrys

.pre-commit-config.yaml

@@ -1,3 +1,4 @@
+exclude: ^src/neptune_query/generated/
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v6.0.0

pyproject.toml

@@ -11,11 +11,16 @@ pattern = "default-unprefixed"
 [tool.poetry.dependencies]
 python = "^3.10"
 
-# Base neptune package
-neptune-api = ">=0.26.0,<0.27.0"
 azure-storage-blob = "^12.7.0"
 pandas = ">=1.4.0"
 
+# neptune-api dependencies:
+httpx = { version = ">=0.15.4,<0.28.2", extras = ["http2"] }
+attrs = ">=21.3.0"
+python-dateutil = "^2.8.0"
+PyJWT = "^2.0.0"
+protobuf = ">=4.21.1,<7"
+
 # Optional for default progress update handling
 tqdm = { version = ">=4.66.0" }
 

src/neptune_query/generated/neptune_api/__init__.py

@@ -0,0 +1,26 @@
+#
+# Copyright (c) 2025, Neptune Labs Sp. z o.o.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""A client library for accessing Neptune Leaderboard REST API"""
+
+from .client import (
+    AuthenticatedClient,
+    Client,
+)
+
+__all__ = (
+    "AuthenticatedClient",
+    "Client",
+)

src/neptune_query/generated/neptune_api/api/__init__.py

@@ -0,0 +1,16 @@
+#
+# Copyright (c) 2025, Neptune Labs Sp. z o.o.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Contains methods for accessing the API"""

src/neptune_query/generated/neptune_api/api/backend/__init__.py

@@ -0,0 +1,14 @@
+#
+# Copyright (c) 2025, Neptune Labs Sp. z o.o.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

src/neptune_query/generated/neptune_api/api/backend/exchange_api_token.py

@@ -0,0 +1,199 @@
+#
+# Copyright (c) 2025, Neptune Labs Sp. z o.o.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from http import HTTPStatus
+from typing import (
+    Any,
+    Dict,
+    Optional,
+    Union,
+    cast,
+)
+
+import httpx
+
+from ... import errors
+from ...client import (
+    AuthenticatedClient,
+    Client,
+)
+from ...models.error import Error
+from ...models.neptune_oauth_token import NeptuneOauthToken
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    x_neptune_api_token: str,
+) -> Dict[str, Any]:
+    headers: Dict[str, Any] = {}
+    headers["X-Neptune-Api-Token"] = x_neptune_api_token
+
+    _kwargs: Dict[str, Any] = {
+        "method": "get",
+        "url": "/api/backend/v1/authorization/oauth-token",
+    }
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[Union[Any, Error, NeptuneOauthToken]]:
+    try:
+        if response.status_code == HTTPStatus.OK:
+            response_200 = NeptuneOauthToken.from_dict(response.json())
+
+            return response_200
+        if response.status_code == HTTPStatus.BAD_REQUEST:
+            response_400 = Error.from_dict(response.json())
+
+            return response_400
+        if response.status_code == HTTPStatus.UNAUTHORIZED:
+            response_401 = cast(Any, None)
+            return response_401
+        if response.status_code == HTTPStatus.FORBIDDEN:
+            response_403 = cast(Any, None)
+            return response_403
+        if response.status_code == HTTPStatus.NOT_FOUND:
+            response_404 = cast(Any, None)
+            return response_404
+        if response.status_code == HTTPStatus.REQUEST_TIMEOUT:
+            response_408 = cast(Any, None)
+            return response_408
+        if response.status_code == HTTPStatus.UNPROCESSABLE_ENTITY:
+            response_422 = cast(Any, None)
+            return response_422
+    except Exception as e:
+        raise errors.UnableToParseResponse(e, response) from e
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[Union[Any, Error, NeptuneOauthToken]]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+    x_neptune_api_token: str,
+) -> Response[Union[Any, Error, NeptuneOauthToken]]:
+    """
+    Args:
+        x_neptune_api_token (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Union[Any, Error, NeptuneOauthToken]]
+    """
+
+    kwargs = _get_kwargs(
+        x_neptune_api_token=x_neptune_api_token,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+    x_neptune_api_token: str,
+) -> Optional[Union[Any, Error, NeptuneOauthToken]]:
+    """
+    Args:
+        x_neptune_api_token (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Union[Any, Error, NeptuneOauthToken]
+    """
+
+    return sync_detailed(
+        client=client,
+        x_neptune_api_token=x_neptune_api_token,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+    x_neptune_api_token: str,
+) -> Response[Union[Any, Error, NeptuneOauthToken]]:
+    """
+    Args:
+        x_neptune_api_token (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Union[Any, Error, NeptuneOauthToken]]
+    """
+
+    kwargs = _get_kwargs(
+        x_neptune_api_token=x_neptune_api_token,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+    x_neptune_api_token: str,
+) -> Optional[Union[Any, Error, NeptuneOauthToken]]:
+    """
+    Args:
+        x_neptune_api_token (str):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Union[Any, Error, NeptuneOauthToken]
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            x_neptune_api_token=x_neptune_api_token,
+        )
+    ).parsed