Mechanical: Apply formatting to Python files

Author: josemontespg

agent_sdks/python/src/a2ui/parser/streaming.py

@@ -1046,7 +1046,10 @@ def traverse(obj, parent_key=None):
                   if not any(ec["id"] == placeholder_id for ec in extra_components):
                     extra_components.append(placeholder_comp)
 
-              if not valid_children and field in ("children", "explicitList"):
+              if not valid_children and field in (
+                  "children",
+                  "explicitList",
+              ):
                 # If list is empty, check if it was partial in the buffer
                 # (meaning it's a sequence that started but hasn't yielded items yet)
                 term = f'"{field}"'

agent_sdks/python/src/a2ui/parser/streaming_v08.py

@@ -33,9 +33,9 @@ def __init__(self, catalog=None):
   def _placeholder_component(self) -> Dict[str, Any]:
     """Returns the placeholder component."""
     return {
-        'component': {
-            'Row': {
-                'children': {'explicitList': []},
+        "component": {
+            "Row": {
+                "children": {"explicitList": []},
             }
         }
     }
@@ -75,9 +75,9 @@ def get_latest_value(key: str) -> Optional[str]:
         if match:
           return match.group(1)
 
-    self.surface_id = get_latest_value('surfaceId')
+    self.surface_id = get_latest_value("surfaceId")
 
-    parsed_root = get_latest_value('root')
+    parsed_root = get_latest_value("root")
     if parsed_root is not None:
       self.root_id = parsed_root
 
@@ -121,7 +121,7 @@ def _handle_complete_object(
         surface_id = val.get(SURFACE_ID_KEY) or surface_id
 
     self.surface_id = surface_id
-    sid = self.surface_id or 'unknown'
+    sid = self.surface_id or "unknown"
 
     if MSG_TYPE_DELETE_SURFACE in obj:
       if sid in self._yielded_surfaces_set or self._buffered_start_message:
@@ -144,7 +144,7 @@ def _handle_complete_object(
       br_val = obj[MSG_TYPE_BEGIN_RENDERING]
       if isinstance(br_val, dict):
         self.surface_id = br_val.get(SURFACE_ID_KEY, self.surface_id)
-      self.root_id = br_val.get('root', self.root_id or DEFAULT_ROOT_ID)
+      self.root_id = br_val.get("root", self.root_id or DEFAULT_ROOT_ID)
       self._buffered_start_message = obj
 
       # Yield beginRendering immediately when it completes
@@ -164,10 +164,10 @@ def _handle_complete_object(
 
     if MSG_TYPE_SURFACE_UPDATE in obj:
       self.add_msg_type(MSG_TYPE_SURFACE_UPDATE)
-      components = obj[MSG_TYPE_SURFACE_UPDATE].get('components', [])
+      components = obj[MSG_TYPE_SURFACE_UPDATE].get("components", [])
       for comp in components:
-        if isinstance(comp, dict) and 'id' in comp:
-          self._seen_components[comp['id']] = comp
+        if isinstance(comp, dict) and "id" in comp:
+          self._seen_components[comp["id"]] = comp
       self.yield_reachable(messages, check_root=True, raise_on_orphans=False)
       return True
 
@@ -209,17 +209,17 @@ def _get_active_msg_type_for_components(self) -> Optional[str]:
   def _deduplicate_data_model(self, m: Dict[str, Any], strict_integrity: bool) -> bool:
     if MSG_TYPE_DATA_MODEL_UPDATE in m:
       dm = m[MSG_TYPE_DATA_MODEL_UPDATE]
-      raw_contents = dm.get('contents', {})
+      raw_contents = dm.get("contents", {})
       contents_dict = {}
       if isinstance(raw_contents, list):
         for entry in raw_contents:
-          if isinstance(entry, dict) and 'key' in entry:
-            key = entry['key']
+          if isinstance(entry, dict) and "key" in entry:
+            key = entry["key"]
             val = (
-                entry.get('valueString')
-                or entry.get('valueNumber')
-                or entry.get('valueBoolean')
-                or entry.get('valueMap')
+                entry.get("valueString")
+                or entry.get("valueNumber")
+                or entry.get("valueBoolean")
+                or entry.get("valueMap")
             )
             if key and val is not None:
               contents_dict[key] = val

agent_sdks/python/src/a2ui/parser/streaming_v09.py

@@ -34,8 +34,8 @@ def __init__(self, catalog=None):
   def _placeholder_component(self) -> Dict[str, Any]:
     """Returns a v0.9 flat style placeholder component specification."""
     return {
-        'component': 'Row',
-        'children': [],
+        "component": "Row",
+        "children": [],
     }
 
   @property
@@ -67,9 +67,9 @@ def get_latest_value(key: str) -> Optional[str]:
         if match:
           return match.group(1)
 
-    self.surface_id = get_latest_value('surfaceId')
+    self.surface_id = get_latest_value("surfaceId")
 
-    parsed_root = get_latest_value('root')
+    parsed_root = get_latest_value("root")
     if parsed_root is not None:
       self.root_id = parsed_root
 
@@ -105,13 +105,13 @@ def _handle_complete_object(
         surface_id = val.get(SURFACE_ID_KEY) or surface_id
 
     self.surface_id = surface_id
-    sid = self.surface_id or 'unknown'
+    sid = self.surface_id or "unknown"
 
     # v0.9 Specific Handling
     if MSG_TYPE_CREATE_SURFACE in obj:
       val = obj[MSG_TYPE_CREATE_SURFACE]
       if isinstance(val, dict):
-        self.root_id = val.get('root', self.root_id or DEFAULT_ROOT_ID)
+        self.root_id = val.get("root", self.root_id or DEFAULT_ROOT_ID)
       self._buffered_start_message = obj
 
       # Yield createSurface immediately when it completes
@@ -131,12 +131,12 @@ def _handle_complete_object(
     if MSG_TYPE_UPDATE_COMPONENTS in obj:
       self.add_msg_type(MSG_TYPE_UPDATE_COMPONENTS)
       self.root_id = obj[MSG_TYPE_UPDATE_COMPONENTS].get(
-          'root', self.root_id or DEFAULT_ROOT_ID
+          "root", self.root_id or DEFAULT_ROOT_ID
       )
-      components = obj[MSG_TYPE_UPDATE_COMPONENTS].get('components', [])
+      components = obj[MSG_TYPE_UPDATE_COMPONENTS].get("components", [])
       for comp in components:
-        if isinstance(comp, dict) and 'id' in comp:
-          self._seen_components[comp['id']] = comp
+        if isinstance(comp, dict) and "id" in comp:
+          self._seen_components[comp["id"]] = comp
       self.yield_reachable(messages, check_root=True, raise_on_orphans=False)
       return True
 
@@ -161,7 +161,7 @@ def _construct_sniffed_data_model_message(
       self, active_msg_type: str, delta_msg_payload: Dict[str, Any]
   ) -> Dict[str, Any]:
     """Returns the message to yield for a partial data model update for v0.9."""
-    return {'version': 'v0.9', active_msg_type: delta_msg_payload}
+    return {"version": "v0.9", active_msg_type: delta_msg_payload}
 
   def _sniff_partial_data_model(self, messages: List[ResponsePart]) -> None:
     """Sniffs for partial data model updates in v0.9 (value property)."""
@@ -170,7 +170,7 @@ def _sniff_partial_data_model(self, messages: List[ResponsePart]) -> None:
       return
 
     for b_type, start_idx in reversed(self._brace_stack):
-      if b_type != '{':
+      if b_type != "{":
         continue
       raw_fragment = self._json_buffer[start_idx:]
       if not raw_fragment:
@@ -183,8 +183,8 @@ def _sniff_partial_data_model(self, messages: List[ResponsePart]) -> None:
       except json.JSONDecodeError:
         # Fallback: iteratively strip from the last comma
         trimmed = raw_fragment
-        while ',' in trimmed:
-          trimmed = trimmed.rsplit(',', 1)[0]
+        while "," in trimmed:
+          trimmed = trimmed.rsplit(",", 1)[0]
           try:
             fixed_trimmed = self._fix_json(trimmed)
             if fixed_trimmed:
@@ -196,8 +196,8 @@ def _sniff_partial_data_model(self, messages: List[ResponsePart]) -> None:
       if obj and isinstance(obj, dict) and msg_type in obj:
 
         dm_obj = obj[msg_type]
-        if isinstance(dm_obj, dict) and 'value' in dm_obj:
-          value_map = dm_obj['value']
+        if isinstance(dm_obj, dict) and "value" in dm_obj:
+          value_map = dm_obj["value"]
           if isinstance(value_map, dict):
             # Find delta against yielded data model
             delta = {}
@@ -206,10 +206,10 @@ def _sniff_partial_data_model(self, messages: List[ResponsePart]) -> None:
                 delta[k] = v
 
             if delta:
-              sid = dm_obj.get(SURFACE_ID_KEY) or self._surface_id or 'default'
+              sid = dm_obj.get(SURFACE_ID_KEY) or self._surface_id or "default"
               delta_msg_payload = {
                   SURFACE_ID_KEY: sid,
-                  'value': delta,
+                  "value": delta,
               }
               delta_msg = self._construct_sniffed_data_model_message(
                   msg_type, delta_msg_payload
@@ -230,12 +230,12 @@ def _construct_partial_message(
     }
     if self.surface_id:
       payload[SURFACE_ID_KEY] = self.surface_id
-    return {'version': 'v0.9', MSG_TYPE_UPDATE_COMPONENTS: payload}
+    return {"version": "v0.9", MSG_TYPE_UPDATE_COMPONENTS: payload}
 
   @property
   def _yielded_surfaces_set(self) -> Set[str]:
     """Provides access to version-specific yielded surfaces set."""
-    if not hasattr(self, '_yielded_create_surfaces'):
+    if not hasattr(self, "_yielded_create_surfaces"):
       self._yielded_create_surfaces: Set[str] = set()
     return self._yielded_create_surfaces
 
@@ -255,13 +255,13 @@ def _deduplicate_data_model(self, m: Dict[str, Any], strict_integrity: bool) ->
       if isinstance(udm, dict):
         is_new = False
         for k, v in udm.items():
-          if k not in (SURFACE_ID_KEY, 'root') and self._yielded_data_model.get(k) != v:
+          if k not in (SURFACE_ID_KEY, "root") and self._yielded_data_model.get(k) != v:
             is_new = True
             break
         if not is_new and strict_integrity:
           return False
         # Update yielded model
         for k, v in udm.items():
-          if k not in (SURFACE_ID_KEY, 'root'):
+          if k not in (SURFACE_ID_KEY, "root"):
             self._yielded_data_model[k] = v
     return True

agent_sdks/python/src/a2ui/schema/common_modifiers.py

@@ -17,10 +17,10 @@ def remove_strict_validation(schema):
   if isinstance(schema, dict):
     new_schema = {k: remove_strict_validation(v) for k, v in schema.items()}
     if (
-        'additionalProperties' in new_schema
-        and new_schema['additionalProperties'] is False
+        "additionalProperties" in new_schema
+        and new_schema["additionalProperties"] is False
     ):
-      del new_schema['additionalProperties']
+      del new_schema["additionalProperties"]
     return new_schema
   elif isinstance(schema, list):
     return [remove_strict_validation(item) for item in schema]

agent_sdks/python/src/a2ui/schema/validator.py

@@ -305,7 +305,10 @@ def validate(
           ref_map = extract_component_ref_fields(self._catalog)
           root_id = _find_root_id(messages, surface_id)
           _validate_component_integrity(
-              root_id, components, ref_map, skip_root_check=not strict_integrity
+              root_id,
+              components,
+              ref_map,
+              skip_root_check=not strict_integrity,
           )
           analyze_topology(
               root_id, components, ref_map, raise_on_orphans=strict_integrity