initial_commit

frontend/src/stores/chat.ts

@@ -0,0 +1,84 @@
+import { defineStore } from 'pinia'
+import { ref, computed } from 'vue'
+import { fetchSessions, deleteSession, streamChat, type SessionInfo } from '../api/client'
+
+export interface Message {
+  role: 'user' | 'assistant'
+  content: string
+  pending?: boolean
+}
+
+export const useChatStore = defineStore('chat', () => {
+  const sessions = ref<SessionInfo[]>([])
+  const currentSessionId = ref<string | null>(null)
+  const messages = ref<Message[]>([])
+  const isStreaming = ref(false)
+  const selectedModel = ref('gpt-4o')
+
+  const currentSession = computed(() =>
+    sessions.value.find((s) => s.id === currentSessionId.value) ?? null,
+  )
+
+  async function loadSessions() {
+    sessions.value = await fetchSessions()
+  }
+
+  function selectSession(sessionId: string | null) {
+    currentSessionId.value = sessionId
+    messages.value = []
+  }
+
+  async function removeSession(sessionId: string) {
+    await deleteSession(sessionId)
+    if (currentSessionId.value === sessionId) {
+      currentSessionId.value = null
+      messages.value = []
+    }
+    await loadSessions()
+  }
+
+  function sendMessage(text: string): void {
+    if (isStreaming.value || !text.trim()) return
+
+    messages.value.push({ role: 'user', content: text })
+    const assistantMsg: Message = { role: 'assistant', content: '', pending: true }
+    messages.value.push(assistantMsg)
+    isStreaming.value = true
+
+    streamChat(
+      text,
+      currentSessionId.value,
+      selectedModel.value,
+      (chunk) => {
+        if (chunk.type === 'session_id' && chunk.value) {
+          currentSessionId.value = chunk.value
+          loadSessions()
+        } else if (chunk.type === 'chunk' && chunk.value) {
+          assistantMsg.content += chunk.value
+        } else if (chunk.type === 'done') {
+          assistantMsg.pending = false
+          isStreaming.value = false
+          loadSessions()
+        }
+      },
+      (_err) => {
+        assistantMsg.content = 'エラーが発生しました。'
+        assistantMsg.pending = false
+        isStreaming.value = false
+      },
+    )
+  }
+
+  return {
+    sessions,
+    currentSessionId,
+    messages,
+    isStreaming,
+    selectedModel,
+    currentSession,
+    loadSessions,
+    selectSession,
+    removeSession,
+    sendMessage,
+  }
+})