chore(tasks): publish assignment to generate assignment hub id

superiorsd10 · May 2, 2024 · 737853c · 737853c
1 parent c660539
commit 737853c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/server/app/celery/tasks/assignment_tasks.py b/server/app/celery/tasks/assignment_tasks.py
@@ -645,6 +645,7 @@ def process_assignment_generation(
     types_of_questions: dict,
     generate_assignment_id: str,
     assignments_count: int,
+    hub_id: str,
 ) -> None:
     """
     Process assignment generation task asynchronously.
@@ -677,6 +678,8 @@ def process_assignment_generation(
             ]
         )
 
+        print(types_of_questions_string)
+
         if assignments_count == 0:
             generated_assignment = generate_assignment_llama(
                 title=title,
@@ -704,9 +707,10 @@ def process_assignment_generation(
         if assignments_dict:
             redis_client = Config.REDIS_CLIENT
             generate_assignment_key = f"generate_assignment_id_{generate_assignment_id}"
+            generate_assignment_hub_key = f"generate_assignment_hub_id_{hub_id}"
             assignments_dict_data = json.dumps(assignments_dict)
             redis_client.set(generate_assignment_key, assignments_dict_data)
-            redis_client.publish(generate_assignment_key, assignments_dict_data)
+            redis_client.publish(generate_assignment_hub_key, assignments_dict_data)
         else:
             print("assignments_dict is None or empty, skipping Redis operations.")
 

diff --git a/server/app/sockets/__init__.py b/server/app/sockets/__init__.py
@@ -13,3 +13,7 @@
     handle_leave_hub,
     handle_send_message,
 )
+
+from .assignment_sockets import (
+    generate_assignment,
+)