@@ -278,8 +278,17 @@ def get_seed(self, session: int, task_index: int, hash_cache: dict = {}) -> int:
278
278
method = "GET"
279
279
url = "http://209.126.9.130:18000/api/v1/task/seed"
280
280
response = requests .request (method , url , params = {"session" : session , "task_number" : task_index })
281
- if response .status_code == 200 and response .json ()["seed" ] is not None :
282
- return response .json ()["seed" ]
281
+ if response .status_code == 200 :
282
+ response_json = response .json ()
283
+ success = response_json ["success" ]
284
+ if not success :
285
+ raise Exception (f"Failed to get seed from API: { seed } " )
286
+
287
+ seed = response_json ["seed" ], response_json ["task_id_seed" ]
288
+
289
+ if seed is None :
290
+ raise Exception (f"Seed is None" )
291
+ return seed
283
292
else :
284
293
raise Exception (f"Failed to get seed from API: { response .status_code } " )
285
294
@@ -306,15 +315,19 @@ async def forward(self):
306
315
return
307
316
308
317
bt .logging .info (f"Forwarding task #{ task_index } in session #{ session } " )
309
- seed = self .get_seed (session , task_index )
318
+ seed , task_id_seed = self .get_seed (session , task_index )
310
319
311
- bt .logging .info (f"Init random with seed: { seed } " )
320
+ bt .logging .info (f"Random seed: { seed } | task_id_seed: { task_id_seed } " )
312
321
random .seed (seed )
313
322
set_seed (seed )
314
323
324
+ number_of_tries = 0
315
325
while True :
316
326
try :
317
327
await self .synthensize_task ()
328
+ task , _ = self .synthetic_tasks [- 1 ]
329
+ number_of_tries += 1
330
+ task .task_id = f"{ task_id_seed } _{ number_of_tries } "
318
331
break
319
332
except Exception as e :
320
333
bt .logging .error (
0 commit comments