boulder-opal-scale-up-sdk 1.0.2__py3-none-any.whl → 1.0.4__py3-none-any.whl

boulderopalscaleupsdk/experiments/resonator_spectroscopy_by_power.py

@@ -33,23 +33,22 @@ class ResonatorSpectroscopyByPower(Experiment):
     Parameters
     ----------
     resonator : str
-        The reference for the resonator to target in the experiment.
+        The reference for the resonator to target.
     frequencies : list[int] or LinspaceIterable or RangeIterable or CWSIterable
                   or HypIterable or None, optional
-        The frequencies at which to scan.
-        If None, frequencies around the readout frequency will be used.
-    powers : list[int] or None, optional
-        The powers at which to scan the readout pulse.
+        The frequencies at which to scan, in Hz.
+        Defaults to a scan around the readout frequency.
+    powers_dbm : list[float] or None, optional
+        The powers at which to scan the readout pulse, in dBm.
         If None, a default power scan will be used.
     duration_ns : int, optional
-        The duration of the readout pulse, in nanoseconds.
-        Defaults to 2000 ns.
+        The duration of the readout pulse, in nanoseconds. Defaults to 2,000 ns.
     recycle_delay_ns : int, optional
-        The delay time between consecutive shots of the experiment, in nanoseconds.
-        Defaults to 1000 ns.
+        The delay between consecutive shots, in nanoseconds. Defaults to 1,000 ns.
     shot_count : int, optional
-        The number of shots to be taken in the experiment.
-        Defaults to 100.
+        The number of shots to take. Defaults to 100.
+    run_mixer_calibration: bool
+        Whether to run mixer calibrations before running a program. Defaults to False.
     """
 
     _experiment_name: str = PrivateAttr("resonator_spectroscopy_by_power")
@@ -58,7 +57,8 @@ class ResonatorSpectroscopyByPower(Experiment):
     frequencies: list[int] | LinspaceIterable | RangeIterable | CWSIterable | HypIterable | None = (
         None
     )
-    powers: list[float] | None = None
+    powers_dbm: list[float] | None = None
     duration_ns: int = DEFAULT_DURATION_NS
     recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
     shot_count: int = DEFAULT_SHOT_COUNT
+    run_mixer_calibration: bool = False

boulderopalscaleupsdk/experiments/t1.py

@@ -0,0 +1,46 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
+from pydantic import PrivateAttr
+
+from .common import Experiment, LogspaceIterable
+
+DEFAULT_RECYCLE_DELAY_NS = 200_000
+DEFAULT_SHOT_COUNT = 400
+
+
+class T1(Experiment):
+    """
+    Parameters for running a T1 experiment.
+
+    Parameters
+    ----------
+    transmon : str
+        The reference for the transmon to target.
+    delays_ns : LogspaceIterable
+        The delay times, in nanoseconds.
+    recycle_delay_ns : int, optional
+        The delay between consecutive shots, in nanoseconds. Defaults to 200,000 ns.
+    shot_count : int, optional
+        The number of shots to take. Defaults to 400.
+    run_mixer_calibration: bool
+        Whether to run mixer calibrations before running a program. Defaults to False.
+    """
+
+    _experiment_name: str = PrivateAttr("t1")
+
+    transmon: str
+    delays_ns: LogspaceIterable
+    recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
+    shot_count: int = DEFAULT_SHOT_COUNT
+    run_mixer_calibration: bool = False

boulderopalscaleupsdk/experiments/t2.py

@@ -0,0 +1,49 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
+from pydantic import PrivateAttr
+
+from .common import Experiment, RangeIterable
+
+DEFAULT_RECYCLE_DELAY_NS = 100_000
+DEFAULT_SHOT_COUNT = 400
+
+
+class T2(Experiment):
+    """
+    Parameters for running a Ramsey experiment.
+
+    Parameters
+    ----------
+    transmon : str
+        The reference for the transmon to target.
+    delays_ns : list[int] or RangeIterable
+        The delay times, in nanoseconds.
+    virtual_detuning : float
+        The virtual detuning added between SX pulses, in Hz.
+    recycle_delay_ns : int, optional
+        The delay between consecutive shots, in nanoseconds. Defaults to 100,000 ns.
+    shot_count : int, optional
+        The number of shots to take. Defaults to 400.
+    run_mixer_calibration: bool
+        Whether to run mixer calibrations before running a program. Defaults to False.
+    """
+
+    _experiment_name: str = PrivateAttr("t2")
+
+    transmon: str
+    delays_ns: list[int] | RangeIterable
+    virtual_detuning: float
+    recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
+    shot_count: int = DEFAULT_SHOT_COUNT
+    run_mixer_calibration: bool = False

boulderopalscaleupsdk/experiments/t2_echo.py

@@ -0,0 +1,49 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
+from pydantic import PrivateAttr
+
+from .common import Experiment, RangeIterable
+
+DEFAULT_RECYCLE_DELAY_NS = 100_000
+DEFAULT_SHOT_COUNT = 400
+
+
+class T2Echo(Experiment):
+    """
+    Parameters for running a T2 echo experiment.
+
+    Parameters
+    ----------
+    transmon : str
+        The reference for the transmon to target.
+    delays_ns : list[int] or RangeIterable
+        The delay times, in nanoseconds.
+    virtual_detuning : float
+        The virtual detuning added between SX pulses, in Hz.
+    recycle_delay_ns : int, optional
+        The delay between consecutive shots, in nanoseconds. Defaults to 100,000 ns.
+    shot_count : int, optional
+        The number of shots to take. Defaults to 400.
+    run_mixer_calibration: bool
+        Whether to run mixer calibrations before running a program. Defaults to False.
+    """
+
+    _experiment_name: str = PrivateAttr("t2_echo")
+
+    transmon: str
+    delays_ns: list[int] | RangeIterable
+    virtual_detuning: float
+    recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
+    shot_count: int = DEFAULT_SHOT_COUNT
+    run_mixer_calibration: bool = False

boulderopalscaleupsdk/experiments/transmon_anharmonicity.py

@@ -13,9 +13,15 @@
 
 from pydantic import PrivateAttr
 
-from .common import CWSIterable, Experiment, HypIterable, LinspaceIterable, RangeIterable
+from .common import (
+    CWSIterable,
+    Experiment,
+    HypIterable,
+    LinspaceIterable,
+    RangeIterable,
+)
+from .waveforms import ConstantWaveform
 
-DEFAULT_DURATION_NS = 2_000  # ns
 DEFAULT_RECYCLE_DELAY_NS = 1_000  # ns
 DEFAULT_SHOT_COUNT = 100
 DEFAULT_ANHARMONICITY_GUESS = -200e6  # Hz, typical anharmonicity for transmons
@@ -28,29 +34,27 @@ class TransmonAnharmonicity(Experiment):
     Parameters
     ----------
     transmon : str
-        The transmon that is the target of the experiment.
+        The reference for the transmon to target.
     frequencies : list[int] or LinspaceIterable or RangeIterable or CWSIterable
                   or HypIterable or None, optional
-        The drive frequencies to be scanned in the experiment.
-        If None, a default scan will be used based on the transmon's frequency and anharmonicity.
-    spec_amplitude : float
-        The amplitude of the spectroscopy pulse, in volts.
-    spec_duration : int
-        The duration of the spectroscopy pulse, in nanoseconds.
-    anharmonicity_guess_hz : float
-        The guessed anharmonicity of the qubit, in Hz. This is used to determine the
-        range of frequencies to scan if `frequencies` is not provided.
-        Defaults to -200 MHz.
-    readout_amplitude : float
-        The amplitude of the readout pulse.
-    duration_ns : int, optional
-        The duration of the readout pulse, in nanoseconds.
-        Defaults to 2000 ns.
+        The frequencies at which to scan, in Hz.
+        Defaults to a scan based on the transmon's frequency and anharmonicity.
+    anharmonicity : float, optional
+        The guessed anharmonicity of the transmon, in Hz.
+        Used to determine the scan frequencies if not provided. Defaults to -200 MHz.
     recycle_delay_ns : int, optional
-        The delay time between consecutive shots of the experiment, in nanoseconds.
-        Defaults to 1000 ns.
+        The delay between consecutive shots, in nanoseconds. Defaults to 1,000 ns.
     shot_count : int, optional
-        The number of shots to be taken in the experiment. Defaults to 100.
+        The number of shots to take. Defaults to 100.
+    spectroscopy_waveform : ConstantWaveform or None, optional
+        The waveform to use in the spectroscopy pulse.
+        Defaults to a constant waveform with a duration of 2,000 ns
+        and an amplitude of 1.5 times the transmon's x_vp.
+    measure_waveform : ConstantWaveform or None, optional
+        The waveform to use for the measurement pulse.
+        Defaults to the measurement defcal.
+    run_mixer_calibration: bool
+        Whether to run mixer calibrations before running a program. Defaults to False.
     """
 
     _experiment_name: str = PrivateAttr("transmon_anharmonicity")
