chalk-remote-call-python 0.0.0__tar.gz → 1.1.2__tar.gz

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalk-remote-call-python
-Version: 0.0.0
+Version: 1.1.2
 Summary: Chalk remote call Python runtime interface client
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai
@@ -65,15 +65,18 @@ def handler(event: dict[str, pa.Array], context: dict) -> pa.Array:
 
     Args:
         event: dict mapping column names to pyarrow.Array values.
-        context: dict with request metadata:
-            - "peer": remote address string
-            - "metadata": dict of gRPC headers
+        context: dict with request metadata
 
     Returns:
         A pyarrow.Array, pyarrow.RecordBatch, pyarrow.Table, list, dict, or scalar.
         The framework auto-converts the result to Arrow IPC for the response.
     """
     return pc.multiply(event["x"], event["y"])
+
+
+def on_shutdown():
+    """Optional -- runs once after the server stops accepting requests."""
+    print("Releasing resources...")
 ```
 
 ### 2. Start the server
@@ -97,6 +100,7 @@ python -m chalk_remote_call --handler my_handler.handler
 | `--host` | `CHALK_REMOTE_CALL_HOST` | `[::]` | Host to bind to |
 | `--workers` | `CHALK_REMOTE_CALL_WORKERS` | `10` | Tokio runtime worker threads |
 | `--on-startup` | | | Dotted path to a startup function |
+| `--on-shutdown` | | | Dotted path to a shutdown function |
 | `--log-level` | | `INFO` | `DEBUG`, `INFO`, `WARNING`, or `ERROR` |
 
 ### Environment variables
@@ -115,12 +119,12 @@ WORKDIR /app
 RUN pip install chalk-remote-call-python
 
 # Copy your handler code
-COPY my_handler.py .
+COPY my_handler.py /app
 
-# The server listens on port 6666 by default
 EXPOSE 6666
 
-ENTRYPOINT ["chalk-remote-call", "--handler", "my_handler.handler"]
+ENTRYPOINT ["chalk-remote-call"]
+CMD ["--handler", "handler.handler"]
 ```
 
 Build and run:

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/README.md

@@ -32,15 +32,18 @@ def handler(event: dict[str, pa.Array], context: dict) -> pa.Array:
 
     Args:
         event: dict mapping column names to pyarrow.Array values.
-        context: dict with request metadata:
-            - "peer": remote address string
-            - "metadata": dict of gRPC headers
+        context: dict with request metadata
 
     Returns:
         A pyarrow.Array, pyarrow.RecordBatch, pyarrow.Table, list, dict, or scalar.
         The framework auto-converts the result to Arrow IPC for the response.
     """
     return pc.multiply(event["x"], event["y"])
+
+
+def on_shutdown():
+    """Optional -- runs once after the server stops accepting requests."""
+    print("Releasing resources...")
 ```
 
 ### 2. Start the server
@@ -64,6 +67,7 @@ python -m chalk_remote_call --handler my_handler.handler
 | `--host` | `CHALK_REMOTE_CALL_HOST` | `[::]` | Host to bind to |
 | `--workers` | `CHALK_REMOTE_CALL_WORKERS` | `10` | Tokio runtime worker threads |
 | `--on-startup` | | | Dotted path to a startup function |
+| `--on-shutdown` | | | Dotted path to a shutdown function |
 | `--log-level` | | `INFO` | `DEBUG`, `INFO`, `WARNING`, or `ERROR` |
 
 ### Environment variables
@@ -82,12 +86,12 @@ WORKDIR /app
 RUN pip install chalk-remote-call-python
 
 # Copy your handler code
-COPY my_handler.py .
+COPY my_handler.py /app
 
-# The server listens on port 6666 by default
 EXPOSE 6666
 
-ENTRYPOINT ["chalk-remote-call", "--handler", "my_handler.handler"]
+ENTRYPOINT ["chalk-remote-call"]
+CMD ["--handler", "handler.handler"]
 ```
 
 Build and run:

chalk_remote_call_python-1.1.2/chalk_remote_call/__init__.py

@@ -0,0 +1,4 @@
+from chalk_remote_call._version import __version__
+from chalk_remote_call.server import serve
+
+__all__ = ["__version__", "serve"]

chalk_remote_call_python-1.1.2/chalk_remote_call/_version.py

@@ -0,0 +1 @@
+__version__ = "1.1.2"

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/chalk_remote_call/cli.py

