WebSocketモードでemotionを受け取れるようにする

tegnike · Oct 25, 2024 · fda8686 · fda8686
1 parent cde6500
commit fda8686
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/src/components/useWebSocket.tsx b/src/components/useWebSocket.tsx
@@ -17,18 +17,26 @@ interface Params {
   handleReceiveTextFromWs: (
     text: string,
     role?: string,
+    emotion?: string,
     state?: string
   ) => Promise<void>
 }
 
 const useWebSocket = ({ handleReceiveTextFromWs }: Params) => {
   const { t } = useTranslation()
   const webSocketMode = settingsStore((s) => s.webSocketMode)
-  const [receivedMessages, setReceivedMessages] = useState<ReceivedMessage[]>([])
+  const [receivedMessages, setReceivedMessages] = useState<ReceivedMessage[]>(
+    []
+  )
 
   const processMessage = useCallback(
     async (message: ReceivedMessage) => {
-      await handleReceiveTextFromWs(message.text, message.role, message.state)
+      await handleReceiveTextFromWs(
+        message.text,
+        message.role,
+        message.emotion,
+        message.state
+      )
     },
     [handleReceiveTextFromWs]
   )

diff --git a/src/features/chat/handlers.ts b/src/features/chat/handlers.ts
@@ -492,7 +492,13 @@ export const handleSendChatFn =
  * WebSocketからのテキストを受信したときの処理
  */
 export const handleReceiveTextFromWsFn =
-  () => async (text: string, role?: string, state?: string) => {
+  () =>
+  async (
+    text: string,
+    role?: string,
+    emotion: string = 'neutral',
+    state?: string
+  ) => {
     if (text === null || role === undefined) return
 
     const ss = settingsStore.getState()
@@ -528,7 +534,7 @@ export const handleReceiveTextFromWsFn =
       }
 
       if (role === 'assistant' && text !== '') {
-        let aiText = `${'[neutral]'} ${text}`
+        let aiText = `[${emotion}] ${text}`
         try {
           const aiTalks = textsToScreenplay([aiText], ss.koeiroParam)