@@ -59,10 +63,9 @@ class TransmonAnharmonicity(Experiment):
     frequencies: list[int] | LinspaceIterable | RangeIterable | CWSIterable | HypIterable | None = (
         None
     )
-    spec_amplitude: float
-    spec_duration: int
-    anharmonicity_guess_hz: float = DEFAULT_ANHARMONICITY_GUESS
-    readout_amplitude: float
-    duration_ns: int = DEFAULT_DURATION_NS
+    anharmonicity: float = DEFAULT_ANHARMONICITY_GUESS
     recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
     shot_count: int = DEFAULT_SHOT_COUNT
+    spectroscopy_waveform: ConstantWaveform | None = None
+    measure_waveform: ConstantWaveform | None = None
+    run_mixer_calibration: bool = False

boulderopalscaleupsdk/experiments/transmon_spectroscopy.py

@@ -20,8 +20,8 @@ from .common import (
     LinspaceIterable,
     RangeIterable,
 )
+from .waveforms import ConstantWaveform
 
-DEFAULT_DURATION_NS = 2000  # ns
 DEFAULT_RECYCLE_DELAY_NS = 1000  # ns
 DEFAULT_SHOT_COUNT = 100
 
@@ -33,26 +33,22 @@ class TransmonSpectroscopy(Experiment):
     Parameters
     ----------
     transmon : str
-        The reference for the transmon to target in the experiment.
+        The reference for the transmon to target.
     frequencies : list[int] or LinspaceIterable or RangeIterable or CWSIterable
                   or HypIterable or None, optional
-        The frequencies at which to scan.
-        If None, frequencies around the readout frequency will be used.
-    spec_amplitude : float
-        The amplitude of the spectroscopy pulse.
-    spec_duration : int
-        The duration of the spectroscopy pulse, in nanoseconds.
-    readout_amplitude : float
-        The amplitude of the readout pulse.
-    duration_ns : int, optional
-        The duration of the readout pulse, in nanoseconds.
-        Defaults to 2000 ns.
+        The frequencies at which to scan, in Hz.
+        Default to a scan around the transmon frequency.
     recycle_delay_ns : int, optional
-        The delay time between consecutive shots of the experiment, in nanoseconds.
-        Defaults to 1000 ns.
+        The delay between consecutive shots, in nanoseconds. Defaults to 1,000 ns.
     shot_count : int, optional
