Adding and threading through protocol_version

Author: blast-hardcheese

src/replit_river/codegen/client.py

@@ -802,6 +802,7 @@ def render_library_call(
     schema_name: str,
     name: str,
     procedure: RiverProcedure,
+    protocol_version: Literal["v1.1", "v2.0"],
     init_meta: tuple[RiverType, TypeExpression, str] | None,
     input_meta: tuple[RiverType, TypeExpression, str] | None,
     output_meta: tuple[RiverType, TypeExpression, str] | None,
@@ -1022,6 +1023,7 @@ def generate_individual_service(
     schema_name: str,
     schema: RiverService,
     input_base_class: Literal["TypedDict"] | Literal["BaseModel"],
+    protocol_version: Literal["v1.1", "v2.0"],
 ) -> tuple[ModuleName, ClassName, dict[RenderedPath, FileContents]]:
     serdes: list[tuple[list[TypeName], list[ModuleName], list[FileContents]]] = []
 
@@ -1254,6 +1256,7 @@ def combine_or_none(
                 schema_name=schema_name,
                 name=name,
                 procedure=procedure,
+                protocol_version=protocol_version,
                 init_meta=combine_or_none(
                     procedure.init, init_type, render_init_method
                 ),
@@ -1308,6 +1311,7 @@ def generate_river_client_module(
     client_name: str,
     schema_root: RiverSchema,
     typed_dict_inputs: bool,
+    protocol_version: Literal["v1.1", "v2.0"],
 ) -> dict[RenderedPath, FileContents]:
     files: dict[RenderedPath, FileContents] = {}
 
@@ -1332,7 +1336,10 @@ def generate_river_client_module(
     )
     for schema_name, schema in schema_root.services.items():
         module_name, class_name, emitted_files = generate_individual_service(
-            schema_name, schema, input_base_class
+            schema_name,
+            schema,
+            input_base_class,
+            protocol_version,
         )
         files.update(emitted_files)
         modules.append((module_name, class_name))
@@ -1356,7 +1363,10 @@ def schema_to_river_client_codegen(
     with read_schema() as f:
         schemas = RiverSchemaFile(json.load(f))
     for subpath, contents in generate_river_client_module(
-        client_name, schemas.root, typed_dict_inputs
+        client_name,
+        schemas.root,
+        typed_dict_inputs,
+        "v1.1",
     ).items():
         module_path = Path(target_path).joinpath(subpath)
         module_path.parent.mkdir(mode=0o755, parents=True, exist_ok=True)