[11.x] Add vector column support to migrations (#52884)

* Add vector column support to Blueprint and Schema grammars

* FIX sql not defined

* MOD remove redundant param on doc

* FIX code styling

* FIX code styling

* formatting

* remove method from sqlite grammar

---------

Co-authored-by: Taylor Otwell <[email protected]>

Author: Jim-Webfox

src/Illuminate/Database/Schema/Blueprint.php

@@ -1452,6 +1452,18 @@ public function computed($column, $expression)
         return $this->addColumn('computed', $column, compact('expression'));
     }
 
+    /**
+     * Create a new vector column on the table.
+     *
+     * @param  string  $column
+     * @param  int  $dimensions
+     * @return \Illuminate\Database\Schema\ColumnDefinition
+     */
+    public function vector($column, $dimensions)
+    {
+        return $this->addColumn('vector', $column, compact('dimensions'));
+    }
+
     /**
      * Add the proper columns for a polymorphic table.
      *

src/Illuminate/Database/Schema/Grammars/Grammar.php

@@ -234,6 +234,17 @@ protected function typeComputed(Fluent $column)
         throw new RuntimeException('This database driver does not support the computed type.');
     }
 
+    /**
+     * Create the column definition for a vector type.
+     *
+     * @param  \Illuminate\Support\Fluent  $column
+     * @return string
+     */
+    protected function typeVector(Fluent $column)
+    {
+        throw new RuntimeException('This database driver does not support the vector type.');
+    }
+
     /**
      * Add the column modifiers to the definition.
      *

src/Illuminate/Database/Schema/Grammars/MySqlGrammar.php

@@ -1103,6 +1103,17 @@ protected function typeComputed(Fluent $column)
         throw new RuntimeException('This database driver requires a type, see the virtualAs / storedAs modifiers.');
     }
 
+    /**
+     * Create the column definition for a vector type.
+     *
+     * @param  \Illuminate\Support\Fluent  $column
+     * @return string
+     */
+    protected function typeVector(Fluent $column)
+    {
+        return "vector($column->dimensions)";
+    }
+
     /**
      * Get the SQL for a generated virtual column modifier.
      *

src/Illuminate/Database/Schema/Grammars/PostgresGrammar.php

@@ -1053,6 +1053,17 @@ protected function typeGeography(Fluent $column)
         return 'geography';
     }
 
+    /**
+     * Create the column definition for a vector type.
+     *
+     * @param  \Illuminate\Support\Fluent  $column
+     * @return string
+     */
+    protected function typeVector(Fluent $column)
+    {
+        return "vector($column->dimensions)";
+    }
+
     /**
      * Get the SQL for a collation column modifier.
      *

tests/Database/DatabaseMySqlSchemaGrammarTest.php

@@ -1322,6 +1322,16 @@ public function testAddingComment()
         $this->assertSame("alter table `users` add `foo` varchar(255) not null comment 'Escape \\' when using words like it\\'s'", $statements[0]);
     }
 
+    public function testAddingVector()
+    {
+        $blueprint = new Blueprint('embeddings');
+        $blueprint->vector('embedding', 384);
+        $statements = $blueprint->toSql($this->getConnection(), $this->getGrammar());
+
+        $this->assertCount(1, $statements);
+        $this->assertSame('alter table `embeddings` add `embedding` vector(384) not null', $statements[0]);
+    }
+
     public function testCreateDatabase()
     {
         $connection = $this->getConnection();

tests/Database/DatabasePostgresSchemaGrammarTest.php

@@ -42,6 +42,16 @@ public function testBasicCreateTable()
         ], $statements);
     }
 
+    public function testAddingVector()
+    {
+        $blueprint = new Blueprint('embeddings');
+        $blueprint->vector('embedding', 384);
+        $statements = $blueprint->toSql($this->getConnection(), $this->getGrammar());
+
+        $this->assertCount(1, $statements);
+        $this->assertSame('alter table "embeddings" add column "embedding" vector(384) not null', $statements[0]);
+    }
+
     public function testCreateTableWithAutoIncrementStartingValue()
     {
         $blueprint = new Blueprint('users');