From adf40d1b16aa25594e965ab06b4f659f662d7bb3 Mon Sep 17 00:00:00 2001
From: Eva Lott <evalott100@gmail.com>
Date: Mon, 7 Oct 2024 08:49:31 +0100
Subject: [PATCH] exposed all documents in `event_model` namespace

---
 docs/user/reference/release-history.rst |  1 +
 event_model/__init__.py                 | 37 +++++++++++++++++++++----
 2 files changed, 33 insertions(+), 5 deletions(-)

diff --git a/docs/user/reference/release-history.rst b/docs/user/reference/release-history.rst
index cbf9d4c9..d4a85494 100644
--- a/docs/user/reference/release-history.rst
+++ b/docs/user/reference/release-history.rst
@@ -12,6 +12,7 @@ Changed
 * Added a `Limits` type for storing EPICS limit data and a `NotRequired[Limits]` `limits` field to `DataKey`.
 * Added a `NotRequired` `dtype_numpy` field to `DataKey` for specifying the numpy dtype of data.
 * Bumped from `jsonschema 3.*` to `jsonschema 4.*`.
+* Added all documents to `event_model` namespace.
 
 v1.21.0 (2024-05-21)
 ====================
diff --git a/event_model/__init__.py b/event_model/__init__.py
index 784e33ee..ac3c466a 100644
--- a/event_model/__init__.py
+++ b/event_model/__init__.py
@@ -34,19 +34,21 @@
 import numpy
 from typing_extensions import Literal
 
-from .documents import Dtype
 from .documents.datum import Datum
 from .documents.datum_page import DatumPage
-from .documents.event import Event
+from .documents.event import Event, PartialEvent
 from .documents.event_descriptor import (
     Configuration,
     DataKey,
+    Dtype,
     EventDescriptor,
+    Limits,
+    LimitsRange,
     PerObjectHint,
 )
-from .documents.event_page import EventPage
-from .documents.resource import Resource
-from .documents.run_start import RunStart
+from .documents.event_page import EventPage, PartialEventPage
+from .documents.resource import PartialResource, Resource
+from .documents.run_start import Calculation, Hints, Projection, Projections, RunStart
 from .documents.run_stop import RunStop
 from .documents.stream_datum import StreamDatum, StreamRange
 from .documents.stream_resource import StreamResource
@@ -62,7 +64,32 @@
 
 
 __all__ = [
+    # Document types
+    "Datum",
+    "DatumPage",
+    "Event",
+    "PartialEvent",
+    "Configuration",
+    "DataKey",
     "Dtype",
+    "EventDescriptor",
+    "Limits",
+    "LimitsRange",
+    "PerObjectHint",
+    "EventPage",
+    "PartialEventPage",
+    "PartialResource",
+    "Resource",
+    "Calculation",
+    "Hints",
+    "Projection",
+    "Projections",
+    "RunStart",
+    "RunStop",
+    "StreamDatum",
+    "StreamRange",
+    "StreamResource",
+    # Schema and version
     "DocumentNames",
     "schemas",
     "schema_validators",