@@ -170,44 +170,44 @@ def __init__(
170170 self ._pool = requests .Session ()
171171
172172 def get_plans (self ) -> PlanResponse :
173- return self ._request_and_deserialize ("/plans" , PlanResponse )
173+ return self ._request_and_deserialize ("/api/v1/ plans" , PlanResponse )
174174
175175 def get_plan (self , name : str ) -> PlanModel :
176- return self ._request_and_deserialize (f"/plans/{ name } " , PlanModel )
176+ return self ._request_and_deserialize (f"/api/v1/ plans/{ name } " , PlanModel )
177177
178178 def get_devices (self ) -> DeviceResponse :
179- return self ._request_and_deserialize ("/devices" , DeviceResponse )
179+ return self ._request_and_deserialize ("/api/v1/ devices" , DeviceResponse )
180180
181181 def get_device (self , name : str ) -> DeviceModel :
182- return self ._request_and_deserialize (f"/devices/{ name } " , DeviceModel )
182+ return self ._request_and_deserialize (f"/api/v1/ devices/{ name } " , DeviceModel )
183183
184184 def get_state (self ) -> WorkerState :
185- return self ._request_and_deserialize ("/worker/state" , WorkerState )
185+ return self ._request_and_deserialize ("/api/v1/ worker/state" , WorkerState )
186186
187187 def set_state (
188188 self ,
189189 state : Literal [WorkerState .RUNNING , WorkerState .PAUSED ],
190190 defer : bool | None = False ,
191191 ):
192192 return self ._request_and_deserialize (
193- "/worker/state" ,
193+ "/api/v1/ worker/state" ,
194194 target_type = WorkerState ,
195195 method = "PUT" ,
196196 data = {"new_state" : state , "defer" : defer },
197197 )
198198
199199 def get_task (self , task_id : str ) -> TrackableTask :
200- return self ._request_and_deserialize (f"/tasks/{ task_id } " , TrackableTask )
200+ return self ._request_and_deserialize (f"/api/v1/ tasks/{ task_id } " , TrackableTask )
201201
202202 def get_all_tasks (self ) -> TasksListResponse :
203- return self ._request_and_deserialize ("/tasks" , TasksListResponse )
203+ return self ._request_and_deserialize ("/api/v1/ tasks" , TasksListResponse )
204204
205205 def get_active_task (self ) -> WorkerTask :
206- return self ._request_and_deserialize ("/worker/task" , WorkerTask )
206+ return self ._request_and_deserialize ("/api/v1/ worker/task" , WorkerTask )
207207
208208 def create_task (self , task : TaskRequest ) -> TaskResponse :
209209 return self ._request_and_deserialize (
210- "/tasks" ,
210+ "/api/v1/ tasks" ,
211211 TaskResponse ,
212212 method = "POST" ,
213213 get_exception = _create_task_exceptions ,
@@ -216,12 +216,12 @@ def create_task(self, task: TaskRequest) -> TaskResponse:
216216
217217 def clear_task (self , task_id : str ) -> TaskResponse :
218218 return self ._request_and_deserialize (
219- f"/tasks/{ task_id } " , TaskResponse , method = "DELETE"
219+ f"/api/v1/ tasks/{ task_id } " , TaskResponse , method = "DELETE"
220220 )
221221
222222 def update_worker_task (self , task : WorkerTask ) -> WorkerTask :
223223 return self ._request_and_deserialize (
224- "/worker/task" ,
224+ "/api/v1/ worker/task" ,
225225 WorkerTask ,
226226 method = "PUT" ,
227227 data = task .model_dump (),
@@ -233,18 +233,18 @@ def cancel_current_task(
233233 reason : str | None = None ,
234234 ):
235235 return self ._request_and_deserialize (
236- "/worker/state" ,
236+ "/api/v1/ worker/state" ,
237237 target_type = WorkerState ,
238238 method = "PUT" ,
239239 data = {"new_state" : state , "reason" : reason },
240240 )
241241
242242 def get_environment (self ) -> EnvironmentResponse :
243- return self ._request_and_deserialize ("/environment" , EnvironmentResponse )
243+ return self ._request_and_deserialize ("/api/v1/ environment" , EnvironmentResponse )
244244
245245 def delete_environment (self ) -> EnvironmentResponse :
246246 return self ._request_and_deserialize (
247- "/environment" , EnvironmentResponse , method = "DELETE"
247+ "/api/v1/ environment" , EnvironmentResponse , method = "DELETE"
248248 )
249249
250250 def get_oidc_config (self ) -> OIDCConfig | None :
@@ -258,7 +258,7 @@ def get_python_environment(
258258 self , name : str | None = None , source : SourceInfo | None = None
259259 ) -> PythonEnvironmentResponse :
260260 return self ._request_and_deserialize (
261- "/python_environment" ,
261+ "/api/v1/ python_environment" ,
262262 PythonEnvironmentResponse ,
263263 params = {"name" : name , "source" : source },
264264 )
0 commit comments