Question about limitations of task function argument type.

[wyl1232792]

Is there any limit that task function can take in prefect?

@task
async def foo(arg1: int, arg2: AnyAllowedType):
    pass

I'm working on prefect workflow, of which tasks often make requests through sqlConnection and natsClient.

My pattern is like below.

@task
async def foo(nats_client: NatsClient, sql_connection: SqlConnection):
    # make some requests and waits for imcoming message
    pass

@flow
async def bar():
    nats_client = await create_nc()
    sql_connection = await create_sql()

    # do something before
    await foo(nats_client, sql_connection)
    # recycle connections
    await nats_client.drain()
    await sql_connection.close()

Though I found very strange behaviours like request timeout, no connections etc, while those function works fine in normal runtime.
I understand there might be some complicated runner which makes connection pool unusable.
So should I only pass basic, stateless arguments into task functions? Is there any apis that enables sharing connection pools among task runs?