diff options
| -rw-r--r-- | src/infini/core.py | 2 | ||||
| -rw-r--r-- | src/infini/loader.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/src/infini/core.py b/src/infini/core.py index 0de6034c..5694caff 100644 --- a/src/infini/core.py +++ b/src/infini/core.py @@ -1,3 +1,4 @@ +from infini.doc import Doc from infini.input import Input from infini.interceptor import Interceptor from infini.generator import Generator @@ -14,6 +15,7 @@ class Core: generator: Generator interceptor: Interceptor injector: Injector + doc: Doc def input(self, input: Input) -> GeneratorT[Union[str, Output], Any, None]: for pre_intercepted_stream in self.pre_intercept(input): diff --git a/src/infini/loader.py b/src/infini/loader.py index c57e8de9..13c26fdb 100644 --- a/src/infini/loader.py +++ b/src/infini/loader.py @@ -200,6 +200,8 @@ class Loader: self._core.interceptor = interceptor self._core.injector = injector + self._core.doc = self.doc + return self._core def inject_register(self, register: Register): |