David Snelling
da760a34ed
- Added `cli-package/brainy-wrapper.js` to patch global `TextEncoder` and `TextDecoder` for Node.js environments, ensuring compatibility with TensorFlow.js. - Implemented unified text encoding utilities in `cli-package/src/utils/textEncoding.ts` for cross-environment consistency. - Enhanced `src/worker.js` and introduced `src/worker.ts` to improve worker execution with better function serialization and error handling. - Updated `package.json`: - Modified the `build` script to include a patch for `TextEncoder`. - Added `test-all` script for multi-environment testing. - Introduced the Puppeteer dependency for browser testing. - Created a favicon generation script `scripts/create-favicon.js` using a base64-encoded icon. - Added `scripts/test-all-environments.js` to run automated tests across Node.js, browser, and CLI environments. - Enhanced `src/utils/workerUtils.ts` to support improved fallback mechanisms and robust worker pooling. This update improves the project's compatibility across environments, refines worker functionalities, and adds comprehensive testing support for reliability.
60 lines
2 KiB
TypeScript
60 lines
2 KiB
TypeScript
/**
|
|
* Unified Text Encoding Utilities for CLI
|
|
*
|
|
* This module provides a consistent way to handle text encoding/decoding across all environments
|
|
* without relying on TextEncoder/TextDecoder polyfills or patches.
|
|
*/
|
|
|
|
/**
|
|
* Apply the TensorFlow.js platform patch if needed
|
|
* This function patches the global object to provide a PlatformNode class
|
|
* that uses our text encoding utilities instead of relying on TextEncoder/TextDecoder
|
|
*/
|
|
export function applyTensorFlowPatch(): void {
|
|
// Only apply in Node.js environment
|
|
if (
|
|
typeof global !== 'undefined' &&
|
|
typeof process !== 'undefined' &&
|
|
process.versions &&
|
|
process.versions.node
|
|
) {
|
|
try {
|
|
// Define a custom PlatformNode class
|
|
class PlatformNode {
|
|
util: any
|
|
textEncoder: any
|
|
textDecoder: any
|
|
|
|
constructor() {
|
|
// Create a util object with necessary methods and constructors
|
|
this.util = {
|
|
isFloat32Array: (arr: any) =>
|
|
!!(
|
|
arr instanceof Float32Array ||
|
|
(arr &&
|
|
Object.prototype.toString.call(arr) ===
|
|
'[object Float32Array]')
|
|
),
|
|
isTypedArray: (arr: any) =>
|
|
!!(ArrayBuffer.isView(arr) && !(arr instanceof DataView)),
|
|
// Use native TextEncoder and TextDecoder
|
|
TextEncoder: TextEncoder,
|
|
TextDecoder: TextDecoder
|
|
}
|
|
|
|
// Initialize using the constructors from util
|
|
this.textEncoder = new this.util.TextEncoder()
|
|
this.textDecoder = new this.util.TextDecoder()
|
|
}
|
|
}
|
|
|
|
// Assign the PlatformNode class to the global object
|
|
;(global as any).PlatformNode = PlatformNode
|
|
|
|
// Also create an instance and assign it to global.platformNode (lowercase p)
|
|
;(global as any).platformNode = new PlatformNode()
|
|
} catch (error) {
|
|
console.warn('Failed to apply TensorFlow.js platform patch:', error)
|
|
}
|
|
}
|
|
}
|