Expose protocol-version to CLI

Author: blast-hardcheese

src/replit_river/codegen/run.py

@@ -38,6 +38,13 @@ def main() -> None:
         action="store_true",
         default=False,
     )
+    client.add_argument(
+        "--protocol-version",
+        help="Generate river v2 clients",
+        action="store",
+        default="v1.1",
+        choices=["v1.1", "v2.0"],
+    )
     client.add_argument("schema", help="schema file")
     args = parser.parse_args()
 
@@ -62,7 +69,7 @@ def file_opener(path: Path) -> TextIO:
             client_name=args.client_name,
             typed_dict_inputs=args.typed_dict_inputs,
             file_opener=file_opener,
-            protocol_version="v1.1",
+            protocol_version=args.protocol_version,
         )
     else:
         raise NotImplementedError(f"Unknown command {args.command}")