From a3a3569e48a8bfef41bf216ff0dfcb058c9d2333 Mon Sep 17 00:00:00 2001
From: Mirian Okradze <okradzemirian@gmail.com>
Date: Thu, 2 Nov 2023 22:41:44 +0400
Subject: [PATCH] build: worker

---
 apps/worker/main.py | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/apps/worker/main.py b/apps/worker/main.py
index 89f952bc5..1741f9b89 100644
--- a/apps/worker/main.py
+++ b/apps/worker/main.py
@@ -1,7 +1,5 @@
-import asyncio
 from datetime import timedelta
 
-import httpx
 import requests
 from celery import Celery
 
@@ -17,11 +15,11 @@
 CELERY_BEAT_SCHEDULE = {
     "register-scheduled-run-tasks": {
         "task": "register-scheduled-run-tasks",
-        "schedule": timedelta(seconds=15),
+        "schedule": timedelta(minutes=2),
     },
     "register-fine-tuning-tasks": {
         "task": "register-fine-tuning-tasks",
-        "schedule": timedelta(seconds=15),
+        "schedule": timedelta(minutes=2),
     },
 }
 
@@ -41,12 +39,8 @@ def execute_scheduled_runs_task():
         headers={"Authorization": f"Bearer {Config.SERVER_AUTH_TOKEN}"},
     )
 
-    print(res.text)
-
     schedules_with_configs = res.json()
 
-    print(schedules_with_configs)
-
     for schedule in schedules_with_configs:
         execute_single_schedule_task.apply_async(args=[schedule["schedule"]["id"]])
 
@@ -80,12 +74,13 @@ def register_fine_tunings_task():
         headers={"Authorization": f"Bearer {Config.SERVER_AUTH_TOKEN}"},
     )
 
-    return res.text
+    fine_tunings = res.json()
 
-    # fine_tunings = res.json()
+    for fine_tuning in fine_tunings:
+        check_single_fine_tuning_task.apply_async(args=[fine_tuning["id"]])
 
-    # for fine_tuning in fine_tunings:
-    #     check_single_fine_tuning_task.apply_async(args=[fine_tuning["id"]])
+    fine_tuning_ids = [fine_tuning["id"] for fine_tuning in fine_tunings]
+    return fine_tuning_ids
 
 
 @app.task(