UNPKG

@2501-ai/cli

Version:

[![npm version](https://img.shields.io/npm/v/@2501-ai/cli.svg)](https://www.npmjs.com/package/@2501-ai/cli) [![HumanEval Score](https://img.shields.io/badge/HumanEval-96.95%25-brightgreen.svg)](https://www.2501.ai/research/full-humaneval-benchmark) [![Lic

63 lines (62 loc) 3.04 kB
"use strict"; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); const fs_1 = __importDefault(require("fs")); const path_1 = __importDefault(require("path")); const streams_1 = require("../streams"); describe('streams', () => { let streamChunks; beforeAll(() => { const streamChunksRaw = fs_1.default.readFileSync(`${__dirname}${path_1.default.sep}mocks/streams.mock.jsonl`, 'utf-8'); expect(streamChunksRaw).toBeTruthy(); streamChunks = streamChunksRaw.split('\n'); expect(streamChunks.length).toEqual(3); }); it('should parseChunkedMessages with multiple chunks correctly ', () => { const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(streamChunks[0]); expect(remaining).toBe(''); expect(parsed.length).toBe(3); }); it('should parseChunkedMessages with escaped double quotes correctly ', () => { const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(streamChunks[1]); expect(remaining).toBe(''); expect(parsed.length).toBe(1); }); it('should parseChunks with curly braces in it', () => { const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(streamChunks[2]); expect(remaining).toBe(''); expect(parsed.length).toBe(5); }); it('should have remains when content is incomplete', () => { const content = '{"status":"chunked_message","message":"'; const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(content); expect(remaining).toBe(content); expect(parsed.length).toBe(0); }); it('should parse content fully', () => { const content = '{"status":"chunked_message","message":"\\\\\\""}{"status":"chunked_message","message":"\\\\\\""}'; const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(content); expect(remaining).toBe(''); expect(parsed.length).toBe(2); }); it('should have remains when trying to parse an incomplete content', () => { const content = '{"status":"chunked_message","message":"}'; const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(content); expect(remaining).toBe(content); expect(parsed.length).toBe(0); }); it('should have remains when invalid content is given', () => { const content = `{"status":"chunked_message","message":"\\"}`; const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(content); expect(remaining).toBe(content); expect(parsed.length).toBe(0); }); it('should have remains when invalid content is given #2', () => { const content = `{"status":"chunked_message","message":"}`; const { remaining, parsed } = (0, streams_1.parseChunkedMessages)(content); expect(remaining).toBe(content); expect(parsed.length).toBe(0); }); });