From 0bcbf060c0cbe4e45c7d013fde3edba519f35e71 Mon Sep 17 00:00:00 2001
From: David Whittaker <dwhittaker@netflix.com>
Date: Thu, 17 Apr 2025 22:02:28 -0700
Subject: [PATCH 1/7] now when adding a new entity type, it determines all
 entities for that new type

---
 src/dispatch/entity_type/service.py           | 21 ++++++++++-
 .../entity_type/EntityTypeCreateDialogV2.vue  |  2 +-
 .../src/signal/NewRawSignalViewer.vue         | 36 ++++++++++++-------
 .../static/dispatch/src/util/jpath.ts         |  2 +-
 4 files changed, 46 insertions(+), 15 deletions(-)

diff --git a/src/dispatch/entity_type/service.py b/src/dispatch/entity_type/service.py
index c4dbe8775fec..e092a86abe5c 100644
--- a/src/dispatch/entity_type/service.py
+++ b/src/dispatch/entity_type/service.py
@@ -57,7 +57,7 @@ def get_all(*, db_session: Session, scope: str = None) -> Query:
 
 
 def create(*, db_session: Session, entity_type_in: EntityTypeCreate) -> EntityType:
-    """Creates a new entity type."""
+    """Creates a new entity type and extracts entities from existing signal instances."""
     project = project_service.get_by_name_or_raise(
         db_session=db_session, project_in=entity_type_in.project
     )
@@ -75,6 +75,25 @@ def create(*, db_session: Session, entity_type_in: EntityTypeCreate) -> EntityTy
 
     db_session.add(entity_type)
     db_session.commit()
+
+    # Extract entities for all relevant signal instances
+    from dispatch.signal.models import SignalInstance
+    from dispatch.entity.service import find_entities
+
+    for signal in signals:
+        # Find all signal instances for this signal
+        signal_instances = (
+            db_session.query(SignalInstance).filter(SignalInstance.signal_id == signal.id).all()
+        )
+        for signal_instance in signal_instances:
+            # Extract and create entities for this instance using only the new entity_type
+            new_entities = find_entities(db_session, signal_instance, [entity_type])
+            # Associate new entities with the signal_instance
+            for entity in new_entities:
+                if entity not in signal_instance.entities:
+                    signal_instance.entities.append(entity)
+    db_session.commit()
+
     return entity_type
 
 
diff --git a/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue b/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
index 78eda3e07136..1c61e1658d53 100644
--- a/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
+++ b/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
@@ -60,7 +60,7 @@
             :options="editorOptions"
             :editorMounted="editorMounted"
             language="json"
-            style="width: 100%; height: 100%"
+            style="width: 100%; height: 240px"
           />
         </div>
 
