canvas 0.5.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

{canvas-0.5.0.dist-info → canvas-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: canvas
-Version: 0.5.0
+Version: 0.6.0
 Summary: SDK to customize event-driven actions in your Canvas instance
 License: MIT
 Author: Canvas Team
@@ -25,6 +25,7 @@ Requires-Dist: psycopg[binary] (>=3.2.2,<4.0.0)
 Requires-Dist: pydantic (>=2.6.1,<3.0.0)
 Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
+Requires-Dist: rapidfuzz (>=3.10.1,<4.0.0)
 Requires-Dist: redis (>=5.0.4,<6.0.0)
 Requires-Dist: requests
 Requires-Dist: restrictedpython (>=7.1,<8.0)

{canvas-0.5.0.dist-info → canvas-0.6.0.dist-info}/RECORD

@@ -104,9 +104,9 @@ canvas_generated/services/plugin_runner_pb2.py,sha256=RfAo_imYoSuoexq-1IHhMhXZgQ
 canvas_generated/services/plugin_runner_pb2.pyi,sha256=1w-Pa4k7HtlmQAr7B6sgV64zdZplBKQKHN-S8bjwO3w,265
 canvas_generated/services/plugin_runner_pb2_grpc.py,sha256=EzJJVkP_AZ3dwBA7OxUito0NSalRmjjg8q9TZ_P18ww,4549
 canvas_sdk/__init__.py,sha256=d7V1Qsp4hSqp8opmvqp-0J33uibArUjwENMfzSDAdZg,102
-canvas_sdk/base.py,sha256=tcmP3zxp24kAoplT-jqrIlSpiYsJ6qRMNY-HiuJYFGc,1834
+canvas_sdk/base.py,sha256=Qp_bcUFaoGBRtDXuYyTZyQC1Q6-El3w-J5Gkg8U3ps8,2040
 canvas_sdk/commands/__init__.py,sha256=ApJy2i8IRCL2YFUumk1q-YihLL2YcmLYKKiWFK04evY,2365
-canvas_sdk/commands/base.py,sha256=lc6OizuXTB5mDzGVpYGx9nz_OrJUKBaffmmCPxs2B0s,5155
+canvas_sdk/commands/base.py,sha256=WX6nnP1OlH4LT28N_xmE55gS_FbFcZuLk_dIvSK49VY,5226
 canvas_sdk/commands/commands/allergy.py,sha256=_S0tY43sErRlTeUGCC6Thb8gMSD_JiOmRSFdLaDlr48,1296
 canvas_sdk/commands/commands/assess.py,sha256=IjsvkhzKinRaykMRzbnk6G4JAkzDhyqGG6ZvjDhNAuE,1723
 canvas_sdk/commands/commands/close_goal.py,sha256=RTmi8fGPEEag8YSrlsPn7sy9Y-WymVuU8IWpXRKgNjg,792
@@ -147,14 +147,14 @@ canvas_sdk/data/staff.py,sha256=Po-NH2UdGTTP4LnUzfItfzriJKdul-2Cd31oq9uFHQE,210
 canvas_sdk/data/task.py,sha256=UpjWjB1QqClR83zeqFcM7b7gg2w0NjfzNKsWrgShN_s,1995
 canvas_sdk/effects/__init__.py,sha256=oChQi5y_8mK2zDiNMwWYp6CaQX1_zbmwXILPF7iSt-4,69
 canvas_sdk/effects/banner_alert/__init__.py,sha256=8z5l9rwonKEcRkvD5q4DWvHPYhKJzwyGuUYoUNKYwX4,158
-canvas_sdk/effects/banner_alert/add_banner_alert.py,sha256=RhMEIc62T5rLO_FltGvOq0XVU_wXIdCcBpxfaQeqCJY,1497
-canvas_sdk/effects/banner_alert/remove_banner_alert.py,sha256=9oLetlX_22B3iRwVZqB4IpA9e9YqZ_1cmWV4vJRNDmE,559
-canvas_sdk/effects/banner_alert/tests.py,sha256=idaXkAAUm_pnquXn1ZbWAPmoTf16MsbXrnM6cpaJZDQ,7712
-canvas_sdk/effects/base.py,sha256=cyTEQ4p3IxZ3mPu6Ancr4Y7RcrVirWBbNUs9V6g7Odc,674
+canvas_sdk/effects/banner_alert/add_banner_alert.py,sha256=W4Fv9IHGCWvVtFiH3s68JBRpUb8b3xsncGt2WfgMFc0,1681
+canvas_sdk/effects/banner_alert/remove_banner_alert.py,sha256=PK_LzXBQ-rOH3fX182ISyutu4W69xyGgMX-1G4g4XhY,613
+canvas_sdk/effects/banner_alert/tests.py,sha256=6vid5_pDeBSqgqDoeUeaVsflE0bnuSFgLpfSFV7pXLY,8531
+canvas_sdk/effects/base.py,sha256=znIaDNkCYbzJxf4Q80hLRYUG5H73c8IMq1UCzeBcXjw,714
 canvas_sdk/effects/patient_chart_summary_configuration.py,sha256=_Gx7UIp4NLaSYYLLcBSOjtOD4Skqot_kt_x8zI3Ccwo,1184
 canvas_sdk/effects/protocol_card/__init__.py,sha256=ESzUYajBBmcTAad2PIwHuV3SgGj4B9JHtf9uop_V4YA,88
-canvas_sdk/effects/protocol_card/protocol_card.py,sha256=gn-ArEiYWoSG4IDE-at2oaPCbSIrNxz_wpucYaMNC5M,2427
-canvas_sdk/effects/protocol_card/tests.py,sha256=fdFPkORW7e_0FpSlxBQWUPQwF6-JQeBG1g3R7MKj9Jc,6419
+canvas_sdk/effects/protocol_card/protocol_card.py,sha256=R7sHF74nup1nSOikf3o7wWukMvimrlS9AlRL0iDSlAM,2540
+canvas_sdk/effects/protocol_card/tests.py,sha256=6EVFyAPA4YYIQO6UF-Xwe266LLzJRYlmmDiJC7lFoCc,6463
 canvas_sdk/events/__init__.py,sha256=JGZ-3uf0Luzr4kg5UqduupD4w6B2WVEgQ9WlcoMaPwE,81
 canvas_sdk/handlers/__init__.py,sha256=VVz2_Xyb9720R2MZmSbGPG0T8u2a5u6zSDkWlwiJZv4,49
 canvas_sdk/handlers/base.py,sha256=7kw95-nYOsphP2_dVoJxBvi2RsuUpwBo5RcKvo924Yw,599
@@ -171,18 +171,18 @@ canvas_sdk/utils/tests.py,sha256=t3MScbIfzXkQttMIvj0dRzJlFVS8LFU8WgWRrChM-H0,193
 canvas_sdk/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 canvas_sdk/v1/data/__init__.py,sha256=LWAcICOx4fWea_lp_yNt0UIuhgjZMoNOaWx-7ZuM-dg,132
 canvas_sdk/v1/data/allergy_intolerance.py,sha256=KJg7UR8BJMuASWL5zXZ57JTmC0TQT7FAnODPvdq2mFg,2093
-canvas_sdk/v1/data/base.py,sha256=6JgLZsl9NRRS6s9-jjJM8Opxk_SI_sGkyokKIGPCoWI,3504
+canvas_sdk/v1/data/base.py,sha256=FyKcqayXEH4KV1DGMFU63jlyoNT57-eD3TkXk_yICM0,3530
 canvas_sdk/v1/data/command.py,sha256=z9LHw95Vq8efZE5y9r-Ei-Yzt8qj-o-BE3x9BBb4yPQ,945
 canvas_sdk/v1/data/common.py,sha256=9PiJ6oCPsvCvKBHtShIg0cW5Sr8CrnM9te-m2DYJB-g,1590
-canvas_sdk/v1/data/condition.py,sha256=hJ3APJvRh-dKYwujujdo5kugnc6YXBES7YpUIKUTjhI,1525
+canvas_sdk/v1/data/condition.py,sha256=izWzeNBUlFJOzjlp81cN2c6u78bzkFWQgSQgI4WFiC4,1535
 canvas_sdk/v1/data/detected_issue.py,sha256=c1rKZuPV_tTp_1yDJJa88gIsixpPV_F-a3eooje8BW8,1710
 canvas_sdk/v1/data/device.py,sha256=uxeQRT8ljzBmDwsARFDB9MZKAVAKOA-pu4nAa6fcHeU,1655
 canvas_sdk/v1/data/imaging.py,sha256=z6CtVcBYv8EL7fxzT0pWPUuh1mt5XFPlRDNNbOjvLng,4303
 canvas_sdk/v1/data/lab.py,sha256=nGV2FGDtnj7NcO8WS5LRXv5JUiLjuWMtm0i-PUNsKLk,10313
 canvas_sdk/v1/data/medication.py,sha256=XbV_iYgt40n8Q1IcVPtsinW2v-VJEms4T13cFNquBRE,1798
 canvas_sdk/v1/data/observation.py,sha256=BftaK7tOZaTlfUKoTH_A-X23J4DUFRVWpvZhkBJ1DJg,3710
-canvas_sdk/v1/data/patient.py,sha256=YGlp13cSNQHDtUJYTU37zisF94ZEKNYAlc2TJAmhdB8,1930
-canvas_sdk/v1/data/protocol_override.py,sha256=mUEcA7_9VkGsDmxrOxJ1eT8XVG0j8PHkd8JmQN4P3zM,1837
+canvas_sdk/v1/data/patient.py,sha256=QvGJxiqX7be-p1RNw-fSXlomb7gPwCyOZOKZMlJfT2s,1945
+canvas_sdk/v1/data/protocol_override.py,sha256=Dra2qf5-VX7qsK5mWy6U5XU7Y9v4NGkobRU4BAeAxkU,1835
 canvas_sdk/v1/data/questionnaire.py,sha256=wZNj8wg5aEThrHvA6-Oq5USnViyoeYGlcD5cjNmiULM,6899
 canvas_sdk/v1/data/user.py,sha256=DmA8fNNGwiNOmuR-RYxQw1NpxN5yYhgwtRtRBRJ_22g,309
 canvas_sdk/value_set/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -212,7 +212,7 @@ plugin_runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plugin_runner/authentication.py,sha256=SDPso2AogtLAV_H0LuMDp99IMZuF3oTq-Q_AXAvJ8uc,1116
 plugin_runner/plugin_runner.py,sha256=Ae7ApJLeV_wo431wfqXHdE1xEkOcGrnOQ8TvHKQ3mkw,13447
 plugin_runner/plugin_synchronizer.py,sha256=t3zzfDw-bDK_hvUDQ434qaQuKLGgBGndzaCRAnSpuTU,2554
-plugin_runner/sandbox.py,sha256=jEFIzpYHYW5pTSLXTUHjyK3vz6ZrSsAw6HqJOreRWT4,9511
+plugin_runner/sandbox.py,sha256=WYqCVfEL2wOj7wxvCVJb8hRaHiPJ6z1iy5zunI-D6p4,9532
 plugin_runner/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plugin_runner/tests/data/plugins/.gitkeep,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plugin_runner/tests/fixtures/plugins/example_plugin/CANVAS_MANIFEST.json,sha256=J9T_E5vqUX4HITHbFsd6JQpw3YvMS4wR_lhI5JL2KMk,749
@@ -249,7 +249,7 @@ plugin_runner/tests/test_sandbox.py,sha256=I44rz0sbxqtWm6mAG8fGhneE1yu9M-K3PMkE4
 pubsub/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pubsub/pubsub.py,sha256=pyTW0JU8mtaqiAV6g6xjZwel1CVy2EonPMU-_vkmhUM,1044
 settings.py,sha256=to4MXxHXTnDhhPpc8JbPuIMtxPX9cCZFuE5tfBwbROQ,2065
-canvas-0.5.0.dist-info/METADATA,sha256=MQz_rKT4LndF75okGzCix01v0eWhWhQ-FAM_RIt8_7w,4605
-canvas-0.5.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-canvas-0.5.0.dist-info/entry_points.txt,sha256=VSmSo1IZ3aEfL7enmLmlWSraS_IIkoXNVeyXzgRxFiY,46
-canvas-0.5.0.dist-info/RECORD,,
+canvas-0.6.0.dist-info/METADATA,sha256=EF8WA1OYvGaPumByC6Bm1KdOccwx_tqpM-hfGS-Ec5M,4648
+canvas-0.6.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+canvas-0.6.0.dist-info/entry_points.txt,sha256=VSmSo1IZ3aEfL7enmLmlWSraS_IIkoXNVeyXzgRxFiY,46
+canvas-0.6.0.dist-info/RECORD,,

canvas_sdk/base.py

@@ -34,15 +34,18 @@ class Model(BaseModel):
         class_name = self.__repr_name__()  # type: ignore[misc]
 
         class_name_article = "an" if class_name.startswith(("A", "E", "I", "O", "U")) else "a"
-        return [
-            self._create_error_detail(
-                "missing",
-                f"Field '{field}' is required to {method.replace('_', ' ')} {class_name_article} {class_name}",
-                v,
-            )
-            for field in required_fields
-            if (v := getattr(self, field)) is None
-        ]
+
+        error_details = []
+        for field in required_fields:
+            fields = field.split("|")
+            if not all(getattr(self, f) is None for f in fields):
+                continue
+            field_description = " or ".join([f"'{f}'" for f in fields])
+            message = f"Field {field_description} is required to {method.replace('_', ' ')} {class_name_article} {class_name}"
+            error = self._create_error_detail("missing", message, None)
+            error_details.append(error)
+
+        return error_details
 
     def _validate_before_effect(self, method: str) -> None:
         self.model_validate(self)

canvas_sdk/commands/base.py

@@ -83,7 +83,7 @@ class _BaseCommand(Model):
             if name not in base_properties
         }
 
