remove comments

This commit is contained in:
Dinesh Yeduguru 2025-02-11 09:29:47 -08:00
parent 2214de9e54
commit 14a1a04b8e

View file

@ -136,7 +136,6 @@ class SchemaBuilder:
type_schema, type_definitions = self.schema_generator.classdef_to_schema(typ) type_schema, type_definitions = self.schema_generator.classdef_to_schema(typ)
# Add metrics field to all response schemas
if self._is_response_type(typ): if self._is_response_type(typ):
if isinstance(type_schema, dict) and '$ref' in type_schema: if isinstance(type_schema, dict) and '$ref' in type_schema:
# If it's a reference, modify the schema in type_definitions # If it's a reference, modify the schema in type_definitions
@ -144,10 +143,8 @@ class SchemaBuilder:
if ref_name in type_definitions: if ref_name in type_definitions:
self._add_metrics_to_schema(type_definitions[ref_name]) self._add_metrics_to_schema(type_definitions[ref_name])
else: else:
# Direct schema case
self._add_metrics_to_schema(type_schema) self._add_metrics_to_schema(type_schema)
# Register all schemas, including modified ones
for ref, schema in type_definitions.items(): for ref, schema in type_definitions.items():
self._add_ref(ref, schema) self._add_ref(ref, schema)