From 3893085dfe7b614a587a3ae6172775f137e89caf Mon Sep 17 00:00:00 2001
From: Devon Stewart <devon.stewart@repl.it>
Date: Fri, 7 Mar 2025 14:30:45 -0800
Subject: [PATCH] Missing tzinfo

If tzinfo is None, msgpack will explode.
---
 src/replit_river/codegen/server.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/replit_river/codegen/server.py b/src/replit_river/codegen/server.py
index dfebe73e..cd2f706c 100644
--- a/src/replit_river/codegen/server.py
+++ b/src/replit_river/codegen/server.py
@@ -269,7 +269,7 @@ def _{m.name}Encoder(
         chunks.append(f"  match e.WhichOneof({repr(oneof.name)}):")
         for field in oneofs[index]:
             if field.type_name == ".google.protobuf.Timestamp":
-                value = f"e.{field.name}.ToDatetime()"
+                value = f"e.{field.name}.ToDatetime(tzinfo=datetime.timezone.utc)"
             elif field.type == descriptor_pb2.FieldDescriptorProto.TYPE_MESSAGE:
                 encode_method_name = get_encoder_name(field)
                 value = f"{encode_method_name}(e.{field.name})"