nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f1898ef0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f1899970>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '4', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f184fe00>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f187c7a0>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '4', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18c1d00>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18c2750>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '3')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18c3cb0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18d4710>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '3')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18b3d40>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18c07d0>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '2')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18b0e60>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18b1850>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '3')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f189aed0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f189b8f0>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '3', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18b1e20>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18b2870>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '3')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18b2e40>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18b37a0>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '2')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f187ffb0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18984a0>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '5', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18c0da0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18c1730>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '2')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f1899f40>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f189a900>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '4', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18c2d20>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18c3620>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '3')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f189bec0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18b08c0>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '4', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f187eed0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f187fa10>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '4', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f187cd70>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f187d790>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '4', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f18d4da0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f18d5700>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://aws-us-east-2-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://aws-us-east-2-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) ...', 'Content-Length': '22', 'Connection': 'keep-alive', 'Server': 'istio-envoy', 'x-envoy-upstream-service-time': '3')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f187dd60>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>

    @pytest.fixture
    async def tmp_nua_key(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
    ) -> AsyncGenerator[str, None]:
        account_id = TOKENS[nua_config.region].account_id
        pat_client_generator = aiohttp_client(
            base_url=nua_config.url,
            pat_key=TOKENS[nua_config.url].pat_key,
            timeout=30,
        )
        pat_client = await anext(pat_client_generator)
>       nua_client_id, nua_key = await create_nua_key(
            client=pat_client,
            account_id=account_id,
            title=f"E2E DA AGENTS - {nua_config.region}",
        )

nua/e2e/regional/test_da_tasks.py:524: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

client = <aiohttp.client.ClientSession object at 0x7f35f187e900>
account_id = '5cec111b-ea23-4b0c-a82a-d1a666dd1fd2'
title = 'E2E DA AGENTS - https://europe-1.nuclia.cloud'

    async def create_nua_key(
        client: aiohttp.ClientSession, account_id: str, title: str
    ) -> tuple[str, str]:
        body = {
            "title": title,
            "contact": "temporal key, safe to delete",
        }
        resp = await client.post(f"/api/v1/account/{account_id}/nua_clients", json=body)
