62 lines
2.9 KiB
TypeScript
62 lines
2.9 KiB
TypeScript
|
// This file is auto-generated by alef — DO NOT EDIT.
|
||
|
|
// alef:hash:4e15143f4af1ae8bafbdb1506ef057da924484c66a19483966333558ad437e75
|
||
|
|
// To regenerate: alef generate
|
||
|
|
// To verify freshness: alef verify --exit-code
|
||
|
|
// Issues & docs: https://github.com/kreuzberg-dev/alef
|
||
|
|
|
||
|
|
import { describe, expect, it } from 'vitest';import { extractFile, embedTexts, getEmbeddingPreset, listEmbeddingPresets, type EmbeddingConfig, type ExtractionConfig } from '@kreuzberg/node';
|
||
|
|
function _alefE2eText(value: unknown): string {
|
||
|
|
return value == null ? "" : String(value);
|
||
|
|
}
|
||
|
|
|
||
|
|
function _alefE2eItemTexts(item: unknown): string[] {
|
||
|
|
if (item == null || typeof item !== "object") {
|
||
|
|
return [_alefE2eText(item)];
|
||
|
|
}
|
||
|
|
const record = item as Record<string, unknown>;
|
||
|
|
const itemsText = Array.isArray(record.items) ? record.items.map(_alefE2eText).join(" ") : "";
|
||
|
|
return [_alefE2eText(item), _alefE2eText(record.kind), _alefE2eText(record.name), _alefE2eText(record.source), _alefE2eText(record.alias), _alefE2eText(record.text), _alefE2eText(record.signature), itemsText];
|
||
|
|
}
|
||
|
|
|
||
|
|
function _alefE2eFormatMetadataDisplay(fm: unknown): string {
|
||
|
|
if (fm == null) return "";
|
||
|
|
if (typeof fm !== "object") return String(fm);
|
||
|
|
const record = fm as Record<string, unknown>;
|
||
|
|
const formatType = record.format_type;
|
||
|
|
|
||
|
|
// FormatMetadata is a tagged union: { format_type: 'image', image: { format: 'PNG', ... }, ... }
|
||
|
|
// Extract the display string based on the variant type
|
||
|
|
if (formatType === "image" && typeof record.image === "object") {
|
||
|
|
const imageData = record.image as Record<string, unknown>;
|
||
|
|
if (typeof imageData.format === "string") return imageData.format;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fallback: return format_type variant name
|
||
|
|
if (typeof record.format_type === "string") return record.format_type;
|
||
|
|
return "";
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
describe('embeddings', () => { it('embed_texts_different_preset: embed_texts: multilingual preset', () => { const result = embedTexts(["Hello world", "Test"], { model: { name: "multilingual", type: "preset" } } as ExtractionConfig); expect(result.length).toBeGreaterThanOrEqual(2);
|
||
|
|
|
||
|
|
}, 600000);
|
||
|
|
it('get_embedding_preset_known: get_embedding_preset: known preset', () => { getEmbeddingPreset("balanced");
|
||
|
|
}, 600000);
|
||
|
|
it('get_embedding_preset_nominal: get_embedding_preset: nominal case', () => { getEmbeddingPreset("balanced");
|
||
|
|
}, 600000);
|
||
|
|
it('get_embedding_preset_unknown: get_embedding_preset: unknown preset fails', () => { const result = getEmbeddingPreset("nonexistent-xyz"); expect((result ?? "").length).toBe(0);
|
||
|
|
|
||
|
|
}, 600000);
|
||
|
|
it('list_embedding_presets_sanity: list_embedding_presets: returns at least one', () => { const result = listEmbeddingPresets(); {
|
||
|
|
const _v = result;
|
||
|
|
if (typeof _v === "string" || Array.isArray(_v)) {
|
||
|
|
expect(_v.length).toBeGreaterThan(0);
|
||
|
|
} else {
|
||
|
|
expect(_v).toBeDefined();
|
||
|
|
expect(_v).not.toBeNull();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}, 600000);
|
||
|
|
});
|