fix: make shipped module source inspectable for nested remote calls

Author: KAJdev

runpod/runtimes/task/runner.py

@@ -150,9 +150,13 @@ def execute_request(request):
         # exec mirrors deployed-mode module import: the code is the
         # user's full module, so __name__ is set like an import would
         # (main guards stay inert) and decorated handles resolve to
-        # their wrapped functions
-        namespace = {"__name__": "__runpod_live__"}
-        exec(function_code, namespace)  # noqa: S102 - that is the job
+        # their wrapped functions. the source is written to a real
+        # file first so inspect.getsource works inside the function
+        # (nested .remote() calls re-extract sibling source)
+        source_path = _materialize_source(function_code)
+        namespace = {"__name__": "__runpod_live__", "__file__": source_path}
+        code_obj = compile(function_code, source_path, "exec")
+        exec(code_obj, namespace)  # noqa: S102 - that is the job
         if function_name not in namespace:
             return {
                 "success": False,
@@ -188,6 +192,21 @@ def execute_request(request):
         }
 
 
+def _materialize_source(function_code):
+    """persist request source to a stable path for inspect/linecache."""
+    import hashlib
+    import tempfile
+
+    digest = hashlib.sha256(function_code.encode()).hexdigest()[:16]
+    path = os.path.join(
+        tempfile.gettempdir(), f"runpod_live_{digest}.py"
+    )
+    if not os.path.exists(path):
+        with open(path, "w") as f:
+            f.write(function_code)
+    return path
+
+
 class _Tee(io.TextIOBase):
     """write-through to multiple streams, flushing eagerly so log
     followers see lines as they happen."""

tests/test_apps/test_dispatch.py

@@ -304,3 +304,33 @@ def test_execute_request_unwraps_handles(self):
         )
         assert response["success"], response.get("error")
         assert response["json_result"] == "AAAAAA"
+
+    def test_nested_source_extraction_from_shipped_module(self):
+        # inside a live worker, a function calling sibling.remote()
+        # re-extracts that sibling's source via inspect; the shipped
+        # module must be inspectable after exec
+        from runpod.runtimes.task.runner import execute_request
+
+        code = (
+            "import runpod\n"
+            "app = runpod.App('t2')\n"
+            "@app.queue()\n"
+            "def sibling(x):\n"
+            "    return x\n"
+            "@app.queue()\n"
+            "def caller():\n"
+            "    import inspect\n"
+            "    from runpod.apps.serialization import get_function_source\n"
+            "    return get_function_source(sibling._fn)\n"
+        )
+        response = execute_request(
+            {
+                "function_name": "caller",
+                "function_code": code,
+                "args": [],
+                "kwargs": {},
+                "serialization_format": "json",
+            }
+        )
+        assert response["success"], response.get("error")
+        assert "def sibling" in response["json_result"]