android-env 1.2.2__py3-none-any.whl → 1.2.3__py3-none-any.whl

android_env/loader_test.py

@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 DeepMind Technologies Limited.
+# Copyright 2024 DeepMind Technologies Limited.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -20,11 +20,14 @@ import os
 from unittest import mock
 
 from absl.testing import absltest
-from android_env import environment
+from android_env import env_interface
 from android_env import loader
+from android_env.components import config_classes
 from android_env.components import coordinator as coordinator_lib
+from android_env.components import device_settings as device_settings_lib
 from android_env.components import task_manager as task_manager_lib
 from android_env.components.simulators.emulator import emulator_simulator
+from android_env.components.simulators.fake import fake_simulator
 from android_env.proto import task_pb2
 
 
@@ -32,49 +35,116 @@ class LoaderTest(absltest.TestCase):
 
   @mock.patch.object(task_manager_lib, 'TaskManager', autospec=True)
   @mock.patch.object(emulator_simulator, 'EmulatorSimulator', autospec=True)
+  @mock.patch.object(device_settings_lib, 'DeviceSettings', autospec=True)
   @mock.patch.object(coordinator_lib, 'Coordinator', autospec=True)
   @mock.patch.object(builtins, 'open', autospec=True)
-  def test_load(self, mock_open, coordinator, simulator, task_manager):
-
+  def test_load_emulator(
+      self,
+      mock_open,
+      mock_coordinator,
+      mock_device_settings,
+      mock_simulator_class,
+      mock_task_manager,
+  ):
+
+    # Arrange.
     mock_open.return_value.__enter__ = mock_open
     mock_open.return_value.read.return_value = ''
+    config = config_classes.AndroidEnvConfig(
+        task=config_classes.FilesystemTaskConfig(path='some/path/'),
+        simulator=config_classes.EmulatorConfig(
+            emulator_launcher=config_classes.EmulatorLauncherConfig(
+                avd_name='my_avd',
+                android_avd_home='~/.android/avd',
+                android_sdk_root='~/Android/Sdk',
+                emulator_path='~/Android/Sdk/emulator/emulator',
+                run_headless=False,
+            ),
+            adb_controller=config_classes.AdbControllerConfig(
+                adb_path='~/Android/Sdk/platform-tools/adb',
+            ),
+        ),
+    )
 