@@ -5,7 +5,7 @@ import logging
 import os
 import sys
 
-from chalk_remote_call.handler_loader import load_handler, load_startup_function
+from chalk_remote_call.handler_loader import load_function, load_handler
 from chalk_remote_call.input_transform import parse_input_args
 from chalk_remote_call.server import serve
 
@@ -42,6 +42,11 @@ def main() -> None:
         default=None,
         help="Dotted path to a startup function (e.g. my_module.setup)",
     )
+    parser.add_argument(
+        "--on-shutdown",
+        default=None,
+        help="Dotted path to a shutdown function (e.g. my_module.cleanup)",
+    )
     parser.add_argument(
         "--log-level",
         default="INFO",
@@ -57,7 +62,7 @@ def main() -> None:
 
     # Load handler
     try:
-        handler, auto_startup = load_handler(args.handler)
+        handler, auto_startup, auto_shutdown = load_handler(args.handler)
     except Exception as e:
         print(f"Error loading handler: {e}", file=sys.stderr)
         sys.exit(1)
@@ -66,13 +71,24 @@ def main() -> None:
     on_startup = None
     if args.on_startup:
         try:
-            on_startup = load_startup_function(args.on_startup)
+            on_startup = load_function(args.on_startup, label="Startup")
         except Exception as e:
             print(f"Error loading startup function: {e}", file=sys.stderr)
             sys.exit(1)
     elif auto_startup is not None:
         on_startup = auto_startup
 
+    # Determine shutdown function
+    on_shutdown = None
+    if args.on_shutdown:
+        try:
+            on_shutdown = load_function(args.on_shutdown, label="Shutdown")
+        except Exception as e:
+            print(f"Error loading shutdown function: {e}", file=sys.stderr)
+            sys.exit(1)
+    elif auto_shutdown is not None:
+        on_shutdown = auto_shutdown
+
     # Parse input args
     arg_names = parse_input_args()
 
@@ -82,5 +98,6 @@ def main() -> None:
         port=args.port,
         workers=args.workers,
         on_startup=on_startup,
+        on_shutdown=on_shutdown,
         arg_names=arg_names,
     )

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/chalk_remote_call/handler_loader.py

@@ -5,12 +5,14 @@ from collections.abc import Callable
 from typing import Any
 
 
-def load_handler(dotted_path: str) -> tuple[Callable[..., Any], Callable[[], None] | None]:
+def load_handler(
+    dotted_path: str,
+) -> tuple[Callable[..., Any], Callable[[], None] | None, Callable[[], None] | None]:
     """Load a handler function from a dotted module path.
 
-    Returns (handler_fn, on_startup_fn_or_None).
-    The on_startup function is auto-discovered if the handler's module defines
-    a top-level `on_startup` callable.
+    Returns (handler_fn, on_startup_fn_or_None, on_shutdown_fn_or_None).
+    The on_startup/on_shutdown functions are auto-discovered if the handler's
+    module defines top-level callables with those names.
     """
     if "." not in dotted_path:
         raise ValueError(f"Handler path must be a dotted path like 'my_module.handler', got: {dotted_path!r}")
