From 083541fe613a8da7687f14c50bff87b1b88c4475 Mon Sep 17 00:00:00 2001
From: David Gauldie <dgauldie@uncharted.software>
Date: Wed, 2 Oct 2024 15:45:35 -0400
Subject: [PATCH] 4991 task update gollm intervention policy schema to align
 with new intervention policy model (#5010)

---
 .../intervention-policy-operation.ts          |  6 +-
 .../tera-intervention-policy-drilldown.vue    | 75 +++++++++++++++----
 .../tera-intervention-policy-node.vue         | 37 ++++++++-
 .../client/hmi-client/src/services/goLLM.ts   | 17 +++++
 .../src/services/notificationEventHandlers.ts | 14 +++-
 packages/client/hmi-client/src/types/Types.ts |  1 +
 .../prompts/equations_from_image.py           |  8 ++
 .../prompts/interventions_from_document.py    |  9 ++-
 .../gollm_openai/prompts/latex_style_guide.py | 18 +++++
 .../schemas/intervention_policy.json          | 45 ++++++-----
 packages/gollm/setup.py                       |  1 +
 .../controller/gollm/GoLLMController.java     | 22 +++---
 .../hmiserver/models/ClientEventType.java     |  1 +
 .../interventions/Intervention.java           |  4 +-
 ...erventionsFromDocumentResponseHandler.java | 13 ----
 .../tasks/TaskNotificationEventTypes.java     | 15 ++--
 16 files changed, 214 insertions(+), 72 deletions(-)
 create mode 100644 packages/gollm/gollm_openai/prompts/latex_style_guide.py

diff --git a/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/intervention-policy-operation.ts b/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/intervention-policy-operation.ts
index 5d96c4e32b..7b97ab562e 100644
--- a/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/intervention-policy-operation.ts
+++ b/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/intervention-policy-operation.ts
@@ -1,10 +1,11 @@
 import { InterventionPolicy } from '@/types/Types';
+import type { BaseState, Operation } from '@/types/workflow';
 import { WorkflowOperationTypes } from '@/types/workflow';
-import type { Operation, BaseState } from '@/types/workflow';
 import { isEqual, omit } from 'lodash';
 
 export interface InterventionPolicyState extends BaseState {
 	interventionPolicy: InterventionPolicy;
+	taskIds: string[];
 }
 
 export const InterventionPolicyOperation: Operation = {
@@ -24,7 +25,8 @@ export const InterventionPolicyOperation: Operation = {
 			interventionPolicy: {
 				modelId: '',
 				interventions: []
-			}
+			},
+			taskIds: []
 		};
 		return init;
 	}
diff --git a/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-drilldown.vue b/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-drilldown.vue
index 24a4831051..87b2d1d3de 100644
--- a/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-drilldown.vue
+++ b/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-drilldown.vue
@@ -16,14 +16,17 @@
 				<template #content>
 					<section>
 						<nav class="inline-flex">
-							<!-- Disabled until Backend code is complete -->
-							<!-- <Button class="flex-1 mr-1" outlined severity="secondary" label="Extract from inputs" /> -->
 							<Button
-								class="flex-1 ml-1"
-								label="Create New"
-								:disabled="!model?.id"
-								@click="createNewInterventionPolicy"
+								class="flex-1 mr-1"
+								outlined
+								severity="secondary"
+								label="Extract from inputs"
+								icon="pi pi-sparkles"
+								:loading="isLoading"
+								:disabled="!props.node.inputs[0]?.value && !props.node.inputs[1]?.value"
+								@click="extractInterventionPolicyFromInputs"
 							/>
+							<Button class="ml-1" label="Create New" :disabled="!model?.id" @click="createNewInterventionPolicy" />
 						</nav>
 						<tera-input-text v-model="filterInterventionsText" placeholder="Filter" />
 						<ul v-if="!isFetchingPolicies">
@@ -161,7 +164,7 @@
 
 <script setup lang="ts">
 import _, { cloneDeep, groupBy, isEmpty, omit } from 'lodash';
