Added Pgvector.extensions/0 function

pgvector · May 18, 2024 · d5e574a · d5e574a
1 parent aa479fe
commit d5e574a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,6 @@
 ## 0.2.2 (unreleased)
 
+- Added `Pgvector.extensions/0` function
 - Added `l1_distance` function for Ecto
 
 ## 0.2.1 (2023-09-25)

diff --git a/lib/pgvector.ex b/lib/pgvector.ex
@@ -77,6 +77,15 @@ defmodule Pgvector do
       end
     end
   end
+
+  @doc """
+  Extensions for Postgrex
+  """
+  def extensions do
+    [
+      Pgvector.Extensions.Vector
+    ]
+  end
 end
 
 defimpl Inspect, for: Pgvector do

diff --git a/test/test_helper.exs b/test/test_helper.exs
@@ -1,4 +1,4 @@
-Postgrex.Types.define(EctoApp.PostgrexTypes, [Pgvector.Extensions.Vector] ++ Ecto.Adapters.Postgres.extensions(), [])
+Postgrex.Types.define(EctoApp.PostgrexTypes, Pgvector.extensions() ++ Ecto.Adapters.Postgres.extensions(), [])
 
 defmodule Repo do
   use Ecto.Repo,