From c89694f25be9ea966c4e7eedf9dfb6de8f72dbef Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Fri, 26 Jan 2024 17:51:14 +0800
Subject: :sparkles: feat(generator): add generator method

---
 src/infini/core.py      |  2 +-
 src/infini/generator.py | 10 +++++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

(limited to 'src')

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
-- 
cgit v1.2.3-70-g09d2