Updates

src/lib/mariadb/db-generate-type-defs.ts

@@ -63,6 +63,10 @@ export default function generateTypeDefinition({
                 return "0 | 1";
             }
 
+            if (schemaType.dataType?.match(/vector/i) || schemaType.isVector) {
+                return `string | number[] | Buffer<ArrayBufferLike> | Float32Array<ArrayBuffer>`;
+            }
+
             return "string";
         }
 

src/types/index.ts

@@ -1479,6 +1479,7 @@ export type SQLInsertGenValueType =
     | number
     | Float32Array<ArrayBuffer>
     | Buffer<ArrayBuffer>
+    | number[]
     | null;
 
 /**

src/utils/sql-insert-generator.ts

@@ -40,15 +40,21 @@ export default function sqlInsertGenerator({
                         .map((ky) => {
                             const value = item[ky];
 
+                            const is_embedding =
+                                Array.isArray(value) &&
+                                typeof value[0] == "number";
+
                             const finalValue =
                                 typeof value == "string" ||
                                 typeof value == "number"
                                     ? value
                                     : typeof value == "function"
                                       ? value().value
-                                      : value
-                                        ? value
-                                        : null;
+                                      : is_embedding
+                                        ? JSON.stringify(value)
+                                        : value
+                                          ? value
+                                          : null;
 
                             if (!finalValue && typeof value !== "number") {
                                 queryValues.push(null);
@@ -60,7 +66,9 @@ export default function sqlInsertGenerator({
                             const placeholder =
                                 typeof value == "function"
                                     ? value().placeholder
-                                    : "?";
+                                    : is_embedding
+                                      ? "VEC_FromText(?)"
+                                      : "?";
 
                             return placeholder;
                         })