update to new protobuf

Author: xzrderek

eval_protocol/cli_commands/create_rft.py

@@ -660,8 +660,8 @@ def _create_rft_job(
         ("temperature", "temperature"),
         ("topP", "top_p"),
         ("topK", "top_k"),
-        ("maxTokens", "max_output_tokens"),
-        ("n", "response_candidates_count"),
+        ("maxOutputTokens", "max_output_tokens"),
+        ("responseCandidatesCount", "response_candidates_count"),
     ]:
         val = getattr(args, arg_name, None)
         if val is not None:

tests/test_cli_create_rft.py

@@ -182,8 +182,8 @@ def _fake_validate_evaluator_locally(
     assert abs(ip["temperature"] - 0.9) < 1e-12
     assert abs(ip["topP"] - 0.95) < 1e-12
     assert ip["topK"] == 50
-    assert ip["maxTokens"] == 4096
-    assert ip["n"] == 6
+    assert ip["maxOutputTokens"] == 4096
+    assert ip["responseCandidatesCount"] == 6
     assert ip["extraBody"] == '{"foo":"bar"}'
 
     # W&B mapping
@@ -1126,8 +1126,8 @@ def _fake_post(url, json=None, headers=None, timeout=None):
 
     # Inference params mapping
     ip = body["inferenceParameters"]
-    assert ip["n"] == 4
-    assert ip["maxTokens"] == 32768
+    assert ip["responseCandidatesCount"] == 4
+    assert ip["maxOutputTokens"] == 32768
 
     # Other top-level
     assert body["chunkSize"] == 50