Improved code

pgvector · Feb 17, 2025 · 0d09492 · 0d09492
1 parent f1d8f8d
commit 0d09492
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 9 deletions.
diff --git a/Sources/PgvectorNIO/HalfVector.swift b/Sources/PgvectorNIO/HalfVector.swift
@@ -1,10 +1,14 @@
 import Pgvector
 import PostgresNIO
 
-extension HalfVector: @retroactive PostgresEncodable {
-    public static var psqlType: PostgresDataType = PostgresDataType(1)
+extension HalfVector: @retroactive PostgresDynamicTypeEncodable {
+    public static var psqlType: PostgresDataType?
 
-    public static var psqlFormat: PostgresFormat {
+    public var psqlType: PostgresDataType {
+        HalfVector.psqlType!
+    }
+
+    public var psqlFormat: PostgresFormat {
         .binary
     }
 

diff --git a/Sources/PgvectorNIO/SparseVector.swift b/Sources/PgvectorNIO/SparseVector.swift
@@ -1,10 +1,14 @@
 import Pgvector
 import PostgresNIO
 
-extension SparseVector: @retroactive PostgresEncodable {
-    public static var psqlType: PostgresDataType = PostgresDataType(1)
+extension SparseVector: @retroactive PostgresDynamicTypeEncodable {
+    public static var psqlType: PostgresDataType?
 
-    public static var psqlFormat: PostgresFormat {
+    public var psqlType: PostgresDataType {
+        SparseVector.psqlType!
+    }
+
+    public var psqlFormat: PostgresFormat {
         .binary
     }
 

diff --git a/Sources/PgvectorNIO/Vector.swift b/Sources/PgvectorNIO/Vector.swift
@@ -1,10 +1,14 @@
 import Pgvector
 import PostgresNIO
 
-extension Vector: @retroactive PostgresEncodable {
-    public static var psqlType: PostgresDataType = PostgresDataType(1)
+extension Vector: @retroactive PostgresDynamicTypeEncodable {
+    public static var psqlType: PostgresDataType?
 
-    public static var psqlFormat: PostgresFormat {
+    public var psqlType: PostgresDataType {
+        Vector.psqlType!
+    }
+
+    public var psqlFormat: PostgresFormat {
         .binary
     }