Update with correct LDClient type.

Author: kinyoklion

ldclient/plugin.py

@@ -1,9 +1,17 @@
+from __future__ import annotations
 from abc import ABCMeta, abstractmethod
 from dataclasses import dataclass
-from typing import Any, List, Optional
+from typing import List, Optional, TYPE_CHECKING
 
+from ldclient.context import Context
+from ldclient.evaluation import EvaluationDetail, FeatureFlagsState
 from ldclient.hook import Hook
+from ldclient.impl import AnyNum
+from ldclient.impl.evaluator import error_reason
+from ldclient.interfaces import BigSegmentStoreStatusProvider, DataSourceStatusProvider, DataStoreStatusProvider, FlagTracker
 
+if TYPE_CHECKING:
+    from ldclient.client import LDClient
 
 @dataclass(kw_only=True)
 class SdkMetadata:
@@ -70,7 +78,7 @@ def metadata(self) -> PluginMetadata:
         return PluginMetadata(name='UNDEFINED')
 
     @abstractmethod
-    def register(self, client: Any, metadata: EnvironmentMetadata) -> None:
+    def register(self, client: LDClient, metadata: EnvironmentMetadata) -> None:
         """
         Register the plugin with the SDK client.
 

ldclient/testing/test_plugin.py

@@ -115,7 +115,7 @@ def test_example_plugin(self):
 
         # Test hooks
         sdk_metadata = SdkMetadata(name="test-sdk", version="1.0.0")
-        env_metadata = EnvironmentMetadata(sdk=sdk_metadata)
+        env_metadata = EnvironmentMetadata(sdk=sdk_metadata, sdk_key="test-key")
 
         hooks = plugin.get_hooks(env_metadata)
         self.assertEqual(len(hooks), 1)