@@ -105,10 +105,20 @@ def create_computation_cb(url, **kwargs) -> CallbackResult:
105105 "62237c33-8d6c-4709-aa92-c3cf693dd6d2" ,
106106 ],
107107 }
108- returned_computation = ComputationTask .model_validate (
109- ComputationTask .model_config ["json_schema_extra" ]["examples" ][0 ]
110- ).model_copy (
111- update = {
108+
109+ assert "json_schema_extra" in ComputationTask .model_config
110+ assert isinstance (ComputationTask .model_config ["json_schema_extra" ], dict )
111+ assert isinstance (
112+ ComputationTask .model_config ["json_schema_extra" ]["examples" ], list
113+ )
114+ assert isinstance (
115+ ComputationTask .model_config ["json_schema_extra" ]["examples" ][0 ], dict
116+ )
117+ computation : dict [str , Any ] = ComputationTask .model_config ["json_schema_extra" ][
118+ "examples"
119+ ][0 ].copy ()
120+ computation .update (
121+ {
112122 "id" : f"{ kwargs ['json' ]['project_id' ]} " ,
113123 "state" : state ,
114124 "pipeline_details" : {
@@ -118,6 +128,10 @@ def create_computation_cb(url, **kwargs) -> CallbackResult:
118128 },
119129 }
120130 )
131+ returned_computation = ComputationTask .model_validate (
132+ computation
133+ )
134+
121135 return CallbackResult (
122136 status = 201 ,
123137 # NOTE: aioresponses uses json.dump which does NOT encode serialization of UUIDs
@@ -134,10 +148,15 @@ def get_computation_cb(url, **kwargs) -> CallbackResult:
134148 assert isinstance (
135149 ComputationGet .model_config ["json_schema_extra" ]["examples" ], list
136150 )
137- returned_computation = ComputationGet .model_validate (
138- ComputationGet .model_config ["json_schema_extra" ]["examples" ][0 ]
139- ).model_copy (
140- update = {
151+ assert isinstance (
152+ ComputationGet .model_config ["json_schema_extra" ]["examples" ][0 ], dict
153+ )
154+
155+ computation : dict [str , Any ] = ComputationGet .model_config ["json_schema_extra" ][
156+ "examples"
157+ ][0 ].copy ()
158+ computation .update (
159+ {
141160 "id" : Path (url .path ).name ,
142161 "state" : state ,
143162 "pipeline_details" : {
@@ -147,6 +166,7 @@ def get_computation_cb(url, **kwargs) -> CallbackResult:
147166 },
148167 }
149168 )
169+ returned_computation = ComputationGet .model_validate (computation )
150170
151171 return CallbackResult (
152172 status = 200 ,
0 commit comments