fix: pydantic v2 deprecation warning for dict

Closes melvinkcx#54
psychedelicious · Feb 19, 2024 · f1593b3 · f1593b3
1 parent ecef554
commit f1593b3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fastapi_events/dispatcher.py b/fastapi_events/dispatcher.py
@@ -123,7 +123,10 @@ def _derive_event_name_and_payload_from_pydantic_model(
 
     if not payload:
         payload_schema_cls_dict_args = payload_schema_cls_dict_args or DEFAULT_PAYLOAD_SCHEMA_CLS_DICT_ARGS
-        payload = event_name_or_model.dict(**payload_schema_cls_dict_args)
+        if IS_PYDANTIC_V1:
+            payload = event_name_or_model.dict(**payload_schema_cls_dict_args)
+        else:
+            payload = event_name_or_model.model_dump(**payload_schema_cls_dict_args)
 
     return event_name, payload