-import { computed, onMounted, ref, watch, nextTick, ComponentPublicInstance } from 'vue';
+import { ComponentPublicInstance, computed, nextTick, onMounted, ref, watch } from 'vue';
 import TeraDrilldown from '@/components/drilldown/tera-drilldown.vue';
 import TeraDrilldownSection from '@/components/drilldown/tera-drilldown-section.vue';
 import { WorkflowNode } from '@/types/workflow';
@@ -170,17 +173,17 @@ import TeraColumnarPanel from '@/components/widgets/tera-columnar-panel.vue';
 import Button from 'primevue/button';
 import TeraInputText from '@/components/widgets/tera-input-text.vue';
 import { getInterventionPoliciesForModel, getModel } from '@/services/model';
-import { Intervention, InterventionPolicy, Model, AssetType } from '@/types/Types';
+import { AssetType, Intervention, InterventionPolicy, Model, type TaskResponse } from '@/types/Types';
 import { logger } from '@/utils/logger';
 import TeraProgressSpinner from '@/components/widgets/tera-progress-spinner.vue';
 import { useConfirm } from 'primevue/useconfirm';
 import { getParameters, getStates } from '@/model-representation/service';
 import TeraToggleableInput from '@/components/widgets/tera-toggleable-input.vue';
 import {
-	getInterventionPolicyById,
-	updateInterventionPolicy,
 	blankIntervention,
-	flattenInterventionData
+	flattenInterventionData,
+	getInterventionPolicyById,
+	updateInterventionPolicy
 } from '@/services/intervention-policy';
 import Accordion from 'primevue/accordion';
 import AccordionTab from 'primevue/accordiontab';
@@ -192,12 +195,13 @@ import { createInterventionChart } from '@/services/charts';
 import VegaChart from '@/components/widgets/VegaChart.vue';
 import TeraSaveAssetModal from '@/components/project/tera-save-asset-modal.vue';
 import { useProjects } from '@/composables/project';
+import { interventionPolicyFromDocument } from '@/services/goLLM';
 import TeraInterventionCard from './tera-intervention-card.vue';
 import {
 	InterventionPolicyOperation,
 	InterventionPolicyState,
-	isInterventionPoliciesValuesEqual,
-	isInterventionPoliciesEqual
+	isInterventionPoliciesEqual,
+	isInterventionPoliciesValuesEqual
 } from './intervention-policy-operation';
 import TeraInterventionPolicyCard from './tera-intervention-policy-card.vue';
 
@@ -234,6 +238,7 @@ const isSidebarOpen = ref(true);
 const filterInterventionsText = ref('');
 const model = ref<Model | null>(null);
 const isFetchingPolicies = ref(false);