-    def originate(self) -> Effect:
+    def originate(self, line_number: int = -1) -> Effect:
         """Originate a new command in the note body."""
         self._validate_before_effect("originate")
         return Effect(
@@ -92,6 +92,7 @@ class _BaseCommand(Model):
                 {
                     "note": self.note_uuid,
                     "data": self.values,
+                    "line_number": line_number,
                 }
             ),
         )

canvas_sdk/effects/banner_alert/add_banner_alert.py

@@ -13,7 +13,13 @@ class AddBannerAlert(_BaseEffect):
 
     class Meta:
         effect_type = EffectType.ADD_BANNER_ALERT
-        apply_required_fields = ("patient_id", "key", "narrative", "placement", "intent")
+        apply_required_fields = (
+            "patient_id|patient_filter",
+            "key",
+            "narrative",
+            "placement",
+            "intent",
+        )
 
     class Placement(Enum):
         CHART = "chart"
@@ -47,4 +53,9 @@ class AddBannerAlert(_BaseEffect):
     @property
     def effect_payload(self) -> dict[str, Any]:
         """The payload of the effect."""
-        return {"patient": self.patient_id, "key": self.key, "data": self.values}
+        return {
+            "patient": self.patient_id,
+            "patient_filter": self.patient_filter,
+            "key": self.key,
+            "data": self.values,
+        }

