feat: vectors RE from DDL

hackolade · Dec 18, 2024 · 9f85064 · 9f85064
1 parent 93c7f18
commit 9f85064
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/reverse_engineering/helpers/getJsonSchema.js b/reverse_engineering/helpers/getJsonSchema.js
@@ -1,3 +1,6 @@
+const _ = require('lodash');
+const { isVector } = require('../../forward_engineering/ddlProvider/ddlHelpers/typeHelper');
+
 const getJsonSchema = columns => {
 	const properties = columns.reduce((properties, column) => {
 		if (column.properties) {
@@ -10,6 +13,16 @@ const getJsonSchema = columns => {
 			};
 		}
 
+		if (isVector(column.type)) {
+			return {
+				...properties,
+				[column.name]: {
+					...column,
+					items: _.fill(Array(column.dimension), { type: 'number', mode: 'real' }),
+				},
+			};
+		}
+
 		return {
 			...properties,
 			[column.name]: column,