feat: expose raw parsing function

ahoylabs · Apr 26, 2024 · 5d0af69 · 5d0af69
1 parent 14fe924
commit 5d0af69
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -395,7 +395,12 @@ type ParsedMetadata =
   | { error: Error; metadata?: undefined }
   | { error?: undefined; metadata: GGUFMetadata }
 
-const parseMetadata = async (filePath: string): Promise<ParsedMetadata> => {
+type RawMetadata =
+  | { error: Error }
+  | { error?: null; metadata: Record<string, any> }
+export const parseRawMetadata = async (
+  filePath: string,
+): Promise<RawMetadata> => {
   const metadata = await new Promise<
     { error: Error } | { error?: null; metadata: Record<string, any> }
   >((resolve) => {
@@ -554,6 +559,12 @@ const parseMetadata = async (filePath: string): Promise<ParsedMetadata> => {
     })
   })
 
+  return metadata
+}
+
+const parseMetadata = async (filePath: string): Promise<ParsedMetadata> => {
+  const metadata = await parseRawMetadata(filePath)
+
   if (metadata.error) return { error: metadata.error }
 
   const validationError = validateMetadata({