-        The number of shots to be taken in the experiment.
-        Defaults to 100.
+        The number of shots to take. Defaults to 100.
+    spectroscopy_waveform : ConstantWaveform
+        The waveform to use in the spectroscopy pulse.
+    measure_waveform : ConstantWaveform or None, optional
+        The waveform to use for the measurement pulse.
+        Defaults to the measurement defcal.
+    run_mixer_calibration: bool
+        Whether to run mixer calibrations before running a program. Defaults to False.
     """
 
     _experiment_name: str = PrivateAttr("transmon_spectroscopy")
@@ -61,9 +57,8 @@ class TransmonSpectroscopy(Experiment):
     frequencies: list[int] | LinspaceIterable | RangeIterable | CWSIterable | HypIterable | None = (
         None
     )
-    spec_amplitude: float
-    spec_duration: int
-    readout_amplitude: float
-    duration_ns: int = DEFAULT_DURATION_NS
     recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
     shot_count: int = DEFAULT_SHOT_COUNT
+    spectroscopy_waveform: ConstantWaveform
+    measure_waveform: ConstantWaveform | None = None
+    run_mixer_calibration: bool = False

boulderopalscaleupsdk/experiments/waveforms.py

@@ -0,0 +1,63 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
+from dataclasses import replace
+from typing import Annotated, Literal, Protocol, TypeVar
+
+from pydantic import Discriminator
+from pydantic.dataclasses import dataclass
+
+
+class _Waveform(Protocol):  # pragma: no cover
+    duration_ns: int
+    amplitude: float
+
+
+@dataclass
+class ConstantWaveform(_Waveform):  # pragma: no cover
+    duration_ns: int
+    amplitude: float
+    waveform_type: Literal["Constant"] = "Constant"
+
+
+@dataclass
+class GaussianWaveform(_Waveform):  # pragma: no cover
+    duration_ns: int
+    amplitude: float
+    sigma: float
+    waveform_type: Literal["Gaussian"] = "Gaussian"
+
+
+@dataclass
+class DragCosineWaveform(_Waveform):  # pragma: no cover
+    duration_ns: int
+    amplitude: float
+    drag: float
+    buffer_ns: int
+    center: float
+    waveform_type: Literal["DragCosineWaveform"] = "DragCosineWaveform"
+
+
+Waveform = Annotated[
+    ConstantWaveform | GaussianWaveform | DragCosineWaveform,
+    Discriminator("waveform_type"),
+]
+
+T = TypeVar("T", bound=Waveform)
+
+
+def update_amplitude(waveform: T, amplitude: float) -> T:
+    """
+    Update the amplitude of a waveform.
+    """
+    return replace(waveform, amplitude=amplitude)

boulderopalscaleupsdk/plotting/dtypes.py

@@ -71,6 +71,7 @@ class PlotConfig(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
     title: str
+    subtitle: str | None = None
 
     x_label: str
     y_label: str

boulderopalscaleupsdk/protobuf/v1/device_pb2.py

@@ -16,7 +16,7 @@ from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
 from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.boulderopalscaleupsdk/protobuf/v1/device.proto\x12!boulderopalscaleupsdk.protobuf.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1cgoogle/protobuf/struct.proto\"-\n\x14GetJobSummaryRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"Z\n\x15GetJobSummaryResponse\x12\x41\n\x10job_summary_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x0ejobSummaryData\"&\n\rGetJobRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"D\n\x0eGetJobResponse\x12\x32\n\x08job_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x07jobData\"\x96\x01\n\x0fListJobsRequest\x12\x1f\n\x0b\x64\x65vice_name\x18\x01 \x01(\tR\ndeviceName\x12\x19\n\x08job_name\x18\x02 \x01(\tR\x07jobName\x12\x12\n\x04page\x18\x03 \x01(\x05R\x04page\x12\x14\n\x05limit\x18\x04 \x01(\x05R\x05limit\x12\x1d\n\nsort_order\x18\x05 \x01(\x05R\tsortOrder\"?\n\x10ListJobsResponse\x12+\n\x04jobs\x18\x01 \x03(\x0b\x32\x17.google.protobuf.StructR\x04jobs\"\x85\x01\n\rCreateRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\x12\x38\n\x0b\x64\x65vice_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\ndeviceData\"$\n\x0e\x43reateResponse\x12\x12\n\x04\x64one\x18\x01 \x01(\x08R\x04\x64one\"]\n\x0b\x43opyRequest\x12(\n\x10\x66rom_device_name\x18\x01 \x01(\tR\x0e\x66romDeviceName\x12$\n\x0eto_device_name\x18\x02 \x01(\tR\x0ctoDeviceName\"\"\n\x0c\x43opyResponse\x12\x12\n\x04\x64one\x18\x01 \x01(\x08R\x04\x64one\"L\n\x0eGetDataRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\"\x93\x01\n\x0fGetDataResponse\x12>\n\x0eprocessor_data\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructR\rprocessorData\x12@\n\x0f\x63ontroller_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x0e\x63ontrollerData\"\x8b\x01\n\rUpdateRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\x12>\n\x0eprocessor_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\rprocessorData\"\x92\x01\n\x0eUpdateResponse\x12>\n\x0eprocessor_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\rprocessorData\x12@\n\x0f\x63ontroller_data\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x0e\x63ontrollerData\"K\n\rDeleteRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\"$\n\x0e\x44\x65leteResponse\x12\x12\n\x04\x64one\x18\x01 \x01(\x08R\x04\x64one\"5\n\x12GetMetadataRequest\x12\x1f\n\x0b\x64\x65vice_name\x18\x01 \x01(\tR\ndeviceName\"J\n\x13GetMetadataResponse\x12\x33\n\x08metadata\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\"\x1e\n\x1cGetAllDevicesMetadataRequest\"V\n\x1dGetAllDevicesMetadataResponse\x12\x35\n\tmetadatas\x18\x01 \x03(\x0b\x32\x17.google.protobuf.StructR\tmetadatas2\xb0\x0c\n\x14\x44\x65viceManagerService\x12\x8b\x01\n\x06\x43reate\x12\x30.boulderopalscaleupsdk.protobuf.v1.CreateRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.CreateResponse\"\x1c\x82\xd3\xe4\x93\x02\x16\"\x11/v1/device/create:\x01*\x12\x83\x01\n\x04\x43opy\x12..boulderopalscaleupsdk.protobuf.v1.CopyRequest\x1a/.boulderopalscaleupsdk.protobuf.v1.CopyResponse\"\x1a\x82\xd3\xe4\x93\x02\x14\"\x0f/v1/device/copy:\x01*\x12\x97\x01\n\x07GetData\x12\x31.boulderopalscaleupsdk.protobuf.v1.GetDataRequest\x1a\x32.boulderopalscaleupsdk.protobuf.v1.GetDataResponse\"%\x82\xd3\xe4\x93\x02\x1f\"\x1a/v1/device/get_device_data:\x01*\x12\x8b\x01\n\x06Update\x12\x30.boulderopalscaleupsdk.protobuf.v1.UpdateRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.UpdateResponse\"\x1c\x82\xd3\xe4\x93\x02\x16\"\x11/v1/device/update:\x01*\x12\x8b\x01\n\x06\x44\x65lete\x12\x30.boulderopalscaleupsdk.protobuf.v1.DeleteRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.DeleteResponse\"\x1c\x82\xd3\xe4\x93\x02\x16\"\x11/v1/device/delete:\x01*\x12\x94\x01\n\x08ListJobs\x12\x32.boulderopalscaleupsdk.protobuf.v1.ListJobsRequest\x1a\x33.boulderopalscaleupsdk.protobuf.v1.ListJobsResponse\"\x1f\x82\xd3\xe4\x93\x02\x19\"\x14/v1/device/list_jobs:\x01*\x12\x8c\x01\n\x06GetJob\x12\x30.boulderopalscaleupsdk.protobuf.v1.GetJobRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.GetJobResponse\"\x1d\x82\xd3\xe4\x93\x02\x17\"\x12/v1/device/get_job:\x01*\x12\xa9\x01\n\rGetJobSummary\x12\x37.boulderopalscaleupsdk.protobuf.v1.GetJobSummaryRequest\x1a\x38.boulderopalscaleupsdk.protobuf.v1.GetJobSummaryResponse\"%\x82\xd3\xe4\x93\x02\x1f\"\x1a/v1/device/get_job_summary:\x01*\x12\xa7\x01\n\x0bGetMetadata\x12\x35.boulderopalscaleupsdk.protobuf.v1.GetMetadataRequest\x1a\x36.boulderopalscaleupsdk.protobuf.v1.GetMetadataResponse\")\x82\xd3\xe4\x93\x02#\"\x1e/v1/device/get_device_metadata:\x01*\x12\xd1\x01\n\x15GetAllDevicesMetadata\x12?.boulderopalscaleupsdk.protobuf.v1.GetAllDevicesMetadataRequest\x1a@.boulderopalscaleupsdk.protobuf.v1.GetAllDevicesMetadataResponse\"5\x82\xd3\xe4\x93\x02/\"*/v1/device/get_all_devices_metadata_by_org:\x01*B\xaa\x02\n%com.boulderopalscaleupsdk.protobuf.v1B\x0b\x44\x65viceProtoP\x01ZNgithub.com/qctrl/boulder-opal-scale-up/proto/boulderopalscaleupsdk/protobuf/v1\xa2\x02\x03\x42PX\xaa\x02!Boulderopalscaleupsdk.Protobuf.V1\xca\x02!Boulderopalscaleupsdk\\Protobuf\\V1\xe2\x02-Boulderopalscaleupsdk\\Protobuf\\V1\\GPBMetadata\xea\x02#Boulderopalscaleupsdk::Protobuf::V1b\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.boulderopalscaleupsdk/protobuf/v1/device.proto\x12!boulderopalscaleupsdk.protobuf.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1cgoogle/protobuf/struct.proto\"-\n\x14GetJobSummaryRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"Z\n\x15GetJobSummaryResponse\x12\x41\n\x10job_summary_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x0ejobSummaryData\"&\n\rGetJobRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"D\n\x0eGetJobResponse\x12\x32\n\x08job_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x07jobData\"\x96\x01\n\x0fListJobsRequest\x12\x1f\n\x0b\x64\x65vice_name\x18\x01 \x01(\tR\ndeviceName\x12\x19\n\x08job_name\x18\x02 \x01(\tR\x07jobName\x12\x12\n\x04page\x18\x03 \x01(\x05R\x04page\x12\x14\n\x05limit\x18\x04 \x01(\x05R\x05limit\x12\x1d\n\nsort_order\x18\x05 \x01(\x05R\tsortOrder\"?\n\x10ListJobsResponse\x12+\n\x04jobs\x18\x01 \x03(\x0b\x32\x17.google.protobuf.StructR\x04jobs\"\x85\x01\n\rCreateRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\x12\x38\n\x0b\x64\x65vice_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\ndeviceData\"$\n\x0e\x43reateResponse\x12\x12\n\x04\x64one\x18\x01 \x01(\x08R\x04\x64one\"]\n\x0b\x43opyRequest\x12(\n\x10\x66rom_device_name\x18\x01 \x01(\tR\x0e\x66romDeviceName\x12$\n\x0eto_device_name\x18\x02 \x01(\tR\x0ctoDeviceName\"\"\n\x0c\x43opyResponse\x12\x12\n\x04\x64one\x18\x01 \x01(\x08R\x04\x64one\"L\n\x0eGetDataRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\"\xc6\x01\n\x0fGetDataResponse\x12>\n\x0eprocessor_data\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructR\rprocessorData\x12@\n\x0f\x63ontroller_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\x0e\x63ontrollerData\x12\x31\n\x07\x64\x65\x66\x63\x61ls\x18\x04 \x03(\x0b\x32\x17.google.protobuf.StructR\x07\x64\x65\x66\x63\x61ls\"\x8b\x01\n\rUpdateRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\x12>\n\x0eprocessor_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\rprocessorData\"\x92\x01\n\x0eUpdateResponse\x12>\n\x0eprocessor_data\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructR\rprocessorData\x12@\n\x0f\x63ontroller_data\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x0e\x63ontrollerData\"K\n\rDeleteRequest\x12\x19\n\x08\x61pp_name\x18\x01 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x02 \x01(\tR\ndeviceName\"$\n\x0e\x44\x65leteResponse\x12\x12\n\x04\x64one\x18\x01 \x01(\x08R\x04\x64one\"5\n\x12GetMetadataRequest\x12\x1f\n\x0b\x64\x65vice_name\x18\x01 \x01(\tR\ndeviceName\"J\n\x13GetMetadataResponse\x12\x33\n\x08metadata\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x08metadata\"\x1e\n\x1cGetAllDevicesMetadataRequest\"V\n\x1dGetAllDevicesMetadataResponse\x12\x35\n\tmetadatas\x18\x01 \x03(\x0b\x32\x17.google.protobuf.StructR\tmetadatas2\xb0\x0c\n\x14\x44\x65viceManagerService\x12\x8b\x01\n\x06\x43reate\x12\x30.boulderopalscaleupsdk.protobuf.v1.CreateRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.CreateResponse\"\x1c\x82\xd3\xe4\x93\x02\x16\"\x11/v1/device/create:\x01*\x12\x83\x01\n\x04\x43opy\x12..boulderopalscaleupsdk.protobuf.v1.CopyRequest\x1a/.boulderopalscaleupsdk.protobuf.v1.CopyResponse\"\x1a\x82\xd3\xe4\x93\x02\x14\"\x0f/v1/device/copy:\x01*\x12\x97\x01\n\x07GetData\x12\x31.boulderopalscaleupsdk.protobuf.v1.GetDataRequest\x1a\x32.boulderopalscaleupsdk.protobuf.v1.GetDataResponse\"%\x82\xd3\xe4\x93\x02\x1f\"\x1a/v1/device/get_device_data:\x01*\x12\x8b\x01\n\x06Update\x12\x30.boulderopalscaleupsdk.protobuf.v1.UpdateRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.UpdateResponse\"\x1c\x82\xd3\xe4\x93\x02\x16\"\x11/v1/device/update:\x01*\x12\x8b\x01\n\x06\x44\x65lete\x12\x30.boulderopalscaleupsdk.protobuf.v1.DeleteRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.DeleteResponse\"\x1c\x82\xd3\xe4\x93\x02\x16\"\x11/v1/device/delete:\x01*\x12\x94\x01\n\x08ListJobs\x12\x32.boulderopalscaleupsdk.protobuf.v1.ListJobsRequest\x1a\x33.boulderopalscaleupsdk.protobuf.v1.ListJobsResponse\"\x1f\x82\xd3\xe4\x93\x02\x19\"\x14/v1/device/list_jobs:\x01*\x12\x8c\x01\n\x06GetJob\x12\x30.boulderopalscaleupsdk.protobuf.v1.GetJobRequest\x1a\x31.boulderopalscaleupsdk.protobuf.v1.GetJobResponse\"\x1d\x82\xd3\xe4\x93\x02\x17\"\x12/v1/device/get_job:\x01*\x12\xa9\x01\n\rGetJobSummary\x12\x37.boulderopalscaleupsdk.protobuf.v1.GetJobSummaryRequest\x1a\x38.boulderopalscaleupsdk.protobuf.v1.GetJobSummaryResponse\"%\x82\xd3\xe4\x93\x02\x1f\"\x1a/v1/device/get_job_summary:\x01*\x12\xa7\x01\n\x0bGetMetadata\x12\x35.boulderopalscaleupsdk.protobuf.v1.GetMetadataRequest\x1a\x36.boulderopalscaleupsdk.protobuf.v1.GetMetadataResponse\")\x82\xd3\xe4\x93\x02#\"\x1e/v1/device/get_device_metadata:\x01*\x12\xd1\x01\n\x15GetAllDevicesMetadata\x12?.boulderopalscaleupsdk.protobuf.v1.GetAllDevicesMetadataRequest\x1a@.boulderopalscaleupsdk.protobuf.v1.GetAllDevicesMetadataResponse\"5\x82\xd3\xe4\x93\x02/\"*/v1/device/get_all_devices_metadata_by_org:\x01*B\xaa\x02\n%com.boulderopalscaleupsdk.protobuf.v1B\x0b\x44\x65viceProtoP\x01ZNgithub.com/qctrl/boulder-opal-scale-up/proto/boulderopalscaleupsdk/protobuf/v1\xa2\x02\x03\x42PX\xaa\x02!Boulderopalscaleupsdk.Protobuf.V1\xca\x02!Boulderopalscaleupsdk\\Protobuf\\V1\xe2\x02-Boulderopalscaleupsdk\\Protobuf\\V1\\GPBMetadata\xea\x02#Boulderopalscaleupsdk::Protobuf::V1b\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -67,23 +67,23 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_GETDATAREQUEST']._serialized_start=917
   _globals['_GETDATAREQUEST']._serialized_end=993
   _globals['_GETDATARESPONSE']._serialized_start=996
-  _globals['_GETDATARESPONSE']._serialized_end=1143
-  _globals['_UPDATEREQUEST']._serialized_start=1146
-  _globals['_UPDATEREQUEST']._serialized_end=1285
-  _globals['_UPDATERESPONSE']._serialized_start=1288
-  _globals['_UPDATERESPONSE']._serialized_end=1434
-  _globals['_DELETEREQUEST']._serialized_start=1436
-  _globals['_DELETEREQUEST']._serialized_end=1511
-  _globals['_DELETERESPONSE']._serialized_start=1513
-  _globals['_DELETERESPONSE']._serialized_end=1549
-  _globals['_GETMETADATAREQUEST']._serialized_start=1551
-  _globals['_GETMETADATAREQUEST']._serialized_end=1604
-  _globals['_GETMETADATARESPONSE']._serialized_start=1606
-  _globals['_GETMETADATARESPONSE']._serialized_end=1680
-  _globals['_GETALLDEVICESMETADATAREQUEST']._serialized_start=1682
-  _globals['_GETALLDEVICESMETADATAREQUEST']._serialized_end=1712
-  _globals['_GETALLDEVICESMETADATARESPONSE']._serialized_start=1714
-  _globals['_GETALLDEVICESMETADATARESPONSE']._serialized_end=1800
-  _globals['_DEVICEMANAGERSERVICE']._serialized_start=1803
-  _globals['_DEVICEMANAGERSERVICE']._serialized_end=3387
+  _globals['_GETDATARESPONSE']._serialized_end=1194
+  _globals['_UPDATEREQUEST']._serialized_start=1197
+  _globals['_UPDATEREQUEST']._serialized_end=1336
+  _globals['_UPDATERESPONSE']._serialized_start=1339
+  _globals['_UPDATERESPONSE']._serialized_end=1485
+  _globals['_DELETEREQUEST']._serialized_start=1487
+  _globals['_DELETEREQUEST']._serialized_end=1562
+  _globals['_DELETERESPONSE']._serialized_start=1564
+  _globals['_DELETERESPONSE']._serialized_end=1600
+  _globals['_GETMETADATAREQUEST']._serialized_start=1602
+  _globals['_GETMETADATAREQUEST']._serialized_end=1655
+  _globals['_GETMETADATARESPONSE']._serialized_start=1657
+  _globals['_GETMETADATARESPONSE']._serialized_end=1731
+  _globals['_GETALLDEVICESMETADATAREQUEST']._serialized_start=1733
+  _globals['_GETALLDEVICESMETADATAREQUEST']._serialized_end=1763
+  _globals['_GETALLDEVICESMETADATARESPONSE']._serialized_start=1765
+  _globals['_GETALLDEVICESMETADATARESPONSE']._serialized_end=1851
+  _globals['_DEVICEMANAGERSERVICE']._serialized_start=1854
+  _globals['_DEVICEMANAGERSERVICE']._serialized_end=3438
 # @@protoc_insertion_point(module_scope)

boulderopalscaleupsdk/protobuf/v1/device_pb2.pyi

@@ -90,12 +90,14 @@ class GetDataRequest(_message.Message):
     def __init__(self, app_name: _Optional[str] = ..., device_name: _Optional[str] = ...) -> None: ...
 
 class GetDataResponse(_message.Message):
-    __slots__ = ("processor_data", "controller_data")
+    __slots__ = ("processor_data", "controller_data", "defcals")
     PROCESSOR_DATA_FIELD_NUMBER: _ClassVar[int]
     CONTROLLER_DATA_FIELD_NUMBER: _ClassVar[int]
+    DEFCALS_FIELD_NUMBER: _ClassVar[int]
     processor_data: _struct_pb2.Struct
     controller_data: _struct_pb2.Struct
-    def __init__(self, processor_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., controller_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    defcals: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
+    def __init__(self, processor_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., controller_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., defcals: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ...) -> None: ...
 
 class UpdateRequest(_message.Message):
     __slots__ = ("app_name", "device_name", "processor_data")

boulderopalscaleupsdk/protobuf/v1/task_pb2.py

@@ -16,7 +16,7 @@ from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
 from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,boulderopalscaleupsdk/protobuf/v1/task.proto\x12!boulderopalscaleupsdk.protobuf.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x19google/protobuf/any.proto\"\xc7\x01\n\x13StartSessionRequest\x12\x19\n\x08\x61gent_id\x18\x01 \x01(\tR\x07\x61gentId\x12\x19\n\x08\x61pp_name\x18\x02 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x03 \x01(\tR\ndeviceName\x12!\n\x0croutine_name\x18\x04 \x01(\tR\x0broutineName\x12-\n\x04\x64\x61ta\x18\x05 \x01(\x0b\x32\x14.google.protobuf.AnyH\x00R\x04\x64\x61ta\x88\x01\x01\x42\x07\n\x05_data\"\xfc\x01\n\x14UpdateSessionRequest\x12\x1d\n\nsession_id\x18\x01 \x01(\tR\tsessionId\x12R\n\rcurrent_state\x18\x02 \x01(\x0e\x32-.boulderopalscaleupsdk.protobuf.v1.AgentStateR\x0c\x63urrentState\x12G\n\x07results\x18\x03 \x03(\x0b\x32-.boulderopalscaleupsdk.protobuf.v1.TaskResultR\x07results\x12(\n\x10task_in_progress\x18\x04 \x03(\tR\x0etaskInProgress\"\xac\x03\n\x12\x41gentTasksResponse\x12\x1d\n\nsession_id\x18\x01 \x01(\tR\tsessionId\x12U\n\x0ctarget_state\x18\x02 \x01(\x0e\x32-.boulderopalscaleupsdk.protobuf.v1.AgentStateH\x00R\x0btargetState\x88\x01\x01\x12=\n\x05tasks\x18\x03 \x03(\x0b\x32\'.boulderopalscaleupsdk.protobuf.v1.TaskR\x05tasks\x12)\n\x10invalidate_tasks\x18\x04 \x03(\tR\x0finvalidateTasks\x12\x66\n\x0bretry_tasks\x18\x05 \x03(\x0b\x32\x45.boulderopalscaleupsdk.protobuf.v1.AgentTasksResponse.RetryTasksEntryR\nretryTasks\x1a=\n\x0fRetryTasksEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0f\n\r_target_state\"\x94\x02\n\x04Task\x12\x17\n\x07task_id\x18\x01 \x01(\tR\x06taskId\x12\x1d\n\nsession_id\x18\x02 \x01(\tR\tsessionId\x12\x1b\n\ttask_type\x18\x03 \x01(\tR\x08taskType\x12(\n\x04\x64\x61ta\x18\x04 \x01(\x0b\x32\x14.google.protobuf.AnyR\x04\x64\x61ta\x12,\n\x0ftimeout_seconds\x18\x05 \x01(\x05H\x00R\x0etimeoutSeconds\x88\x01\x01\x12\x1f\n\x08\x64\x65\x61\x64line\x18\x06 \x01(\x03H\x01R\x08\x64\x65\x61\x64line\x88\x01\x01\x12\x1d\n\ndepends_on\x18\x07 \x03(\tR\tdependsOnB\x12\n\x10_timeout_secondsB\x0b\n\t_deadline\"\xed\x01\n\nTaskResult\x12\x17\n\x07task_id\x18\x01 \x01(\tR\x06taskId\x12\x1d\n\nsession_id\x18\x03 \x01(\tR\tsessionId\x12\x1f\n\x0bresult_type\x18\x04 \x01(\tR\nresultType\x12.\n\x06result\x18\x05 \x01(\x0b\x32\x14.google.protobuf.AnyH\x00R\x06result\x12J\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x32.boulderopalscaleupsdk.protobuf.v1.TaskErrorDetailH\x00R\x05\x65rrorB\n\n\x08response\"{\n\x0fTaskErrorDetail\x12@\n\x04\x63ode\x18\x01 \x01(\x0e\x32,.boulderopalscaleupsdk.protobuf.v1.TaskErrorR\x04\x63ode\x12\x1b\n\x06\x64\x65tail\x18\x02 \x01(\tH\x00R\x06\x64\x65tail\x88\x01\x01\x42\t\n\x07_detail*\xdd\x01\n\nAgentState\x12\x1b\n\x17\x41GENT_STATE_UNSPECIFIED\x10\x00\x12\x1e\n\x1a\x41GENT_STATE_ACTIVE_SESSION\x10\x01\x12\x1b\n\x17\x41GENT_STATE_ACTIVE_IDLE\x10\x02\x12)\n%AGENT_STATE_SHUTDOWN_CLIENT_INITIATED\x10\x03\x12*\n&AGENT_STATE_SHUTDOWN_MANAGER_INITIATED\x10\x04\x12\x1e\n\x1a\x41GENT_STATE_SHUTDOWN_FAULT\x10\x05*G\n\tTaskError\x12\x1a\n\x16TASK_ERROR_UNSPECIFIED\x10\x00\x12\x1e\n\x1aTASK_ERROR_NOT_IMPLEMENTED\x10\x01\x32\xe0\x02\n\x13\x41gentManagerService\x12\xa1\x01\n\x0cStartSession\x12\x36.boulderopalscaleupsdk.protobuf.v1.StartSessionRequest\x1a\x35.boulderopalscaleupsdk.protobuf.v1.AgentTasksResponse\"\"\x82\xd3\xe4\x93\x02\x1c\"\x17/v1/agent/start_session:\x01*\x12\xa4\x01\n\rUpdateSession\x12\x37.boulderopalscaleupsdk.protobuf.v1.UpdateSessionRequest\x1a\x35.boulderopalscaleupsdk.protobuf.v1.AgentTasksResponse\"#\x82\xd3\xe4\x93\x02\x1d\"\x18/v1/agent/update_session:\x01*B\xa8\x02\n%com.boulderopalscaleupsdk.protobuf.v1B\tTaskProtoP\x01ZNgithub.com/qctrl/boulder-opal-scale-up/proto/boulderopalscaleupsdk/protobuf/v1\xa2\x02\x03\x42PX\xaa\x02!Boulderopalscaleupsdk.Protobuf.V1\xca\x02!Boulderopalscaleupsdk\\Protobuf\\V1\xe2\x02-Boulderopalscaleupsdk\\Protobuf\\V1\\GPBMetadata\xea\x02#Boulderopalscaleupsdk::Protobuf::V1b\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,boulderopalscaleupsdk/protobuf/v1/task.proto\x12!boulderopalscaleupsdk.protobuf.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x19google/protobuf/any.proto\"\xc7\x01\n\x13StartSessionRequest\x12\x19\n\x08\x61gent_id\x18\x01 \x01(\tR\x07\x61gentId\x12\x19\n\x08\x61pp_name\x18\x02 \x01(\tR\x07\x61ppName\x12\x1f\n\x0b\x64\x65vice_name\x18\x03 \x01(\tR\ndeviceName\x12!\n\x0croutine_name\x18\x04 \x01(\tR\x0broutineName\x12-\n\x04\x64\x61ta\x18\x05 \x01(\x0b\x32\x14.google.protobuf.AnyH\x00R\x04\x64\x61ta\x88\x01\x01\x42\x07\n\x05_data\"\xfc\x01\n\x14UpdateSessionRequest\x12\x1d\n\nsession_id\x18\x01 \x01(\tR\tsessionId\x12R\n\rcurrent_state\x18\x02 \x01(\x0e\x32-.boulderopalscaleupsdk.protobuf.v1.AgentStateR\x0c\x63urrentState\x12G\n\x07results\x18\x03 \x03(\x0b\x32-.boulderopalscaleupsdk.protobuf.v1.TaskResultR\x07results\x12(\n\x10task_in_progress\x18\x04 \x03(\tR\x0etaskInProgress\"\xc3\x03\n\x12\x41gentTasksResponse\x12\x1d\n\nsession_id\x18\x01 \x01(\tR\tsessionId\x12U\n\x0ctarget_state\x18\x02 \x01(\x0e\x32-.boulderopalscaleupsdk.protobuf.v1.AgentStateH\x00R\x0btargetState\x88\x01\x01\x12=\n\x05tasks\x18\x03 \x03(\x0b\x32\'.boulderopalscaleupsdk.protobuf.v1.TaskR\x05tasks\x12)\n\x10invalidate_tasks\x18\x04 \x03(\tR\x0finvalidateTasks\x12\x66\n\x0bretry_tasks\x18\x05 \x03(\x0b\x32\x45.boulderopalscaleupsdk.protobuf.v1.AgentTasksResponse.RetryTasksEntryR\nretryTasks\x12\x15\n\x06job_id\x18\x06 \x01(\tR\x05jobId\x1a=\n\x0fRetryTasksEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0f\n\r_target_state\"\x94\x02\n\x04Task\x12\x17\n\x07task_id\x18\x01 \x01(\tR\x06taskId\x12\x1d\n\nsession_id\x18\x02 \x01(\tR\tsessionId\x12\x1b\n\ttask_type\x18\x03 \x01(\tR\x08taskType\x12(\n\x04\x64\x61ta\x18\x04 \x01(\x0b\x32\x14.google.protobuf.AnyR\x04\x64\x61ta\x12,\n\x0ftimeout_seconds\x18\x05 \x01(\x05H\x00R\x0etimeoutSeconds\x88\x01\x01\x12\x1f\n\x08\x64\x65\x61\x64line\x18\x06 \x01(\x03H\x01R\x08\x64\x65\x61\x64line\x88\x01\x01\x12\x1d\n\ndepends_on\x18\x07 \x03(\tR\tdependsOnB\x12\n\x10_timeout_secondsB\x0b\n\t_deadline\"\xed\x01\n\nTaskResult\x12\x17\n\x07task_id\x18\x01 \x01(\tR\x06taskId\x12\x1d\n\nsession_id\x18\x03 \x01(\tR\tsessionId\x12\x1f\n\x0bresult_type\x18\x04 \x01(\tR\nresultType\x12.\n\x06result\x18\x05 \x01(\x0b\x32\x14.google.protobuf.AnyH\x00R\x06result\x12J\n\x05\x65rror\x18\x06 \x01(\x0b\x32\x32.boulderopalscaleupsdk.protobuf.v1.TaskErrorDetailH\x00R\x05\x65rrorB\n\n\x08response\"{\n\x0fTaskErrorDetail\x12@\n\x04\x63ode\x18\x01 \x01(\x0e\x32,.boulderopalscaleupsdk.protobuf.v1.TaskErrorR\x04\x63ode\x12\x1b\n\x06\x64\x65tail\x18\x02 \x01(\tH\x00R\x06\x64\x65tail\x88\x01\x01\x42\t\n\x07_detail*\xdd\x01\n\nAgentState\x12\x1b\n\x17\x41GENT_STATE_UNSPECIFIED\x10\x00\x12\x1e\n\x1a\x41GENT_STATE_ACTIVE_SESSION\x10\x01\x12\x1b\n\x17\x41GENT_STATE_ACTIVE_IDLE\x10\x02\x12)\n%AGENT_STATE_SHUTDOWN_CLIENT_INITIATED\x10\x03\x12*\n&AGENT_STATE_SHUTDOWN_MANAGER_INITIATED\x10\x04\x12\x1e\n\x1a\x41GENT_STATE_SHUTDOWN_FAULT\x10\x05*G\n\tTaskError\x12\x1a\n\x16TASK_ERROR_UNSPECIFIED\x10\x00\x12\x1e\n\x1aTASK_ERROR_NOT_IMPLEMENTED\x10\x01\x32\xe0\x02\n\x13\x41gentManagerService\x12\xa1\x01\n\x0cStartSession\x12\x36.boulderopalscaleupsdk.protobuf.v1.StartSessionRequest\x1a\x35.boulderopalscaleupsdk.protobuf.v1.AgentTasksResponse\"\"\x82\xd3\xe4\x93\x02\x1c\"\x17/v1/agent/start_session:\x01*\x12\xa4\x01\n\rUpdateSession\x12\x37.boulderopalscaleupsdk.protobuf.v1.UpdateSessionRequest\x1a\x35.boulderopalscaleupsdk.protobuf.v1.AgentTasksResponse\"#\x82\xd3\xe4\x93\x02\x1d\"\x18/v1/agent/update_session:\x01*B\xa8\x02\n%com.boulderopalscaleupsdk.protobuf.v1B\tTaskProtoP\x01ZNgithub.com/qctrl/boulder-opal-scale-up/proto/boulderopalscaleupsdk/protobuf/v1\xa2\x02\x03\x42PX\xaa\x02!Boulderopalscaleupsdk.Protobuf.V1\xca\x02!Boulderopalscaleupsdk\\Protobuf\\V1\xe2\x02-Boulderopalscaleupsdk\\Protobuf\\V1\\GPBMetadata\xea\x02#Boulderopalscaleupsdk::Protobuf::V1b\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -30,24 +30,24 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_AGENTMANAGERSERVICE'].methods_by_name['StartSession']._serialized_options = b'\202\323\344\223\002\034\"\027/v1/agent/start_session:\001*'
   _globals['_AGENTMANAGERSERVICE'].methods_by_name['UpdateSession']._loaded_options = None
   _globals['_AGENTMANAGERSERVICE'].methods_by_name['UpdateSession']._serialized_options = b'\202\323\344\223\002\035\"\030/v1/agent/update_session:\001*'
-  _globals['_AGENTSTATE']._serialized_start=1673
-  _globals['_AGENTSTATE']._serialized_end=1894
-  _globals['_TASKERROR']._serialized_start=1896
-  _globals['_TASKERROR']._serialized_end=1967
+  _globals['_AGENTSTATE']._serialized_start=1696
+  _globals['_AGENTSTATE']._serialized_end=1917
+  _globals['_TASKERROR']._serialized_start=1919
+  _globals['_TASKERROR']._serialized_end=1990
   _globals['_STARTSESSIONREQUEST']._serialized_start=141
   _globals['_STARTSESSIONREQUEST']._serialized_end=340
   _globals['_UPDATESESSIONREQUEST']._serialized_start=343
   _globals['_UPDATESESSIONREQUEST']._serialized_end=595
   _globals['_AGENTTASKSRESPONSE']._serialized_start=598
-  _globals['_AGENTTASKSRESPONSE']._serialized_end=1026
-  _globals['_AGENTTASKSRESPONSE_RETRYTASKSENTRY']._serialized_start=948
-  _globals['_AGENTTASKSRESPONSE_RETRYTASKSENTRY']._serialized_end=1009
-  _globals['_TASK']._serialized_start=1029
-  _globals['_TASK']._serialized_end=1305
-  _globals['_TASKRESULT']._serialized_start=1308
-  _globals['_TASKRESULT']._serialized_end=1545
-  _globals['_TASKERRORDETAIL']._serialized_start=1547
-  _globals['_TASKERRORDETAIL']._serialized_end=1670
-  _globals['_AGENTMANAGERSERVICE']._serialized_start=1970
-  _globals['_AGENTMANAGERSERVICE']._serialized_end=2322
+  _globals['_AGENTTASKSRESPONSE']._serialized_end=1049
+  _globals['_AGENTTASKSRESPONSE_RETRYTASKSENTRY']._serialized_start=971
+  _globals['_AGENTTASKSRESPONSE_RETRYTASKSENTRY']._serialized_end=1032
+  _globals['_TASK']._serialized_start=1052
+  _globals['_TASK']._serialized_end=1328
+  _globals['_TASKRESULT']._serialized_start=1331
+  _globals['_TASKRESULT']._serialized_end=1568
+  _globals['_TASKERRORDETAIL']._serialized_start=1570
+  _globals['_TASKERRORDETAIL']._serialized_end=1693
+  _globals['_AGENTMANAGERSERVICE']._serialized_start=1993
+  _globals['_AGENTMANAGERSERVICE']._serialized_end=2345
 # @@protoc_insertion_point(module_scope)

boulderopalscaleupsdk/protobuf/v1/task_pb2.pyi

@@ -57,7 +57,7 @@ class UpdateSessionRequest(_message.Message):
     def __init__(self, session_id: _Optional[str] = ..., current_state: _Optional[_Union[AgentState, str]] = ..., results: _Optional[_Iterable[_Union[TaskResult, _Mapping]]] = ..., task_in_progress: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class AgentTasksResponse(_message.Message):
-    __slots__ = ("session_id", "target_state", "tasks", "invalidate_tasks", "retry_tasks")
+    __slots__ = ("session_id", "target_state", "tasks", "invalidate_tasks", "retry_tasks", "job_id")
     class RetryTasksEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -70,12 +70,14 @@ class AgentTasksResponse(_message.Message):
     TASKS_FIELD_NUMBER: _ClassVar[int]
     INVALIDATE_TASKS_FIELD_NUMBER: _ClassVar[int]
     RETRY_TASKS_FIELD_NUMBER: _ClassVar[int]
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
     session_id: str
     target_state: AgentState
     tasks: _containers.RepeatedCompositeFieldContainer[Task]
     invalidate_tasks: _containers.RepeatedScalarFieldContainer[str]
     retry_tasks: _containers.ScalarMap[str, str]
-    def __init__(self, session_id: _Optional[str] = ..., target_state: _Optional[_Union[AgentState, str]] = ..., tasks: _Optional[_Iterable[_Union[Task, _Mapping]]] = ..., invalidate_tasks: _Optional[_Iterable[str]] = ..., retry_tasks: _Optional[_Mapping[str, str]] = ...) -> None: ...
+    job_id: str
+    def __init__(self, session_id: _Optional[str] = ..., target_state: _Optional[_Union[AgentState, str]] = ..., tasks: _Optional[_Iterable[_Union[Task, _Mapping]]] = ..., invalidate_tasks: _Optional[_Iterable[str]] = ..., retry_tasks: _Optional[_Mapping[str, str]] = ..., job_id: _Optional[str] = ...) -> None: ...
 
 class Task(_message.Message):
     __slots__ = ("task_id", "session_id", "task_type", "data", "timeout_seconds", "deadline", "depends_on")

boulderopalscaleupsdk/routines/__init__.py

@@ -1,6 +1,21 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
 """Routine library."""
 
-__all__ = ["ResonatorMapping", "Routine"]
+__all__ = ["ResonatorMapping", "Routine", "TransmonDiscovery", "TransmonRetuning"]
 
 from .common import Routine
 from .resonator_mapping import ResonatorMapping
+from .transmon_discovery import TransmonDiscovery
+from .transmon_retuning import TransmonRetuning

boulderopalscaleupsdk/routines/common.py

@@ -1,3 +1,16 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
 from pydantic import BaseModel, ConfigDict