refactor: parameterise stage value

Signed-off-by: Danju Visvanathan <danju.visvanathan@gmail.com>

Author: danjuv

README.md

@@ -196,7 +196,7 @@ The Python SDK includes a `LoggingHook`, which logs detailed information at key
 
 ```python
 from openfeature import api
-from openfeature.hook import LoggingHook
+from openfeature.hook.logging_hook import LoggingHook
 
 api.add_hooks([LoggingHook()])
 ```

openfeature/hook/logging_hook.py

@@ -61,7 +61,6 @@ def error(
         self, hook_context: HookContext, exception: Exception, hints: HookHints
     ) -> None:
         args = self._build_args(hook_context, "error")
-        args["stage"] = "error"
         args["error_code"] = (
             exception.error_code
             if isinstance(exception, OpenFeatureError)

tests/hook/test_logging_hook.py

@@ -141,35 +141,38 @@ def test_build_args_without_metadata():
         client_metadata=None,
         provider_metadata=None,
     )
-    result = hook._build_args(ctx)
+    result = hook._build_args(ctx, "before")
     assert result == {
         "flag_key": "flag",
         "default_value": "default",
         "domain": None,
         "provider_name": None,
+        "stage": "before",
     }
 
 
 def test_build_args_excludes_evaluation_context_by_default(hook_context):
     hook = LoggingHook()
-    result = hook._build_args(hook_context)
+    result = hook._build_args(hook_context, "before")
     assert result == {
         "flag_key": "my-flag",
         "default_value": False,
         "domain": "my-domain",
         "provider_name": "my-provider",
+        "stage": "before",
     }
 
 
 def test_build_args_includes_evaluation_context_when_enabled(hook_context):
     hook = LoggingHook(include_evaluation_context=True)
-    result = hook._build_args(hook_context)
+    result = hook._build_args(hook_context, "after")
     assert result == {
         "flag_key": "my-flag",
         "default_value": False,
         "domain": "my-domain",
         "provider_name": "my-provider",
         "evaluation_context": '{"targeting_key": "user-1", "attributes": {"env": "prod"}}',
+        "stage": "after",
     }