David Snelling
3ec183dab6
- **Vector Handling Updates**: - Added a `dimensions` property to `BrainyDataConfig` for specifying vector dimensions. - Introduced validation for vector dimensions during database creation and insertion to ensure consistency. - Enhanced error handling and logging for dimension mismatches. - **Model Loading Improvements**: - Implemented retry logic for Universal Sentence Encoder model loading to handle network instability and JSON parsing errors gracefully. - Improved logging and debugging support for failures during model initialization and embedding operations. - **Compatibility Enhancements**: - Updated polyfills to support TensorFlow.js compatibility across diverse server environments (Node.js, serverless, etc.). - Introduced and refactored global `TextEncoder`/`TextDecoder` definitions for seamless operation in non-browser environments. - Simplified TensorFlow.js backend setup with streamlined imports and logging for GPU/WebGL fallback. - **Purpose**: - These updates improve BrainyData's robustness, enforce correct vector usage, and extend compatibility with varied runtime environments. The changes enhance the usability, reliability, and cross-platform readiness of core functionalities.
55 lines
1.7 KiB
TypeScript
55 lines
1.7 KiB
TypeScript
/**
|
|
* Unified entry point for Brainy
|
|
* This file exports everything from index.ts
|
|
* Environment detection is handled here and made available to all components
|
|
*/
|
|
|
|
// CRITICAL: The TensorFlow.js environment patch is now centralized in setup.ts
|
|
// We import setup.ts below which applies the necessary patches
|
|
|
|
// CRITICAL: Import setup.js first to ensure TensorFlow.js environment patching
|
|
// This MUST be the first import to prevent race conditions with TensorFlow.js initialization
|
|
// Moving or removing this import will cause errors like "TextEncoder is not a constructor"
|
|
// when the package is used in Node.js environments
|
|
//
|
|
// The setup.js file applies a patch that ensures TextEncoder/TextDecoder are properly
|
|
// available to TensorFlow.js before it initializes its platform detection
|
|
import './setup.js'
|
|
|
|
// Import environment detection functions
|
|
import { isBrowser, isNode } from './utils/environment.js'
|
|
|
|
// Export environment information with lazy evaluation
|
|
export const environment = {
|
|
get isBrowser() {
|
|
return isBrowser()
|
|
},
|
|
get isNode() {
|
|
return isNode()
|
|
},
|
|
get isServerless() {
|
|
return !isBrowser() && !isNode()
|
|
}
|
|
}
|
|
|
|
// Make environment information available globally
|
|
if (typeof globalThis !== 'undefined') {
|
|
;(globalThis as any).__ENV__ = environment
|
|
}
|
|
|
|
// Log the detected environment
|
|
console.log(
|
|
`Brainy running in ${
|
|
environment.isBrowser
|
|
? 'browser'
|
|
: environment.isNode
|
|
? 'Node.js'
|
|
: 'serverless/unknown'
|
|
} environment`
|
|
)
|
|
|
|
// Re-export everything from index.ts
|
|
export * from './index.js'
|
|
|
|
// Export the TensorFlow patch function for testing and manual use
|
|
export { applyTensorFlowPatch } from './utils/textEncoding.js'
|