:sparkles: feat(generator): add generator method

2 files changed, 10 insertions, 2 deletions

diff --git a/src/infini/core.py b/src/infini/core.py
index 2bb3e7da..4eaeb4f8 100644
--- a/src/infini/core.py
+++ b/src/infini/core.py
@@ -29,7 +29,7 @@ class Core:
             yield output
 
     def generate(self, output: Output) -> str:
-        return output.name  # TODO 生成器实现
+        return self.generator.output(output)
 
     def intercept(self, output: str) -> str:
         return (
diff --git a/src/infini/generator.py b/src/infini/generator.py
index a4d60e27..81f5a69c 100644
--- a/src/infini/generator.py
+++ b/src/infini/generator.py
@@ -1,2 +1,10 @@
+from infini.output import Output
+from infini.typing import Dict, Callable
+
+
 class Generator:
-    ...
+    events: Dict[str, str]
+    global_variables: Dict[str, str | Callable]
+
+    def output(self, output: Output) -> str:
+        return output.name