export * from '../_common/utils/sql';
import { TextEmbeddingModelType, ImageEmbeddingModelType, OpenAITextEmbeddingModelType, CohereTextEmbeddingModelType } from '../_embeddings/model-types';

declare module 'knex' {
  interface Knex {
    generateTextEmbedding: (modelKey: TextEmbeddingModelType, value: string) => Promise<any>;
    generateImageEmbedding: (modelKey: ImageEmbeddingModelType, value: string) => Promise<any>;
  }
  namespace Knex {
    interface SchemaBuilder {
      createLanternExtension: () => Promise<void>;
      createLanternExtrasExtension: () => Promise<void>;
    }
  }
}

export function textEmbedding(modelKey: TextEmbeddingModelType, column: string): any;
export function imageEmbedding(modelKey: ImageEmbeddingModelType, column: string): any;
export function openaiEmbedding(modelKey: OpenAITextEmbeddingModelType, column: string, dimension?: number): any;
export function cohereEmbedding(modelKey: CohereTextEmbeddingModelType, column: string): any;

export function l2Distance(column: string, value: number[] | string): any;
export function cosineDistance(column: string, value: number[] | string): any;
export function hammingDistance(column: string, value: number[] | string): any;