diff --git a/stubs/google-cloud-ndb/METADATA.toml b/stubs/google-cloud-ndb/METADATA.toml
index 2337e58a1..715653337 100644
--- a/stubs/google-cloud-ndb/METADATA.toml
+++ b/stubs/google-cloud-ndb/METADATA.toml
@@ -1,4 +1,4 @@
-version = "2.1.*"
+version = "2.2.*"
 upstream_repository = "https://github.com/googleapis/python-ndb"
 partial_stub = true
 
diff --git a/stubs/google-cloud-ndb/google/cloud/ndb/client.pyi b/stubs/google-cloud-ndb/google/cloud/ndb/client.pyi
index 09812a7ff..8b7e65d98 100644
--- a/stubs/google-cloud-ndb/google/cloud/ndb/client.pyi
+++ b/stubs/google-cloud-ndb/google/cloud/ndb/client.pyi
@@ -14,12 +14,14 @@ class Client:
     client_info: Incomplete
     secure: bool
     stub: Incomplete
+    database: str | None
     def __init__(
         self,
         project: str | None = ...,
         namespace: str | None = ...,
         credentials: Incomplete | None = ...,
         client_options: Incomplete | None = ...,
+        database: str | None = None,
     ) -> None: ...
     @contextmanager
     def context(
diff --git a/stubs/google-cloud-ndb/google/cloud/ndb/key.pyi b/stubs/google-cloud-ndb/google/cloud/ndb/key.pyi
index 5139ae1b7..292e5dd9d 100644
--- a/stubs/google-cloud-ndb/google/cloud/ndb/key.pyi
+++ b/stubs/google-cloud-ndb/google/cloud/ndb/key.pyi
@@ -18,6 +18,7 @@ class Key:
     def namespace(self): ...
     def project(self): ...
     app: Any
+    def database(self) -> str | None: ...
     def id(self): ...
     def string_id(self): ...
     def integer_id(self): ...
diff --git a/stubs/google-cloud-ndb/google/cloud/ndb/model.pyi b/stubs/google-cloud-ndb/google/cloud/ndb/model.pyi
index 1d272f020..38b2eeeca 100644
--- a/stubs/google-cloud-ndb/google/cloud/ndb/model.pyi
+++ b/stubs/google-cloud-ndb/google/cloud/ndb/model.pyi
@@ -320,6 +320,7 @@ class Model(_NotEqualMixin, metaclass=MetaModel):
         max_memcache_items: int | None = ...,
         force_writes: bool | None = ...,
         _options=...,
+        database: str | None = None,
     ) -> Model | None: ...
     @classmethod
     def get_by_id_async(
@@ -344,6 +345,7 @@ class Model(_NotEqualMixin, metaclass=MetaModel):
         max_memcache_items: int | None = ...,
         force_writes: bool | None = ...,
         _options=...,
+        database: str | None = None,
     ) -> tasklets_module.Future: ...
     @classmethod
     def get_or_insert(
diff --git a/stubs/google-cloud-ndb/google/cloud/ndb/query.pyi b/stubs/google-cloud-ndb/google/cloud/ndb/query.pyi
index 12da5f4f1..74748626c 100644
--- a/stubs/google-cloud-ndb/google/cloud/ndb/query.pyi
+++ b/stubs/google-cloud-ndb/google/cloud/ndb/query.pyi
@@ -91,6 +91,7 @@ OR = DisjunctionNode
 class QueryOptions(_options.ReadOptions):
     project: Any
     namespace: Any
+    database: str | None
     def __init__(self, config: Incomplete | None = ..., context: Incomplete | None = ..., **kwargs) -> None: ...
 
 class Query:
@@ -106,6 +107,7 @@ class Query:
     keys_only: Any
     projection: Any
     distinct_on: Any
+    database: str | None
     def __init__(
         self,
         kind: Incomplete | None = ...,