mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 10:12:51 +00:00
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
|