chore(openrpc): Support for errors data model

smagafurov · Oct 14, 2023 · 818c158 · 818c158
1 parent 83ba705
commit 818c158
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/fastapi_jsonrpc/__init__.py b/fastapi_jsonrpc/__init__.py
@@ -1338,6 +1338,20 @@ def openrpc(self):
                 result_model = create_model(f'{route.name}_Result', result=(route.result_model or Any, ...))
                 result_schema = result_model.schema(ref_template=ref_template)
 
+            errors_spec = []
+            for error in route.errors:
+                spec = {
+                    'code': error.CODE,
+                    'message': error.MESSAGE,
+                }
+                error_model = error.get_data_model()
+                if error_model is not None:
+                    error_schema = error_model.schema(ref_template=ref_template)
+                    schemas_spec.update(error_schema.pop('definitions', {}))
+                    spec['data'] = error_schema
+
+                errors_spec.append(spec)
+
             method_spec = {
                 'name': route.name,
                 'summary': route.summary,
@@ -1359,13 +1373,7 @@ def openrpc(self):
                     }
                     for tag in route.tags
                 ],
-                'errors': [
-                    {
-                        'code': error.CODE,
-                        'message': error.MESSAGE,
-                    }
-                    for error in route.errors
-                ],
+                'errors': errors_spec,
             }
             methods_spec.append(method_spec)
             schemas_spec.update(params_schema.get('definitions', {}))