Minor UX improvements

src/dispatch/fastapi.py

@@ -120,11 +120,6 @@ def __init__(
         function_service = _new_app(self, verification_key)
         app.mount("/dispatch.sdk.v1.FunctionService", function_service)
 
-    def batch(self) -> Batch:
-        """Returns a Batch instance that can be used to build
-        a set of calls to dispatch."""
-        return self.client.batch()
-
 
 def parse_verification_key(
     verification_key: Ed25519PublicKey | str | bytes | None,

src/dispatch/function.py

@@ -250,6 +250,11 @@ def set_client(self, client: Client):
         for fn in self.functions.values():
             fn._client = client
 
+    def batch(self) -> Batch:
+        """Returns a Batch instance that can be used to build
+        a set of calls to dispatch."""
+        return self.client.batch()
+
 
 class Client:
     """Client for the Dispatch API."""

src/dispatch/test/__main__.py

@@ -85,7 +85,11 @@ def main():
             print(f'  export DISPATCH_VERIFICATION_KEY="{verification_key}"')
             print()
 
-            server.wait()
+            try:
+                server.wait()
+            except KeyboardInterrupt:
+                print("closing down server after interrupt")
+                pass
 
 
 if __name__ == "__main__":