@openfluke/welvet 0.1.3 → 0.1.4

package/dist/types.d.ts

@@ -1,211 +1,105 @@
 /**
- * LOOM Network instance returned by NewNetwork()
- * All Network methods are dynamically exposed via reflection
+ * Type definitions for LOOM WASM API
  */
-export interface LoomNetwork {
-    ForwardCPU(inputJSON: string): string;
-    ForwardGPU?(inputJSON: string): string;
-    BackwardCPU(gradOutputJSON: string): string;
-    BackwardGPU?(gradOutputJSON: string): string;
-    UpdateWeights(learningRateJSON: string): string;
-    Train(batchesAndConfigJSON: string): string;
-    SetLayer(configJSON: string): string;
-    InitGPU?(argsJSON: string): string;
-    ReleaseGPU?(): string;
-    SaveModelToString(modelIDJSON: string): string;
-    SaveModel?(pathAndIDJSON: string): string;
-    GetMethods(): string;
-    ListMethods(): string;
-    GetMethodSignature(methodNameJSON: string): string;
-    HasMethod(methodName: string): boolean;
-    [key: string]: any;
+export interface LayerConfig {
+    type: string;
+    input_size?: number;
+    output_size?: number;
+    hidden_size?: number;
+    seq_length?: number;
+    activation?: string;
+    combine_mode?: string;
+    grid_output_rows?: number;
+    grid_output_cols?: number;
+    grid_output_layers?: number;
+    grid_positions?: GridPosition[];
+    branches?: LayerConfig[];
 }
-/**
- * Global WASM functions exposed by LOOM
- */
-export interface LoomAPI {
-    /**
-     * Create a new neural network
-     * @param inputSize - Size of input layer
-     * @param gridRows - Number of grid rows
-     * @param gridCols - Number of grid columns
-     * @param layersPerCell - Number of layers per grid cell
-     * @returns LoomNetwork instance with all methods
-     */
-    NewNetwork: (inputSize: number, gridRows: number, gridCols: number, layersPerCell: number) => LoomNetwork;
-    /**
-     * Load a model from JSON string
-     * @param modelJSON - JSON string of the model
-     * @param modelID - Model identifier
-     * @returns LoomNetwork instance
-     */
-    LoadModelFromString: (modelJSON: string, modelID: string) => LoomNetwork;
-    /**
-     * Call any layer initialization function from the registry
-     * @param functionName - Name of the layer init function (e.g., "InitDenseLayer")
-     * @param paramsJSON - JSON string of parameters array
-     * @returns JSON string of layer configuration
-     */
-    CallLayerInit: (functionName: string, paramsJSON: string) => string;
-    /**
-     * Initialize a dense (fully-connected) layer configuration
-     * @param inputSize - Input dimension
-     * @param outputSize - Output dimension
-     * @param activation - Activation type (0=ReLU, 1=Sigmoid, 2=Tanh, 3=Linear)
-     * @returns JSON string of layer configuration
-     */
-    InitDenseLayer: (inputSize: number, outputSize: number, activation: number) => string;
-    /**
-     * Initialize a multi-head attention layer configuration
-     * @param dModel - Model dimension
-     * @param numHeads - Number of attention heads
-     * @param seqLength - Sequence length
-     * @param activation - Activation type
-     * @returns JSON string of layer configuration
-     */
-    InitMultiHeadAttentionLayer: (dModel: number, numHeads: number, seqLength: number, activation: number) => string;
-    /**
-     * Initialize a 2D convolutional layer configuration
-     * @param inputHeight - Input height
-     * @param inputWidth - Input width
-     * @param inputChannels - Number of input channels
-     * @param kernelSize - Kernel size
-     * @param stride - Stride
-     * @param padding - Padding
-     * @param filters - Number of filters
-     * @param activation - Activation type
-     * @returns JSON string of layer configuration
-     */
-    InitConv2DLayer: (inputHeight: number, inputWidth: number, inputChannels: number, kernelSize: number, stride: number, padding: number, filters: number, activation: number) => string;
-    /**
-     * Initialize an RNN layer configuration
-     * @param inputSize - Input size
-     * @param hiddenSize - Hidden size
-     * @param batchSize - Batch size
-     * @param seqLength - Sequence length
-     * @returns JSON string of layer configuration
-     */
-    InitRNNLayer: (inputSize: number, hiddenSize: number, batchSize: number, seqLength: number) => string;
-    /**
-     * Initialize an LSTM layer configuration
-     * @param inputSize - Input size
-     * @param hiddenSize - Hidden size
-     * @param batchSize - Batch size
-     * @param seqLength - Sequence length
-     * @returns JSON string of layer configuration
-     */
-    InitLSTMLayer: (inputSize: number, hiddenSize: number, batchSize: number, seqLength: number) => string;
+export interface NetworkConfig {
+    batch_size: number;
+    grid_rows?: number;
+    grid_cols?: number;
+    layers_per_cell?: number;
+    layers: LayerConfig[];
 }
-export interface InitOptions {
-    /** Override where the WASM is read from. Useful in exotic deploys (Capacitor, CDN). */
-    wasmUrl?: string | URL;
-    /** Set false to skip injecting wasm_exec.js (e.g., if you already included it). Default: true */
-    injectGoRuntime?: boolean;
+export interface GridPosition {
+    branch_index: number;
+    target_row: number;
+    target_col: number;
+    target_layer: number;
 }
-/**
- * Activation function types
- */
-export declare enum ActivationType {
-    ReLU = 0,
-    Sigmoid = 1,
-    Tanh = 2,
-    Softplus = 3,
-    LeakyReLU = 4,
-    Linear = 5
+export interface TrainingBatch {
+    Input: number[];
+    Target: number[];
 }
-/**
- * Result from tokenizer loading
- */
-export interface TokenizerLoadResult {
-    success: boolean;
-    vocab_size?: number;
-    message?: string;
-    error?: string;
+export interface TrainingConfig {
+    Epochs: number;
+    LearningRate: number;
+    LossType?: string;
+    Verbose?: boolean;
+    UseGPU?: boolean;
+    PrintEveryBatch?: number;
+    GradientClip?: number;
 }
-/**
- * Result from transformer model loading
- */
-export interface TransformerLoadResult {
-    success: boolean;
-    num_layers?: number;
-    hidden_size?: number;
-    vocab_size?: number;
-    message?: string;
-    error?: string;
-}
-/**
- * Result from text encoding
- */
-export interface EncodeResult {
-    success: boolean;
-    ids?: number[];
-    error?: string;
-}
-/**
- * Result from token decoding
- */
-export interface DecodeResult {
-    success: boolean;
-    text?: string;
-    error?: string;
-}
-/**
- * Result from text generation
- */
-export interface GenerateResult {
-    success: boolean;
-    generated_text?: string;
-    error?: string;
+export interface TrainingResult {
+    final_loss: number;
+    duration_ms: number;
+    epochs_completed: number;
 }
 /**
- * Result from next token generation
+ * Network interface with all available methods
+ * All methods take a JSON string of parameters (JSON array)
+ * and return a JSON string of results (JSON array)
  */
-export interface NextTokenResult {
-    success: boolean;
-    token?: number;
-    is_eos?: boolean;
-    error?: string;
+export interface Network {
+    ForwardCPU(inputsJSON: string): string;
+    ForwardGPU(inputsJSON: string): string;
+    BackwardCPU(gradientsJSON: string): string;
+    BackwardGPU(gradientsJSON: string): string;
+    UpdateWeights(paramsJSON: string): string;
+    Train(paramsJSON: string): string;
+    ZeroGradients(paramsJSON: string): string;
+    ResetGradients(paramsJSON: string): string;
+    SaveModelToString(paramsJSON: string): string;
+    SerializeModel(paramsJSON: string): string;
+    GetWeights(paramsJSON: string): string;
+    SetWeights(paramsJSON: string): string;
+    GetBiases(paramsJSON: string): string;
+    SetBiases(paramsJSON: string): string;
+    GetLayer(paramsJSON: string): string;
+    SetLayer(paramsJSON: string): string;
+    GetActivation(paramsJSON: string): string;
+    SetActivation(paramsJSON: string): string;
+    GetLayerType(paramsJSON: string): string;
+    GetLayerSizes(paramsJSON: string): string;
+    GetBatchSize(paramsJSON: string): string;
+    SetBatchSize(paramsJSON: string): string;
+    GetGridDimensions(paramsJSON: string): string;
+    GetLayersPerCell(paramsJSON: string): string;
+    TotalLayers(paramsJSON: string): string;
+    GetNetworkInfo(paramsJSON: string): string;
+    GetTotalParameters(paramsJSON: string): string;
+    GetMemoryUsage(paramsJSON: string): string;
+    ValidateArchitecture(paramsJSON: string): string;
+    GetLastOutput(paramsJSON: string): string;
+    GetLastGradients(paramsJSON: string): string;
+    Activations(paramsJSON: string): string;
+    KernelGradients(paramsJSON: string): string;
+    BiasGradients(paramsJSON: string): string;
+    Clone(paramsJSON: string): string;
+    InitializeWeights(paramsJSON: string): string;
+    InitGPU(paramsJSON: string): string;
+    ReleaseGPU(paramsJSON: string): string;
+    GetMethods(paramsJSON: string): string;
+    GetMethodsJSON(paramsJSON: string): string;
+    ListMethods(paramsJSON: string): string;
+    HasMethod(paramsJSON: string): string;
+    GetMethodSignature(paramsJSON: string): string;
 }
 /**
- * Transformer API for LLM inference
+ * Global WASM functions exposed by main.go
+ * Only createLoomNetwork is exposed - use network.SaveModelToString() for saving
  */
-export interface TransformerAPI {
-    /**
-     * Load tokenizer from JSON bytes
-     * @param tokenizerData - Uint8Array of tokenizer.json file
-     */
-    loadTokenizer(tokenizerData: Uint8Array): Promise<TokenizerLoadResult>;
-    /**
-     * Load transformer model from config and weights bytes
-     * @param configData - Uint8Array of config.json file
-     * @param weightsData - Uint8Array of model.safetensors file
-     */
-    loadModel(configData: Uint8Array, weightsData: Uint8Array): Promise<TransformerLoadResult>;
-    /**
-     * Encode text to token IDs
-     * @param text - Input text to encode
-     * @param addSpecialTokens - Whether to add special tokens (default: true)
-     */
-    encode(text: string, addSpecialTokens?: boolean): Promise<EncodeResult>;
-    /**
-     * Decode token IDs to text
-     * @param tokenIds - Array of token IDs
-     * @param skipSpecialTokens - Whether to skip special tokens (default: true)
-     */
-    decode(tokenIds: number[], skipSpecialTokens?: boolean): Promise<DecodeResult>;
-    /**
-     * Generate text from prompt (blocking, all tokens at once)
-     * @param prompt - Input prompt
-     * @param maxTokens - Maximum tokens to generate (default: 50)
-     * @param temperature - Sampling temperature (default: 0.7)
-     */
-    generate(prompt: string, maxTokens?: number, temperature?: number): Promise<GenerateResult>;
-    /**
-     * Generate text token-by-token (streaming)
-     * @param prompt - Input prompt
-     * @param maxTokens - Maximum tokens to generate (default: 50)
-     * @param temperature - Sampling temperature (default: 0.7)
-     * @yields Token text strings
-     */
-    generateStream(prompt: string, maxTokens?: number, temperature?: number): AsyncGenerator<string, void, unknown>;
+declare global {
+    function createLoomNetwork(jsonConfig: string): Network;
 }

package/dist/types.js

@@ -1,12 +1,4 @@
 /**
- * Activation function types
+ * Type definitions for LOOM WASM API
  */
-export var ActivationType;
-(function (ActivationType) {
-    ActivationType[ActivationType["ReLU"] = 0] = "ReLU";
-    ActivationType[ActivationType["Sigmoid"] = 1] = "Sigmoid";
-    ActivationType[ActivationType["Tanh"] = 2] = "Tanh";
-    ActivationType[ActivationType["Softplus"] = 3] = "Softplus";
-    ActivationType[ActivationType["LeakyReLU"] = 4] = "LeakyReLU";
-    ActivationType[ActivationType["Linear"] = 5] = "Linear";
-})(ActivationType || (ActivationType = {}));
+export {};