canvas_sdk/effects/banner_alert/remove_banner_alert.py

@@ -10,7 +10,7 @@ class RemoveBannerAlert(_BaseEffect):
 
     class Meta:
         effect_type = EffectType.REMOVE_BANNER_ALERT
-        apply_required_fields = ("patient_id", "key")
+        apply_required_fields = ("patient_id|patient_filter", "key")
 
     patient_id: str | None = None
     key: str | None = None
@@ -18,4 +18,4 @@ class RemoveBannerAlert(_BaseEffect):
     @property
     def effect_payload(self) -> dict[str, Any]:
         """The payload of the effect."""
-        return {"patient": self.patient_id, "key": self.key}
+        return {"patient": self.patient_id, "patient_filter": self.patient_filter, "key": self.key}

canvas_sdk/effects/banner_alert/tests.py

@@ -178,12 +178,28 @@ def test_protocol_that_adds_banner_alert(
                 "placement": [AddBannerAlert.Placement.APPOINTMENT_CARD],
                 "intent": AddBannerAlert.Intent.INFO,
             },
-            '{"patient": "uuid", "key": "test-key", "data": {"narrative": "hellooo", "placement": ["appointment_card"], "intent": "info", "href": null}}',
+            '{"patient": "uuid", "patient_filter": null, "key": "test-key", "data": {"narrative": "hellooo", "placement": ["appointment_card"], "intent": "info", "href": null}}',
+        ),
+        (
+            AddBannerAlert,
+            {
+                "patient_filter": {"active": True},
+                "key": "test-key",
+                "narrative": "hellooo",
+                "placement": [AddBannerAlert.Placement.APPOINTMENT_CARD],
+                "intent": AddBannerAlert.Intent.INFO,
+            },
+            '{"patient": null, "patient_filter": {"active": true}, "key": "test-key", "data": {"narrative": "hellooo", "placement": ["appointment_card"], "intent": "info", "href": null}}',
         ),
         (
             RemoveBannerAlert,
             {"patient_id": "uuid", "key": "testeroo"},
-            '{"patient": "uuid", "key": "testeroo"}',
+            '{"patient": "uuid", "patient_filter": null, "key": "testeroo"}',
+        ),
+        (
+            RemoveBannerAlert,
+            {"patient_filter": {"active": True}, "key": "testeroo"},
+            '{"patient": null, "patient_filter": {"active": true}, "key": "testeroo"}',
         ),
     ],
 )
