@quidgest/chatbot 0.5.8 → 0.5.9

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@quidgest/chatbot",
     "private": false,
-    "version": "0.5.8",
+    "version": "0.5.9",
     "type": "module",
     "license": "UNLICENSED",
     "main": "dist/index.cjs",

package/src/composables/__tests__/useSSE.spec.ts

@@ -129,4 +129,100 @@ describe('useSSE', () => {
 		expect(consoleWarnSpy).toHaveBeenCalledWith('Unknown event type: unknown_event')
 		consoleWarnSpy.mockRestore()
 	})
+
+	// Tests for handling events split across chunks (bug reproduction)
+	describe('Events split across multiple chunks', () => {
+		it('should handle event split in the middle of data field', async () => {
+			// Simulate network splitting an event across two chunks
+			const stream = createMockStream(['event: message\ndata: {"val', 'ue":"hello"}\n\n'])
+			mockedAxios.mockResolvedValue({ data: stream })
+
+			await useSSE({ url: '/sse' }, handlers)
+
+			expect(handlers.onMessage).toHaveBeenCalledWith('hello')
+			expect(handlers.onDone).toHaveBeenCalled()
+		})
+
+		it('should handle multiple events split across multiple chunks', async () => {
+			// Simulate the real scenario from the bug report
+			const stream = createMockStream([
+				'event: message\ndata: {"value":"The"}\n\nevent: message\ndata: {"val',
+				'ue":" capital"}\n\nevent: message\ndata: {"value":" of"}\n\n',
+				'event: message\ndata: {"value":" France"}\n\n'
+			])
+			mockedAxios.mockResolvedValue({ data: stream })
+
+			await useSSE({ url: '/sse' }, handlers)
+
+			// Should receive all 4 messages
+			expect(handlers.onMessage).toHaveBeenCalledTimes(4)
+			expect(handlers.onMessage).toHaveBeenNthCalledWith(1, 'The')
+			expect(handlers.onMessage).toHaveBeenNthCalledWith(2, ' capital')
+			expect(handlers.onMessage).toHaveBeenNthCalledWith(3, ' of')
+			expect(handlers.onMessage).toHaveBeenNthCalledWith(4, ' France')
+			expect(handlers.onDone).toHaveBeenCalled()
+		})
+
+		it('should handle event split between event name and data', async () => {
+			const stream = createMockStream([
+				'event: message\n',
+				'data: {"value":"split event"}\n\n'
+			])
+			mockedAxios.mockResolvedValue({ data: stream })
+
+			await useSSE({ url: '/sse' }, handlers)
+
+			expect(handlers.onMessage).toHaveBeenCalledWith('split event')
+			expect(handlers.onDone).toHaveBeenCalled()
+		})
+
+		it('should handle event split right after event separator', async () => {
+			const stream = createMockStream([
+				'event: message\ndata: {"value":"first"}\n\n',
+				'event: message\ndata: {"value":"second"}\n\n'
+			])
+			mockedAxios.mockResolvedValue({ data: stream })
+
+			await useSSE({ url: '/sse' }, handlers)
+
+			expect(handlers.onMessage).toHaveBeenCalledTimes(2)
+			expect(handlers.onMessage).toHaveBeenNthCalledWith(1, 'first')
+			expect(handlers.onMessage).toHaveBeenNthCalledWith(2, 'second')
+			expect(handlers.onDone).toHaveBeenCalled()
+		})
+
+		it('should handle complex JSON split across chunks', async () => {
+			const stream = createMockStream([
+				'event: field_metadata\ndata: {"foo":"bar","nested":{"key',
+				'":"value"}}\n\n'
+			])
+			mockedAxios.mockResolvedValue({ data: stream })
+
+			await useSSE({ url: '/sse' }, handlers)
+
+			expect(handlers.onFieldMetadata).toHaveBeenCalledWith({
+				foo: 'bar',
+				nested: { key: 'value' }
+			})
+			expect(handlers.onDone).toHaveBeenCalled()
+		})
+
+		it('should handle event split in very small chunks', async () => {
+			// Extreme case: each character in a separate chunk
+			const stream = createMockStream([
+				'event: ',
+				'message\n',
+				'data: ',
+				'{"value":',
+				'"test"}',
+				'\n\n'
+			])
+			mockedAxios.mockResolvedValue({ data: stream })
+
+			await useSSE({ url: '/sse' }, handlers)
+
+			expect(handlers.onMessage).toHaveBeenCalledWith('test')
+			expect(handlers.onDone).toHaveBeenCalled()
+		})
+	})
 })

package/src/composables/useSSE.ts

@@ -30,6 +30,7 @@ export async function useSSE(config: AxiosRequestConfig, handlers: SSEvents) {
 
 	const reader = stream.getReader()
 	const decoder = new TextDecoder()
+	let buffer = '' // Buffer to accumulate incomplete events across chunks
 
 	while (true) {
 		const { done, value } = await reader.read()
@@ -39,7 +40,12 @@ export async function useSSE(config: AxiosRequestConfig, handlers: SSEvents) {
 		}
 
 		const chunk = decoder.decode(value, { stream: true })
-		const events = chunk.split(/\n\n+/)
+		buffer += chunk // Append chunk to buffer
+		const events = buffer.split(/\n\n+/)
+
+		// Keep the last element in buffer (might be incomplete)
+		// Process all complete events (all except the last one)
+		buffer = events.pop() || ''
 
 		for (const eventBlock of events) {
 			const lines = eventBlock.trim().split('\n')