Update tasks

1
from anduril import Lattice, EntityIdsSelector, TaskStatus, Principal, System
2
import asyncio, os, sys
3
4
lattice_endpoint = os.getenv('LATTICE_ENDPOINT')
5
environment_token = os.getenv('ENVIRONMENT_TOKEN')
6
7
# Remove sandboxes_token from the following statements if you are not developing on Sandboxes.
8
sandboxes_token = os.getenv('SANDBOXES_TOKEN')
9
if not environment_token or not lattice_endpoint or not sandboxes_token:
10
    print("Missing required environment variables.")
11
    sys.exit(1)
12
    
13
client = Lattice(
14
    base_url=f"https://{lattice_endpoint}",
15
    token=environment_token,
16
    # Remove the following header if you are not developing on Sandboxes.
17
    headers={ "anduril-sandbox-authorization": f"Bearer {sandboxes_token}" }
18
)
19
async def listen_as_agent(agent_entity_id):
20
    try:
21
        requests = client.tasks.listen_as_agent(
22
            agent_selector=EntityIdsSelector(
23
                entity_ids=[agent_entity_id]
24
            )
25
        )
26
        return requests
27
    except Exception as error:
28
        print(f"Encountered the following error while listening for tasks: {error}")
29
30
async def start_task(task_id, task_status_version, agent_entity_id):
31
    try:
32
        task_status_version += 1
33
        response = await client.tasks.update_task_status(
34
            task_id=task_id,
35
            status_version=task_status_version,
36
            new_status=TaskStatus(
37
                status="STATUS_EXECUTING"
38
            ),
39
            author=Principal(
40
                system=System(
41
                    entity_id=agent_entity_id
42
                )
43
            )
44
        )
45
        return response
46
    except Exception as error:
47
        print(f"Encountered the following error while starting task: {error}")
48
49
async def main(entity_id):
50
    while True:
51
        try:
52
            print(f"Listening for tasks for entity: {entity_id}...")
53
            request = await listen_as_agent(entity_id)
54
            if request.execute_request:
55
                # Parse task-specific information.
56
                task_id = request.execute_request.task.version.task_id
57
                task_status_version = request.execute_request.task.version.status_version
58
59
                print(f"Starting task {task_id}, version {task_status_version}.")
60
                result = start_task(task_id, task_status_version, entity_id)
61
                print (f"Started task with status version: {result.version.status_version}.")
62
                
63
            elif request.complete_request:
64
                print(f"Completing task: {request.complete_request.task_id}")
65
            elif request.cancel_request:
66
                print(f"Cancelling task:, {request.cancel_request.task_id}")
67
        except Exception as error:
68
            print("Encountered the following error while processing task:", error)
69
70
if __name__ == "__main__":
71
    raise SystemExit(asyncio.run(main(entity_id="ENTITY_ID")))