mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 10:12:51 +00:00
28 lines
689 B
Python
28 lines
689 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,
|
||
_status=status.HTTP_200_OK):
|
||
pass
|
||
|
||
@abstractmethod
|
||
def to_stream_chunk_response(self, chat_id, chat_record_id, content, is_end, completion_tokens, prompt_tokens):
|
||
pass
|
||
|
||
@staticmethod
|
||
def format_stream_chunk(response_str):
|
||
return 'data: ' + response_str + '\n\n'
|