quickstart.py

# Copyright 2026 Google LLC
#
# 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.

"""End-to-end example against a running ate-env-api (default localhost:7777).

Port-forward the service first:

    kubectl port-forward -n ate-env svc/ate-env-api 7777:7777
"""

import asyncio

from ate_env import Client


async def main():
    # One Client can be shared across tasks/agents; close it on shutdown.
    client = Client("localhost:7777")
    try:
        env = await client.create("demo-env")

        result = await env.shell("echo hello && uname -a")
        print(result.exit_code, result.stdout)

        await env.write_file("/workspace/notes.txt", b"hi\n", mode=0o644)
        print(await env.read_file_bytes("/workspace/notes.txt"))

        pid = await env.start_process(
            ["sh", "-c", "for i in 1 2 3; do echo $i; sleep 1; done"]
        )
        async for chunk in env.stream_outputs(pid, follow=True):
            print(chunk.source.name, chunk.data.decode())

        await env.delete()
    finally:
        await client.close()


if __name__ == "__main__":
    asyncio.run(main())