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

pipecat-ai · Sep 24, 2024 · 5eaf9a7 · 5eaf9a7
1 parent 35512cc
commit 5eaf9a7
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/pipecat/processors/frameworks/rtvi.py b/src/pipecat/processors/frameworks/rtvi.py
@@ -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