pgvector 0.1.0 → 0.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: 1699eea3d6ebb0ede69e22c38fe8596482577b2258904a76c877295bc9008e8f
4
- data.tar.gz: d7ecf8fb6dc2d3bf781a5d65d07bc5abea5f768a268720fd93e6de6e4b778885
3
+ metadata.gz: a0729ce3ec22ea08be365fc6c7a5c5e0b038860d5c9a57ccd2554d9d8b57a3f2
4
+ data.tar.gz: c36eabc18ca6db3dd47204df3c48b58f7ac635e230bb9039269506d7d8afa834
5
5
  SHA512:
6
- metadata.gz: 010f1f0c74ac1ef69bba4953a50b24d106f359b9e77c8f7bc282d1272008b0376f1b9d8fe72a67c1325bf49b1bd0fa29f6fb5aa139966a46eda0478b3957d56a
7
- data.tar.gz: e21326c03271c19bd8babcf43d29b3896ef30f0bd1011284c52e6f980e40400f9bc4b56345130968d401075d615ab3d5df7198730899bebaab8e36b6ccfb6f47
6
+ metadata.gz: ff4dc7d6a7a26b1ff3502d2305b62fc25fff38f3bf314261dcb6aa0f40cc520963cfd9562af03b11678cad7819aef9951b480a61e138ae2ae02459ed668a5298
7
+ data.tar.gz: 6aa39019d5566f83331f3c95bb6404b065b7f7a29be62e84be2f2b6ea58bc27d20002543d93dfec7a17256c970e32b9676e2d319d07758cf49fc31b89d3d4d7b
data/CHANGELOG.md CHANGED
@@ -1,3 +1,12 @@
1
+ ## 0.2.0 (2023-05-11)
2
+
3
+ - Dropped support for Ruby < 3
4
+
5
+ ## 0.1.1 (2022-02-08)
6
+
7
+ - Added autoloading
8
+ - Added binary decoder
9
+
1
10
  ## 0.1.0 (2022-02-07)
2
11
 
3
12
  - First release
data/LICENSE.txt CHANGED
@@ -1,6 +1,6 @@
1
1
  The MIT License (MIT)
2
2
 
3
- Copyright (c) 2022 Andrew Kane
3
+ Copyright (c) 2022-2023 Andrew Kane
4
4
 
5
5
  Permission is hereby granted, free of charge, to any person obtaining a copy
6
6
  of this software and associated documentation files (the "Software"), to deal
data/README.md CHANGED
@@ -25,8 +25,6 @@ And follow the instructions for your database library:
25
25
  Register the vector type with your connection
26
26
 
27
27
  ```ruby
28
- require "pgvector/pg"
29
-
30
28
  registry = PG::BasicTypeRegistry.new.define_default_types
31
29
  Pgvector::PG.register_vector(registry)
32
30
  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
35
33
  Insert a vector
36
34
 
37
35
  ```ruby
38
- factors = [1, 2, 3]
39
- conn.exec_params("INSERT INTO items (factors) VALUES ($1)", [factors])
36
+ embedding = [1, 2, 3]
37
+ conn.exec_params("INSERT INTO items (embedding) VALUES ($1)", [embedding])
40
38
  ```
41
39
 
42
40
  Get the nearest neighbors to a vector
43
41
 
44
42
  ```ruby
45
- conn.exec_params("SELECT * FROM items ORDER BY factors <-> $1 LIMIT 5", [factors]).to_a
43
+ conn.exec_params("SELECT * FROM items ORDER BY embedding <-> $1 LIMIT 5", [embedding]).to_a
46
44
  ```
47
45
 
48
46
  ## History
data/lib/pgvector/pg.rb CHANGED
@@ -4,6 +4,17 @@ module Pgvector
4
4
  module PG
5
5
  def self.register_vector(registry)
6
6
  registry.register_type(0, "vector", nil, TextDecoder::Vector)
7
+ registry.register_type(1, "vector", nil, BinaryDecoder::Vector)
8
+ end
9
+
10
+ module BinaryDecoder
11
+ class Vector < ::PG::SimpleDecoder
12
+ def decode(string, tuple = nil, field = nil)
13
+ dim, unused = string[0, 4].unpack("nn")
14
+ raise "expected unused to be 0" if unused != 0
15
+ string[4..-1].unpack("g#{dim}")
16
+ end
17
+ end
7
18
  end
8
19
 
9
20
  module TextDecoder
@@ -1,3 +1,3 @@
1
1
  module Pgvector
2
- VERSION = "0.1.0"
2
+ VERSION = "0.2.0"
3
3
  end
data/lib/pgvector.rb CHANGED
@@ -1 +1,5 @@
1
1
  require "pgvector/version"
2
+
3
+ module Pgvector
4
+ autoload :PG, "pgvector/pg"
5
+ end
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: pgvector
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.1.0
4
+ version: 0.2.0
5
5
  platform: ruby
6
6
  authors:
7
7
  - Andrew Kane
8
8
  autorequire:
9
9
  bindir: bin
10
10
  cert_chain: []
11
- date: 2022-02-07 00:00:00.000000000 Z
11
+ date: 2023-05-12 00:00:00.000000000 Z
12
12
  dependencies: []
13
13
  description:
14
14
  email: andrew@ankane.org
@@ -34,14 +34,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
34
34
  requirements:
35
35
  - - ">="
36
36
  - !ruby/object:Gem::Version
37
- version: '2.6'
37
+ version: '3'
38
38
  required_rubygems_version: !ruby/object:Gem::Requirement
39
39
  requirements:
40
40
  - - ">="
41
41
  - !ruby/object:Gem::Version
42
42
  version: '0'
43
43
  requirements: []
44
- rubygems_version: 3.3.3
44
+ rubygems_version: 3.4.10
45
45
  signing_key:
46
46
  specification_version: 4
47
47
  summary: pgvector support for Ruby