Macro external data endpoint (#162)

* perf: retrieve macro execution conversation

* perf: SQL select conversation ID only

* perf: add all() method to enums.MacroTypes

* perf: add task_queue wait for exec group id

Author: andhreljaKern

business_objects/task_queue.py

@@ -83,6 +83,23 @@ def get_waiting_by_information_source(project_id: str, source_id: str) -> TaskQu
     )
 
 
+def get_waiting_by_macro_group_execution_ids(
+    project_id: str, source_ids: List[str]
+) -> TaskQueue:
+    source_ids = prevent_sql_injection(source_ids, isinstance(source_ids, list))
+    return (
+        session.query(TaskQueue)
+        .filter(
+            TaskQueue.task_type == enums.TaskType.RUN_COGNITION_MACRO.value,
+            text(
+                f"task_info->>'group_execution_id' IN ({','.join(map(repr, source_ids))})"
+            ),
+            text(f"task_info->>'project_id' = '{project_id}'"),
+        )
+        .first()
+    )
+
+
 def get_by_tokenization(project_id: str) -> TaskQueue:
     # could have multiple tokenization tasks in queue
     # if active => something is running else it's queued

cognition_objects/macro.py

@@ -626,6 +626,22 @@ def get_macro_execution_data_for_message_queue(
     return []
 
 
+def get_macro_execution_group_conversation_id(group_id: str):
+    group_id = prevent_sql_injection(group_id, isinstance(group_id, str))
+    query = f"""
+    SELECT
+        c.id
+    FROM
+        cognition.conversation c
+    WHERE
+        c.scope_dict->>'group_execution_id' = '{group_id}';
+    """
+    result = general.execute_first(query)
+    if result and result[0]:
+        return result[0]
+    return None
+
+
 def delete_by_exec_groups(
     macro_id: str,
     group_ids: List[str],

enums.py

@@ -755,6 +755,12 @@ class MacroType(Enum):
     DOCUMENT_MESSAGE_QUEUE = "DOCUMENT_MESSAGE_QUEUE"
     FOLDER_MESSAGE_QUEUE = "FOLDER_MESSAGE_QUEUE"
 
+    def all():
+        return [
+            MacroType.DOCUMENT_MESSAGE_QUEUE.value,
+            MacroType.FOLDER_MESSAGE_QUEUE.value,
+        ]
+
 
 # currently only one option, but could be extended in the future
 class MacroNodeContentType(Enum):