+const isLoading = ref(false);
 const interventionsPolicyList = ref<InterventionPolicy[]>([]);
 const interventionPoliciesFiltered = computed(() =>
 	interventionsPolicyList.value
@@ -269,6 +274,13 @@ const isSaveDisabled = computed(() => {
 	return hasSelectedPolicy && (isPolicyIdDifferent || arePoliciesEqual || !arePolicyValuesEqual);
 });
 
+const documentIds = computed(() =>
+	props.node.inputs
+		.filter((input) => input.type === 'documentId' && input.status === 'connected')
+		.map((input) => input.value?.[0]?.documentId)
+		.filter((id): id is string => id !== undefined)
+);
+
 const parameterOptions = computed(() => {
 	if (!model.value) return [];
 	return getParameters(model.value).map((parameter) => ({
@@ -473,6 +485,29 @@ const createNewInterventionPolicy = () => {
 	showSaveModal.value = true;
 };
 
+const extractInterventionPolicyFromInputs = async () => {
+	const state = cloneDeep(props.node.state);
+	if (!model.value?.id) {
+		return;
+	}
+
+	if (documentIds.value) {
+		const promiseList = [] as Promise<TaskResponse | null>[];
+		documentIds.value.forEach((documentId) => {
+			promiseList.push(
+				interventionPolicyFromDocument(documentId, model.value?.id as string, props.node.workflowId, props.node.id)
+			);
+		});
+		const responsesRaw = await Promise.all(promiseList);
+		responsesRaw.forEach((resp) => {
+			if (resp) {
+				state.taskIds.push(resp.id);
+			}
+		});
+	}
+	emit('update-state', state);
+};
+
 watch(
 	() => knobs.value,
 	async () => {
@@ -493,6 +528,20 @@ watch(
 	}
 );
 
+watch(
+	() => props.node.state.taskIds,
+	async (watchVal) => {
+		if (watchVal.length > 0) {
+			isLoading.value = true;
+		} else {
+			isLoading.value = false;
+			const modelId = props.node.inputs[0].value?.[0];
+			if (!modelId) return;
+			await fetchInterventionPolicies(modelId);
+		}
+	}
+);
+
 onMounted(() => {
 	if (props.node.active) {
 		selectedOutputId.value = props.node.active;
diff --git a/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-node.vue b/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-node.vue
index e1507220eb..1e5afe433c 100644
--- a/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-node.vue
+++ b/packages/client/hmi-client/src/components/workflow/ops/intervention-policy/tera-intervention-policy-node.vue
@@ -6,6 +6,7 @@
 			</li>
 		</ul>
 		<tera-operator-placeholder :node="node" v-else />
+		<tera-progress-spinner is-centered :font-size="2" v-if="isLoading" />
 		<Button
 			:label="isModelInputConnected ? 'Open' : 'Attach a model'"
 			@click="emit('open-drilldown')"
@@ -17,7 +18,7 @@
 </template>
 
 <script setup lang="ts">
-import { computed, watch } from 'vue';
+import { computed, ref, watch } from 'vue';
 import { WorkflowNode, WorkflowPortStatus } from '@/types/workflow';
 import Button from 'primevue/button';
 import TeraOperatorPlaceholder from '@/components/operator/tera-operator-placeholder.vue';
@@ -25,6 +26,9 @@ import _, { cloneDeep, groupBy } from 'lodash';
 import { blankIntervention, flattenInterventionData } from '@/services/intervention-policy';
 import { createInterventionChart } from '@/services/charts';
 import VegaChart from '@/components/widgets/VegaChart.vue';
+import { useClientEvent } from '@/composables/useClientEvent';
+import { type ClientEvent, ClientEventType, type TaskResponse, TaskStatus } from '@/types/Types';
+import TeraProgressSpinner from '@/components/widgets/tera-progress-spinner.vue';
 import { InterventionPolicyState } from './intervention-policy-operation';
 
 const emit = defineEmits(['open-drilldown', 'update-state']);
@@ -32,6 +36,19 @@ const props = defineProps<{
 	node: WorkflowNode<InterventionPolicyState>;
 }>();
 
+const taskIds = ref<string[]>([]);
+
+const interventionEventHandler = async (event: ClientEvent<TaskResponse>) => {
+	if (!taskIds.value.includes(event.data?.id)) return;
+	if ([TaskStatus.Success, TaskStatus.Cancelled, TaskStatus.Failed].includes(event.data.status)) {
+		taskIds.value = taskIds.value.filter((id) => id !== event.data.id);
+	}
+};
+
+useClientEvent(ClientEventType.TaskGollmInterventionsFromDocument, interventionEventHandler);
+
+const isLoading = computed(() => taskIds.value.length > 0);
+
 const modelInput = props.node.inputs.find((input) => input.type === 'modelId');
 const isModelInputConnected = computed(() => modelInput?.status === WorkflowPortStatus.CONNECTED);
 
@@ -69,6 +86,24 @@ watch(
 	},
 	{ deep: true }
 );
+
+watch(
+	() => props.node.state.taskIds,
+	() => {
+		taskIds.value = props.node.state.taskIds ?? [];
+	}
+);
+
+watch(
+	() => isLoading.value,
+	() => {
+		if (!isLoading.value) {
+			const state = cloneDeep(props.node.state);
+			state.taskIds = [];
+			emit('update-state', state);
+		}
+	}
+);
 </script>
 
 <style scoped>
diff --git a/packages/client/hmi-client/src/services/goLLM.ts b/packages/client/hmi-client/src/services/goLLM.ts
index 836afe600f..8ca5a94466 100644
--- a/packages/client/hmi-client/src/services/goLLM.ts
+++ b/packages/client/hmi-client/src/services/goLLM.ts
@@ -34,6 +34,23 @@ export async function modelCard(modelId: string, documentId?: string): Promise<v
 	}
 }
 
+export async function interventionPolicyFromDocument(
+	documentId: string,
+	modelId: string,
+	workflowId?: string,
+	nodeId?: string
+): Promise<TaskResponse> {
+	const { data } = await API.get<TaskResponse>('/gollm/interventions-from-document', {
+		params: {
+			'model-id': modelId,
+			'document-id': documentId,
+			'workflow-id': workflowId,
+			'node-id': nodeId
+		}
+	});
+	return data;
+}
+
 export async function configureModelFromDocument(
 	documentId: string,
 	modelId: string,
diff --git a/packages/client/hmi-client/src/services/notificationEventHandlers.ts b/packages/client/hmi-client/src/services/notificationEventHandlers.ts
index 46633b7e3b..6f1ce6ccf6 100644
--- a/packages/client/hmi-client/src/services/notificationEventHandlers.ts
+++ b/packages/client/hmi-client/src/services/notificationEventHandlers.ts
@@ -162,7 +162,7 @@ export const createNotificationEventHandlers = (notificationItems: Ref<Notificat
 	});
 	registerHandler<TaskResponse>(ClientEventType.TaskGollmConfigureModelFromDocument, (event, created) => {
 		created.supportCancel = true;
-		created.sourceName = 'Configure model';
+		created.sourceName = 'Model Configuration from Document';
 		created.assetId = event.data.additionalProperties.workflowId as string;
 		created.pageType = AssetType.Workflow;
 		created.nodeId = event.data.additionalProperties.nodeId as string;
@@ -172,7 +172,7 @@ export const createNotificationEventHandlers = (notificationItems: Ref<Notificat
 	});
 	registerHandler<TaskResponse>(ClientEventType.TaskGollmConfigureModelFromDataset, (event, created) => {
 		created.supportCancel = true;
-		created.sourceName = 'Configure model';
+		created.sourceName = 'Model Configuration from Dataset';
 		created.assetId = event.data.additionalProperties.workflowId as string;
 		created.pageType = AssetType.Workflow;
 		created.nodeId = event.data.additionalProperties.nodeId as string;
@@ -190,6 +190,16 @@ export const createNotificationEventHandlers = (notificationItems: Ref<Notificat
 			Object.assign(created, { context: workflow?.name || '' })
 		);
 	});
+	registerHandler<TaskResponse>(ClientEventType.TaskGollmInterventionsFromDocument, (event, created) => {
+		created.supportCancel = true;
+		created.sourceName = 'Intervention Policies from Document';
+		created.assetId = event.data.additionalProperties.workflowId as string;
+		created.pageType = AssetType.Workflow;
+		created.nodeId = event.data.additionalProperties.nodeId as string;
+		getWorkflow(created.assetId, created.projectId).then((workflow) =>
+			Object.assign(created, { context: workflow?.name || '' })
+		);
+	});
 	registerHandler<StatusUpdate<SimulationNotificationData>>(
 		ClientEventType.SimulationNotification,
 		(event, created) => {
diff --git a/packages/client/hmi-client/src/types/Types.ts b/packages/client/hmi-client/src/types/Types.ts
index d28a201240..e936121e1e 100644
--- a/packages/client/hmi-client/src/types/Types.ts
+++ b/packages/client/hmi-client/src/types/Types.ts
@@ -1076,6 +1076,7 @@ export enum ClientEventType {
     TaskGollmEnrichAmr = "TASK_GOLLM_ENRICH_AMR",
     TaskGollmEquationsFromImage = "TASK_GOLLM_EQUATIONS_FROM_IMAGE",
     TaskGollmGenerateSummary = "TASK_GOLLM_GENERATE_SUMMARY",
+    TaskGollmInterventionsFromDocument = "TASK_GOLLM_INTERVENTIONS_FROM_DOCUMENT",
     TaskGollmModelCard = "TASK_GOLLM_MODEL_CARD",
     TaskMiraAmrToMmt = "TASK_MIRA_AMR_TO_MMT",
     TaskMiraGenerateModelLatex = "TASK_MIRA_GENERATE_MODEL_LATEX",
diff --git a/packages/gollm/gollm_openai/prompts/equations_from_image.py b/packages/gollm/gollm_openai/prompts/equations_from_image.py
index b240341b02..64007d243e 100644
--- a/packages/gollm/gollm_openai/prompts/equations_from_image.py
+++ b/packages/gollm/gollm_openai/prompts/equations_from_image.py
@@ -4,5 +4,13 @@
 
 Do not respond in full sentences; only create a JSON object that satisfies the JSON schema specified in the response format.
 
+LaTeX equations need to conform to a standard form. Use the following guidelines to ensure that your LaTeX equations are correctly formatted:
+
+--- STYLE GUIDE ---
+
+{style_guide}
+
+--- STYLE GUIDE END ---
+
 Answer:
 """
diff --git a/packages/gollm/gollm_openai/prompts/interventions_from_document.py b/packages/gollm/gollm_openai/prompts/interventions_from_document.py
index d594c2db99..c160cfe17a 100644
--- a/packages/gollm/gollm_openai/prompts/interventions_from_document.py
+++ b/packages/gollm/gollm_openai/prompts/interventions_from_document.py
@@ -2,7 +2,7 @@
 You are a helpful agent designed to find intervention policies for a given AMR model described in a research paper.
 For context, intervention policies can include multiple interventions that include only static interventions or dynamic interventions.
 Static interventions are applied at a specific point in time and permanently change the value of a specific parameter or state.
-Dynamic interventions try to keep a specific parameter from going above or below a threshold value.
+Dynamic interventions are applied when the value of a state crosses a threshold value.
 
 Use the following AMR model JSON file as a reference:
 
@@ -26,9 +26,10 @@
 3.	`model_id` id a UUID. If the AMR model has an id, you can use it. Otherwise, you can set as the nil UUID "00000000-0000-0000-0000-000000000000".
 4.	For each intervention specified in the policy create an intervention object with the following rules.
     a.	Create a value for `name` from the user-provided text.
-    b.	`appliedTo` should reference the id of the parameter or initial state of the AMR Model. If you cannot find an initial state or parameter that matches the intervention, do not create an intervention object.
-    c.	`type` should be either "state" or "parameter" depending on what the intervention is applied to.
-    d.	create a list of either static or dynamic interventions, but not both.
+    d.	create a list of either static or dynamic interventions, but not both. The other list should be empty.
+        i. `appliedTo` should reference the id of a parameter or state in the AMR Model. If you cannot find a state or parameter in the AMR model that matches, do not create an intervention object.
+        ii. `type` should be either "state" or "parameter" depending on what the intervention is applied to.
+        iii. For dynamic interventions, `parameter` should be the id of a state that the threshold is applied to.
 
 Do not respond in full sentences; only create a JSON object that satisfies the JSON schema specified in the response format.
 
diff --git a/packages/gollm/gollm_openai/prompts/latex_style_guide.py b/packages/gollm/gollm_openai/prompts/latex_style_guide.py
new file mode 100644
index 0000000000..4cecc6166f
--- /dev/null
+++ b/packages/gollm/gollm_openai/prompts/latex_style_guide.py
@@ -0,0 +1,18 @@
+LATEXT_STYLE_GUIDE = """
+1) Derivatives must be written in Leibniz notation
+2) First-order derivative must be on the left of the equal sign
+3) Use whitespace to indicate multiplication
+    a) "*" is optional but probably should be avoided
+4) "(t)" is optional and probably should be avoided
+5) Avoid superscripts and LaTeX superscripts "^", particularly to denote sub-variables
+6) Subscripts using LaTeX "_" are permitted
+    a) Ensure that all characters used in the subscript are surrounded by a pair of curly brackets "{...}"
+7) Avoid mathematical constants like pi or Euler's number
+    a) Replace them as floats with 3 decimal places of precision
+8) Avoid parentheses
+9) Avoid capital sigma and pi notations for summation and product
+10) Avoid non-ASCII characters when possible
+11) Avoid using homoglyphs
+12) Avoid words or multi-character names for variables and names
+    a) Use camel case to express multi-word or multi-character names
+"""
diff --git a/packages/gollm/gollm_openai/schemas/intervention_policy.json b/packages/gollm/gollm_openai/schemas/intervention_policy.json
index a69783c97f..78ce944312 100644
--- a/packages/gollm/gollm_openai/schemas/intervention_policy.json
+++ b/packages/gollm/gollm_openai/schemas/intervention_policy.json
@@ -8,16 +8,6 @@
         "name": {
           "type": "string"
         },
-        "appliedTo": {
-          "type": "string"
-        },
-        "type": {
-          "type": "string",
-          "enum": [
-            "state",
-            "parameter"
-          ]
-        },
         "staticInterventions": {
           "type": "array",
           "description": "A list of static interventions that can be applied to the model.",
@@ -31,11 +21,24 @@
               "value": {
                 "type": "number",
                 "description": "The new value of the state or parameter"
+              },
+              "appliedTo": {
+                "type": "string",
+                "description": "The state or parameter to which the intervention is applied. This should reference the id of a parameter or initial state of the AMR Model."
+              },
+              "type": {
+                "type": "string",
+                "enum": [
+                  "state",
+                  "parameter"
+                ]
               }
             },
             "required": [
               "timestep",
-              "value"
+              "value",
+              "appliedTo",
+              "type"
             ],
             "additionalProperties": false
           }
@@ -48,26 +51,34 @@
             "properties": {
               "parameter": {
                 "type": "string",
-                "description": "The parameter to which the intervention is applied"
+                "description": "The state to which the threshold is applied. This can only be a state, not a parameter. This should reference the id of an initial state of the AMR Model."
               },
               "threshold": {
                 "type": "string",
-                "description": "The threshold value for the which the parameter can't go above or below"
+                "description": "The threshold that once the value of the state crosses, the intervention is applied"
               },
               "value": {
                 "type": "string",
                 "description": "The new value of the state or parameter"
               },
-              "isGreaterThan": {
+              "appliedTo": {
+                "type": "string",
+                "description": "The state or parameter to which the intervention is applied. This should reference the id of a parameter or initial state of the AMR Model."
+              },
+              "type": {
                 "type": "string",
-                "description": "Whether the parameter should be greater than or less than the threshold"
+                "enum": [
+                  "state",
+                  "parameter"
+                ]
               }
             },
             "required": [
               "parameter",
               "threshold",
               "value",
-              "isGreaterThan"
+              "appliedTo",
+              "type"
             ],
             "additionalProperties": false
           }
@@ -75,8 +86,6 @@
       },
       "required": [
         "name",
-        "appliedTo",
-        "type",
         "staticInterventions",
         "dynamicInterventions"
       ],
diff --git a/packages/gollm/setup.py b/packages/gollm/setup.py
index b86b004612..a62bf4a7b3 100644
--- a/packages/gollm/setup.py
+++ b/packages/gollm/setup.py
@@ -25,6 +25,7 @@
             "gollm:equations_from_image=tasks.equations_from_image:main",
             "gollm:generate_response=tasks.generate_response:main",
             "gollm:generate_summary=tasks.generate_summary:main",
+            "gollm:interventions_from_document=tasks.interventions_from_document:main",
             "gollm:model_card=tasks.model_card:main",
         ],
     },
diff --git a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/controller/gollm/GoLLMController.java b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/controller/gollm/GoLLMController.java
index 4f9d47dd38..858de1433b 100644
--- a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/controller/gollm/GoLLMController.java
+++ b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/controller/gollm/GoLLMController.java
@@ -79,27 +79,29 @@ public class GoLLMController {
 	private final ProjectService projectService;
 	private final CurrentUserService currentUserService;
 
-	private final ModelCardResponseHandler modelCardResponseHandler;
-	private final ConfigureModelFromDocumentResponseHandler configureModelFromDocumentResponseHandler;
-	private final ConfigureModelFromDatasetResponseHandler configureModelFromDatasetResponseHandler;
 	private final CompareModelsResponseHandler compareModelsResponseHandler;
-	private final GenerateSummaryHandler generateSummaryHandler;
-	private final GenerateResponseHandler generateResponseHandler;
+	private final ConfigureModelFromDatasetResponseHandler configureModelFromDatasetResponseHandler;
+	private final ConfigureModelFromDocumentResponseHandler configureModelFromDocumentResponseHandler;
 	private final EnrichAmrResponseHandler enrichAmrResponseHandler;
 	private final EquationsFromImageResponseHandler equationsFromImageResponseHandler;
+	private final GenerateResponseHandler generateResponseHandler;
+	private final GenerateSummaryHandler generateSummaryHandler;
+	private final InterventionsFromDocumentResponseHandler interventionsFromDocumentResponseHandler;
+	private final ModelCardResponseHandler modelCardResponseHandler;
 
 	private final Messages messages;
 
 	@PostConstruct
 	void init() {
-		taskService.addResponseHandler(modelCardResponseHandler);
-		taskService.addResponseHandler(configureModelFromDocumentResponseHandler);
 		taskService.addResponseHandler(compareModelsResponseHandler);
 		taskService.addResponseHandler(configureModelFromDatasetResponseHandler);
-		taskService.addResponseHandler(generateSummaryHandler);
-		taskService.addResponseHandler(generateResponseHandler);
+		taskService.addResponseHandler(configureModelFromDocumentResponseHandler);
 		taskService.addResponseHandler(enrichAmrResponseHandler);
 		taskService.addResponseHandler(equationsFromImageResponseHandler);
+		taskService.addResponseHandler(generateResponseHandler);
+		taskService.addResponseHandler(generateSummaryHandler);
+		taskService.addResponseHandler(interventionsFromDocumentResponseHandler);
+		taskService.addResponseHandler(modelCardResponseHandler);
 	}
 
 	@PostMapping("/model-card")
@@ -471,7 +473,7 @@ public ResponseEntity<TaskResponse> createConfigureModelFromDatasetTask(
 		return ResponseEntity.ok().body(resp);
 	}
 
-	@PostMapping("/interventions-from-document")
+	@GetMapping("/interventions-from-document")
 	@Secured(Roles.USER)
 	@Operation(summary = "Dispatch a `GoLLM interventions-from-document` task")
 	@ApiResponses(
diff --git a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/ClientEventType.java b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/ClientEventType.java
index 90bb095ceb..155df32511 100644
--- a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/ClientEventType.java
+++ b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/ClientEventType.java
@@ -21,6 +21,7 @@ public enum ClientEventType {
 	TASK_GOLLM_ENRICH_AMR,
 	TASK_GOLLM_EQUATIONS_FROM_IMAGE,
 	TASK_GOLLM_GENERATE_SUMMARY,
+	TASK_GOLLM_INTERVENTIONS_FROM_DOCUMENT,
 	TASK_GOLLM_MODEL_CARD,
 	TASK_MIRA_AMR_TO_MMT,
 	TASK_MIRA_GENERATE_MODEL_LATEX,
diff --git a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/simulationservice/interventions/Intervention.java b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/simulationservice/interventions/Intervention.java
index e97d909341..e2986e561d 100644
--- a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/simulationservice/interventions/Intervention.java
+++ b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/models/simulationservice/interventions/Intervention.java
@@ -10,8 +10,8 @@
 public class Intervention {
 
 	private String name;
-	private List<StaticIntervention> staticInterventions;
-	private List<DynamicIntervention> dynamicInterventions;
+	private List<StaticIntervention> staticInterventions = new ArrayList<>();
+	private List<DynamicIntervention> dynamicInterventions = new ArrayList<>();
 
 	@Override
 	public Intervention clone() {
diff --git a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/InterventionsFromDocumentResponseHandler.java b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/InterventionsFromDocumentResponseHandler.java
index 934e501fc8..d0c38c2043 100644
--- a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/InterventionsFromDocumentResponseHandler.java
+++ b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/InterventionsFromDocumentResponseHandler.java
@@ -8,9 +8,6 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Component;
-import software.uncharted.terarium.hmiserver.models.dataservice.provenance.Provenance;
-import software.uncharted.terarium.hmiserver.models.dataservice.provenance.ProvenanceRelationType;
-import software.uncharted.terarium.hmiserver.models.dataservice.provenance.ProvenanceType;
 import software.uncharted.terarium.hmiserver.models.simulationservice.interventions.InterventionPolicy;
 import software.uncharted.terarium.hmiserver.models.task.TaskResponse;
 import software.uncharted.terarium.hmiserver.service.data.DocumentAssetService;
@@ -79,16 +76,6 @@ public TaskResponse onSuccess(final TaskResponse resp) {
 					props.projectId,
 					ASSUME_WRITE_PERMISSION_ON_BEHALF_OF_USER
 				);
-
-				// add provenance
-				provenanceService.createProvenance(
-					new Provenance()
-						.setLeft(newPolicy.getId())
-						.setLeftType(ProvenanceType.INTERVENTION_POLICY)
-						.setRight(props.documentId)
-						.setRightType(ProvenanceType.DOCUMENT)
-						.setRelationType(ProvenanceRelationType.EXTRACTED_FROM)
-				);
 			}
 		} catch (final Exception e) {
 			log.error("Failed to extract intervention policy", e);
diff --git a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/TaskNotificationEventTypes.java b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/TaskNotificationEventTypes.java
index 09b94e90e4..cc26ec0892 100644
--- a/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/TaskNotificationEventTypes.java
+++ b/packages/server/src/main/java/software/uncharted/terarium/hmiserver/service/tasks/TaskNotificationEventTypes.java
@@ -8,18 +8,19 @@
 public class TaskNotificationEventTypes {
 
 	private static final Map<String, ClientEventType> clientEventTypes = Map.ofEntries(
-		Map.entry(ModelCardResponseHandler.NAME, ClientEventType.TASK_GOLLM_MODEL_CARD),
-		Map.entry(ConfigureModelFromDocumentResponseHandler.NAME, ClientEventType.TASK_GOLLM_CONFIGURE_MODEL_FROM_DOCUMENT),
-		Map.entry(ConfigureModelFromDatasetResponseHandler.NAME, ClientEventType.TASK_GOLLM_CONFIGURE_MODEL_FROM_DATASET),
+		Map.entry(AMRToMMTResponseHandler.NAME, ClientEventType.TASK_MIRA_AMR_TO_MMT),
 		Map.entry(CompareModelsResponseHandler.NAME, ClientEventType.TASK_GOLLM_COMPARE_MODEL),
-		Map.entry(GenerateSummaryHandler.NAME, ClientEventType.TASK_GOLLM_GENERATE_SUMMARY),
-		Map.entry(ValidateModelConfigHandler.NAME, ClientEventType.TASK_FUNMAN_VALIDATION),
+		Map.entry(ConfigureModelFromDatasetResponseHandler.NAME, ClientEventType.TASK_GOLLM_CONFIGURE_MODEL_FROM_DATASET),
+		Map.entry(ConfigureModelFromDocumentResponseHandler.NAME, ClientEventType.TASK_GOLLM_CONFIGURE_MODEL_FROM_DOCUMENT),
 		Map.entry(EnrichAmrResponseHandler.NAME, ClientEventType.TASK_GOLLM_ENRICH_AMR),
-		Map.entry(AMRToMMTResponseHandler.NAME, ClientEventType.TASK_MIRA_AMR_TO_MMT),
+		Map.entry(EquationsFromImageResponseHandler.NAME, ClientEventType.TASK_GOLLM_EQUATIONS_FROM_IMAGE),
+		Map.entry(InterventionsFromDocumentResponseHandler.NAME, ClientEventType.TASK_GOLLM_INTERVENTIONS_FROM_DOCUMENT),
 		Map.entry(ExtractEquationsResponseHandler.NAME, ClientEventType.TASK_EXTRACT_EQUATION_PDF),
 		Map.entry(ExtractTablesResponseHandler.NAME, ClientEventType.TASK_EXTRACT_TABLE_PDF),
 		Map.entry(ExtractTextResponseHandler.NAME, ClientEventType.TASK_EXTRACT_TEXT_PDF),
-		Map.entry(EquationsFromImageResponseHandler.NAME, ClientEventType.TASK_GOLLM_EQUATIONS_FROM_IMAGE)
+		Map.entry(GenerateSummaryHandler.NAME, ClientEventType.TASK_GOLLM_GENERATE_SUMMARY),
+		Map.entry(ModelCardResponseHandler.NAME, ClientEventType.TASK_GOLLM_MODEL_CARD),
+		Map.entry(ValidateModelConfigHandler.NAME, ClientEventType.TASK_FUNMAN_VALIDATION)
 	);
 
 	public static ClientEventType getTypeFor(final String taskName) {