refactor code

Author: Abraham Lazaro Martinez

py/plugins/google-genai/src/genkit/plugins/google_genai/models/gemini.py

@@ -544,17 +544,23 @@ def _convert_schema_property(
         if defs is None:
             defs = input_schema.get('$defs') if '$defs' in input_schema else {}
 
+        if '$ref' in input_schema:
+            ref_path = input_schema['$ref']
+            ref_tokens = ref_path.split('/')
+            ref_name = ref_tokens[-1]
+
+            if ref_name not in defs:
+                raise ValueError(f'Failed to resolve schema for {ref_tokens[2]}')
+
+            schema = self._convert_schema_property(defs[ref_tokens[2]], defs)
+
+            if input_schema.get('description'):
+                schema.description = input_schema['description']
+
+            return schema
+
         if 'type' not in input_schema:
-            if '$ref' in input_schema:
-                ref_tokens = input_schema['$ref'].split('/')
-                if ref_tokens[2] not in defs:
-                    raise ValueError(f'Failed to resolve schema for {ref_tokens[2]}')
-                schema = self._convert_schema_property(defs[ref_tokens[2]], defs)
-                if input_schema.get('description'):
-                    schema.description = input_schema['description']
-                return schema
-            else:
-                return None
+            return None
 
         schema = genai_types.Schema()
         if input_schema.get('description'):