[SHIP-768] Receive headers in invocations from proxy (#507)

Receive headers in the invocation context from the proxy.

Depends on https://github.com/nludb/nludb/pull/603

Co-authored-by: Ted Benson <edward.benson@gmail.com>

Author: dkolas

src/steamship/invocable/invocable_request.py

@@ -30,6 +30,7 @@ class InvocationContext(CamelModel):
     invocable_url: str = None
     invocable_owner_handle: Optional[str] = None
     workspace_handle: Optional[str] = None
+    headers: Dict[str, str] = None
 
 
 class InvocableRequest(CamelModel):

tests/assets/packages/echo_test_header.py

@@ -0,0 +1,9 @@
+import json
+
+from steamship.invocable import PackageService, get
+
+
+class EchoTestHeader(PackageService):
+    @get("echo_test_header")
+    def echo_test_header(self) -> str:
+        return json.dumps(self.context.headers)

tests/steamship_tests/app/integration/test_header_passing.py

@@ -0,0 +1,28 @@
+import json
+
+import pytest
+import requests
+from steamship_tests import PACKAGES_PATH
+from steamship_tests.utils.deployables import deploy_package
+
+from steamship import Steamship
+
+
+@pytest.mark.usefixtures("client")
+def test_instance_init_is_called_on_package(client: Steamship):
+    demo_package_path = PACKAGES_PATH / "echo_test_header.py"
+
+    with deploy_package(client, demo_package_path) as (_, _, instance):
+        url = instance.invocation_url + "/echo_test_header"
+        response = requests.get(
+            url,
+            headers={
+                "authorization": f"bearer {client.config.api_key.get_secret_value()}",
+                "test1": "testValue1",
+                "test2": "testValue2",
+            },
+        )
+        result = json.loads(response.text)
+        assert result is not None
+        assert result["test1"] == "testValue1"
+        assert result["test2"] == "testValue2"