feat: pass request-headers as metadata

jina/constants.py

@@ -53,6 +53,7 @@
 __args_executor_func__ = {
     'docs',
     'parameters',
+    'metadata',
     'docs_matrix',
 }
 __args_executor_init__ = {'metas', 'requests', 'runtime_args'}

jina/proto/docarray_v1/jina.proto

@@ -122,6 +122,8 @@ message DataRequestProto {
     }
 
     DataContentProto data = 4; // container for docs and groundtruths
+
+    google.protobuf.Struct metadata = 5; // extra kwargs that will be used in executor
 }
 
 
@@ -138,6 +140,8 @@ message SingleDocumentRequestProto {
 
     docarray.DocumentProto document = 4; // the document in this request
 
+    google.protobuf.Struct metadata = 5; // extra kwargs that will be used in executor
+
 }
 
 message DataRequestProtoWoData {
@@ -148,6 +152,8 @@ message DataRequestProtoWoData {
 
     repeated RouteProto routes = 3; // status info on every routes
 
+    google.protobuf.Struct metadata = 4; // extra kwargs that will be used in executor
+
 }
 
 

jina/serve/runtimes/gateway/http_fastapi_app_docarrayv2.py

@@ -187,8 +187,10 @@ def add_post_route(
         )
         app_kwargs['response_class'] = DocArrayResponse
 
+        from fastapi import Request
+
         @app.api_route(**app_kwargs)
-        async def post(body: input_model, response: Response):
+        async def post(body: input_model, response: Response, request: Request):
             target_executor = None
             req_id = None
             if body.header is not None:
@@ -208,6 +210,11 @@ async def post(body: input_model, response: Response):
                     docs,
                     exec_endpoint=endpoint_path,
                     parameters=body.parameters,
+                    metadata=dict(
+                        request.headers or {
+                            "no_headers": "true"
+                        }
+                    ),
                     target_executor=target_executor,
                     request_id=req_id,
                     return_results=True,
@@ -245,8 +252,6 @@ def add_streaming_routes(
         endpoint_path,
         input_doc_model=None,
     ):
-        from fastapi import Request
-
         @app.api_route(
             path=f'/{endpoint_path.strip("/")}',
             methods=['GET'],