Add custom serializer for CreateRequest + tests

This commit is contained in:
ParthSareen
2025-01-13 11:02:00 -08:00
parent bee11029f7
commit 4f9fb88137
2 changed files with 68 additions and 4 deletions

View File

@@ -401,9 +401,9 @@ class PushRequest(BaseStreamableRequest):
class CreateRequest(BaseStreamableRequest):
@model_serializer
def serialize_model(self):
output = {k: v for k, v in self.__dict__.items() if v is not None}
@model_serializer(mode='wrap')
def serialize_model(self, nxt):
output = nxt(self)
if 'from_' in output:
output['from'] = output.pop('from_')
return output