chore: remove aiostream dependency (jina-ai#5891)

Signed-off-by: Joan Fontanals Martinez <[email protected]>

Author: JoanFM

extra-requirements.txt

@@ -37,7 +37,6 @@ jina-hubble-sdk>=0.30.4:    core
 jcloud>=0.0.35:             core
 opentelemetry-api>=1.12.0:  core
 opentelemetry-instrumentation-grpc>=0.35b0:  core 
-aiostream:  core
 uvloop:                     perf,standard,devel
 prometheus_client>=0.12.0:          perf,standard,devel
 opentelemetry-sdk>=1.14.0:   perf,standard,devel

jina/clients/request/asyncio.py

@@ -41,10 +41,19 @@ async def request_generator(
                 endpoint=exec_endpoint, target=target_executor, parameters=parameters
             )
         else:
-            with ImportExtensions(required=True):
-                import aiostream
-
-            async for batch in aiostream.stream.chunks(data, request_size):
+            batch = []
+            async for d in data:
+                batch.append(d)
+                if len(batch) >= request_size:
+                    yield _new_data_request_from_batch(
+                        batch=batch,
+                        data_type=data_type,
+                        endpoint=exec_endpoint,
+                        target=target_executor,
+                        parameters=parameters,
+                    )
+                    batch = []
+            if len(batch) > 0:
                 yield _new_data_request_from_batch(
                     batch=batch,
                     data_type=data_type,

jina/resources/extra-requirements.txt

@@ -37,7 +37,6 @@ jina-hubble-sdk>=0.30.4:    core
 jcloud>=0.0.35:             core
 opentelemetry-api>=1.12.0:  core
 opentelemetry-instrumentation-grpc>=0.35b0:  core 
-aiostream:  core
 uvloop:                     perf,standard,devel
 prometheus_client>=0.12.0:          perf,standard,devel
 opentelemetry-sdk>=1.14.0:   perf,standard,devel

jina/serve/stream/__init__.py

@@ -10,8 +10,6 @@
     Union,
 )
 
-from aiostream.aiter_utils import anext
-
 from jina.excepts import InternalNetworkError
 from jina.logging.logger import JinaLogger
 from jina.serve.stream.helper import AsyncRequestsIterator, _RequestsCounter
@@ -281,4 +279,4 @@ async def process_single_data(
         :param context: grpc context
         :return: response DataRequest
         """
-        return await anext(self.stream(iter([request]), context=context))
+        return await self.stream(iter([request]), context=context).__anext__()