>       assert resp.status == 201, await resp.text()
E       AssertionError: {"detail":"Forbidden"}
E       assert 403 == 201
E        +  where 403 = <ClientResponse(https://europe-1.nuclia.cloud/api/v1/account/5cec111b-ea23-4b0c-a82a-d1a666dd1fd2/nua_clients) [403 Fo...-envoy-upstream-service-time': '4', 'Via': '1.1 google', 'Alt-Svc': 'h3=":443"; ma=2592000,h3-29=":443"; ma=2592000')>\n.status

nua/e2e/regional/test_da_tasks.py:77: AssertionError

self = <ClientResponse(https://europe-1.stashify.cloud/api/v1/dataset/af560139-9117-42ba-8cab-8c9b7ca451bd/task/e74e26fb1a7c47a29110bf91111acdb7/stop) [None None]>
None

connection = Connection<ConnectionKey(host='europe-1.stashify.cloud', port=443, is_ssl=True, ssl=True, proxy=None, proxy_auth=None, proxy_headers_hash=None)>

    async def start(self, connection: "Connection") -> "ClientResponse":
        """Start response processing."""
        self._closed = False
        self._protocol = connection.protocol
        self._connection = connection
    
        with self._timer:
            while True:
                # read response
                try:
                    protocol = self._protocol
>                   message, payload = await protocol.read()  # type: ignore[union-attr]

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1059: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.client_proto.ResponseHandler object at 0x7f35f079a510>

    async def read(self) -> _T:
        if not self._buffer and not self._eof:
            assert not self._waiter
            self._waiter = self._loop.create_future()
            try:
>               await self._waiter
E               asyncio.exceptions.CancelledError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/streams.py:671: CancelledError

The above exception was the direct cause of the following exception:

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f184ce60>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>
tmp_nua_key = 'eyJhbGciOiJSUzI1NiIsImtpZCI6Im51YSIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJodHRwczovL2V1cm9wZS0xLnN0YXNoaWZ5LmNsb3VkLyIsImlhdCI6...dnWkstR1YVWHwv-rHnK5AzMq1aG_ChA-U5-iTHt_8aZgMZBvipGyt41taC4FwocE47fQW_0x9OyG0O6ZBAMu1ecfsKkZ4NaicBdvv8HPyt0yLXx_vVsnN0'
test_input = TestInput(filename='financial-news-kb.arrow', task_name=<TaskName.LABELER: 'labeler'>, parameters=DataAugmentation(nam...rovider='', keys=None, prompts=None)), validate_output=<function validate_labeler_output_text_block at 0x7f35f1bf0400>)

    @pytest.mark.asyncio_cooperative
    @pytest.mark.parametrize(
        "test_input", DA_TEST_INPUTS, ids=lambda test_input: test_input.parameters.name
    )
    async def test_da_agent_tasks(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
        tmp_nua_key: str,
        test_input: TestInput,
    ):
        dataset_id = None
        task_id = None
        start_time = asyncio.get_event_loop().time()
        try:
            nua_client_generator = aiohttp_client(
                base_url=nua_config.url, nua_key=tmp_nua_key, timeout=30
            )
            nua_client = await anext(nua_client_generator)
    
            dataset_id = await create_dataset(client=nua_client)
            await push_data_to_dataset(
                client=nua_client, dataset_id=dataset_id, filename=test_input.filename
            )
    
            task_id = await start_task(
                client=nua_client,
                dataset_id=dataset_id,
                task_name=test_input.task_name,
                parameters=test_input.parameters,
            )
            print(f"{test_input.parameters.name} task_id: {task_id}")
            task_request = await wait_for_task_completion(
                client=nua_client, dataset_id=dataset_id, task_id=task_id
            )
            assert task_request["completed"] is True
            assert task_request["failed"] is False
    
            await validate_task_output(
                client=nua_client,
                validation=test_input.validate_output,
            )
        finally:
            if dataset_id is not None:
                if task_id is not None:
>                   await stop_task(
                        client=nua_client, dataset_id=dataset_id, task_id=task_id
                    )

nua/e2e/regional/test_da_tasks.py:581: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
nua/e2e/regional/test_da_tasks.py:138: in stop_task
    resp = await client.post(f"/api/v1/dataset/{dataset_id}/task/{task_id}/stop")
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client.py:730: in _request
    await resp.start(conn)
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1054: in start
    with self._timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.helpers.TimerContext object at 0x7f35f185d040>
exc_type = <class 'asyncio.exceptions.CancelledError'>
exc_val = CancelledError(), exc_tb = <traceback object at 0x7f35f017b380>

    def __exit__(
        self,
        exc_type: Optional[Type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> Optional[bool]:
        enter_task: Optional[asyncio.Task[Any]] = None
        if self._tasks:
            enter_task = self._tasks.pop()
    
        if exc_type is asyncio.CancelledError and self._cancelled:
            assert enter_task is not None
            # The timeout was hit, and the task was cancelled
            # so we need to uncancel the last task that entered the context manager
            # since the cancellation should not leak out of the context manager
            if sys.version_info >= (3, 11):
                # If the task was already cancelling don't raise
                # asyncio.TimeoutError and instead return None
                # to allow the cancellation to propagate
                if enter_task.uncancel() > self._cancelling:
                    return None
>           raise asyncio.TimeoutError from exc_val
E           TimeoutError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/helpers.py:671: TimeoutError

self = <ClientResponse(https://europe-1.stashify.cloud/api/v1/dataset/9f341eb2-3c51-460a-be1f-1ac5d53ebd82/task/a0feb376691645c48bf65ea1b6f6346b/stop) [None None]>
None

connection = Connection<ConnectionKey(host='europe-1.stashify.cloud', port=443, is_ssl=True, ssl=True, proxy=None, proxy_auth=None, proxy_headers_hash=None)>

    async def start(self, connection: "Connection") -> "ClientResponse":
        """Start response processing."""
        self._closed = False
        self._protocol = connection.protocol
        self._connection = connection
    
        with self._timer:
            while True:
                # read response
                try:
                    protocol = self._protocol
>                   message, payload = await protocol.read()  # type: ignore[union-attr]

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1059: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.client_proto.ResponseHandler object at 0x7f35f06eb410>

    async def read(self) -> _T:
        if not self._buffer and not self._eof:
            assert not self._waiter
            self._waiter = self._loop.create_future()
            try:
>               await self._waiter
E               asyncio.exceptions.CancelledError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/streams.py:671: CancelledError

The above exception was the direct cause of the following exception:

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f184e120>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>
tmp_nua_key = 'eyJhbGciOiJSUzI1NiIsImtpZCI6Im51YSIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJodHRwczovL2V1cm9wZS0xLnN0YXNoaWZ5LmNsb3VkLyIsImlhdCI6...Kc-782f1DDO34oExNjygSpasLkQk-5S1a-uXwCDnZA_BMyKcENm10mGpdm2oOYk9QwP2UY7Rn38PJNq0cwEjvQ4iSZkc-oQhOlLxoXIfe-chXN1iZetB7M'
test_input = TestInput(filename='jailbreak-kb.arrow', task_name=<TaskName.PROMPT_GUARD: 'prompt-guard'>, parameters=DataAugmentatio...er='', keys=None, prompts=None)), validate_output=<function validate_prompt_guard_output_text_block at 0x7f35f1bf00e0>)

    @pytest.mark.asyncio_cooperative
    @pytest.mark.parametrize(
        "test_input", DA_TEST_INPUTS, ids=lambda test_input: test_input.parameters.name
    )
    async def test_da_agent_tasks(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
        tmp_nua_key: str,
        test_input: TestInput,
    ):
        dataset_id = None
        task_id = None
        start_time = asyncio.get_event_loop().time()
        try:
            nua_client_generator = aiohttp_client(
                base_url=nua_config.url, nua_key=tmp_nua_key, timeout=30
            )
            nua_client = await anext(nua_client_generator)
    
            dataset_id = await create_dataset(client=nua_client)
            await push_data_to_dataset(
                client=nua_client, dataset_id=dataset_id, filename=test_input.filename
            )
    
            task_id = await start_task(
                client=nua_client,
                dataset_id=dataset_id,
                task_name=test_input.task_name,
                parameters=test_input.parameters,
            )
            print(f"{test_input.parameters.name} task_id: {task_id}")
            task_request = await wait_for_task_completion(
                client=nua_client, dataset_id=dataset_id, task_id=task_id
            )
            assert task_request["completed"] is True
            assert task_request["failed"] is False
    
            await validate_task_output(
                client=nua_client,
                validation=test_input.validate_output,
            )
        finally:
            if dataset_id is not None:
                if task_id is not None:
>                   await stop_task(
                        client=nua_client, dataset_id=dataset_id, task_id=task_id
                    )

nua/e2e/regional/test_da_tasks.py:581: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
nua/e2e/regional/test_da_tasks.py:138: in stop_task
    resp = await client.post(f"/api/v1/dataset/{dataset_id}/task/{task_id}/stop")
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client.py:730: in _request
    await resp.start(conn)
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1054: in start
    with self._timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.helpers.TimerContext object at 0x7f35f07a7340>
exc_type = <class 'asyncio.exceptions.CancelledError'>
exc_val = CancelledError(), exc_tb = <traceback object at 0x7f35f0170100>

    def __exit__(
        self,
        exc_type: Optional[Type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> Optional[bool]:
        enter_task: Optional[asyncio.Task[Any]] = None
        if self._tasks:
            enter_task = self._tasks.pop()
    
        if exc_type is asyncio.CancelledError and self._cancelled:
            assert enter_task is not None
            # The timeout was hit, and the task was cancelled
            # so we need to uncancel the last task that entered the context manager
            # since the cancellation should not leak out of the context manager
            if sys.version_info >= (3, 11):
                # If the task was already cancelling don't raise
                # asyncio.TimeoutError and instead return None
                # to allow the cancellation to propagate
                if enter_task.uncancel() > self._cancelling:
                    return None
>           raise asyncio.TimeoutError from exc_val
E           TimeoutError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/helpers.py:671: TimeoutError

self = <ClientResponse(https://europe-1.stashify.cloud/api/v1/dataset/feb3d656-6172-4065-8b0b-652a8d0b7112/task/f3c424f56f144562b17a8d00a9b93905/stop) [None None]>
None

connection = Connection<ConnectionKey(host='europe-1.stashify.cloud', port=443, is_ssl=True, ssl=True, proxy=None, proxy_auth=None, proxy_headers_hash=None)>

    async def start(self, connection: "Connection") -> "ClientResponse":
        """Start response processing."""
        self._closed = False
        self._protocol = connection.protocol
        self._connection = connection
    
        with self._timer:
            while True:
                # read response
                try:
                    protocol = self._protocol
>                   message, payload = await protocol.read()  # type: ignore[union-attr]

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1059: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.client_proto.ResponseHandler object at 0x7f35f03b80b0>

    async def read(self) -> _T:
        if not self._buffer and not self._eof:
            assert not self._waiter
            self._waiter = self._loop.create_future()
            try:
>               await self._waiter
E               asyncio.exceptions.CancelledError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/streams.py:671: CancelledError

The above exception was the direct cause of the following exception:

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f184be00>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>
tmp_nua_key = 'eyJhbGciOiJSUzI1NiIsImtpZCI6Im51YSIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJodHRwczovL2V1cm9wZS0xLnN0YXNoaWZ5LmNsb3VkLyIsImlhdCI6...MZrh6o-MK_oF0_QtiQ_WrDRMGUsVmfHUST3j1e4hK07iPlwIGurNCP7Ey7cnExNoVOKOw2Xt3of2T_rRvolCDtusRLLYIbgQLch6eeX0jEjpzGLGDEM9ks'
test_input = TestInput(filename='legal-text-kb.arrow', task_name=<TaskName.SYNTHETIC_QUESTIONS: 'synthetic-questions'>, parameters=...ovider='', keys=None, prompts=None)), validate_output=<function validate_synthetic_questions_output at 0x7f35f1bf0360>)

    @pytest.mark.asyncio_cooperative
    @pytest.mark.parametrize(
        "test_input", DA_TEST_INPUTS, ids=lambda test_input: test_input.parameters.name
    )
    async def test_da_agent_tasks(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
        tmp_nua_key: str,
        test_input: TestInput,
    ):
        dataset_id = None
        task_id = None
        start_time = asyncio.get_event_loop().time()
        try:
            nua_client_generator = aiohttp_client(
                base_url=nua_config.url, nua_key=tmp_nua_key, timeout=30
            )
            nua_client = await anext(nua_client_generator)
    
            dataset_id = await create_dataset(client=nua_client)
            await push_data_to_dataset(
                client=nua_client, dataset_id=dataset_id, filename=test_input.filename
            )
    
            task_id = await start_task(
                client=nua_client,
                dataset_id=dataset_id,
                task_name=test_input.task_name,
                parameters=test_input.parameters,
            )
            print(f"{test_input.parameters.name} task_id: {task_id}")
            task_request = await wait_for_task_completion(
                client=nua_client, dataset_id=dataset_id, task_id=task_id
            )
            assert task_request["completed"] is True
            assert task_request["failed"] is False
    
            await validate_task_output(
                client=nua_client,
                validation=test_input.validate_output,
            )
        finally:
            if dataset_id is not None:
                if task_id is not None:
>                   await stop_task(
                        client=nua_client, dataset_id=dataset_id, task_id=task_id
                    )

nua/e2e/regional/test_da_tasks.py:581: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
nua/e2e/regional/test_da_tasks.py:138: in stop_task
    resp = await client.post(f"/api/v1/dataset/{dataset_id}/task/{task_id}/stop")
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client.py:730: in _request
    await resp.start(conn)
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1054: in start
    with self._timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.helpers.TimerContext object at 0x7f35f04910e0>
exc_type = <class 'asyncio.exceptions.CancelledError'>
exc_val = CancelledError(), exc_tb = <traceback object at 0x7f35f07862c0>

    def __exit__(
        self,
        exc_type: Optional[Type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> Optional[bool]:
        enter_task: Optional[asyncio.Task[Any]] = None
        if self._tasks:
            enter_task = self._tasks.pop()
    
        if exc_type is asyncio.CancelledError and self._cancelled:
            assert enter_task is not None
            # The timeout was hit, and the task was cancelled
            # so we need to uncancel the last task that entered the context manager
            # since the cancellation should not leak out of the context manager
            if sys.version_info >= (3, 11):
                # If the task was already cancelling don't raise
                # asyncio.TimeoutError and instead return None
                # to allow the cancellation to propagate
                if enter_task.uncancel() > self._cancelling:
                    return None
>           raise asyncio.TimeoutError from exc_val
E           TimeoutError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/helpers.py:671: TimeoutError

self = <ClientResponse(https://europe-1.stashify.cloud/api/v1/dataset/148eff78-3841-4bfd-82ad-c83f99075d3d/task/ca9b941b7c1a40109993e74d446d522d/stop) [None None]>
None

connection = Connection<ConnectionKey(host='europe-1.stashify.cloud', port=443, is_ssl=True, ssl=True, proxy=None, proxy_auth=None, proxy_headers_hash=None)>

    async def start(self, connection: "Connection") -> "ClientResponse":
        """Start response processing."""
        self._closed = False
        self._protocol = connection.protocol
        self._connection = connection
    
        with self._timer:
            while True:
                # read response
                try:
                    protocol = self._protocol
>                   message, payload = await protocol.read()  # type: ignore[union-attr]

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1059: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.client_proto.ResponseHandler object at 0x7f35f052aab0>

    async def read(self) -> _T:
        if not self._buffer and not self._eof:
            assert not self._waiter
            self._waiter = self._loop.create_future()
            try:
>               await self._waiter
E               asyncio.exceptions.CancelledError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/streams.py:671: CancelledError

The above exception was the direct cause of the following exception:

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f1832b10>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>
tmp_nua_key = 'eyJhbGciOiJSUzI1NiIsImtpZCI6Im51YSIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJodHRwczovL2V1cm9wZS0xLnN0YXNoaWZ5LmNsb3VkLyIsImlhdCI6...KnYL9AlBWqTBtIvgUk4hGTpQiEAWT40Marp7yb-X0QbXESCsMUyg788iW_FMXdJC4XFfYCq7uSXtZKrStv8Wg1Z3C_j-4thetOc0t8lMTj8sRBTLw82buY'
test_input = TestInput(filename='financial-news-kb.arrow', task_name=<TaskName.LABELER: 'labeler'>, parameters=DataAugmentation(nam...4o-mini', provider='', keys=None, prompts=None)), validate_output=<function validate_labeler_output at 0x7f35f1befec0>)

    @pytest.mark.asyncio_cooperative
    @pytest.mark.parametrize(
        "test_input", DA_TEST_INPUTS, ids=lambda test_input: test_input.parameters.name
    )
    async def test_da_agent_tasks(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
        tmp_nua_key: str,
        test_input: TestInput,
    ):
        dataset_id = None
        task_id = None
        start_time = asyncio.get_event_loop().time()
        try:
            nua_client_generator = aiohttp_client(
                base_url=nua_config.url, nua_key=tmp_nua_key, timeout=30
            )
            nua_client = await anext(nua_client_generator)
    
            dataset_id = await create_dataset(client=nua_client)
            await push_data_to_dataset(
                client=nua_client, dataset_id=dataset_id, filename=test_input.filename
            )
    
            task_id = await start_task(
                client=nua_client,
                dataset_id=dataset_id,
                task_name=test_input.task_name,
                parameters=test_input.parameters,
            )
            print(f"{test_input.parameters.name} task_id: {task_id}")
            task_request = await wait_for_task_completion(
                client=nua_client, dataset_id=dataset_id, task_id=task_id
            )
            assert task_request["completed"] is True
            assert task_request["failed"] is False
    
            await validate_task_output(
                client=nua_client,
                validation=test_input.validate_output,
            )
        finally:
            if dataset_id is not None:
                if task_id is not None:
>                   await stop_task(
                        client=nua_client, dataset_id=dataset_id, task_id=task_id
                    )

nua/e2e/regional/test_da_tasks.py:581: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
nua/e2e/regional/test_da_tasks.py:138: in stop_task
    resp = await client.post(f"/api/v1/dataset/{dataset_id}/task/{task_id}/stop")
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client.py:730: in _request
    await resp.start(conn)
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1054: in start
    with self._timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.helpers.TimerContext object at 0x7f35f03c0690>
exc_type = <class 'asyncio.exceptions.CancelledError'>
exc_val = CancelledError(), exc_tb = <traceback object at 0x7f35f06ab400>

    def __exit__(
        self,
        exc_type: Optional[Type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> Optional[bool]:
        enter_task: Optional[asyncio.Task[Any]] = None
        if self._tasks:
            enter_task = self._tasks.pop()
    
        if exc_type is asyncio.CancelledError and self._cancelled:
            assert enter_task is not None
            # The timeout was hit, and the task was cancelled
            # so we need to uncancel the last task that entered the context manager
            # since the cancellation should not leak out of the context manager
            if sys.version_info >= (3, 11):
                # If the task was already cancelling don't raise
                # asyncio.TimeoutError and instead return None
                # to allow the cancellation to propagate
                if enter_task.uncancel() > self._cancelling:
                    return None
>           raise asyncio.TimeoutError from exc_val
E           TimeoutError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/helpers.py:671: TimeoutError

self = <ClientResponse(https://europe-1.stashify.cloud/api/v1/dataset/394b8a1f-0bb3-473c-8028-e64a3e0b9966/task/b60303094a8848439cfa77cd61a72e1c/stop) [None None]>
None

connection = Connection<ConnectionKey(host='europe-1.stashify.cloud', port=443, is_ssl=True, ssl=True, proxy=None, proxy_auth=None, proxy_headers_hash=None)>

    async def start(self, connection: "Connection") -> "ClientResponse":
        """Start response processing."""
        self._closed = False
        self._protocol = connection.protocol
        self._connection = connection
    
        with self._timer:
            while True:
                # read response
                try:
                    protocol = self._protocol
>                   message, payload = await protocol.read()  # type: ignore[union-attr]

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1059: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.client_proto.ResponseHandler object at 0x7f35f06e93d0>

    async def read(self) -> _T:
        if not self._buffer and not self._eof:
            assert not self._waiter
            self._waiter = self._loop.create_future()
            try:
>               await self._waiter
E               asyncio.exceptions.CancelledError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/streams.py:671: CancelledError

The above exception was the direct cause of the following exception:

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f1833cb0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>
tmp_nua_key = 'eyJhbGciOiJSUzI1NiIsImtpZCI6Im51YSIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJodHRwczovL2V1cm9wZS0xLnN0YXNoaWZ5LmNsb3VkLyIsImlhdCI6...MeeOednGtkx6kAlZegyaIux2XOdoqHIsuC2OKW7qP5vHa6hjxOesyACvRboNVzvXLWFj36j06UkZgVw1dkarRgznp2Zxs9Ra0_jl1-TJRxifiLWk6IuGZM'
test_input = TestInput(filename='legal-text-kb.arrow', task_name=<TaskName.LLM_GRAPH: 'llm-graph'>, parameters=DataAugmentation(nam...-mini', provider='', keys=None, prompts=None)), validate_output=<function validate_llm_graph_output at 0x7f35f1beff60>)

    @pytest.mark.asyncio_cooperative
    @pytest.mark.parametrize(
        "test_input", DA_TEST_INPUTS, ids=lambda test_input: test_input.parameters.name
    )
    async def test_da_agent_tasks(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
        tmp_nua_key: str,
        test_input: TestInput,
    ):
        dataset_id = None
        task_id = None
        start_time = asyncio.get_event_loop().time()
        try:
            nua_client_generator = aiohttp_client(
                base_url=nua_config.url, nua_key=tmp_nua_key, timeout=30
            )
            nua_client = await anext(nua_client_generator)
    
            dataset_id = await create_dataset(client=nua_client)
            await push_data_to_dataset(
                client=nua_client, dataset_id=dataset_id, filename=test_input.filename
            )
    
            task_id = await start_task(
                client=nua_client,
                dataset_id=dataset_id,
                task_name=test_input.task_name,
                parameters=test_input.parameters,
            )
            print(f"{test_input.parameters.name} task_id: {task_id}")
            task_request = await wait_for_task_completion(
                client=nua_client, dataset_id=dataset_id, task_id=task_id
            )
            assert task_request["completed"] is True
            assert task_request["failed"] is False
    
            await validate_task_output(
                client=nua_client,
                validation=test_input.validate_output,
            )
        finally:
            if dataset_id is not None:
                if task_id is not None:
>                   await stop_task(
                        client=nua_client, dataset_id=dataset_id, task_id=task_id
                    )

nua/e2e/regional/test_da_tasks.py:581: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
nua/e2e/regional/test_da_tasks.py:138: in stop_task
    resp = await client.post(f"/api/v1/dataset/{dataset_id}/task/{task_id}/stop")
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client.py:730: in _request
    await resp.start(conn)
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1054: in start
    with self._timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.helpers.TimerContext object at 0x7f35f185f570>
exc_type = <class 'asyncio.exceptions.CancelledError'>
exc_val = CancelledError(), exc_tb = <traceback object at 0x7f35f07878c0>

    def __exit__(
        self,
        exc_type: Optional[Type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> Optional[bool]:
        enter_task: Optional[asyncio.Task[Any]] = None
        if self._tasks:
            enter_task = self._tasks.pop()
    
        if exc_type is asyncio.CancelledError and self._cancelled:
            assert enter_task is not None
            # The timeout was hit, and the task was cancelled
            # so we need to uncancel the last task that entered the context manager
            # since the cancellation should not leak out of the context manager
            if sys.version_info >= (3, 11):
                # If the task was already cancelling don't raise
                # asyncio.TimeoutError and instead return None
                # to allow the cancellation to propagate
                if enter_task.uncancel() > self._cancelling:
                    return None
>           raise asyncio.TimeoutError from exc_val
E           TimeoutError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/helpers.py:671: TimeoutError

self = <ClientResponse(https://europe-1.stashify.cloud/api/v1/dataset/bd873f6e-6bf0-45fd-9087-422a3c749a37/task/09c167b1f2e340c687f0134dcf2b5c95/stop) [None None]>
None

connection = Connection<ConnectionKey(host='europe-1.stashify.cloud', port=443, is_ssl=True, ssl=True, proxy=None, proxy_auth=None, proxy_headers_hash=None)>

    async def start(self, connection: "Connection") -> "ClientResponse":
        """Start response processing."""
        self._closed = False
        self._protocol = connection.protocol
        self._connection = connection
    
        with self._timer:
            while True:
                # read response
                try:
                    protocol = self._protocol
>                   message, payload = await protocol.read()  # type: ignore[union-attr]

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1059: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.client_proto.ResponseHandler object at 0x7f35f06eb5f0>

    async def read(self) -> _T:
        if not self._buffer and not self._eof:
            assert not self._waiter
            self._waiter = self._loop.create_future()
            try:
>               await self._waiter
E               asyncio.exceptions.CancelledError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/streams.py:671: CancelledError

The above exception was the direct cause of the following exception:

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f184adb0>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>
tmp_nua_key = 'eyJhbGciOiJSUzI1NiIsImtpZCI6Im51YSIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJodHRwczovL2V1cm9wZS0xLnN0YXNoaWZ5LmNsb3VkLyIsImlhdCI6...rWqLrqT9yOZ4aBEVySXZE78bYVLtPSovCWJ58InA8zlLTHbcocwWcs0IMemPOD2J3DkKL_Jw5FmzNTt_d9ewro7-3IosktHnnjUxH1kjTHPHLzNfrOMZnw'
test_input = TestInput(filename='legal-text-kb.arrow', task_name=<TaskName.ASK: 'ask'>, parameters=DataAugmentation(name='e2e-test-...ure-4o-mini', provider='', keys=None, prompts=None)), validate_output=<function validate_ask_output at 0x7f35f1bf02c0>)

    @pytest.mark.asyncio_cooperative
    @pytest.mark.parametrize(
        "test_input", DA_TEST_INPUTS, ids=lambda test_input: test_input.parameters.name
    )
    async def test_da_agent_tasks(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
        tmp_nua_key: str,
        test_input: TestInput,
    ):
        dataset_id = None
        task_id = None
        start_time = asyncio.get_event_loop().time()
        try:
            nua_client_generator = aiohttp_client(
                base_url=nua_config.url, nua_key=tmp_nua_key, timeout=30
            )
            nua_client = await anext(nua_client_generator)
    
            dataset_id = await create_dataset(client=nua_client)
            await push_data_to_dataset(
                client=nua_client, dataset_id=dataset_id, filename=test_input.filename
            )
    
            task_id = await start_task(
                client=nua_client,
                dataset_id=dataset_id,
                task_name=test_input.task_name,
                parameters=test_input.parameters,
            )
            print(f"{test_input.parameters.name} task_id: {task_id}")
            task_request = await wait_for_task_completion(
                client=nua_client, dataset_id=dataset_id, task_id=task_id
            )
            assert task_request["completed"] is True
            assert task_request["failed"] is False
    
            await validate_task_output(
                client=nua_client,
                validation=test_input.validate_output,
            )
        finally:
            if dataset_id is not None:
                if task_id is not None:
>                   await stop_task(
                        client=nua_client, dataset_id=dataset_id, task_id=task_id
                    )

nua/e2e/regional/test_da_tasks.py:581: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
nua/e2e/regional/test_da_tasks.py:138: in stop_task
    resp = await client.post(f"/api/v1/dataset/{dataset_id}/task/{task_id}/stop")
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client.py:730: in _request
    await resp.start(conn)
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1054: in start
    with self._timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.helpers.TimerContext object at 0x7f35f07a42d0>
exc_type = <class 'asyncio.exceptions.CancelledError'>
exc_val = CancelledError(), exc_tb = <traceback object at 0x7f35f0179f80>

    def __exit__(
        self,
        exc_type: Optional[Type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> Optional[bool]:
        enter_task: Optional[asyncio.Task[Any]] = None
        if self._tasks:
            enter_task = self._tasks.pop()
    
        if exc_type is asyncio.CancelledError and self._cancelled:
            assert enter_task is not None
            # The timeout was hit, and the task was cancelled
            # so we need to uncancel the last task that entered the context manager
            # since the cancellation should not leak out of the context manager
            if sys.version_info >= (3, 11):
                # If the task was already cancelling don't raise
                # asyncio.TimeoutError and instead return None
                # to allow the cancellation to propagate
                if enter_task.uncancel() > self._cancelling:
                    return None
>           raise asyncio.TimeoutError from exc_val
E           TimeoutError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/helpers.py:671: TimeoutError

self = <ClientResponse(https://europe-1.stashify.cloud/api/v1/dataset/dc314c0d-be40-45e8-8fa6-79c58e4572da/task/aec868a5d5d949818bc7d035eaea5518/stop) [None None]>
None

connection = Connection<ConnectionKey(host='europe-1.stashify.cloud', port=443, is_ssl=True, ssl=True, proxy=None, proxy_auth=None, proxy_headers_hash=None)>

    async def start(self, connection: "Connection") -> "ClientResponse":
        """Start response processing."""
        self._closed = False
        self._protocol = connection.protocol
        self._connection = connection
    
        with self._timer:
            while True:
                # read response
                try:
                    protocol = self._protocol
>                   message, payload = await protocol.read()  # type: ignore[union-attr]

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1059: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.client_proto.ResponseHandler object at 0x7f35f03b9010>

    async def read(self) -> _T:
        if not self._buffer and not self._eof:
            assert not self._waiter
            self._waiter = self._loop.create_future()
            try:
>               await self._waiter
E               asyncio.exceptions.CancelledError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/streams.py:671: CancelledError

The above exception was the direct cause of the following exception:

nua_config = <nuclia.lib.nua.AsyncNuaClient object at 0x7f35f184f200>
aiohttp_client = <function aiohttp_client.<locals>.create_aiohttp_client at 0x7f35f181e520>
tmp_nua_key = 'eyJhbGciOiJSUzI1NiIsImtpZCI6Im51YSIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJodHRwczovL2V1cm9wZS0xLnN0YXNoaWZ5LmNsb3VkLyIsImlhdCI6...PhtmYbmi0-zGNpdmw0lmf09KAVF71JwhhcIweEeyRfWlwqbPYDJw_Z35WGh26QGOjnvDfRpFd4hl5f12BfGjtgpIZl980sXAZ6-6ulEdB8mEY0cTmyP9Cw'
test_input = TestInput(filename='toxic-kb.arrow', task_name=<TaskName.LLAMA_GUARD: 'llama-guard'>, parameters=DataAugmentation(name...der='', keys=None, prompts=None)), validate_output=<function validate_llama_guard_output_text_block at 0x7f35f1bf0220>)

    @pytest.mark.asyncio_cooperative
    @pytest.mark.parametrize(
        "test_input", DA_TEST_INPUTS, ids=lambda test_input: test_input.parameters.name
    )
    async def test_da_agent_tasks(
        nua_config: AsyncNuaClient,
        aiohttp_client: AsyncGenerator[aiohttp.ClientSession, None],
        tmp_nua_key: str,
        test_input: TestInput,
    ):
        dataset_id = None
        task_id = None
        start_time = asyncio.get_event_loop().time()
        try:
            nua_client_generator = aiohttp_client(
                base_url=nua_config.url, nua_key=tmp_nua_key, timeout=30
            )
            nua_client = await anext(nua_client_generator)
    
            dataset_id = await create_dataset(client=nua_client)
            await push_data_to_dataset(
                client=nua_client, dataset_id=dataset_id, filename=test_input.filename
            )
    
            task_id = await start_task(
                client=nua_client,
                dataset_id=dataset_id,
                task_name=test_input.task_name,
                parameters=test_input.parameters,
            )
            print(f"{test_input.parameters.name} task_id: {task_id}")
            task_request = await wait_for_task_completion(
                client=nua_client, dataset_id=dataset_id, task_id=task_id
            )
            assert task_request["completed"] is True
            assert task_request["failed"] is False
    
            await validate_task_output(
                client=nua_client,
                validation=test_input.validate_output,
            )
        finally:
            if dataset_id is not None:
                if task_id is not None:
>                   await stop_task(
                        client=nua_client, dataset_id=dataset_id, task_id=task_id
                    )

nua/e2e/regional/test_da_tasks.py:581: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
nua/e2e/regional/test_da_tasks.py:138: in stop_task
    resp = await client.post(f"/api/v1/dataset/{dataset_id}/task/{task_id}/stop")
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client.py:730: in _request
    await resp.start(conn)
/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/client_reqrep.py:1054: in start
    with self._timer:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <aiohttp.helpers.TimerContext object at 0x7f35f041fd90>
exc_type = <class 'asyncio.exceptions.CancelledError'>
exc_val = CancelledError(), exc_tb = <traceback object at 0x7f35f1a07c00>

    def __exit__(
        self,
        exc_type: Optional[Type[BaseException]],
        exc_val: Optional[BaseException],
        exc_tb: Optional[TracebackType],
    ) -> Optional[bool]:
        enter_task: Optional[asyncio.Task[Any]] = None
        if self._tasks:
            enter_task = self._tasks.pop()
    
        if exc_type is asyncio.CancelledError and self._cancelled:
            assert enter_task is not None
            # The timeout was hit, and the task was cancelled
            # so we need to uncancel the last task that entered the context manager
            # since the cancellation should not leak out of the context manager
            if sys.version_info >= (3, 11):
                # If the task was already cancelling don't raise
                # asyncio.TimeoutError and instead return None
                # to allow the cancellation to propagate
                if enter_task.uncancel() > self._cancelling:
                    return None
>           raise asyncio.TimeoutError from exc_val
E           TimeoutError

/opt/hostedtoolcache/Python/3.12.8/x64/lib/python3.12/site-packages/aiohttp/helpers.py:671: TimeoutError