pgvector 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1699eea3d6ebb0ede69e22c38fe8596482577b2258904a76c877295bc9008e8f
-  data.tar.gz: d7ecf8fb6dc2d3bf781a5d65d07bc5abea5f768a268720fd93e6de6e4b778885
+  metadata.gz: a0729ce3ec22ea08be365fc6c7a5c5e0b038860d5c9a57ccd2554d9d8b57a3f2
+  data.tar.gz: c36eabc18ca6db3dd47204df3c48b58f7ac635e230bb9039269506d7d8afa834
 SHA512:
-  metadata.gz: 010f1f0c74ac1ef69bba4953a50b24d106f359b9e77c8f7bc282d1272008b0376f1b9d8fe72a67c1325bf49b1bd0fa29f6fb5aa139966a46eda0478b3957d56a
-  data.tar.gz: e21326c03271c19bd8babcf43d29b3896ef30f0bd1011284c52e6f980e40400f9bc4b56345130968d401075d615ab3d5df7198730899bebaab8e36b6ccfb6f47
+  metadata.gz: ff4dc7d6a7a26b1ff3502d2305b62fc25fff38f3bf314261dcb6aa0f40cc520963cfd9562af03b11678cad7819aef9951b480a61e138ae2ae02459ed668a5298
+  data.tar.gz: 6aa39019d5566f83331f3c95bb6404b065b7f7a29be62e84be2f2b6ea58bc27d20002543d93dfec7a17256c970e32b9676e2d319d07758cf49fc31b89d3d4d7b

data/CHANGELOG.md

@@ -1,3 +1,12 @@
+## 0.2.0 (2023-05-11)
+
+- Dropped support for Ruby < 3
+
+## 0.1.1 (2022-02-08)
+
+- Added autoloading
+- Added binary decoder
+
 ## 0.1.0 (2022-02-07)
 
 - First release

data/LICENSE.txt

@@ -1,6 +1,6 @@
 The MIT License (MIT)
 
-Copyright (c) 2022 Andrew Kane
+Copyright (c) 2022-2023 Andrew Kane
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

data/README.md

@@ -25,8 +25,6 @@ And follow the instructions for your database library:
 Register the vector type with your connection
 
 ```ruby
-require "pgvector/pg"
-
 registry = PG::BasicTypeRegistry.new.define_default_types
 Pgvector::PG.register_vector(registry)
 conn.type_map_for_results = PG::BasicTypeMapForResults.new(conn, registry: registry)
@@ -35,14 +33,14 @@ conn.type_map_for_results = PG::BasicTypeMapForResults.new(conn, registry: regis
 Insert a vector
 
 ```ruby
-factors = [1, 2, 3]
-conn.exec_params("INSERT INTO items (factors) VALUES ($1)", [factors])
+embedding = [1, 2, 3]
+conn.exec_params("INSERT INTO items (embedding) VALUES ($1)", [embedding])
 ```
 
 Get the nearest neighbors to a vector
 
 ```ruby
-conn.exec_params("SELECT * FROM items ORDER BY factors <-> $1 LIMIT 5", [factors]).to_a
+conn.exec_params("SELECT * FROM items ORDER BY embedding <-> $1 LIMIT 5", [embedding]).to_a
 ```
 
 ## History

data/lib/pgvector/pg.rb

@@ -4,6 +4,17 @@ module Pgvector
   module PG
     def self.register_vector(registry)
       registry.register_type(0, "vector", nil, TextDecoder::Vector)
+      registry.register_type(1, "vector", nil, BinaryDecoder::Vector)
+    end
+
+    module BinaryDecoder
+      class Vector < ::PG::SimpleDecoder
+        def decode(string, tuple = nil, field = nil)
+          dim, unused = string[0, 4].unpack("nn")
+          raise "expected unused to be 0" if unused != 0
+          string[4..-1].unpack("g#{dim}")
+        end
+      end
     end
 
     module TextDecoder

data/lib/pgvector/version.rb

@@ -1,3 +1,3 @@
 module Pgvector
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/pgvector.rb

@@ -1 +1,5 @@
 require "pgvector/version"
+
+module Pgvector
+  autoload :PG, "pgvector/pg"
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pgvector
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-02-07 00:00:00.000000000 Z
+date: 2023-05-12 00:00:00.000000000 Z
 dependencies: []
 description:
 email: andrew@ankane.org
@@ -34,14 +34,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '2.6'
+      version: '3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.3
+rubygems_version: 3.4.10
 signing_key:
 specification_version: 4
 summary: pgvector support for Ruby