rtvi: add llm-text and tts-text server messages

This commit is contained in:
Aleix Conchillo Flaqué
2024-09-23 23:39:05 -07:00
parent 016da177db
commit cb49b6a0d6

View File

@@ -242,6 +242,22 @@ class RTVILLMFunctionCallResultData(BaseModel):
result: dict | str
class RTVITextMessageData(BaseModel):
text: str
class RTVILLMTextMessage(BaseModel):
label: Literal["rtvi-ai"] = "rtvi-ai"
type: Literal["llm-text"] = "llm-text"
data: RTVITextMessageData
class RTVITTSTextMessage(BaseModel):
label: Literal["rtvi-ai"] = "rtvi-ai"
type: Literal["tts-text"] = "tts-text"
data: RTVITextMessageData
class RTVITranscriptionMessageData(BaseModel):
text: str
user_id: str