luna-quantum 0.0.37__py3-none-any.whl → 0.10.1__py3-none-any.whl

{luna_quantum-0.0.37.dist-info → luna_quantum-0.10.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: luna-quantum
-Version: 0.0.37
+Version: 0.10.1
 Summary: Python SDK for Aqarios' Luna Platform
 Home-page: https://aqarios.com/
 License: Apache-2.0

{luna_quantum-0.0.37.dist-info → luna_quantum-0.10.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 luna_sdk/__init__.py,sha256=9s6JkcSeRF4WtFo80TelzXXHFhX8w7fXiaJr59V9ScM,100
 luna_sdk/controllers/__init__.py,sha256=ivTapH8np5mQeVEevxwsWawCtVSG4Wep72rNp7nEZXw,60
 luna_sdk/controllers/luna_http_client.py,sha256=_q3_o-bjC61uQMnNTfbbGei0eNR_AoIv-2R1cYRfEgU,819
-luna_sdk/controllers/luna_platform_client.py,sha256=EH1jxruuRKwaJ1Vos79PGT4-hwp7jdoSaRFKqjUMbZA,2406
+luna_sdk/controllers/luna_platform_client.py,sha256=TUeK9hlbY3LdYkl0UO2FJtCzGFuU8lshA-kdutkDZlM,2406
 luna_sdk/controllers/luna_q.py,sha256=cv28o2fYDSaI8iAsfMYWtR68mfuUDxY3jmBWY2KnkRk,1199
 luna_sdk/controllers/luna_solve.py,sha256=ioMdB7Rlx-1-XB5WE5OT8oVlO_T1l5T64R-ccaNUCAA,1797
 luna_sdk/controllers/luna_transform.py,sha256=7kdIp5oRMG1NQZ1hBSkcNrj-z6g_2dWvQZ4j8BK3M0A,1337
@@ -24,9 +24,9 @@ luna_sdk/interfaces/cplex_repo_i.py,sha256=a2CWG1m4YmQAlcM022Xfk09BHVmcd3y2H-GMr
 luna_sdk/interfaces/info_repo_i.py,sha256=hCUYNoqjAKwz6l-61XtymX-UboosDX1RaTTiX64EXXc,1345
 luna_sdk/interfaces/lp_repo_i.py,sha256=hixIKvJ8jz9VOUi33B3K2VvFLcR7eKfeooJwktquyhc,3052
 luna_sdk/interfaces/optimization_repo_i.py,sha256=4JZikxWTJZTX4tG85ydSP711WROQ5B8RtbvOp1EDvxE,6202
-luna_sdk/interfaces/qpu_token_repo_i.py,sha256=OeF9PRcGBjHzNwrllt266M2ELJ7h03c7gt_16FPh4yg,4351
+luna_sdk/interfaces/qpu_token_repo_i.py,sha256=oTxYxoeMOPJUV4-U1Fd1kD76EVPGmi1Sp9C1J4cvffY,11418
 luna_sdk/interfaces/repository_i.py,sha256=vJz8pbggkuLLC333qzjepj3TMPh49LUnyh3pJDlcGi0,287
-luna_sdk/interfaces/solutions_repo_i.py,sha256=Xu-8qQkPoedIp47Uc0ogv5HS45Plsi6zRBEQouxRws0,7463
+luna_sdk/interfaces/solutions_repo_i.py,sha256=uSAClqt_zfg-k9VvVn8DfkG2vE2NXyCqaxRlaK8mCb4,8012
 luna_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 luna_sdk/repositories/__init__.py,sha256=HxD02fpzlcIt5YkBHd7LwIp_eE6eFuAz2VDnlrakwJQ,169
 luna_sdk/repositories/circuit_repo.py,sha256=gpjEmCRrl8UghJrelOt6VOULIJmM-thJAhf8oaA5Ckg,1964
@@ -34,14 +34,15 @@ luna_sdk/repositories/cplex_repo.py,sha256=6vlx6YTJoDSWFSZDDNrPk3E7pa5WWGqG25bhT
 luna_sdk/repositories/info_repo.py,sha256=GzRW5RwhRJrNjujw8Y7vJY8vV_tHzbfwXo0viHuxqUM,1270
 luna_sdk/repositories/lp_repo.py,sha256=T44XMbdbW7k9a4z6JgIj1kb0bBZxepn-93vxdZOQa80,4370
 luna_sdk/repositories/optimization_repo.py,sha256=xHjMdXhzdwt9DtZG7AvGtUgOqWH142veOpAzaNe-nxk,6078
-luna_sdk/repositories/qpu_token_repo.py,sha256=Mn3ojxXhLIC-zbjBzEJis6WKNHUYUVkex_qTcuXeQ3Y,4685
-luna_sdk/repositories/solutions_repo.py,sha256=xI1KQ1JeleBJrBRFxfuxdusb_pgjBLeLhdpFi7nqdXY,8825
+luna_sdk/repositories/qpu_token_repo.py,sha256=_qM9dJB23AR1-Kblx_yJcovYmuYwGcZMjH2va19F8yg,8840
+luna_sdk/repositories/solutions_repo.py,sha256=W-VHmhN6wkL3jSg4FE0IyVLydatLE7LjqI-RhP21XFs,9073
 luna_sdk/schemas/__init__.py,sha256=lTix3zUl2Z8ZLA2BTF4pls5S-kqlVZ3cIdAZwGxPSUI,248
 luna_sdk/schemas/circuit.py,sha256=r3Bv0lyhrvFoI3Gf8eq-PS3kyivTWY3IJmRH0X2LoCk,1067
-luna_sdk/schemas/create/__init__.py,sha256=7G2yflImpZJxHn3l4Op5CsQAxwKGmvWJSttq9kmiHq0,90
+luna_sdk/schemas/create/__init__.py,sha256=WTthzwyS0PI2zSENoHp7EWLciLz_DyFebCzSCFVensU,144
 luna_sdk/schemas/create/circuit.py,sha256=zYtM6SV1QH4JzjYaAhrGNjdNn8EDYsdG1QKwayTqkEY,643
 luna_sdk/schemas/create/optimization.py,sha256=Z9pKoD0y6BjKvl8gQ6XK76bdvq8etvcoNU4RwxnUvHA,759
 luna_sdk/schemas/create/qpu_token.py,sha256=_D_QtYOeZQ6b0lG5ALTDmyr4bU42bUZ8OJ0VdwadMNA,380
+luna_sdk/schemas/create/qpu_token_time_quota.py,sha256=5HyfYuXGZmpqB-j4yuStFc7V3b7niPYHYb0KIuvJAzQ,650
 luna_sdk/schemas/create/qubo.py,sha256=5Y_jWicEnoV82Ubc-f0a_qy9hCuPBumhsGWY8wzMIuA,293
 luna_sdk/schemas/create/solution.py,sha256=zPPgm0wGwfn23ccgBoYs6SMTZNmbT_uO1po_je15WjA,386
 luna_sdk/schemas/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -50,10 +51,10 @@ luna_sdk/schemas/enums/optimization.py,sha256=XtFdcfLxGe-MAL64itjxszhRkTcUzUf83t
 luna_sdk/schemas/enums/problem.py,sha256=HWL6Lc1pqPtJ0J-rXkUTL6WHe2fiBY1ctuZvDnx1hmM,754
 luna_sdk/schemas/enums/qpu_token_type.py,sha256=KBRrB8ExuTTmpp48obxdJGWwWEHb7xgq9Ajjwptl1VY,105
 luna_sdk/schemas/enums/solution.py,sha256=Vax6MX62UDwApbwzf1nrhuFWw1Iiq77gUch7_1roCYo,114
-luna_sdk/schemas/enums/status.py,sha256=KHaOgePqB3GNhFYLo2Sp-JAEXBPN_ZVDmj2WobrT5N4,204
+luna_sdk/schemas/enums/status.py,sha256=D0ZEu6Bg0GspDrjujwrtRAmkF50m2cUT-b0MOiAi4sg,227
 luna_sdk/schemas/enums/timeframe.py,sha256=vwoZsgAYGvqF67qdcXMEnHkqbiRcVY5uK-4-ZTFcurI,238
 luna_sdk/schemas/error_message.py,sha256=svCs-mWQsp3BMEhzZFqskbyP6iS78eVUOPaiWFc4sEQ,307
-luna_sdk/schemas/optimization.py,sha256=X7Eb7K7ximWpmtIItn2sJEgxgN8dmlvuioQV-Q71BrY,3174
+luna_sdk/schemas/optimization.py,sha256=juFzE7UUAnLoLvZneOwpXzzhsSUQ2dpXIX48Bm0Sdzg,3172
 luna_sdk/schemas/optimization_formats/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 luna_sdk/schemas/optimization_formats/bqm.py,sha256=rrmiZYe_iEdgBFusJeJfY8PVSxgf1Wv79FgMQUm2YVs,910
 luna_sdk/schemas/optimization_formats/cqm.py,sha256=VZiNJy4yc-8o3gHoH0E-uaNOHj7UfAGCqznTzKBv0NQ,4257
@@ -62,12 +63,13 @@ luna_sdk/schemas/optimization_formats/qm.py,sha256=yUEZF7FOvXv1Mb5FMiX1q0Rk5w2lt
 luna_sdk/schemas/optimization_formats/qubo.py,sha256=YN-VbPGTgqOKJuJ7t_TT8H_WIWr1ifU974K98W9Blao,135
 luna_sdk/schemas/pretty_base.py,sha256=ZyVdwdY5JfmNlfqZDm6MdTs5roNGtv0Pe805r4N25Z4,1845
 luna_sdk/schemas/qpu_token.py,sha256=CFKbAhFgo0G-LO_lameHH4-iOFUJWx9hmMWi3xQwURw,1508
+luna_sdk/schemas/qpu_token_time_quota.py,sha256=tIlYU2i43m93BOaCUrv62baY3RSIScWBqbD0BBlwES4,656
 luna_sdk/schemas/representation.py,sha256=0RKVV0XxHOJZ1o9grxTq5ri3cGLziyCFObDwQuXIOEY,364
 luna_sdk/schemas/rest/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 luna_sdk/schemas/rest/qpu_token/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 luna_sdk/schemas/rest/qpu_token/qpu_token_source.py,sha256=EPMXHpgxqsQ4uJ1hlwe5gPdSsjuHEuKKAIMtBFgPJ1A,583
 luna_sdk/schemas/rest/qpu_token/token_provider.py,sha256=3Gecn2AmN7E_wAPUac_c55Fg-LwdDKsgbwAN2-0msJU,2985
-luna_sdk/schemas/solution.py,sha256=3I4J-7UXNV95kpK0yjB4iPdq2lx3KB_cZ81m4WNbWwg,8684
+luna_sdk/schemas/solution.py,sha256=bNgd0UpsaQM_TrgTB-CJJiXuTg1ubMJP33y813yd5WY,8818
 luna_sdk/schemas/solver_info.py,sha256=ZUCdTI8zpPZ8EquLYyrLU35pZ4VkzcPUSYeQWkdijHM,799
 luna_sdk/schemas/solver_parameters/aws/__init__.py,sha256=CXjKWzgnP_d1_RvQUfWyaaWvsZ1FxLJN5K61QmRa-uw,33
 luna_sdk/schemas/solver_parameters/aws/optimizer_params.py,sha256=w0eg1SnTpKnuq48YG8JAC1LY0bxsTlZIJ2XbT2ySTHw,1374
@@ -160,7 +162,7 @@ luna_sdk/schemas/wrappers/datetime_wrapper.py,sha256=6usmPJNRYA4Fe_-0NGwe7RYams4
 luna_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 luna_sdk/utils/parameter_finder.py,sha256=LKkd1upH3Djd6eqkcSQ2feNwPUiKi8soYkx6JT1_VSg,2937
 luna_sdk/utils/qpu_tokens.py,sha256=xrh5KSoWzbul-6eRa0ELH9iwEENanE7PIms7jlCt3NY,1641
-luna_quantum-0.0.37.dist-info/LICENSE,sha256=rwwuFPLz36oRvjWu2oEeX42Qtn9gmbh7zRC2OqCFNaI,11342
-luna_quantum-0.0.37.dist-info/METADATA,sha256=199H7vfTIgzdGHF0Bz6eFhCtJnhHMiQuxl93YicnhYk,2311
-luna_quantum-0.0.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-luna_quantum-0.0.37.dist-info/RECORD,,
+luna_quantum-0.10.1.dist-info/LICENSE,sha256=rwwuFPLz36oRvjWu2oEeX42Qtn9gmbh7zRC2OqCFNaI,11342
+luna_quantum-0.10.1.dist-info/METADATA,sha256=BD5AD8c3uY615XODFmYbFvnf9_WsLt4P_e5K457-KsM,2311
+luna_quantum-0.10.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+luna_quantum-0.10.1.dist-info/RECORD,,

luna_sdk/controllers/luna_platform_client.py

@@ -1,6 +1,6 @@
 import os
 from enum import Enum
-from typing import Optional, Literal
+from typing import Literal, Optional
 
 import httpx
 

luna_sdk/interfaces/qpu_token_repo_i.py

@@ -1,9 +1,11 @@
 from abc import ABC, abstractmethod
+from datetime import datetime
 from typing import Dict, List, Optional
 
 from luna_sdk.interfaces.repository_i import IRepository
 from luna_sdk.schemas import QpuTokenOut
 from luna_sdk.schemas.enums.qpu_token_type import QpuTokenTypeEnum
+from luna_sdk.schemas.qpu_token_time_quota import QpuTokenTimeQuotaOut
 
 
 class IQpuTokenRepo(IRepository, ABC):
@@ -145,3 +147,201 @@ class IQpuTokenRepo(IRepository, ABC):
             Parameters to pass to `httpx.request`.
         """
         raise NotImplementedError
+
+    @abstractmethod
+    def create_group_time_quota(
+        self,
+        qpu_token_name: str,
+        quota: int,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        """Create a time quota policy for a shared QPU token that affects every user of
+        the group.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token. Currently, only DWave tokens are supported.
+        quota : int
+        quota : int
+            The amount of quota to add. For DWave Quantum Annealing, which is currently
+            the only algorithm that supports time quota, this is the qpu access time in
+            nanoseconds.
+        start : Optional[datetime], optional
+            The date and time from which the policy is active. If None, the current date
+            and time will be used. If the policy is currently not effective, the token
+            cannot be used at all.
+            Default: None
+        end : Optional[datetime], optional
+            The date and time until which the policy is active. If None, the policy if
+            effective until 265 days after the start date. If the policy is currently
+            not effective, the token cannot be used at all.
+            Default: None
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_group_time_quota(
+        self, qpu_token_name: str, **kwargs
+    ) -> Optional[QpuTokenTimeQuotaOut]:
+        """Get the group time quota policy for a qpu token.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token.
+
+        Returns
+        -------
+        Optional[QpuTokenTimeQuotaOut]
+            The token policy. None, if no group policy is set on this token.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def update_group_time_quota(
+        self,
+        qpu_token_name: str,
+        quota: Optional[int] = None,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        """Update the details on a group qpu time quota policy.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token.
+        quota : Optional[int], optional
+            The amount of quota. For DWave Quantum Annealing, which is currently the
+            only supported solver, this is the qpu access time in nanoseconds. If None,
+            the available quota won't be updated.
+            Default: None
+        start : Optional[datetime], optional
+            The date and time from which the policy is active. If None, the start date
+            won't be updated.
+            Default: None
+        end : Optional[datetime], optional
+            The date and time until which the policy is active. If None, the end date
+            won't be updated.
+            Default: None
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_group_time_quota(self, qpu_token_name: str, **kwargs) -> None:
+        """Delete the group policy set on a qpu token.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def create_user_time_quota(
+        self,
+        qpu_token_name: str,
+        user_email: str,
+        quota: int,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        """Create a time quota policy for a shared QPU token that affects a single user
+        of the group.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token. Currently, only DWave tokens are supported.
+        user_email : str
+            Email of the user for whom to add the policy.
+        quota : int
+            The amount of quota to add. For DWave Quantum Annealing, which is currently
+            the only algorithm that supports time quota, this is the qpu access time in
+            nanoseconds.
+        start : Optional[datetime], optional
+            The date and time from which the policy is active. If None, the current date
+            and time will be used. If the policy is currently not effective, the token
+            cannot be used at all.
+            Default: None
+        end : Optional[datetime], optional
+            The date and time until which the policy is active. If None, the policy if
+            effective until 265 days after the start date. If the policy is currently
+            not effective, the token cannot be used at all.
+            Default: None
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def get_user_time_quota(
+        self, qpu_token_name: str, user_email: str, **kwargs
+    ) -> Optional[QpuTokenTimeQuotaOut]:
+        """Get a user-specific time quota policy for a qpu token.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token.
+        user_email : str
+            Email of the user for whom to get the policy.
+
+        Returns
+        -------
+        Optional[QpuTokenTimeQuotaOut]
+            The token policy. None, if no policy is set on this token for the specified
+            user.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def update_user_time_quota(
+        self,
+        qpu_token_name: str,
+        user_email: str,
+        quota: Optional[int] = None,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        """Update the details on a user-specific qpu time quota policy.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token.
+        user_email : str
+            Email of the user for whom to update the policy.
+        quota : Optional[int], optional
+            The amount of quota. For DWave Quantum Annealing, which is currently the
+            only supported solver, this is the qpu access time in nanoseconds. If None,
+            the available quota won't be updated.
+            Default: None
+        start : Optional[datetime], optional
+            The date and time from which the policy is active. If None, the start date
+            won't be updated.
+            Default: None
+        end : Optional[datetime], optional
+            The date and time until which the policy is active. If None, the end date
+            won't be updated.
+            Default: None
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_user_time_quota(
+        self, qpu_token_name: str, user_email: str, **kwargs
+    ) -> None:
+        """Delete a user-specific policy set on a qpu token.
+
+        Parameters
+        ----------
+        qpu_token_name : str
+            The name of the qpu token.
+        """
+        raise NotImplementedError

luna_sdk/interfaces/solutions_repo_i.py

@@ -231,3 +231,26 @@ class ISolutionsRepo(IRepository, ABC):
             not (yet) available or the solution sense is `None`, `None` is returned.
         """
         raise NotImplementedError
+
+    @abstractmethod
+    def cancel(
+        self,
+        solution_id: str,
+        **kwargs,
+    ) -> Solution:
+        """
+        Cancel a solve job for an optimization.
+
+        Parameters
+        ----------
+        solution_id: str
+            The id of the solution which should be canceled.
+        **kwargs
+            Parameters to pass to `httpx.request`.
+
+        Returns
+        -------
+        SolutionOut
+            The location where the solution can be found once solving is complete.
+        """
+        raise NotImplementedError

luna_sdk/repositories/qpu_token_repo.py

@@ -1,5 +1,5 @@
 import json
-import os
+from datetime import datetime
 from typing import Dict, List, Optional
 
 from httpx import Response
@@ -7,9 +7,9 @@ from pydantic import TypeAdapter
 
 from luna_sdk.interfaces.qpu_token_repo_i import IQpuTokenRepo
 from luna_sdk.schemas import QpuTokenOut
-from luna_sdk.schemas.create import QpuTokenIn
+from luna_sdk.schemas.create import QpuTokenIn, QpuTokenTimeQuotaIn
 from luna_sdk.schemas.enums.qpu_token_type import QpuTokenTypeEnum
-
+from luna_sdk.schemas.qpu_token_time_quota import QpuTokenTimeQuotaOut
 
 _ORGANIZATION_QPU_TOKENS_BACKEND = "shared"
 _PERSONAL_QPU_TOKENS_BACKEND = "private"
@@ -141,3 +141,125 @@ class QpuTokenRepo(IQpuTokenRepo):
             f"{self._get_endpoint_by_type(token_type)}/{name}", **kwargs
         )
         response.raise_for_status()
+
+    def create_group_time_quota(
+        self,
+        qpu_token_name: str,
+        quota: int,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        time_quota = QpuTokenTimeQuotaIn(quota=quota, start=start, end=end)
+
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.post(
+            f"{endpoint}/quota/group/{qpu_token_name}",
+            content=time_quota.model_dump_json(),
+            **kwargs,
+        )
+        response.raise_for_status()
+
+    def get_group_time_quota(
+        self, qpu_token_name: str, **kwargs
+    ) -> Optional[QpuTokenTimeQuotaOut]:
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.get(
+            f"{endpoint}/quota/group/{qpu_token_name}", **kwargs
+        )
+        response.raise_for_status()
+
+        time_quota_data = response.json()
+
+        if time_quota_data is None:
+            return None
+        return QpuTokenTimeQuotaOut.model_validate(time_quota_data)
+
+    def update_group_time_quota(
+        self,
+        qpu_token_name: str,
+        quota: Optional[int] = None,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        data = {"quota": quota, "start": start, "end": end}
+
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.patch(
+            f"{endpoint}/quota/group/{qpu_token_name}",
+            content=json.dumps(data),
+            **kwargs,
+        )
+        response.raise_for_status()
+
+    def delete_group_time_quota(self, qpu_token_name: str, **kwargs) -> None:
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.delete(
+            f"{endpoint}/quota/group/{qpu_token_name}",
+            **kwargs,
+        )
+        response.raise_for_status()
+
+    def create_user_time_quota(
+        self,
+        qpu_token_name: str,
+        user_email: str,
+        quota: int,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        time_quota = QpuTokenTimeQuotaIn(quota=quota, start=start, end=end)
+
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.post(
+            f"{endpoint}/quota/user/{qpu_token_name}/{user_email}",
+            content=time_quota.model_dump_json(),
+            **kwargs,
+        )
+        response.raise_for_status()
+
+    def get_user_time_quota(
+        self, qpu_token_name: str, user_email: str, **kwargs
+    ) -> Optional[QpuTokenTimeQuotaOut]:
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.get(
+            f"{endpoint}/quota/user/{qpu_token_name}/{user_email}", **kwargs
+        )
+        response.raise_for_status()
+
+        time_quota_data = response.json()
+
+        if time_quota_data is None:
+            return None
+        return QpuTokenTimeQuotaOut.model_validate(time_quota_data)
+
+    def update_user_time_quota(
+        self,
+        qpu_token_name: str,
+        user_email: str,
+        quota: Optional[int] = None,
+        start: Optional[datetime] = None,
+        end: Optional[datetime] = None,
+        **kwargs,
+    ) -> None:
+        data = {"quota": quota, "start": start, "end": end}
+
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.patch(
+            f"{endpoint}/quota/user/{qpu_token_name}/{user_email}",
+            content=json.dumps(data),
+            **kwargs,
+        )
+        response.raise_for_status()
+
+    def delete_user_time_quota(
+        self, qpu_token_name: str, user_email: str, **kwargs
+    ) -> None:
+        endpoint = self._get_endpoint_by_type(QpuTokenTypeEnum.GROUP)
+        response = self._client.delete(
+            f"{endpoint}/quota/user/{qpu_token_name}/{user_email}",
+            **kwargs,
+        )
+        response.raise_for_status()

luna_sdk/repositories/solutions_repo.py

@@ -265,3 +265,9 @@ class SolutionsRepo(ISolutionsRepo):
                 "Solving with Luna can still fail due to the parameters."
             )
         return params
+
+    def cancel(self, solution_id: str, **kwargs) -> Solution:
+        response = self._client.post(f"{self._endpoint}/{solution_id}/cancel", **kwargs)
+
+        response.raise_for_status()
+        return Solution.model_validate_json(response.text)

luna_sdk/schemas/create/__init__.py

@@ -1,3 +1,4 @@
 from .circuit import CircuitIn
 from .qpu_token import QpuTokenIn
+from .qpu_token_time_quota import QpuTokenTimeQuotaIn
 from .qubo import QUBOIn

luna_sdk/schemas/create/qpu_token_time_quota.py

@@ -0,0 +1,25 @@
+from datetime import datetime
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class QpuTokenTimeQuotaIn(BaseModel):
+    """
+    Pydantic model for creating a time quota on a qpu token.
+
+    Attributes
+    ----------
+    quota: int
+        The amount of quota.
+    start: datetime | None
+        Effective start date of the time quota policy.
+        If None, policy will be in effect immediately.
+    end: datetime | None
+        Effective end date of the time quota policy.
+        If None, policy will be in effect until 365 days after the start date.
+    """
+
+    quota: int
+    start: Optional[datetime]
+    end: Optional[datetime]

luna_sdk/schemas/enums/status.py

@@ -7,4 +7,5 @@ class StatusEnum(str, Enum):
     IN_PROGRESS = "IN_PROGRESS"
     DONE = "DONE"
     FAILED = "FAILED"
-    CANCELLED = "CANCELED"
+    CANCELED = "CANCELED"
+    TIMEOUT = "TIMEOUT"

luna_sdk/schemas/optimization.py

@@ -56,6 +56,7 @@ class Optimization(PrettyBase):
             "id": data_subset.pop("id"),
             "name": data_subset.pop("name"),
             "original_format": data_subset.pop("original_format"),
+            "created_date": data_subset.pop("created_date"),
             **data_subset,
         }
         output += self._pretty_print(ordered_subset)
@@ -63,16 +64,14 @@ class Optimization(PrettyBase):
 
     def subset(self):
         trimmed_keys = [
-            "created_by",
-            "modified_date",
-            "modified_by",
-            "use_case_name",
-            "params",
-            "verbose",
+            "id",
+            "name",
+            "original_format",
+            "created_date",
         ]
         data = self.model_dump()
         output = ""
-        data_subset = {key: data[key] for key in data if key not in trimmed_keys}
+        data_subset = {key: data[key] for key in data if key in trimmed_keys}
 
         ordered_subset = {
             "id": data_subset.pop("id"),

luna_sdk/schemas/qpu_token_time_quota.py

@@ -0,0 +1,26 @@
+from datetime import datetime
+
+from pydantic import BaseModel
+
+
+class QpuTokenTimeQuotaOut(BaseModel):
+    """
+    Pydantic model for QPU token time quota OUT.
+    It contains the data received from the API call.
+
+    Attributes
+    ----------
+    quota: int
+        The total amount of quota available on a qpu token.
+    start: datetime
+        Effective start date of the time quota policy.
+    end: datetime
+        Effective end date of the time quota policy.
+    quota_used: int
+        How much quota has already been used from the totally available amount of quota.
+    """
+
+    quota: int
+    start: datetime
+    end: datetime
+    quota_used: int

luna_sdk/schemas/solution.py

@@ -99,7 +99,7 @@ class Solution(PrettyBase):
     error_message: Optional[str] = None
     solver_job_info: Optional[str] = None
 
-    results: Optional[List[Result]]
+    results: Optional[List[Result]] = None
     params: Dict[str, Any]
     runtime: Optional[Runtime]
     sense: Optional[SenseEnum]
@@ -114,6 +114,9 @@ class Solution(PrettyBase):
 
     verbose: bool = False
 
+    is_cancelable: bool = True
+    is_cancellation_requested: bool = False
+
     def __str__(self):
         if self.verbose:
             return self.details()
@@ -196,6 +199,7 @@ class Solution(PrettyBase):
         results = data.pop("results")  # Extract and remove results from data
         metadata = data.pop("metadata")  # Extract and remove metadata from data
         provider = data["provider"]
+        data.pop("verbose")  # Remove verbose field from data
 
         divider = "--------------------------------------------------------------------------------\n"
 
@@ -239,7 +243,7 @@ class Solution(PrettyBase):
             "created_date",
         ]
         data = self.model_dump()
-        data["optimization_name"] = data["optimization"]["original_format"]
+        data["optimization_name"] = data["optimization"]["name"]
         output = ""
         data_subset = {key: data.get(key) for key in subset_keys if key in data}
         output += self._pretty_print(data_subset)