@celhive/tool.js 0.0.6 → 0.0.8

package/lib/index.d.ts

@@ -2,4 +2,5 @@ export * from './formatCitation';
 export * from './get-pdf-viewer-urll';
 export * from './markdown-patch';
 export * from './mock-stream';
+export * from './stable-streaming';
 export * from './token-exceeded';

package/lib/index.js

@@ -18,4 +18,5 @@ __exportStar(require("./formatCitation"), exports);
 __exportStar(require("./get-pdf-viewer-urll"), exports);
 __exportStar(require("./markdown-patch"), exports);
 __exportStar(require("./mock-stream"), exports);
+__exportStar(require("./stable-streaming"), exports);
 __exportStar(require("./token-exceeded"), exports);

package/lib/markdown-patch.js

@@ -42,7 +42,33 @@ const markdownPatch = (md) => {
     // 写一个正则，如果 ```mermaid 前面只有一个换行符，就添加一个换行符
     const mendMermaid = mendBold.replace(/(\n)```mermaid/g, '$1\n```mermaid');
     // 在连续的 # 符号（1个以上）前面插入换行符，但不处理行首的 #
-    const result = mendMermaid.replace(/([^\n#])(#{2,})/g, '$1\n$2');
+    const mendHashNewline = mendMermaid.replace(/([^\n#])(#{2,})/g, '$1\n$2');
+    // 再次处理新行中的标题格式，确保 # 和内容之间有空格
+    const lines2 = mendHashNewline.split('\n');
+    let inCodeBlock2 = false;
+    const result = lines2
+        .map((line) => {
+        // 检查是否是代码块的开始或结束
+        if (line.trim().startsWith('```')) {
+            inCodeBlock2 = !inCodeBlock2;
+            return line;
+        }
+        // 如果在代码块内，跳过处理
+        if (inCodeBlock2) {
+            return line;
+        }
+        const match = line.match(/^(\s*)(#+)(\s*)(.*)$/);
+        if (match) {
+            const [, leadingSpaces, hashes, spaces, content] = match;
+            if (hashes.length > 0 &&
+                spaces.length === 0 &&
+                !content.startsWith('#')) {
+                return `${leadingSpaces}${hashes} ${content}`;
+            }
+        }
+        return line;
+    })
+        .join('\n');
     return result;
 };
 exports.markdownPatch = markdownPatch;

package/lib/stable-streaming.d.ts

@@ -0,0 +1,20 @@
+interface stableStreamingProps {
+    /** Number of characters output per millisecond */
+    ratePerMs: number;
+    /** Callback invoked with each character */
+    callback: (char: string) => void;
+}
+/**
+ * StableStreaming class to stream characters at a stable rate.
+ * It ensures that characters are output at the specified rate per millisecond,
+ */
+export declare class StableStreaming {
+    private options;
+    private queue;
+    private timer;
+    private pendingResolvers;
+    constructor(options: stableStreamingProps);
+    streaming(chars: string): Promise<void>;
+    private start;
+}
+export {};

package/lib/stable-streaming.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StableStreaming = void 0;
+/**
+ * StableStreaming class to stream characters at a stable rate.
+ * It ensures that characters are output at the specified rate per millisecond,
+ */
+class StableStreaming {
+    constructor(options) {
+        this.queue = [];
+        this.timer = null;
+        this.pendingResolvers = [];
+        if (!options || options.ratePerMs <= 0) {
+            throw new Error('ratePerMs must be greater than 0');
+        }
+        this.options = options;
+    }
+    streaming(chars) {
+        if (!chars)
+            return Promise.resolve();
+        this.queue.push(...chars.split(''));
+        return new Promise((resolve) => {
+            this.pendingResolvers.push(resolve);
+            if (!this.timer) {
+                this.start();
+            }
+        });
+    }
+    start() {
+        const charsPerTick = Math.max(1, Math.floor(this.options.ratePerMs));
+        this.timer = setInterval(() => {
+            let remaining = charsPerTick;
+            while (remaining > 0 && this.queue.length) {
+                const char = this.queue.shift();
+                if (char !== undefined) {
+                    this.options.callback(char);
+                }
+                remaining -= 1;
+            }
+            if (!this.queue.length) {
+                if (this.timer) {
+                    clearInterval(this.timer);
+                    this.timer = null;
+                }
+                this.pendingResolvers.forEach((resolver) => resolver());
+                this.pendingResolvers = [];
+            }
+        }, 1);
+    }
+}
+exports.StableStreaming = StableStreaming;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@celhive/tool.js",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "private": false,
   "publishConfig": {
     "access": "public"