David Snelling
f8c45f2d8d
Open source vector database with HNSW indexing, graph relationships, and metadata facets. Features CLI with professional augmentation registry integration for discovering extensions and capabilities.
105 lines
No EOL
4.1 KiB
JavaScript
105 lines
No EOL
4.1 KiB
JavaScript
/**
|
|
* Augmentation Registry
|
|
*
|
|
* This module provides a registry for augmentations that are loaded at build time.
|
|
* It replaces the dynamic loading mechanism in pluginLoader.ts.
|
|
*/
|
|
import { AugmentationType } from './types/augmentations.js';
|
|
// Forward declaration of the pipeline instance to avoid circular dependency
|
|
// The actual pipeline will be provided when initializeAugmentationPipeline is called
|
|
let defaultPipeline = null;
|
|
/**
|
|
* Sets the default pipeline instance
|
|
* This function should be called from pipeline.ts after the pipeline is created
|
|
*/
|
|
export function setDefaultPipeline(pipeline) {
|
|
defaultPipeline = pipeline;
|
|
}
|
|
/**
|
|
* Registry of all available augmentations
|
|
*/
|
|
export const availableAugmentations = [];
|
|
/**
|
|
* Registers an augmentation with the registry
|
|
*
|
|
* @param augmentation The augmentation to register
|
|
* @returns The augmentation that was registered
|
|
*/
|
|
export function registerAugmentation(augmentation) {
|
|
// Set enabled to true by default if not specified
|
|
if (augmentation.enabled === undefined) {
|
|
augmentation.enabled = true;
|
|
}
|
|
// Add to the registry
|
|
availableAugmentations.push(augmentation);
|
|
return augmentation;
|
|
}
|
|
/**
|
|
* Initializes the augmentation pipeline with all registered augmentations
|
|
*
|
|
* @param pipeline Optional custom pipeline to use instead of the default
|
|
* @returns The pipeline that was initialized
|
|
* @throws Error if no pipeline is provided and the default pipeline hasn't been set
|
|
*/
|
|
export function initializeAugmentationPipeline(pipelineInstance) {
|
|
// Use the provided pipeline or fall back to the default
|
|
const pipeline = pipelineInstance || defaultPipeline;
|
|
if (!pipeline) {
|
|
throw new Error('No pipeline provided and default pipeline not set. Call setDefaultPipeline first.');
|
|
}
|
|
// Register all augmentations with the pipeline
|
|
for (const augmentation of availableAugmentations) {
|
|
if (augmentation.enabled) {
|
|
pipeline.register(augmentation);
|
|
}
|
|
}
|
|
return pipeline;
|
|
}
|
|
/**
|
|
* Enables or disables an augmentation by name
|
|
*
|
|
* @param name The name of the augmentation to enable/disable
|
|
* @param enabled Whether to enable or disable the augmentation
|
|
* @returns True if the augmentation was found and updated, false otherwise
|
|
*/
|
|
export function setAugmentationEnabled(name, enabled) {
|
|
const augmentation = availableAugmentations.find(aug => aug.name === name);
|
|
if (augmentation) {
|
|
augmentation.enabled = enabled;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
/**
|
|
* Gets all augmentations of a specific type
|
|
*
|
|
* @param type The type of augmentation to get
|
|
* @returns An array of all augmentations of the specified type
|
|
*/
|
|
export function getAugmentationsByType(type) {
|
|
return availableAugmentations.filter(aug => {
|
|
// Check if the augmentation is of the specified type
|
|
// This is a simplified check and may need to be updated based on how types are determined
|
|
switch (type) {
|
|
case AugmentationType.SENSE:
|
|
return 'processRawData' in aug && 'listenToFeed' in aug;
|
|
case AugmentationType.CONDUIT:
|
|
return 'establishConnection' in aug && 'readData' in aug && 'writeData' in aug;
|
|
case AugmentationType.COGNITION:
|
|
return 'reason' in aug && 'infer' in aug && 'executeLogic' in aug;
|
|
case AugmentationType.MEMORY:
|
|
return 'storeData' in aug && 'retrieveData' in aug && 'updateData' in aug;
|
|
case AugmentationType.PERCEPTION:
|
|
return 'interpret' in aug && 'organize' in aug && 'generateVisualization' in aug;
|
|
case AugmentationType.DIALOG:
|
|
return 'processUserInput' in aug && 'generateResponse' in aug && 'manageContext' in aug;
|
|
case AugmentationType.ACTIVATION:
|
|
return 'triggerAction' in aug && 'generateOutput' in aug && 'interactExternal' in aug;
|
|
case AugmentationType.WEBSOCKET:
|
|
return 'connectWebSocket' in aug && 'sendWebSocketMessage' in aug && 'onWebSocketMessage' in aug;
|
|
default:
|
|
return false;
|
|
}
|
|
});
|
|
}
|
|
//# sourceMappingURL=augmentationRegistry.js.map
|