@celhive/tool.js 0.0.3 → 0.0.5

package/lib/index.d.ts

@@ -2,3 +2,4 @@ export * from './formatCitation';
 export * from './getPDFViewerUrl';
 export * from './mendMarkdown';
 export * from './mockStream';
+export * from './token-exceeded';

package/lib/index.js

@@ -18,3 +18,4 @@ __exportStar(require("./formatCitation"), exports);
 __exportStar(require("./getPDFViewerUrl"), exports);
 __exportStar(require("./mendMarkdown"), exports);
 __exportStar(require("./mockStream"), exports);
+__exportStar(require("./token-exceeded"), exports);

package/lib/mendMarkdown.js

@@ -39,6 +39,8 @@ const mendMarkdown = (md) => {
             ? `<b>${p1}</b>`
             : match;
     });
-    return mendBold;
+    // 写一个正则，如果 ```mermaid 前面只有一个换行符，就添加一个换行符
+    const result = mendBold.replace(/(\n)```mermaid/g, '$1\n```mermaid');
+    return result;
 };
 exports.mendMarkdown = mendMarkdown;

package/lib/token-exceeded.d.ts

@@ -0,0 +1 @@
+export declare const tokenExceeded: (prompt: string, maxToken: number) => boolean;

package/lib/token-exceeded.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tokenExceeded = void 0;
+const tokenExceeded = (prompt, maxToken) => {
+    const chineseChars = prompt.match(/[\u4e00-\u9fa5]/g) || [];
+    const otherChars = prompt.replace(/[\u4e00-\u9fa5]/g, '');
+    const estimatedTokens = Math.ceil(chineseChars.length / 1.5 + otherChars.length / 1.7);
+    return estimatedTokens > maxToken;
+};
+exports.tokenExceeded = tokenExceeded;

package/package.json

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