put the right files

Author: jackyzha0

pyproject.toml

@@ -60,7 +60,7 @@ lint = { workspace = true }
 
 [tool.ruff]
 lint.select = ["F", "E", "W", "I001"]
-exclude = ["*/generated/*", "*/snapshots/*"]
+exclude = ["*/generated/*", "*/snapshots/*", "*/generated_special_chars/*"]
 
 # Should be kept in sync with mypy.ini in the project root.
 # The VSCode mypy extension can only read /mypy.ini.

tests/v1/codegen/rpc/generated_special_chars/__init__.py

@@ -0,0 +1,13 @@
+# Code generated by river.codegen. DO NOT EDIT.
+from pydantic import BaseModel
+from typing import Literal
+
+import replit_river as river
+
+
+from .test_service import Test_ServiceService
+
+
+class SpecialCharsClient:
+    def __init__(self, client: river.Client[Literal[None]]):
+        self.test_service = Test_ServiceService(client)

tests/v1/codegen/rpc/generated_special_chars/test_service/__init__.py

@@ -0,0 +1,38 @@
+# Code generated by river.codegen. DO NOT EDIT.
+from collections.abc import AsyncIterable, AsyncIterator
+from typing import Any
+import datetime
+
+from pydantic import TypeAdapter
+
+from replit_river.error_schema import RiverError, RiverErrorTypeAdapter
+import replit_river as river
+
+
+from .rpc_method import Rpc_MethodInput, encode_Rpc_MethodInput
+
+boolTypeAdapter: TypeAdapter[bool] = TypeAdapter(bool)
+
+
+class Test_ServiceService:
+    def __init__(self, client: river.Client[Any]):
+        self.client = client
+
+    async def rpc_method(
+        self,
+        input: Rpc_MethodInput,
+        timeout: datetime.timedelta,
+    ) -> bool:
+        return await self.client.send_rpc(
+            "test_service",
+            "rpc_method",
+            input,
+            encode_Rpc_MethodInput,
+            lambda x: boolTypeAdapter.validate_python(
+                x  # type: ignore[arg-type]
+            ),
+            lambda x: RiverErrorTypeAdapter.validate_python(
+                x  # type: ignore[arg-type]
+            ),
+            timeout,
+        )

tests/v1/codegen/rpc/generated_special_chars/test_service/rpc_method.py

@@ -0,0 +1,50 @@
+# Code generated by river.codegen. DO NOT EDIT.
+from collections.abc import AsyncIterable, AsyncIterator
+import datetime
+from typing import (
+    Any,
+    Literal,
+    Mapping,
+    NotRequired,
+    TypedDict,
+)
+from typing_extensions import Annotated
+
+from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
+from replit_river.error_schema import RiverError
+from replit_river.client import (
+    RiverUnknownError,
+    translate_unknown_error,
+    RiverUnknownValue,
+    translate_unknown_value,
+)
+
+import replit_river as river
+
+
+def encode_Rpc_MethodInput(
+    x: "Rpc_MethodInput",
+) -> Any:
+    return {
+        k: v
+        for (k, v) in (
+            {
+                "data field6": x.get("data_field6"),
+                "data-field1": x.get("data_field1"),
+                "data.field3": x.get("data_field3"),
+                "data/field4": x.get("data_field4"),
+                "data:field2": x.get("data_field2"),
+                "data@field5": x.get("data_field5"),
+            }
+        ).items()
+        if v is not None
+    }
+
+
+class Rpc_MethodInput(TypedDict):
+    data_field6: NotRequired[str | None]
+    data_field1: str
+    data_field3: NotRequired[bool | None]
+    data_field4: NotRequired[str | None]
+    data_field2: float
+    data_field5: NotRequired[int | None]

tests/v1/codegen/test_rpc.py

@@ -104,8 +104,8 @@ def deserialize_special_chars_request(request: dict) -> dict:
     return request
 
 
-def serialize_special_chars_response(response: bool) -> dict:
-    """Serialize response for special chars test."""
+def serialize_special_chars_response(response: bool) -> bool:
+    """Serialize response for special chars test - pass through boolean."""
     return response