-    env = loader.load(
-        task_path='some/path/',
-        avd_name='my_avd',
-        android_avd_home='~/.android/avd',
-        android_sdk_root='~/Android/Sdk',
-        emulator_path='~/Android/Sdk/emulator/emulator',
-        adb_path='~/Android/Sdk/platform-tools/adb',
-        run_headless=False,
+    # Act.
+    env = loader.load(config)
+
+    # Assert.
+    self.assertIsInstance(env, env_interface.AndroidEnvInterface)
+    mock_simulator_class.assert_called_with(
+        config=config_classes.EmulatorConfig(
+            emulator_launcher=config_classes.EmulatorLauncherConfig(
+                avd_name='my_avd',
+                android_avd_home=os.path.expanduser('~/.android/avd'),
+                android_sdk_root=os.path.expanduser('~/Android/Sdk'),
+                emulator_path=os.path.expanduser(
+                    '~/Android/Sdk/emulator/emulator'
+                ),
+                run_headless=False,
+                gpu_mode='swangle_indirect',
+            ),
+            adb_controller=config_classes.AdbControllerConfig(
+                adb_path=os.path.expanduser('~/Android/Sdk/platform-tools/adb'),
+                adb_server_port=5037,
+            ),
+        )
+    )
+    mock_coordinator.assert_called_with(
+        mock_simulator_class.return_value,
+        mock_task_manager.return_value,
+        mock_device_settings.return_value,
     )
 
-    self.assertIsInstance(env, environment.AndroidEnv)
-    simulator.assert_called_with(
-        adb_controller_args=dict(
-            adb_path=os.path.expanduser('~/Android/Sdk/platform-tools/adb'),
-            adb_server_port=5037,
+  @mock.patch.object(task_manager_lib, 'TaskManager', autospec=True)
+  @mock.patch.object(fake_simulator, 'FakeSimulator', autospec=True)
+  @mock.patch.object(device_settings_lib, 'DeviceSettings', autospec=True)
+  @mock.patch.object(coordinator_lib, 'Coordinator', autospec=True)
+  @mock.patch.object(builtins, 'open', autospec=True)
+  def test_load_fake_simulator(
+      self,
+      mock_open,
+      mock_coordinator,
+      mock_device_settings,
+      mock_simulator_class,
+      mock_task_manager,
+  ):
+
+    # Arrange.
+    mock_open.return_value.__enter__ = mock_open
+    mock_open.return_value.read.return_value = ''
+    config = config_classes.AndroidEnvConfig(
+        task=config_classes.FilesystemTaskConfig(path='some/path/'),
+        simulator=config_classes.FakeSimulatorConfig(
+            screen_dimensions=(1234, 5678)
         ),
-        emulator_launcher_args=dict(
-            avd_name='my_avd',
-            android_avd_home=os.path.expanduser('~/.android/avd'),
-            android_sdk_root=os.path.expanduser('~/Android/Sdk'),
-            emulator_path=os.path.expanduser('~/Android/Sdk/emulator/emulator'),
-            run_headless=False,
-            gpu_mode='swiftshader_indirect'),
     )
-    coordinator.assert_called_with(
-        simulator.return_value,
-        task_manager.return_value,
+
+    # Act.
+    env = loader.load(config)
+
+    # Assert.
+    self.assertIsInstance(env, env_interface.AndroidEnvInterface)
+    mock_simulator_class.assert_called_with(
+        config=config_classes.FakeSimulatorConfig(
+            screen_dimensions=(1234, 5678)
+        )
+    )
+    mock_coordinator.assert_called_with(
+        mock_simulator_class.return_value,
+        mock_task_manager.return_value,
+        mock_device_settings.return_value,
     )
 
   @mock.patch.object(task_manager_lib, 'TaskManager', autospec=True)
   @mock.patch.object(emulator_simulator, 'EmulatorSimulator', autospec=True)
   @mock.patch.object(coordinator_lib, 'Coordinator', autospec=True)
   @mock.patch.object(builtins, 'open', autospec=True)
-  def test_task(self, mock_open, coordinator, simulator, task_manager):
+  def test_task(
+      self, mock_open, mock_coordinator, mock_simulator, mock_task_manager
+  ):
 
-    del coordinator, simulator
+    # Arrange.
+    del mock_coordinator, mock_simulator
     mock_open.return_value.__enter__ = mock_open
     mock_open.return_value.read.return_value = r'''
 id: "fake_task"
@@ -82,20 +152,30 @@ name: "Fake Task"
 description: "Task for testing loader."
 max_episode_sec: 0
 '''
-
-    env = loader.load(
-        task_path='some/path/',
-        avd_name='my_avd',
+    config = config_classes.AndroidEnvConfig(
+        task=config_classes.FilesystemTaskConfig(path='some/path/'),
+        simulator=config_classes.EmulatorConfig(
+            emulator_launcher=config_classes.EmulatorLauncherConfig(
+                avd_name='my_avd'
+            ),
+            adb_controller=config_classes.AdbControllerConfig(
+                adb_path='~/Android/Sdk/platform-tools/adb',
+            ),
+        ),
     )
 
+    # Act.
+    env = loader.load(config)
+
+    # Assert.
     expected_task = task_pb2.Task()
     expected_task.id = 'fake_task'
     expected_task.name = 'Fake Task'
     expected_task.description = 'Task for testing loader.'
     expected_task.max_episode_sec = 0
 
-    task_manager.assert_called_with(expected_task)
-    assert isinstance(env, environment.AndroidEnv)
+    mock_task_manager.assert_called_with(expected_task)
+    self.assertIsInstance(env, env_interface.AndroidEnvInterface)
 
 
 if __name__ == '__main__':

android_env/proto/__init__.py

@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2023 DeepMind Technologies Limited.
+# Copyright 2024 DeepMind Technologies Limited.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

android_env/proto/a11y/__init__.py

@@ -0,0 +1,15 @@
+# coding=utf-8
+# Copyright 2024 DeepMind Technologies Limited.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+

android_env/proto/a11y/a11y.proto

@@ -0,0 +1,75 @@
+// Copyright 2024 DeepMind Technologies Limited.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+syntax = "proto3";
+
+package android_env;
+
+import "android_env/proto/a11y/android_accessibility_forest.proto";
+
+option java_multiple_files = true;
+option java_package = "com.google.androidenv.accessibilityforwarder";
+
+// A service to send Accessibility information to a remote server.
+//
+// The client is assumed to be running inside an Android device (e.g. emulator
+// or real device) while the server is assumed to be running outside (e.g. in a
+// Python process).
+service A11yService {
+  // Sends a forest of Accessibility trees to a server.
+  rpc SendForest(AndroidAccessibilityForest) returns (ForestResponse) {}
+  // Sends an a11y event to a server.
+  rpc SendEvent(EventRequest) returns (EventResponse) {}
+
+  // Long-lived bidirection communication between the client and the server.
+  rpc Bidi(stream ClientToServer) returns (stream ServerToClient) {}
+}
+
+// TODO(b/334952387): Remove `ForestResponse`, `EventRequest` and
+// `EventResponse` once bidi communication is in-place.
+message ForestResponse {
+  // The error if anything.
+  string error = 1;
+}
+
+// An Accessibility event.
+message EventRequest {
+  // A single event as a dictionary.
+  map<string, string> event = 1;
+}
+
+message EventResponse {
+  // The error if anything.
+  string error = 1;
+}
+
+// The message sent from the Android device to the server running outside of the
+// device.
+message ClientToServer {
+  oneof payload {
+    EventRequest event = 1;
+    AndroidAccessibilityForest forest = 2;
+  }
+}
+
+// The message sent from the server running outside of the device to the Android
+// device.
+message ServerToClient {
+  // A request to obtain the Accessibility forest.
+  message GetA11yForest {}
+
+  oneof payload {
+    GetA11yForest get_forest = 1;
+  }
+}

android_env/proto/a11y/a11y_pb2.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: android_env/proto/a11y/a11y.proto
+# Protobuf Python Version: 5.29.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'android_env/proto/a11y/a11y.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from android_env.proto.a11y import android_accessibility_forest_pb2 as android__env_dot_proto_dot_a11y_dot_android__accessibility__forest__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!android_env/proto/a11y/a11y.proto\x12\x0b\x61ndroid_env\x1a\x39\x61ndroid_env/proto/a11y/android_accessibility_forest.proto\"\x1f\n\x0e\x46orestResponse\x12\r\n\x05\x65rror\x18\x01 \x01(\t\"q\n\x0c\x45ventRequest\x12\x33\n\x05\x65vent\x18\x01 \x03(\x0b\x32$.android_env.EventRequest.EventEntry\x1a,\n\nEventEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x1e\n\rEventResponse\x12\r\n\x05\x65rror\x18\x01 \x01(\t\"\x82\x01\n\x0e\x43lientToServer\x12*\n\x05\x65vent\x18\x01 \x01(\x0b\x32\x19.android_env.EventRequestH\x00\x12\x39\n\x06\x66orest\x18\x02 \x01(\x0b\x32\'.android_env.AndroidAccessibilityForestH\x00\x42\t\n\x07payload\"m\n\x0eServerToClient\x12?\n\nget_forest\x18\x01 \x01(\x0b\x32).android_env.ServerToClient.GetA11yForestH\x00\x1a\x0f\n\rGetA11yForestB\t\n\x07payload2\xf1\x01\n\x0b\x41\x31\x31yService\x12T\n\nSendForest\x12\'.android_env.AndroidAccessibilityForest\x1a\x1b.android_env.ForestResponse\"\x00\x12\x44\n\tSendEvent\x12\x19.android_env.EventRequest\x1a\x1a.android_env.EventResponse\"\x00\x12\x46\n\x04\x42idi\x12\x1b.android_env.ClientToServer\x1a\x1b.android_env.ServerToClient\"\x00(\x01\x30\x01\x42\x30\n,com.google.androidenv.accessibilityforwarderP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.a11y.a11y_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n,com.google.androidenv.accessibilityforwarderP\001'
+  _globals['_EVENTREQUEST_EVENTENTRY']._loaded_options = None
+  _globals['_EVENTREQUEST_EVENTENTRY']._serialized_options = b'8\001'
+  _globals['_FORESTRESPONSE']._serialized_start=109
+  _globals['_FORESTRESPONSE']._serialized_end=140
+  _globals['_EVENTREQUEST']._serialized_start=142
+  _globals['_EVENTREQUEST']._serialized_end=255
+  _globals['_EVENTREQUEST_EVENTENTRY']._serialized_start=211
+  _globals['_EVENTREQUEST_EVENTENTRY']._serialized_end=255
+  _globals['_EVENTRESPONSE']._serialized_start=257
+  _globals['_EVENTRESPONSE']._serialized_end=287
+  _globals['_CLIENTTOSERVER']._serialized_start=290
+  _globals['_CLIENTTOSERVER']._serialized_end=420
+  _globals['_SERVERTOCLIENT']._serialized_start=422
+  _globals['_SERVERTOCLIENT']._serialized_end=531
+  _globals['_SERVERTOCLIENT_GETA11YFOREST']._serialized_start=505
+  _globals['_SERVERTOCLIENT_GETA11YFOREST']._serialized_end=520
+  _globals['_A11YSERVICE']._serialized_start=534
+  _globals['_A11YSERVICE']._serialized_end=775
+# @@protoc_insertion_point(module_scope)

android_env/proto/a11y/a11y_pb2.pyi

@@ -0,0 +1,49 @@
+from android_env.proto.a11y import android_accessibility_forest_pb2 as _android_accessibility_forest_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ForestResponse(_message.Message):
+    __slots__ = ("error",)
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    error: str
+    def __init__(self, error: _Optional[str] = ...) -> None: ...
+
+class EventRequest(_message.Message):
+    __slots__ = ("event",)
+    class EventEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    EVENT_FIELD_NUMBER: _ClassVar[int]
+    event: _containers.ScalarMap[str, str]
+    def __init__(self, event: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class EventResponse(_message.Message):
+    __slots__ = ("error",)
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    error: str
+    def __init__(self, error: _Optional[str] = ...) -> None: ...
+
+class ClientToServer(_message.Message):
+    __slots__ = ("event", "forest")
+    EVENT_FIELD_NUMBER: _ClassVar[int]
+    FOREST_FIELD_NUMBER: _ClassVar[int]
+    event: EventRequest
+    forest: _android_accessibility_forest_pb2.AndroidAccessibilityForest
+    def __init__(self, event: _Optional[_Union[EventRequest, _Mapping]] = ..., forest: _Optional[_Union[_android_accessibility_forest_pb2.AndroidAccessibilityForest, _Mapping]] = ...) -> None: ...
+
+class ServerToClient(_message.Message):
+    __slots__ = ("get_forest",)
+    class GetA11yForest(_message.Message):
+        __slots__ = ()
+        def __init__(self) -> None: ...
+    GET_FOREST_FIELD_NUMBER: _ClassVar[int]
+    get_forest: ServerToClient.GetA11yForest
+    def __init__(self, get_forest: _Optional[_Union[ServerToClient.GetA11yForest, _Mapping]] = ...) -> None: ...

android_env/proto/a11y/a11y_pb2_grpc.py

@@ -0,0 +1,202 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from android_env.proto.a11y import a11y_pb2 as android__env_dot_proto_dot_a11y_dot_a11y__pb2
+from android_env.proto.a11y import android_accessibility_forest_pb2 as android__env_dot_proto_dot_a11y_dot_android__accessibility__forest__pb2
+
+GRPC_GENERATED_VERSION = '1.71.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in android_env/proto/a11y/a11y_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class A11yServiceStub(object):
+    """A service to send Accessibility information to a remote server.
+
+    The client is assumed to be running inside an Android device (e.g. emulator
+    or real device) while the server is assumed to be running outside (e.g. in a
+    Python process).
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.SendForest = channel.unary_unary(
+                '/android_env.A11yService/SendForest',
+                request_serializer=android__env_dot_proto_dot_a11y_dot_android__accessibility__forest__pb2.AndroidAccessibilityForest.SerializeToString,
+                response_deserializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.ForestResponse.FromString,
+                _registered_method=True)
+        self.SendEvent = channel.unary_unary(
+                '/android_env.A11yService/SendEvent',
+                request_serializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.EventRequest.SerializeToString,
+                response_deserializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.EventResponse.FromString,
+                _registered_method=True)
+        self.Bidi = channel.stream_stream(
+                '/android_env.A11yService/Bidi',
+                request_serializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.ClientToServer.SerializeToString,
+                response_deserializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.ServerToClient.FromString,
+                _registered_method=True)
+
+
+class A11yServiceServicer(object):
+    """A service to send Accessibility information to a remote server.
+
+    The client is assumed to be running inside an Android device (e.g. emulator
+    or real device) while the server is assumed to be running outside (e.g. in a
+    Python process).
+    """
+
+    def SendForest(self, request, context):
+        """Sends a forest of Accessibility trees to a server.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def SendEvent(self, request, context):
+        """Sends an a11y event to a server.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Bidi(self, request_iterator, context):
+        """Long-lived bidirection communication between the client and the server.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_A11yServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'SendForest': grpc.unary_unary_rpc_method_handler(
+                    servicer.SendForest,
+                    request_deserializer=android__env_dot_proto_dot_a11y_dot_android__accessibility__forest__pb2.AndroidAccessibilityForest.FromString,
+                    response_serializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.ForestResponse.SerializeToString,
+            ),
+            'SendEvent': grpc.unary_unary_rpc_method_handler(
+                    servicer.SendEvent,
+                    request_deserializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.EventRequest.FromString,
+                    response_serializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.EventResponse.SerializeToString,
+            ),
+            'Bidi': grpc.stream_stream_rpc_method_handler(
+                    servicer.Bidi,
+                    request_deserializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.ClientToServer.FromString,
+                    response_serializer=android__env_dot_proto_dot_a11y_dot_a11y__pb2.ServerToClient.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'android_env.A11yService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('android_env.A11yService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class A11yService(object):
+    """A service to send Accessibility information to a remote server.
+
+    The client is assumed to be running inside an Android device (e.g. emulator
+    or real device) while the server is assumed to be running outside (e.g. in a
+    Python process).
+    """
+
+    @staticmethod
+    def SendForest(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/android_env.A11yService/SendForest',
+            android__env_dot_proto_dot_a11y_dot_android__accessibility__forest__pb2.AndroidAccessibilityForest.SerializeToString,
+            android__env_dot_proto_dot_a11y_dot_a11y__pb2.ForestResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def SendEvent(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/android_env.A11yService/SendEvent',
+            android__env_dot_proto_dot_a11y_dot_a11y__pb2.EventRequest.SerializeToString,
+            android__env_dot_proto_dot_a11y_dot_a11y__pb2.EventResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def Bidi(request_iterator,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.stream_stream(
+            request_iterator,
+            target,
+            '/android_env.A11yService/Bidi',
+            android__env_dot_proto_dot_a11y_dot_a11y__pb2.ClientToServer.SerializeToString,
+            android__env_dot_proto_dot_a11y_dot_a11y__pb2.ServerToClient.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

android_env/proto/a11y/android_accessibility_action.proto

@@ -0,0 +1,32 @@
+// Copyright 2024 DeepMind Technologies Limited.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+syntax = "proto3";
+
+package android_env;
+
+option java_multiple_files = true;
+option java_package = "com.google.androidenv.accessibilityforwarder";
+
+// An Android Accessibility Action.
+// Next index: 3
+message AndroidAccessibilityAction {
+  // Required ID that uniquely identifies the action for this node.
+  // Can be one of the standard action IDs listed in the documentation.
+  // https://developer.android.com/reference/android/view/accessibility/AccessibilityNodeInfo.AccessibilityAction
+  int32 id = 1;
+
+  // Optional label describing what the action is.
+  string label = 2;
+}