31 lines
861 B
Python
31 lines
861 B
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: MaxKB
|
|||
|
|
@Author：虎
|
|||
|
|
@file： base_to_response.py
|
|||
|
|
@date：2024/9/6 16:04
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
from abc import ABC, abstractmethod
|
|||
|
|
|
|||
|
|
from rest_framework import status
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BaseToResponse(ABC):
|
|||
|
|
|
|||
|
|
@abstractmethod
|
|||
|
|
def to_block_response(self, chat_id, chat_record_id, content, is_end, completion_tokens,
|
|||
|
|
prompt_tokens, other_params: dict = None,
|
|||
|
|
_status=status.HTTP_200_OK):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
@abstractmethod
|
|||
|
|
def to_stream_chunk_response(self, chat_id, chat_record_id, node_id, up_node_id_list, content, is_end,
|
|||
|
|
completion_tokens,
|
|||
|
|
prompt_tokens, other_params: dict = None):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def format_stream_chunk(response_str):
|
|||
|
|
return 'data: ' + response_str + '\n\n'
|