Add vector support

docs/entities.md

@@ -393,6 +393,8 @@ or
 `tsrange`, `tstzrange`, `daterange`, `int4multirange`, `int8multirange`, `nummultirange`,
 `tsmultirange`, `tstzmultirange`, `multidaterange`, `geometry`, `geography`, `cube`, `ltree`
 
+> Note: `vector` is also available when the [pgvector](https://github.com/pgvector/pgvector) extension is supported by the postgres installation. You can follow installation instructions from the link in this note.
+
 ### Column types for `cockroachdb`
 
 `array`, `bool`, `boolean`, `bytes`, `bytea`, `blob`, `date`, `numeric`, `decimal`, `dec`, `float`,

src/driver/postgres/PostgresDriver.ts

@@ -190,6 +190,7 @@ export class PostgresDriver implements Driver {
         "geography",
         "cube",
         "ltree",
+        "vector",
     ]
 
     /**
@@ -213,6 +214,7 @@ export class PostgresDriver implements Driver {
         "bit",
         "varbit",
         "bit varying",
+        "vector",
     ]
 
     /**
@@ -421,6 +423,7 @@ export class PostgresDriver implements Driver {
             hasCubeColumns,
             hasGeometryColumns,
             hasLtreeColumns,
+            hasVectorColumns,
             hasExclusionConstraints,
         } = extensionsMetadata
 
@@ -500,6 +503,18 @@ export class PostgresDriver implements Driver {
                     "At least one of the entities has a ltree column, but the 'ltree' extension cannot be installed automatically. Please install it manually using superuser rights",
                 )
             }
+        if (hasVectorColumns)
+            try {
+                await this.executeQuery(
+                    connection,
+                    `CREATE EXTENSION IF NOT EXISTS "vector"`,
+                )
+            } catch (_) {
+                logger.log(
+                    "warn",
+                    "At least one of the entities has a vector column, but the 'vector' extension cannot be installed automatically. Please install it manually using superuser rights",
+                )
+            }
         if (hasExclusionConstraints)
             try {
                 // The btree_gist extension provides operator support in PostgreSQL exclusion constraints
@@ -568,6 +583,15 @@ export class PostgresDriver implements Driver {
                 )
             },
         )
+        const hasVectorColumns = this.connection.entityMetadatas.some(
+            (metadata) => {
+                return (
+                    metadata.columns.filter(
+                        (column) => column.type === "vector",
+                    ).length > 0
+                )
+            },
+        )
         const hasExclusionConstraints = this.connection.entityMetadatas.some(
             (metadata) => {
                 return metadata.exclusions.length > 0
@@ -581,6 +605,7 @@ export class PostgresDriver implements Driver {
             hasCubeColumns,
             hasGeometryColumns,
             hasLtreeColumns,
+            hasVectorColumns,
             hasExclusionConstraints,
             hasExtensions:
                 hasUuidColumns ||
@@ -589,6 +614,7 @@ export class PostgresDriver implements Driver {
                 hasGeometryColumns ||
                 hasCubeColumns ||
                 hasLtreeColumns ||
+                hasVectorColumns ||
                 hasExclusionConstraints,
         }
     }

src/driver/postgres/PostgresQueryRunner.ts

@@ -601,10 +601,22 @@ export class PostgresQueryRunner
                 downQueries.push(this.dropIndexSql(table, index))
             })
         }
-        
+
         if (table.comment) {
-            upQueries.push(new Query("COMMENT ON TABLE " + this.escapePath(table) + " IS '" + table.comment + "'"));
-            downQueries.push(new Query("COMMENT ON TABLE " + this.escapePath(table) + " IS NULL"));
+            upQueries.push(
+                new Query(
+                    "COMMENT ON TABLE " +
+                        this.escapePath(table) +
+                        " IS '" +
+                        table.comment +
+                        "'",
+                ),
+            )
+            downQueries.push(
+                new Query(
+                    "COMMENT ON TABLE " + this.escapePath(table) + " IS NULL",
+                ),
+            )
         }
 
         await this.executeQueries(upQueries, downQueries)
@@ -3542,8 +3554,9 @@ export class PostgresQueryRunner
                             // check if column has user-defined data type.
                             // NOTE: if ENUM type defined with "array:true" it comes with ARRAY type instead of USER-DEFINED
                             if (
-                                dbColumn["data_type"] === "USER-DEFINED" ||
-                                dbColumn["data_type"] === "ARRAY"
+                                (dbColumn["data_type"] === "USER-DEFINED" ||
+                                    dbColumn["data_type"] === "ARRAY") &&
+                                dbColumn["udt_name"] !== "vector"
                             ) {
                                 const { name } =
                                     await this.getUserDefinedTypeName(
@@ -3595,6 +3608,20 @@ export class PostgresQueryRunner
                                 }
                             }
 
+                            if (
+                                dbColumn["data_type"] === "USER-DEFINED" &&
+                                tableColumn.type === "vector"
+                            ) {
+                                // vector length is from the format_type e.g. `format_type = vector(n)`
+                                const match = /vector\((\d+)\)/.exec(
+                                    dbColumn["format_type"],
+                                )
+
+                                if (match) {
+                                    tableColumn.length = match[1]
+                                }
+                            }
+
                             if (
                                 tableColumn.type === "geometry" ||
                                 tableColumn.type === "geography"
@@ -4744,7 +4771,7 @@ export class PostgresQueryRunner
 
         newComment = this.escapeComment(newComment)
         const comment = this.escapeComment(table.comment)
-        
+
         if (newComment === comment) {
             return
         }

src/driver/types/ColumnTypes.ts

@@ -75,6 +75,7 @@ export type WithLengthColumnType =
     | "binary" // mssql
     | "varbinary" // mssql, sap
     | "string" // cockroachdb, spanner
+    | "vector" // postgres pgvector
 
 export type WithWidthColumnType =
     | "tinyint" // mysql

src/query-builder/InsertQueryBuilder.ts

@@ -459,10 +459,8 @@ export class InsertQueryBuilder<
         // add VALUES expression
         if (valuesExpression) {
             if (
-                (
-                    this.connection.driver.options.type === "oracle" ||
-                    this.connection.driver.options.type === "sap"
-                ) &&
+                (this.connection.driver.options.type === "oracle" ||
+                    this.connection.driver.options.type === "sap") &&
                 this.getValueSets().length > 1
             ) {
                 query += ` ${valuesExpression}`

src/query-builder/QueryBuilder.ts

@@ -522,7 +522,10 @@ export abstract class QueryBuilder<Entity extends ObjectLiteral> {
      * Uses same query runner as current QueryBuilder.
      */
     createQueryBuilder(queryRunner?: QueryRunner): this {
-        return new (this.constructor as any)(this.connection, queryRunner ?? this.queryRunner)
+        return new (this.constructor as any)(
+            this.connection,
+            queryRunner ?? this.queryRunner,
+        )
     }
 
     /**

src/schema-builder/RdbmsSchemaBuilder.ts

@@ -604,7 +604,7 @@ export class RdbmsSchemaBuilder implements SchemaBuilder {
 
             if (
                 DriverUtils.isMySQLFamily(this.connection.driver) ||
-                this.connection.driver.options.type === 'postgres'
+                this.connection.driver.options.type === "postgres"
             ) {
                 const newComment = metadata.comment
                 await this.queryRunner.changeTableComment(table, newComment)

test/github-issues/10056/entity/message.ts

@@ -0,0 +1,13 @@
+import { Column, Entity, PrimaryGeneratedColumn } from "../../../../src"
+
+@Entity({
+    name: "message",
+})
+export class Message {
+    @PrimaryGeneratedColumn()
+    id: number
+
+    // create a vector embedding with 5 dimensions
+    @Column("vector", { length: 5 })
+    embedding: string
+}