spinner on enrich model (#4824)

Co-authored-by: Cole Blanchard <[email protected]>

packages/client/hmi-client/src/components/widgets/tera-asset-enrichment.vue

@@ -46,8 +46,8 @@ import {
 	mapAssetTypeToProvenanceType,
 	RelationshipType
 } from '@/services/provenance';
-import type { DocumentAsset, ProjectAsset, TerariumAsset } from '@/types/Types';
-import { AssetType, ProvenanceType } from '@/types/Types';
+import type { ClientEvent, DocumentAsset, ProjectAsset, TaskResponse, TerariumAsset } from '@/types/Types';
+import { AssetType, ClientEventType, ProvenanceType, TaskStatus } from '@/types/Types';
 import { isDocumentAsset } from '@/utils/asset';
 import Button from 'primevue/button';
 import RadioButton from 'primevue/radiobutton';
@@ -57,6 +57,7 @@ import { logger } from '@/utils/logger';
 import { modelCard } from '@/services/goLLM';
 import { useProjects } from '@/composables/project';
 import TeraModal from '@/components/widgets/tera-modal.vue';
+import { useClientEvent } from '@/composables/useClientEvent';
 
 const props = defineProps<{
 	assetType: AssetType;
@@ -65,6 +66,18 @@ const props = defineProps<{
 
 const emit = defineEmits(['finished-job']);
 
+// Listen for the task completion event for models
+useClientEvent(ClientEventType.TaskGollmModelCard, (event: ClientEvent<TaskResponse>) => {
+	const { modelId } = event.data?.additionalProperties || {};
+	const { status } = event.data || {};
+
+	if (props.assetType !== AssetType.Model || modelId !== props.assetId) {
+		return;
+	}
+
+	isLoading.value = ![TaskStatus.Success, TaskStatus.Failed, TaskStatus.Cancelled].includes(status);
+});
+
 enum DialogType {
 	ENRICH,
 	EXTRACT