45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: MaxKB
|
|||
|
|
@Author：虎
|
|||
|
|
@file： common.py
|
|||
|
|
@date：2024/12/11 17:57
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Answer:
|
|||
|
|
def __init__(self, content, view_type, runtime_node_id, chat_record_id, child_node, real_node_id,
|
|||
|
|
reasoning_content):
|
|||
|
|
self.view_type = view_type
|
|||
|
|
self.content = content
|
|||
|
|
self.reasoning_content = reasoning_content
|
|||
|
|
self.runtime_node_id = runtime_node_id
|
|||
|
|
self.chat_record_id = chat_record_id
|
|||
|
|
self.child_node = child_node
|
|||
|
|
self.real_node_id = real_node_id
|
|||
|
|
|
|||
|
|
def to_dict(self):
|
|||
|
|
return {'view_type': self.view_type, 'content': self.content, 'runtime_node_id': self.runtime_node_id,
|
|||
|
|
'chat_record_id': self.chat_record_id,
|
|||
|
|
'child_node': self.child_node,
|
|||
|
|
'reasoning_content': self.reasoning_content,
|
|||
|
|
'real_node_id': self.real_node_id}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class NodeChunk:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.status = 0
|
|||
|
|
self.chunk_list = []
|
|||
|
|
|
|||
|
|
def add_chunk(self, chunk):
|
|||
|
|
self.chunk_list.append(chunk)
|
|||
|
|
|
|||
|
|
def end(self, chunk=None):
|
|||
|
|
if chunk is not None:
|
|||
|
|
self.add_chunk(chunk)
|
|||
|
|
self.status = 200
|
|||
|
|
|
|||
|
|
def is_end(self):
|
|||
|
|
return self.status == 200
|