@@ -204,7 +220,7 @@ def test_banner_alert_apply_method_succeeds_with_all_required_fields(
             AddBannerAlert,
             [
                 "5 validation errors for AddBannerAlert",
-                "Field 'patient_id' is required to apply an AddBannerAlert [type=missing",
+                "Field 'patient_id' or 'patient_filter' is required to apply an AddBannerAlert [type=missing",
                 "Field 'key' is required to apply an AddBannerAlert [type=missing",
                 "Field 'narrative' is required to apply an AddBannerAlert [type=missing",
                 "Field 'placement' is required to apply an AddBannerAlert [type=missing",
@@ -215,7 +231,7 @@ def test_banner_alert_apply_method_succeeds_with_all_required_fields(
             RemoveBannerAlert,
             [
                 "2 validation errors for RemoveBannerAlert",
-                "Field 'patient_id' is required to apply a RemoveBannerAlert [type=missing",
+                "Field 'patient_id' or 'patient_filter' is required to apply a RemoveBannerAlert [type=missing",
                 "Field 'key' is required to apply a RemoveBannerAlert [type=missing",
             ],
         ),

canvas_sdk/effects/base.py

@@ -10,6 +10,8 @@ class _BaseEffect(Model):
     A Canvas Effect that changes user behavior or autonomously performs activities on behalf of users.
     """
 
+    patient_filter: dict | None = None
+
     class Meta:
         effect_type = EffectType.UNKNOWN_EFFECT
 

canvas_sdk/effects/protocol_card/protocol_card.py

@@ -42,7 +42,7 @@ class ProtocolCard(_BaseEffect):
 
     class Meta:
         effect_type = EffectType.ADD_OR_UPDATE_PROTOCOL_CARD
-        apply_required_fields = ("patient_id", "key")
+        apply_required_fields = ("patient_id|patient_filter", "key")
 
     patient_id: str | None = None
     key: str | None = None
@@ -68,7 +68,12 @@ class ProtocolCard(_BaseEffect):
     @property
     def effect_payload(self) -> dict[str, Any]:
         """The payload of the effect."""
-        return {"patient": self.patient_id, "key": self.key, "data": self.values}
+        return {
+            "patient": self.patient_id,
+            "patient_filter": self.patient_filter,
+            "key": self.key,
+            "data": self.values,
+        }
 
     def add_recommendation(
         self,

canvas_sdk/effects/protocol_card/tests.py

@@ -25,7 +25,7 @@ def test_apply_method_succeeds_with_patient_id_and_key() -> None:
     applied = p.apply()
     assert (
         applied.payload
-        == '{"patient": "uuid", "key": "something-unique", "data": {"title": "", "narrative": "", "recommendations": [], "status": "due", "feedback_enabled": false}}'
+        == '{"patient": "uuid", "patient_filter": null, "key": "something-unique", "data": {"title": "", "narrative": "", "recommendations": [], "status": "due", "feedback_enabled": false}}'
     )
 
 
@@ -38,7 +38,7 @@ def test_apply_method_raises_error_without_patient_id_and_key() -> None:
 
     assert "2 validation errors for ProtocolCard" in err_msg
     assert (
-        "Field 'patient_id' is required to apply a ProtocolCard [type=missing, input_value=None, input_type=NoneType]"
+        "Field 'patient_id' or 'patient_filter' is required to apply a ProtocolCard [type=missing, input_value=None, input_type=NoneType]"
         in err_msg
     )
     assert (

canvas_sdk/v1/data/base.py

@@ -1,5 +1,5 @@
 from collections.abc import Container
-from typing import TYPE_CHECKING, Type, cast
+from typing import TYPE_CHECKING, Self, Type, cast
 
 from django.db import models
 from django.db.models import Q
@@ -11,25 +11,28 @@ if TYPE_CHECKING:
 class CommittableModelManager(models.Manager):
     """A manager for commands that can be committed."""
 
-    def get_queryset(self) -> "models.QuerySet":
+    def get_queryset(self) -> "CommittableQuerySet":
         """Return a queryset that filters out deleted objects."""
         # TODO: Should we just filter these out at the view level?
-        return super().get_queryset().filter(deleted=False)
+        return CommittableQuerySet(self.model, using=self._db).filter(deleted=False)
 
-    def committed(self) -> "models.QuerySet":
+
+class CommittableQuerySet(models.QuerySet):
+    """A queryset for committable objects."""
+
+    def committed(self) -> "Self":
         """Return a queryset that filters for objects that have been committed."""
-        # The committer_id IS set, and the entered_in_error_id IS NOT set
         return self.filter(committer_id__isnull=False, entered_in_error_id__isnull=True)
 
-    def for_patient(self, patient_id: str) -> "models.QuerySet":
+    def for_patient(self, patient_id: str) -> "Self":
         """Return a queryset that filters objects for a specific patient."""
         return self.filter(patient__id=patient_id)
 
 
-class ValueSetLookupQuerySet(models.QuerySet):
+class ValueSetLookupQuerySet(CommittableQuerySet):
     """A QuerySet that can filter objects based on a ValueSet."""
 
-    def find(self, value_set: Type["ValueSet"]) -> models.QuerySet:
+    def find(self, value_set: Type["ValueSet"]) -> "Self":
         """
         Filters conditions, medications, etc. to those found in the inherited ValueSet class that is passed.
 

canvas_sdk/v1/data/condition.py

@@ -1,6 +1,10 @@
 from django.db import models
 
-from canvas_sdk.v1.data.base import CommittableModelManager, ValueSetLookupQuerySet
+from canvas_sdk.v1.data.base import (
+    CommittableModelManager,
+    CommittableQuerySet,
+    ValueSetLookupQuerySet,
+)
 from canvas_sdk.v1.data.patient import Patient
 from canvas_sdk.v1.data.user import CanvasUser
 
@@ -19,7 +23,7 @@ class Condition(models.Model):
         app_label = "canvas_sdk"
         db_table = "canvas_sdk_data_api_condition_001"
 
-    objects = CommittableModelManager.from_queryset(ConditionQuerySet)()
+    objects = ConditionQuerySet.as_manager()
 
     id = models.UUIDField()
     dbid = models.BigIntegerField(primary_key=True)

canvas_sdk/v1/data/patient.py

@@ -1,4 +1,4 @@
-from typing import Self
+from typing import TYPE_CHECKING, Self
 
 from django.db import models
 

canvas_sdk/v1/data/protocol_override.py

@@ -40,7 +40,7 @@ class ProtocolOverride(models.Model):
     patient = models.ForeignKey(
         Patient,
         on_delete=models.DO_NOTHING,
-        related_name="allergy_intolerances",
+        related_name="protocol_overrides",
     )
     protocol_key = models.CharField()
     is_adjustment = models.BooleanField()

plugin_runner/sandbox.py

@@ -59,6 +59,7 @@ ALLOWED_MODULES = frozenset(
         "operator",
         "pickletools",
         "random",
+        "rapidfuzz",
         "re",
         "requests",
         "string",