Fix reward_protocol submodule imports

Author: benjibc

reward_protocol/__init__.py

@@ -11,15 +11,16 @@
 tool-augmented models using self-contained task bundles.
 """
 
-# Re-export everything from reward_kit
-from reward_kit import *
-from reward_kit import __version__
-
-# Make sure all public symbols are available
-from reward_kit import __all__
+# Map reward_protocol submodules to the underlying reward_kit modules
+import sys
 
 # Additional convenience imports for common submodules
+# Make sure all public symbols are available
+# Re-export everything from reward_kit
+from reward_kit import *  # noqa: F401,F403
 from reward_kit import (
+    __all__,
+    __version__,
     adapters,
     agent,
     auth,
@@ -47,4 +48,37 @@
     server,
     typed_interface,
     utils,
-) 
+)
+
+_SUBMODULES = [
+    "adapters",
+    "agent",
+    "auth",
+    "cli",
+    "cli_commands",
+    "common_utils",
+    "config",
+    "datasets",
+    "evaluation",
+    "execution",
+    "gcp_tools",
+    "generation",
+    "generic_server",
+    "integrations",
+    "mcp",
+    "mcp_agent",
+    "models",
+    "packaging",
+    "platform_api",
+    "playback_policy",
+    "resources",
+    "reward_function",
+    "rewards",
+    "rl_processing",
+    "server",
+    "typed_interface",
+    "utils",
+]
+
+for _name in _SUBMODULES:
+    sys.modules[f"{__name__}.{_name}"] = getattr(sys.modules["reward_kit"], _name)