blob: 2bb3e7da112c812228ec076e053c7c1a558db822 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
from infini.input import Input
from infini.interceptor import Interceptor
from infini.generator import Generator
from infini.handler import Handler
from infini.output import Output
class Core:
pre_interceptor: Interceptor
handler: Handler
generator: Generator
interceptor: Interceptor
def input(self, input: Input):
if isinstance(pre_intercepted_stream := self.pre_intercept(input), Output):
yield self.generate(pre_intercepted_stream)
return
for handled_stream in self.handle(pre_intercepted_stream):
if handled_stream.is_empty():
return # TODO 处理未找到 Handler
yield self.intercept(self.generate(handled_stream))
def pre_intercept(self, input: Input) -> Input | Output:
return self.pre_interceptor.input(input)
def handle(self, input: Input):
iterator = self.handler.input(input)
for output in iterator:
yield output
def generate(self, output: Output) -> str:
return output.name # TODO 生成器实现
def intercept(self, output: str) -> str:
return (
self.generate(callback)
if isinstance(callback := self.interceptor.output(output), Output)
else callback
)
|