Add idempotent logger middleware

.vscode/settings.json

@@ -6,5 +6,6 @@
   },
   "editor.formatOnSave": true,
   "isort.args": ["--profile", "black"],
-  "isort.check": true
+  "isort.check": true,
+  "mypy-type-checker.preferDaemon": true
 }

examples/flask/app.py

@@ -1,21 +1,16 @@
-import logging
-
 import flask
-import src.inngest
+from src.inngest import inngest_client
 
-import examples.functions
 import inngest.flask
+from examples import functions
 
 app = flask.Flask(__name__)
-
-
-log = logging.getLogger("werkzeug")
-log.setLevel(logging.ERROR)
+inngest_client.set_logger(app.logger)
 
 
 inngest.flask.serve(
     app,
-    src.inngest.inngest_client,
-    examples.functions.functions_sync,
+    inngest_client,
+    functions.functions_sync,
 )
 app.run(port=8000)

inngest/__init__.py

@@ -11,15 +11,15 @@
     TriggerCron,
     TriggerEvent,
 )
-from ._internal.middleware_lib import CallInputTransform
 
 # TODO: Uncomment when middleware is ready for external use.
 # from ._internal.middleware_lib import (
 #     Middleware,
 #     MiddlewareSync,
 # )
+# from ._internal.execution import TransformableCallInput
 from ._internal.step_lib import Step, StepSync
-from ._internal.types import Serializable
+from ._internal.types import Logger, Serializable
 
 __all__ = [
     "Batch",
@@ -28,10 +28,11 @@
     "Event",
     "Function",
     "Inngest",
-    "CallInputTransform",
+    "Logger",
     # TODO: Uncomment when middleware is ready for external use.
     # "Middleware",
     # "MiddlewareSync",
+    # "TransformableCallInput",
     "NonRetriableError",
     "RateLimit",
     "Serializable",

inngest/_internal/client_lib.py

@@ -3,32 +3,39 @@
 import logging
 import os
 import time
+import typing
 import urllib.parse
 
 import httpx
 
-from . import const, env, errors, event_lib, middleware_lib, net, result
+from . import const, env, errors, event_lib, middleware_lib, net, result, types
 
 
 class Inngest:
+    middleware: list[
+        typing.Type[middleware_lib.Middleware | middleware_lib.MiddlewareSync]
+    ]
+
     def __init__(
         self,
         *,
         app_id: str,
         base_url: str | None = None,
         event_key: str | None = None,
         is_production: bool | None = None,
-        logger: logging.Logger | None = None,
+        logger: types.Logger | None = None,
         middleware: list[
-            middleware_lib.Middleware | middleware_lib.MiddlewareSync
+            typing.Type[
+                middleware_lib.Middleware | middleware_lib.MiddlewareSync
+            ]
         ]
         | None = None,
     ) -> None:
         self.app_id = app_id
         self.base_url = base_url
         self.is_production = is_production or env.is_prod()
         self.logger = logger or logging.getLogger(__name__)
-        self.middleware = middleware_lib.MiddlewareManager(middleware or [])
+        self.middleware = middleware or []
 
         if event_key is None:
             if not self.is_production:
@@ -80,6 +87,14 @@ def _build_send_request(
             )
         )
 
+    def add_middleware(
+        self,
+        middleware: typing.Type[
+            middleware_lib.Middleware | middleware_lib.MiddlewareSync
+        ],
+    ) -> None:
+        self.middleware = [*self.middleware, middleware]
+
     async def send(
         self,
         events: event_lib.Event | list[event_lib.Event],
@@ -110,6 +125,9 @@ def send_sync(
                     raise err
         return ids
 
+    def set_logger(self, logger: types.Logger) -> None:
+        self.logger = logger
+
 
 def _extract_ids(body: object) -> list[str]:
     if not isinstance(body, dict) or "ids" not in body: