mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-11-04 01:17:52 +00:00
25 lines
654 B
Python
25 lines
654 B
Python
"""Base Assembler Operator."""
|
|
from abc import abstractmethod
|
|
|
|
from dbgpt.core.awel import MapOperator
|
|
from dbgpt.core.awel.task.base import IN, OUT
|
|
|
|
|
|
class AssemblerOperator(MapOperator[IN, OUT]):
|
|
"""The Base Assembler Operator."""
|
|
|
|
async def map(self, input_value: IN) -> OUT:
|
|
"""Map input value to output value.
|
|
|
|
Args:
|
|
input_value (IN): The input value.
|
|
|
|
Returns:
|
|
OUT: The output value.
|
|
"""
|
|
return await self.blocking_func_to_async(self.assemble, input_value)
|
|
|
|
@abstractmethod
|
|
def assemble(self, input_value: IN) -> OUT:
|
|
"""Assemble knowledge for input value."""
|