@@ -27,13 +29,22 @@ def load_handler(dotted_path: str) -> tuple[Callable[..., Any], Callable[[], Non
     if on_startup is not None and not callable(on_startup):
         on_startup = None
 
-    return handler, on_startup
+    on_shutdown = getattr(module, "on_shutdown", None)
+    if on_shutdown is not None and not callable(on_shutdown):
+        on_shutdown = None
 
+    return handler, on_startup, on_shutdown
 
-def load_startup_function(dotted_path: str) -> Callable[[], None]:
-    """Load a startup function from a dotted module path."""
+
+def load_function(dotted_path: str, label: str = "Function") -> Callable[[], None]:
+    """Load a callable from a dotted module path.
+
+    Args:
+        dotted_path: e.g. 'my_module.setup'
+        label: Human-readable label for error messages (e.g. "Startup", "Shutdown").
+    """
     if "." not in dotted_path:
-        raise ValueError(f"Startup path must be a dotted path like 'my_module.setup', got: {dotted_path!r}")
+        raise ValueError(f"{label} path must be a dotted path like 'my_module.func', got: {dotted_path!r}")
 
     module_path, func_name = dotted_path.rsplit(".", 1)
     module = importlib.import_module(module_path)

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/chalk_remote_call/server.py

@@ -16,6 +16,7 @@ def serve(
     port: int = 6666,
     workers: int = 10,
     on_startup: Callable[[], None] | None = None,
+    on_shutdown: Callable[[], None] | None = None,
     arg_names: list[str] | None = None,
 ) -> None:
     """Start the gRPC server implementing RemoteCallService.
@@ -26,6 +27,7 @@ def serve(
         port: The port to bind to.
         workers: Number of worker threads for the tokio runtime.
         on_startup: Optional startup hook called before the server starts accepting requests.
+        on_shutdown: Optional shutdown hook called after the server stops accepting requests.
         arg_names: Parsed CHALK_INPUT_ARGS column names, or None.
     """
     # Run startup hook before starting the Rust server
@@ -34,11 +36,16 @@ def serve(
         on_startup()
 
     logger.info("Starting server on %s:%d", host, port)
-    start_server(
-        handler=handler,
-        process_fn=process_batches,
-        host=host,
-        port=port,
-        workers=workers,
-        arg_names=arg_names,
-    )
+    try:
+        start_server(
+            handler=handler,
+            process_fn=process_batches,
+            host=host,
+            port=port,
+            workers=workers,
+            arg_names=arg_names,
+        )
+    finally:
+        if on_shutdown is not None:
+            logger.info("Running shutdown hook...")
+            on_shutdown()

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/chalk_remote_call_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalk-remote-call-python
-Version: 0.0.0
+Version: 1.1.2
 Summary: Chalk remote call Python runtime interface client
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai
@@ -65,15 +65,18 @@ def handler(event: dict[str, pa.Array], context: dict) -> pa.Array:
 
     Args:
         event: dict mapping column names to pyarrow.Array values.
-        context: dict with request metadata:
-            - "peer": remote address string
-            - "metadata": dict of gRPC headers
+        context: dict with request metadata
 
     Returns:
         A pyarrow.Array, pyarrow.RecordBatch, pyarrow.Table, list, dict, or scalar.
         The framework auto-converts the result to Arrow IPC for the response.
     """
     return pc.multiply(event["x"], event["y"])
+
+
+def on_shutdown():
+    """Optional -- runs once after the server stops accepting requests."""
+    print("Releasing resources...")
 ```
 
 ### 2. Start the server
@@ -97,6 +100,7 @@ python -m chalk_remote_call --handler my_handler.handler
 | `--host` | `CHALK_REMOTE_CALL_HOST` | `[::]` | Host to bind to |
 | `--workers` | `CHALK_REMOTE_CALL_WORKERS` | `10` | Tokio runtime worker threads |
 | `--on-startup` | | | Dotted path to a startup function |
+| `--on-shutdown` | | | Dotted path to a shutdown function |
 | `--log-level` | | `INFO` | `DEBUG`, `INFO`, `WARNING`, or `ERROR` |
 
 ### Environment variables
@@ -115,12 +119,12 @@ WORKDIR /app
 RUN pip install chalk-remote-call-python
 
 # Copy your handler code
-COPY my_handler.py .
+COPY my_handler.py /app
 
-# The server listens on port 6666 by default
 EXPOSE 6666
 
-ENTRYPOINT ["chalk-remote-call", "--handler", "my_handler.handler"]
+ENTRYPOINT ["chalk-remote-call"]
+CMD ["--handler", "handler.handler"]
 ```
 
 Build and run:

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/chalk_remote_call_python.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ setup.py
 chalk_remote_call/__init__.py
 chalk_remote_call/__main__.py
 chalk_remote_call/_native.pyi
+chalk_remote_call/_version.py
 chalk_remote_call/arrow_utils.py
 chalk_remote_call/cli.py
 chalk_remote_call/handler_loader.py
@@ -25,5 +26,6 @@ chalk_remote_call_python.egg-info/top_level.txt
 tests/test_arrow_utils.py
 tests/test_end_to_end.py
 tests/test_error_paths.py
+tests/test_handler_loader.py
 tests/test_input_transform.py
 tests/test_servicer.py

{chalk_remote_call_python-0.0.0 → chalk_remote_call_python-1.1.2}/pyproject.toml

@@ -52,6 +52,9 @@ Changelog = "https://docs.chalk.ai/docs/changelog"
 [project.scripts]
 chalk-remote-call = "chalk_remote_call.cli:main"
 
+[tool.setuptools.dynamic]
+version = {attr = "chalk_remote_call._version.__version__"}
+
 [tool.setuptools.packages.find]
 where = ["."]
 include = ["chalk_remote_call*"]

chalk_remote_call_python-1.1.2/tests/test_handler_loader.py

@@ -0,0 +1,99 @@
+"""Tests for handler_loader: load_handler and load_function."""
+
+from __future__ import annotations
+
+import types
+
+import pytest
+
+from chalk_remote_call.handler_loader import load_function, load_handler
+
+
+def _make_module(name: str, **attrs: object) -> types.ModuleType:
+    """Create a throwaway module with the given attributes."""
+    mod = types.ModuleType(name)
+    for k, v in attrs.items():
+        setattr(mod, k, v)
+    return mod
+
+
+# ---------------------------------------------------------------------------
+# load_function
+# ---------------------------------------------------------------------------
+
+
+class TestLoadFunction:
+    def test_missing_dot_raises(self):
+        with pytest.raises(ValueError, match="dotted path"):
+            load_function("nodot", label="Test")
+
+    def test_nonexistent_module_raises(self):
+        with pytest.raises(ModuleNotFoundError):
+            load_function("nonexistent_module_xyz.func", label="Test")
+
+    def test_nonexistent_attr_raises(self):
+        with pytest.raises(AttributeError, match="no attribute"):
+            load_function("os.nonexistent_func_xyz", label="Test")
+
+    def test_not_callable_raises(self):
+        with pytest.raises(TypeError, match="not callable"):
+            load_function("os.sep", label="Test")
+
+    def test_loads_callable(self):
+        fn = load_function("os.path.exists", label="Test")
+        assert callable(fn)
+
+    def test_label_appears_in_error(self):
+        with pytest.raises(ValueError, match="Shutdown"):
+            load_function("nodot", label="Shutdown")
+
+
+# ---------------------------------------------------------------------------
+# load_handler — auto-discovery of on_startup / on_shutdown
+# ---------------------------------------------------------------------------
+
+
+class TestLoadHandlerAutoDiscovery:
+    def test_discovers_on_startup(self, monkeypatch):
+        startup = lambda: None  # noqa: E731
+        mod = _make_module("fake_mod", handler=lambda e, c: e, on_startup=startup)
+        monkeypatch.setitem(__import__("sys").modules, "fake_mod", mod)
+
+        _handler, on_startup, on_shutdown = load_handler("fake_mod.handler")
+        assert on_startup is startup
+        assert on_shutdown is None
+
+    def test_discovers_on_shutdown(self, monkeypatch):
+        shutdown = lambda: None  # noqa: E731
+        mod = _make_module("fake_mod", handler=lambda e, c: e, on_shutdown=shutdown)
+        monkeypatch.setitem(__import__("sys").modules, "fake_mod", mod)
+
+        _handler, on_startup, on_shutdown = load_handler("fake_mod.handler")
+        assert on_startup is None
+        assert on_shutdown is shutdown
+
+    def test_discovers_both(self, monkeypatch):
+        startup = lambda: None  # noqa: E731
+        shutdown = lambda: None  # noqa: E731
+        mod = _make_module("fake_mod", handler=lambda e, c: e, on_startup=startup, on_shutdown=shutdown)
+        monkeypatch.setitem(__import__("sys").modules, "fake_mod", mod)
+
+        _handler, on_startup, on_shutdown = load_handler("fake_mod.handler")
+        assert on_startup is startup
+        assert on_shutdown is shutdown
+
+    def test_ignores_non_callable(self, monkeypatch):
+        mod = _make_module("fake_mod", handler=lambda e, c: e, on_startup="not callable", on_shutdown=42)
+        monkeypatch.setitem(__import__("sys").modules, "fake_mod", mod)
+
+        _handler, on_startup, on_shutdown = load_handler("fake_mod.handler")
+        assert on_startup is None
+        assert on_shutdown is None
+
+    def test_neither_defined(self, monkeypatch):
+        mod = _make_module("fake_mod", handler=lambda e, c: e)
+        monkeypatch.setitem(__import__("sys").modules, "fake_mod", mod)
+
+        _handler, on_startup, on_shutdown = load_handler("fake_mod.handler")
+        assert on_startup is None
+        assert on_shutdown is None

chalk_remote_call_python-0.0.0/chalk_remote_call/__init__.py

@@ -1,6 +0,0 @@
-from importlib.metadata import version
-
-from chalk_remote_call.server import serve
-
-__version__ = version("chalk-remote-call-python")
-__all__ = ["serve"]