release-v-1-15-0 (#17)

* missing web sockets messages

* Adds org id to hook

---------

Co-authored-by: JWittmeyer <jens.wittmeyer@kern.ai>

Author: lumburovskalina

hooks/web-socket/constants.ts

@@ -104,7 +104,7 @@ export const WHITELIST_LOOKUP_REFINERY = {
         [CurrentPageSubKey.NONE]: ['labeling_task_updated', 'labeling_task_created', 'label_created', 'label_deleted', 'labeling_task_deleted', 'information_source_deleted', 'information_source_updated', 'model_callback_update_statistics', 'embedding_deleted', 'embedding', 'payload_finished', 'payload_failed', 'payload_created', 'payload_update_statistics']
     },
     [CurrentPage.HEURISTICS]: {
-        [CurrentPageSubKey.NONE]: ['labeling_task_updated', 'labeling_task_created', 'labeling_task_deleted', 'information_source_created', 'information_source_updated', 'information_source_deleted', 'payload_finished', 'payload_failed', 'payload_created', 'payload_update_statistics', 'embedding_deleted'],
+        [CurrentPageSubKey.NONE]: ['labeling_task_updated', 'labeling_task_created', 'labeling_task_deleted', 'information_source_created', 'information_source_updated', 'information_source_deleted', 'payload_finished', 'payload_failed', 'payload_created', 'payload_update_statistics', 'embedding_deleted', 'weak_supervision_started', 'weak_supervision_finished'],
 
     },
     [CurrentPage.EDIT_RECORDS]: {

hooks/web-socket/useWebsocket.tsx

@@ -3,13 +3,13 @@ import { WebSocketsService } from './WebSocketsService';
 import { NotificationSubscription, getConstWhitelist } from './web-sockets-helper';
 import { Application, CurrentPage, CurrentPageSubKey } from './constants';
 
-export function useWebsocket(application: Application, currentPage: CurrentPage, handleFunction: (msgParts: string[]) => void, projectId?: string, subKey?: CurrentPageSubKey) {
+export function useWebsocket(userOrgId: string, application: Application, currentPage: CurrentPage, handleFunction: (msgParts: string[]) => void, projectId?: string, subKey?: CurrentPageSubKey) {
 
     const _projectId = useMemo(() => projectId || "GLOBAL", [projectId]);
     const _subKey = useMemo(() => subKey || CurrentPageSubKey.NONE, [subKey]);
     const __whitelist = useMemo(() => getConstWhitelist(application), [application]);
-
     useEffect(() => {
+        if (!userOrgId) return;
         if (!__whitelist[currentPage]) {
             console.error(`The combination of ${currentPage} and ${application} does not exist in the whitelist`);
             return;
@@ -22,12 +22,13 @@ export function useWebsocket(application: Application, currentPage: CurrentPage,
 
         WebSocketsService.subscribeToNotification(currentPage, nos, _subKey);
 
-        return () => WebSocketsService.unsubscribeFromNotification(currentPage, projectId, _subKey);
-    }, [_projectId, _subKey, application]);
+        return () => { if (userOrgId) WebSocketsService.unsubscribeFromNotification(currentPage, projectId, _subKey) };
+    }, [userOrgId, _projectId, _subKey, application]);
 
     useEffect(() => {
+        if (!userOrgId) return;
         WebSocketsService.updateFunctionPointer(projectId, currentPage, handleFunction, subKey)
-    }, [handleFunction]);
+    }, [userOrgId, handleFunction]);
 
 
 }