char-fork/llama.cpp

git clone https://git.t4t.associates/char-fork/llama.cpp

Aleksander Grygierui: Linting & Formatting scripts (#26819)92d1bb0c9

char/aliceai-t5-moe
brancheschar/aliceai-t5-moeall refs →
3.5 KiB81 linesraw
1import { ChatService } from '$lib/services/chat.service';
2import type { ApiStreamSession } from '$lib/types';
3import { describe, expect, it } from 'vitest';
4
5function makeSession(overrides: Partial<ApiStreamSession>): ApiStreamSession {
6	return {
7		completed_at: 0,
8		conversation_id: 'conv',
9		is_done: true,
10		started_at: 0,
11		total_bytes: 0,
12		...overrides
13	};
14}
15
16describe('selectActiveStream', () => {
17	it('returns null on empty input', () => {
18		expect(ChatService.selectActiveStream([])).toBeNull();
19	});
20
21	it('returns null on null or undefined input', () => {
22		expect(ChatService.selectActiveStream(null)).toBeNull();
23		expect(ChatService.selectActiveStream(undefined)).toBeNull();
24	});
25
26	it('returns the single session when it is running', () => {
27		const s = makeSession({ conversation_id: 'only', is_done: false, started_at: 42 });
28
29		expect(ChatService.selectActiveStream([s])).toBe(s);
30	});
31
32	it('returns null when the single session is finalized', () => {
33		const s = makeSession({ conversation_id: 'only', is_done: true, started_at: 42 });
34
35		expect(ChatService.selectActiveStream([s])).toBeNull();
36	});
37
38	it('prefers a still running session over a finalized one regardless of started_at', () => {
39		const finalized = makeSession({ conversation_id: 'old', is_done: true, started_at: 1000 });
40		const running = makeSession({ conversation_id: 'new', is_done: false, started_at: 10 });
41
42		expect(ChatService.selectActiveStream([finalized, running])?.conversation_id).toBe('new');
43		expect(ChatService.selectActiveStream([running, finalized])?.conversation_id).toBe('new');
44	});
45
46	it('among running sessions, picks the most recently started one', () => {
47		const a = makeSession({ conversation_id: 'a', is_done: false, started_at: 100 });
48		const b = makeSession({ conversation_id: 'b', is_done: false, started_at: 200 });
49		const c = makeSession({ conversation_id: 'c', is_done: false, started_at: 150 });
50
51		expect(ChatService.selectActiveStream([a, b, c])?.conversation_id).toBe('b');
52		expect(ChatService.selectActiveStream([c, a, b])?.conversation_id).toBe('b');
53	});
54
55	it('returns null when all sessions are finalized, the DB already holds the content', () => {
56		const a = makeSession({ conversation_id: 'a', is_done: true, started_at: 10 });
57		const b = makeSession({ conversation_id: 'b', is_done: true, started_at: 30 });
58		const c = makeSession({ conversation_id: 'c', is_done: true, started_at: 20 });
59
60		expect(ChatService.selectActiveStream([a, b, c])).toBeNull();
61	});
62
63	it('keeps the first match on ties when both are running with identical started_at', () => {
64		// reduce visits left to right, the initial accumulator stays unless a strictly greater value appears
65		const a = makeSession({ conversation_id: 'first', is_done: false, started_at: 50 });
66		const b = makeSession({ conversation_id: 'second', is_done: false, started_at: 50 });
67
68		expect(ChatService.selectActiveStream([a, b])?.conversation_id).toBe('first');
69	});
70
71	it('handles a typical realistic mix: two finalized old, one freshly running, one freshly finalized', () => {
72		const old1 = makeSession({ conversation_id: 'old1', is_done: true, started_at: 100 });
73		const old2 = makeSession({ conversation_id: 'old2', is_done: true, started_at: 200 });
74		const freshFin = makeSession({ conversation_id: 'freshFin', is_done: true, started_at: 500 });
75		const running = makeSession({ conversation_id: 'running', is_done: false, started_at: 400 });
76
77		expect(ChatService.selectActiveStream([old1, old2, freshFin, running])?.conversation_id).toBe(
78			'running'
79		);
80	});
81});