Merge pull request #2 from reward-protocol/codex/move-files-from-reward_kit-to-reward_protocol

Switch examples to reward_protocol imports

Author: benjibc

examples/apps_coding_example/main.py

@@ -8,11 +8,11 @@
 
 from typing import Any, Dict, List, Optional, Union
 
-from reward_kit import EvaluateResult, reward_function
-from reward_kit.models import Message
+from reward_protocol import EvaluateResult, reward_function
+from reward_protocol.models import Message
 
 # Import the existing reward function from reward-kit
-from reward_kit.rewards.apps_coding_reward import evaluate_apps_solution
+from reward_protocol.rewards.apps_coding_reward import evaluate_apps_solution
 
 
 @reward_function

examples/braintrust_integration.py

@@ -1,6 +1,6 @@
 from braintrust import Eval
 
-from reward_kit.adapters.braintrust import scorer_to_reward_fn
+from reward_protocol.adapters.braintrust import scorer_to_reward_fn
 
 
 def equality_scorer(input: str, output: str, expected: str) -> float:

examples/e2b_reward_example.py

@@ -14,7 +14,7 @@
 import argparse
 import os
 
-from reward_kit.rewards.code_execution import e2b_code_execution_reward
+from reward_protocol.rewards.code_execution import e2b_code_execution_reward
 
 
 def main():

examples/frozen_lake/client/reward.py

@@ -4,8 +4,8 @@
 
 from typing import List, Optional
 
-from reward_kit.models import EvaluateResult, Message, MetricResult, StepOutput
-from reward_kit.typed_interface import reward_function
+from reward_protocol.models import EvaluateResult, Message, MetricResult, StepOutput
+from reward_protocol.typed_interface import reward_function
 
 
 @reward_function

examples/frozen_lake/server/http_rollout_server.py

@@ -20,7 +20,7 @@
 from pydantic import BaseModel
 
 # Import standardized HTTP rollout protocol types from reward-kit
-from reward_kit.agent.resources import (
+from reward_protocol.agent.resources import (
     EndEpisodeRequest,
     EndEpisodeResponse,
     HealthResponse,

examples/math_example/main.py

@@ -9,11 +9,11 @@
 import re
 from typing import Any, Dict, List, Optional, Union
 
-from reward_kit import EvaluateResult, MetricResult, reward_function
-from reward_kit.models import Message
+from reward_protocol import EvaluateResult, MetricResult, reward_function
+from reward_protocol.models import Message
 
 # Import the existing reward function from reward-kit
-from reward_kit.rewards.math import math_reward
+from reward_protocol.rewards.math import math_reward
 
 
 def check_think_answer_format(text: str) -> bool:

examples/rollout_control_plane_demo.py

@@ -22,9 +22,9 @@
 # Add the mcp directory to path
 sys.path.insert(0, str(Path(__file__).parent.parent))
 
-from reward_kit.mcp.execution.rollout import RolloutManager
-from reward_kit.mcp.session.manager import GeneralMCPVectorEnv, SessionManager
-from reward_kit.mcp.types import DatasetRow, MCPSession, MCPToolCall, Trajectory
+from reward_protocol.mcp.execution.rollout import RolloutManager
+from reward_protocol.mcp.session.manager import GeneralMCPVectorEnv, SessionManager
+from reward_protocol.mcp.types import DatasetRow, MCPSession, MCPToolCall, Trajectory
 
 
 class DemoPolicy:

examples/taxi_mcp_complete/mcp_server/simulation_server.py

@@ -15,8 +15,8 @@
 from taxi_adapter import TaxiAdapter
 from taxi_mcp_server import TaxiProdServer
 
-from reward_kit.mcp import SimulationServerBase
-from reward_kit.mcp.simulation_server import simulation_resource, simulation_tool
+from reward_protocol.mcp import SimulationServerBase
+from reward_protocol.mcp.simulation_server import simulation_resource, simulation_tool
 
 
 class TaxiSimServer(TaxiAdapter, SimulationServerBase):

examples/taxi_mcp_complete/mcp_server/taxi_adapter.py

@@ -10,7 +10,7 @@
 
 from gymnasium.envs.toy_text.taxi import TaxiEnv
 
-from reward_kit.mcp import EnvironmentAdapter
+from reward_protocol.mcp import EnvironmentAdapter
 
 
 class TaxiAdapter(EnvironmentAdapter):

examples/taxi_mcp_complete/mcp_server/taxi_mcp_server.py

@@ -25,7 +25,7 @@
 from mcp.server.fastmcp import Context
 from taxi_adapter import TaxiAdapter
 
-from reward_kit.mcp import GymProductionServer
+from reward_protocol.mcp import GymProductionServer
 
 
 class TaxiProdServer(GymProductionServer):