mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-22 23:23:55 +08:00
27 lines
764 B
Python
27 lines
764 B
Python
|
import uuid
|
||
|
|
from typing import List, Dict
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
|
||
|
|
class Message:
|
||
|
|
def __init__(self, role: str, content: str = None, uniq_id: str = None):
|
||
|
|
self.uniq_id = uniq_id if uniq_id is not None else str(uuid.uuid4())
|
||
|
|
self.role = role
|
||
|
|
self.content = content
|
||
|
|
|
||
|
|
|
||
|
|
class Dialogue:
|
||
|
|
def __init__(self):
|
||
|
|
self.dialogue: List[Message] = []
|
||
|
|
# 获取当前时间
|
||
|
|
self.current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
||
|
|
|
||
|
|
def put(self, message: Message):
|
||
|
|
self.dialogue.append(message)
|
||
|
|
|
||
|
|
def get_llm_dialogue(self) -> List[Dict[str, str]]:
|
||
|
|
dialogue = []
|
||
|
|
for m in self.dialogue:
|
||
|
|
dialogue.append({"role": m.role, "content": m.content})
|
||
|
|
return dialogue
|