From b5fdc156b15ad727c0106e71a383a727e57f899e Mon Sep 17 00:00:00 2001
From: 苏向夜 <fu050409@163.com>
Date: Fri, 26 Jan 2024 23:27:48 +0800
Subject: :bug: fix(interceptor): fix output interception have no input
 defination

---
 src/infini/interceptor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/infini/interceptor.py b/src/infini/interceptor.py
index c227fef7..28a3f869 100644
--- a/src/infini/interceptor.py
+++ b/src/infini/interceptor.py
@@ -17,7 +17,8 @@ class Interceptor:
         return input
 
     def output(self, output_text: str) -> str | Output:
-        queue = self.match(output_text)
+        queue = self.match(output_text)  # TODO 需要测试输出拦截情况
+        input = Input(output_text)
         while not queue.is_empty():
             if isinstance(intercepted := queue.pop()(input), Output):
                 return intercepted
-- 
cgit v1.2.3-70-g09d2