56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: maxkb
|
|||
|
|
@Author：虎
|
|||
|
|
@file： I_base_chat_pipeline.py
|
|||
|
|
@date：2024/1/9 17:25
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
import time
|
|||
|
|
from abc import abstractmethod
|
|||
|
|
from typing import Type
|
|||
|
|
|
|||
|
|
from rest_framework import serializers
|
|||
|
|
|
|||
|
|
|
|||
|
|
class IBaseChatPipelineStep:
|
|||
|
|
def __init__(self):
|
|||
|
|
# 当前步骤上下文,用于存储当前步骤信息
|
|||
|
|
self.context = {}
|
|||
|
|
|
|||
|
|
@abstractmethod
|
|||
|
|
def get_step_serializer(self, manage) -> Type[serializers.Serializer]:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def valid_args(self, manage):
|
|||
|
|
step_serializer_clazz = self.get_step_serializer(manage)
|
|||
|
|
step_serializer = step_serializer_clazz(data=manage.context)
|
|||
|
|
step_serializer.is_valid(raise_exception=True)
|
|||
|
|
self.context['step_args'] = step_serializer.data
|
|||
|
|
|
|||
|
|
def run(self, manage):
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
:param manage: 步骤管理器
|
|||
|
|
:return: 执行结果
|
|||
|
|
"""
|
|||
|
|
start_time = time.time()
|
|||
|
|
# 校验参数,
|
|||
|
|
self.valid_args(manage)
|
|||
|
|
self._run(manage)
|
|||
|
|
self.context['start_time'] = start_time
|
|||
|
|
self.context['run_time'] = time.time() - start_time
|
|||
|
|
|
|||
|
|
def _run(self, manage):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def execute(self, **kwargs):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def get_details(self, manage, **kwargs):
|
|||
|
|
"""
|
|||
|
|
运行详情
|
|||
|
|
:return: 步骤详情
|
|||
|
|
"""
|
|||
|
|
return None
|