diff --git a/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue b/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue
index f7bffe68670e..b456bbf53575 100644
--- a/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue
+++ b/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue
@@ -113,6 +113,9 @@ const editorMounted = (editor, monaco) => {
 
     glyphLines.clear()
 
+    // Combine existing and new entities for checking
+    const allEntities = [...props.value.entities, ...newEntities.value]
+
     function traverse(node) {
       // Traverse `Object` and `Array` nodes to reach their child `Property` nodes.
       if (node.type === "Object" || node.type === "Array") {
@@ -122,19 +125,28 @@ const editorMounted = (editor, monaco) => {
         const { key, value } = node
         // Check for `Literal` type and string value.
         if (value.type === "Literal" && typeof value.value === "string") {
-          const start = model.getPositionAt(value.loc.start.offset)
-          const end = model.getPositionAt(value.loc.end.offset)
-
-          decorations.push({
-            range: new monaco.Range(start.lineNumber, 1, end.lineNumber, 1),
-            options: {
-              isWholeLine: true,
-              glyphMarginClassName: "myGlyphMarginClass",
-              glyphMarginHoverMessage: { value: `Extract new entity from "**${key.value}**"` },
-            },
+          // Skip adding glyph if this value already exists as an entity
+          const stringValue = value.value
+          const entityExists = allEntities.some((entity) => {
+            // Handle both entity objects and entity_type objects
+            return entity.value === stringValue
           })
-          // Record line numbers where glyphs were added.
-          glyphLines.add(start.lineNumber)
+
+          if (!entityExists) {
+            const start = model.getPositionAt(value.loc.start.offset)
+            const end = model.getPositionAt(value.loc.end.offset)
+
+            decorations.push({
+              range: new monaco.Range(start.lineNumber, 1, end.lineNumber, 1),
+              options: {
+                isWholeLine: true,
+                glyphMarginClassName: "myGlyphMarginClass",
+                glyphMarginHoverMessage: { value: `Extract new entity from "**${key.value}**"` },
+              },
+            })
+            // Record line numbers where glyphs were added.
+            glyphLines.add(start.lineNumber)
+          }
         }
 
         // If the value of the property is an object or array, traverse it as well
diff --git a/src/dispatch/static/dispatch/src/util/jpath.ts b/src/dispatch/static/dispatch/src/util/jpath.ts
index e9c8a38bce23..c0df60d4a544 100644
--- a/src/dispatch/static/dispatch/src/util/jpath.ts
+++ b/src/dispatch/static/dispatch/src/util/jpath.ts
@@ -24,7 +24,7 @@ import json_to_ast from "json-to-ast"
 export function findPath<T>(obj: T, key: keyof any, value: any, path: string = "$"): string | null {
   if (Array.isArray(obj)) {
     for (let i = 0; i < obj.length; i++) {
-      const arrayPath = `${path}[${i}]`
+      const arrayPath = `${path}[*]`
       const result = findPath(obj[i], key, value, arrayPath)
       if (result) return result
     }

From 866aee30b7abb3034dbe77fb33936103d36d3c82 Mon Sep 17 00:00:00 2001
From: David Whittaker <dwhittaker@netflix.com>
Date: Fri, 18 Apr 2025 10:03:34 -0700
Subject: [PATCH 2/7] removing unused import

---
 src/dispatch/static/dispatch/src/util/jpath.ts | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/dispatch/static/dispatch/src/util/jpath.ts b/src/dispatch/static/dispatch/src/util/jpath.ts
index c0df60d4a544..5a229bec7c5c 100644
--- a/src/dispatch/static/dispatch/src/util/jpath.ts
+++ b/src/dispatch/static/dispatch/src/util/jpath.ts
@@ -1,5 +1,4 @@
 import jsonpath from "jsonpath"
-import json_to_ast from "json-to-ast"
 
 /**
  * Finds the path to a key in an object hierarchy.

From b2eeda45b73f0d654d53bf6a20c84925327ef1d4 Mon Sep 17 00:00:00 2001
From: David Whittaker <dwhittaker@netflix.com>
Date: Fri, 18 Apr 2025 11:49:51 -0700
Subject: [PATCH 3/7] prevent multiple hover messages from duplicating

---
 .../src/signal/NewRawSignalViewer.vue         | 78 +++++++++++++++++--
 1 file changed, 70 insertions(+), 8 deletions(-)

diff --git a/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue b/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue
index b456bbf53575..c329e6364e3e 100644
--- a/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue
+++ b/src/dispatch/static/dispatch/src/signal/NewRawSignalViewer.vue
@@ -1,5 +1,5 @@
 <script setup>
-import { computed, ref, watch } from "vue"
+import { computed, ref, watch, onBeforeUnmount } from "vue"
 
 // Necessary import for JSON language server
 // eslint-disable-next-line no-unused-vars
@@ -33,25 +33,82 @@ const newEntityTypeJpath = ref("")
 
 const raw_str = computed(() => JSON.stringify(props.value.raw, null, "\t") || "[]")
 
+// Store a reference to the hover provider for cleanup
+let hoverProviderDisposable = null
+
+// Clean up function to dispose of hover providers
+const cleanupHoverProviders = (monacoInstance) => {
+  if (!monacoInstance || !monacoInstance.languages) return
+
+  // Get all registered hover providers for JSON language
+  const providers = monacoInstance.languages.getHoverProviders
+    ? monacoInstance.languages.getHoverProviders("json")
+    : []
+
+  // Dispose all existing hover providers
+  for (const provider of providers) {
+    provider.dispose()
+  }
+
+  // Also dispose our tracked provider if it exists
+  if (hoverProviderDisposable) {
+    hoverProviderDisposable.dispose()
+    hoverProviderDisposable = null
+  }
+}
+
+// Ensure cleanup when component is unmounted
+onBeforeUnmount(() => {
+  // We'll use the global monaco instance if available
+  if (typeof monaco !== "undefined") {
+    cleanupHoverProviders(monaco)
+  }
+})
+
 const editorBeforeMount = (monaco) => {
-  monaco.languages.registerHoverProvider("json", {
+  // Clean up any existing hover providers
+  cleanupHoverProviders(monaco)
+
+  // Register a single new hover provider
+  hoverProviderDisposable = monaco.languages.registerHoverProvider("json", {
     provideHover: (model, position) => {
       const word = model.getWordAtPosition(position)
 
       if (word) {
         const hoveredWord = word.word
 
-        const entity = props.value.entities.find((entity) => entity.value === hoveredWord)
-        if (entity) {
+        // Check in existing entities first
+        const existingEntity = props.value.entities.find((entity) => entity.value === hoveredWord)
+        if (existingEntity) {
           const range = new monaco.Range(
             position.lineNumber,
             word.startColumn,
             position.lineNumber,
             word.endColumn
           )
-          const title = `**Entity Type**: ${entity.entity_type.name}\n`
-          const pattern = `**Pattern**: ${entity.entity_type.jpath}\n`
-          const value = `**Value**: ${entity.value}`
+          const title = `**Entity Type**: ${existingEntity.entity_type.name}\n`
+          const pattern = `**Pattern**: ${existingEntity.entity_type.jpath}\n`
+          const value = `**Value**: ${existingEntity.value}`
+          return {
+            contents: [{ value: title }, { value: pattern }, { value: value }],
+            range: range,
+          }
+        }
+
+        // Then check in new entities
+        const newEntity = newEntities.value.find((entity) => entity.value === hoveredWord)
+        if (newEntity) {
+          const range = new monaco.Range(
+            position.lineNumber,
+            word.startColumn,
+            position.lineNumber,
+            word.endColumn
+          )
+          // Handle different entity structures
+          const entityType = newEntity.entity_type || newEntity
+          const title = `**Entity Type**: ${entityType.name}\n`
+          const pattern = `**Pattern**: ${entityType.jpath}\n`
+          const value = `**Value**: ${newEntity.value}`
           return {
             contents: [{ value: title }, { value: pattern }, { value: value }],
             range: range,
@@ -141,7 +198,7 @@ const editorMounted = (editor, monaco) => {
               options: {
                 isWholeLine: true,
                 glyphMarginClassName: "myGlyphMarginClass",
-                glyphMarginHoverMessage: { value: `Extract new entity from "**${key.value}**"` },
+                glyphMarginHoverMessage: [{ value: `Extract new entity from "**${key.value}**"` }],
               },
             })
             // Record line numbers where glyphs were added.
@@ -237,6 +294,11 @@ const editorOptions = {
   wordWrap: true,
   glyphMargin: true,
   contextmenu: false,
+  // Configure hover settings for immediate response
+  hover: {
+    delay: 0, // Show hover immediately with no delay
+    sticky: true, // Make hovers sticky so they don't disappear quickly
+  },
   scrollbar: {
     vertical: "hidden",
   },

From fc4459c117b8e194419aa2834d923ba9e1fad624 Mon Sep 17 00:00:00 2001
From: David Whittaker <dwhittaker@netflix.com>
Date: Fri, 18 Apr 2025 12:23:03 -0700
Subject: [PATCH 4/7] change api to recalculate on a case

---
 src/dispatch/entity_type/views.py             | 44 ++++++++++++-------
 .../entity_type/EntityTypeCreateDialogV2.vue  |  3 +-
 .../static/dispatch/src/entity_type/api.js    |  4 +-
 3 files changed, 31 insertions(+), 20 deletions(-)

diff --git a/src/dispatch/entity_type/views.py b/src/dispatch/entity_type/views.py
index acd090169231..adaba5cae477 100644
--- a/src/dispatch/entity_type/views.py
+++ b/src/dispatch/entity_type/views.py
@@ -1,9 +1,10 @@
-from typing import Union
+from typing import List, Union
 
 from fastapi import APIRouter, HTTPException, status
 from pydantic.error_wrappers import ErrorWrapper, ValidationError
 from sqlalchemy.exc import IntegrityError
 
+from dispatch.case.service import get as get_case
 from dispatch.database.core import DbSession
 from dispatch.exceptions import ExistsError
 from dispatch.database.service import CommonParameters, search_filter_sort_paginate
@@ -54,11 +55,9 @@ def create_entity_type(db_session: DbSession, entity_type_in: EntityTypeCreate):
     return entity_type
 
 
-@router.put("/recalculate/{entity_type_id}/{signal_instance_id}", response_model=SignalInstanceRead)
-def recalculate(
-    db_session: DbSession, entity_type_id: PrimaryKey, signal_instance_id: Union[str, PrimaryKey]
-):
-    """Recalculates the associated entities for a signal instance."""
+@router.put("/recalculate/{entity_type_id}/{case_id}", response_model=List[SignalInstanceRead])
+def recalculate(db_session: DbSession, entity_type_id: PrimaryKey, case_id: PrimaryKey):
+    """Recalculates the associated entities for all signal instances in a case."""
     entity_type = get(
         db_session=db_session,
         entity_type_id=entity_type_id,
@@ -69,21 +68,32 @@ def recalculate(
             detail=[{"msg": "An entity type with this id does not exist."}],
         )
 
-    signal_instance = get_signal_instance(
-        db_session=db_session,
-        signal_instance_id=signal_instance_id,
-    )
-    if not signal_instance:
+    case = get_case(db_session=db_session, case_id=case_id)
+    if not case:
         raise HTTPException(
             status_code=status.HTTP_404_NOT_FOUND,
-            detail=[{"msg": "A signal instance with this id does not exist."}],
+            detail=[{"msg": "A case with this id does not exist."}],
         )
 
-    return recalculate_entity_flow(
-        db_session=db_session,
-        entity_type=entity_type,
-        signal_instance=signal_instance,
-    )
+    # Get all signal instances associated with the case
+    signal_instances = case.signal_instances
+    if not signal_instances:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail=[{"msg": "No signal instances found for this case."}],
+        )
+
+    # Recalculate entities for each signal instance
+    updated_signal_instances = []
+    for signal_instance in signal_instances:
+        updated_signal_instance = recalculate_entity_flow(
+            db_session=db_session,
+            entity_type=entity_type,
+            signal_instance=signal_instance,
+        )
+        updated_signal_instances.append(updated_signal_instance)
+
+    return updated_signal_instances
 
 
 @router.put("/{entity_type_id}", response_model=EntityTypeRead)
diff --git a/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue b/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
index 1c61e1658d53..23bf108b3b87 100644
--- a/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
+++ b/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
@@ -331,6 +331,8 @@ const saveEntityType = async () => {
   }
   try {
     const newEntityType = await EntityTypeApi.create(entityTypeData)
+    // Use the case_id instead of signal_instance_id for recalculation
+    await EntityTypeApi.recalculate(newEntityType.data.id, selectedCase.value.id)
     emit("new-entity-type", newEntityType.data)
     store.commit(
       "notification_backend/addBeNotification",
@@ -340,7 +342,6 @@ const saveEntityType = async () => {
       },
       { root: true }
     )
-    await EntityTypeApi.recalculate(newEntityType.data.id, props.signalObj.raw.id)
   } catch (error) {
     store.commit(
       "notification_backend/addBeNotification",
diff --git a/src/dispatch/static/dispatch/src/entity_type/api.js b/src/dispatch/static/dispatch/src/entity_type/api.js
index 260d94ddf597..4023a7c18580 100644
--- a/src/dispatch/static/dispatch/src/entity_type/api.js
+++ b/src/dispatch/static/dispatch/src/entity_type/api.js
@@ -27,7 +27,7 @@ export default {
     return API.delete(`${resource}/${entityTypeId}`)
   },
 
-  recalculate(entityTypeId, signalInstanceId) {
-    return API.put(`${resource}/recalculate/${entityTypeId}/${signalInstanceId}`)
+  recalculate(entityTypeId, caseId) {
+    return API.put(`${resource}/recalculate/${entityTypeId}/${caseId}`)
   },
 }

From 3d9a168b3616db2857c506f6d258f571b4c75057 Mon Sep 17 00:00:00 2001
From: David Whittaker <dwhittaker@netflix.com>
Date: Fri, 18 Apr 2025 13:37:29 -0700
Subject: [PATCH 5/7] fixing lint errors

---
 src/dispatch/entity_type/views.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/dispatch/entity_type/views.py b/src/dispatch/entity_type/views.py
index adaba5cae477..ee8a80c4d05d 100644
--- a/src/dispatch/entity_type/views.py
+++ b/src/dispatch/entity_type/views.py
@@ -1,4 +1,4 @@
-from typing import List, Union
+from typing import List
 
 from fastapi import APIRouter, HTTPException, status
 from pydantic.error_wrappers import ErrorWrapper, ValidationError
@@ -9,7 +9,6 @@
 from dispatch.exceptions import ExistsError
 from dispatch.database.service import CommonParameters, search_filter_sort_paginate
 from dispatch.models import PrimaryKey
-from dispatch.signal.service import get_signal_instance
 from dispatch.signal.models import SignalInstanceRead
 
 from .models import (

From f016964005445044c938360e5a8f9303b5c13f28 Mon Sep 17 00:00:00 2001
From: David Whittaker <dwhittaker@netflix.com>
Date: Tue, 22 Apr 2025 11:26:08 -0700
Subject: [PATCH 6/7] pass case_id to reduce number of entities created

---
 src/dispatch/entity_type/service.py           | 19 ++++++++++++-------
 src/dispatch/entity_type/views.py             | 15 +++++++++++++++
 .../entity_type/EntityTypeCreateDialogV2.vue  |  5 ++++-
 .../static/dispatch/src/entity_type/api.js    |  4 ++++
 4 files changed, 35 insertions(+), 8 deletions(-)

diff --git a/src/dispatch/entity_type/service.py b/src/dispatch/entity_type/service.py
index e092a86abe5c..3f4a6cd897d2 100644
--- a/src/dispatch/entity_type/service.py
+++ b/src/dispatch/entity_type/service.py
@@ -56,7 +56,9 @@ def get_all(*, db_session: Session, scope: str = None) -> Query:
     return db_session.query(EntityType)
 
 
-def create(*, db_session: Session, entity_type_in: EntityTypeCreate) -> EntityType:
+def create(
+    *, db_session: Session, entity_type_in: EntityTypeCreate, case_id: int | None = None
+) -> EntityType:
     """Creates a new entity type and extracts entities from existing signal instances."""
     project = project_service.get_by_name_or_raise(
         db_session=db_session, project_in=entity_type_in.project
@@ -80,24 +82,27 @@ def create(*, db_session: Session, entity_type_in: EntityTypeCreate) -> EntityTy
     from dispatch.signal.models import SignalInstance
     from dispatch.entity.service import find_entities
 
-    for signal in signals:
-        # Find all signal instances for this signal
+    if case_id:
+        # Get all signal instances for the case
         signal_instances = (
-            db_session.query(SignalInstance).filter(SignalInstance.signal_id == signal.id).all()
+            db_session.query(SignalInstance).filter(SignalInstance.case_id == case_id).all()
         )
+        # Extract and create entities for these instances using only the new entity_type
         for signal_instance in signal_instances:
-            # Extract and create entities for this instance using only the new entity_type
             new_entities = find_entities(db_session, signal_instance, [entity_type])
             # Associate new entities with the signal_instance
             for entity in new_entities:
                 if entity not in signal_instance.entities:
                     signal_instance.entities.append(entity)
+
     db_session.commit()
 
     return entity_type
 
 
-def get_or_create(*, db_session: Session, entity_type_in: EntityTypeCreate) -> EntityType:
+def get_or_create(
+    *, db_session: Session, entity_type_in: EntityTypeCreate, case_id: int | None = None
+) -> EntityType:
     """Gets or creates a new entity type."""
     q = (
         db_session.query(EntityType)
@@ -109,7 +114,7 @@ def get_or_create(*, db_session: Session, entity_type_in: EntityTypeCreate) -> E
     if instance:
         return instance
 
-    return create(db_session=db_session, entity_type_in=entity_type_in)
+    return create(db_session=db_session, entity_type_in=entity_type_in, case_id=case_id)
 
 
 def update(
diff --git a/src/dispatch/entity_type/views.py b/src/dispatch/entity_type/views.py
index ee8a80c4d05d..ddf274949896 100644
--- a/src/dispatch/entity_type/views.py
+++ b/src/dispatch/entity_type/views.py
@@ -54,6 +54,21 @@ def create_entity_type(db_session: DbSession, entity_type_in: EntityTypeCreate):
     return entity_type
 
 
+@router.post("/{case_id}", response_model=EntityTypeRead)
+def create_entity_type_with_case(
+    db_session: DbSession, case_id: PrimaryKey, entity_type_in: EntityTypeCreate
+):
+    """Create a new entity."""
+    try:
+        entity_type = create(db_session=db_session, entity_type_in=entity_type_in, case_id=case_id)
+    except IntegrityError:
+        raise ValidationError(
+            [ErrorWrapper(ExistsError(msg="An entity with this name already exists."), loc="name")],
+            model=EntityTypeCreate,
+        ) from None
+    return entity_type
+
+
 @router.put("/recalculate/{entity_type_id}/{case_id}", response_model=List[SignalInstanceRead])
 def recalculate(db_session: DbSession, entity_type_id: PrimaryKey, case_id: PrimaryKey):
     """Recalculates the associated entities for all signal instances in a case."""
diff --git a/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue b/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
index 23bf108b3b87..917606e8b092 100644
--- a/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
+++ b/src/dispatch/static/dispatch/src/entity_type/EntityTypeCreateDialogV2.vue
@@ -330,7 +330,10 @@ const saveEntityType = async () => {
     signals: [signalGetResponse.data],
   }
   try {
-    const newEntityType = await EntityTypeApi.create(entityTypeData)
+    const newEntityType = await EntityTypeApi.create_with_case(
+      entityTypeData,
+      selectedCase.value.id
+    )
     // Use the case_id instead of signal_instance_id for recalculation
     await EntityTypeApi.recalculate(newEntityType.data.id, selectedCase.value.id)
     emit("new-entity-type", newEntityType.data)
diff --git a/src/dispatch/static/dispatch/src/entity_type/api.js b/src/dispatch/static/dispatch/src/entity_type/api.js
index 4023a7c18580..c58b93bb8ea4 100644
--- a/src/dispatch/static/dispatch/src/entity_type/api.js
+++ b/src/dispatch/static/dispatch/src/entity_type/api.js
@@ -15,6 +15,10 @@ export default {
     return API.post(`${resource}`, payload)
   },
 
+  create_with_case(payload, caseId) {
+    return API.post(`${resource}/${caseId}`, payload)
+  },
+
   update(entityTypeId, payload) {
     return API.put(`${resource}/${entityTypeId}`, payload)
   },

From 41dfd0bc9ce2f875914b0499b8a479d51b1c881a Mon Sep 17 00:00:00 2001
From: David Whittaker <dwhittaker@netflix.com>
Date: Wed, 23 Apr 2025 08:56:41 -0700
Subject: [PATCH 7/7] adding limit in case too many signal instances

---
 src/dispatch/entity_type/service.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/dispatch/entity_type/service.py b/src/dispatch/entity_type/service.py
index 3f4a6cd897d2..60f7c3a4bb64 100644
--- a/src/dispatch/entity_type/service.py
+++ b/src/dispatch/entity_type/service.py
@@ -85,7 +85,10 @@ def create(
     if case_id:
         # Get all signal instances for the case
         signal_instances = (
-            db_session.query(SignalInstance).filter(SignalInstance.case_id == case_id).all()
+            db_session.query(SignalInstance)
+            .filter(SignalInstance.case_id == case_id)
+            .limit(100)
+            .all()
         )
         # Extract and create entities for these instances using only the new entity_type
         for signal_instance in signal_instances: