llm_memory_pgvector 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2f5ddd879e687f5fbdf0639bcffb131246d0cb621903433ac07ecadfdcfc894c
-  data.tar.gz: a4cb48e9c9920fa999c1195485d1835f88298870c1eb361caecd1fe827c0e0c9
+  metadata.gz: 02a8d906cb751a0dffa89839cc7f293c970c618285c917ca5521ed9091f4cad0
+  data.tar.gz: 7eeced97f49923746900caf68269041f93621f588cf743195aa4acdcdb725d8f
 SHA512:
-  metadata.gz: 0e995c5601a65c2b689301d9722b1137feddcfa7323483e2cd341c840ad3ec2c456ddd27564a34c322ec46cc2e10cc78b72ce792f79afc9d7a893fcbc86d2835
-  data.tar.gz: e2703030b153cbced9406ae08831dd2e4064cb421da3f4f90028ccbe558a1686036ee818afd449c1f593fe67d2aadba23d29166fcccc6385bdcb23529429e52f
+  metadata.gz: c92ec2de76562dca9b5b3a8d0c0414858d58a73fc1cf43a575e96dd446016eb780f42ef4ae04354eec3bb63e615bef7f2efb4b6aae3a4faffc507e0a78efa781
+  data.tar.gz: 8989283b6f74ffb336b479114aa3b5a43790442924f51582c5ab3e6402e7460fe3cabd466591877dfc6c9aaf0f95037a926c24fa8b506aa5ecfd2ce9b2ae3a35

data/Gemfile.lock

@@ -1,8 +1,9 @@
 PATH
   remote: .
   specs:
-    llm_memory_pgvector (0.1.0)
+    llm_memory_pgvector (0.1.1)
       llm_memory (~> 0.1.7)
+      pg (~> 1.5.3)
       pgvector (~> 0.2.0)
 
 GEM

data/lib/llm_memory_pgvector/pgvector_store.rb

@@ -30,20 +30,25 @@ module LlmMemoryPgvector
       @conn.exec("CREATE TABLE #{@index_name} (id bigserial PRIMARY KEY, #{@content_key} TEXT, #{@metadata_key} JSON, #{@vector_key} vector(#{dim}))")
     end
 
+    def index_exists?
+      @conn.exec("SELECT 1 FROM #{@index_name}")
+      true
+    rescue PG::Error
+      false
+    end
+
     def drop_index
       @conn.exec("DROP TABLE IF EXISTS #{@index_name}")
     end
 
     # data = [{ content: "", vector: [], metadata: {} },,]
     def add(data: [])
-      values = data.map { |row|
-        "('#{row[@content_key.to_sym]}', '#{row[@metadata_key.to_sym].to_json}', '#{row[@vector_key.to_sym]}')"
-      }.join(",")
-      sql = <<~SQL
-        INSERT INTO #{@index_name} (#{@content_key}, #{@metadata_key}, #{@vector_key}) 
-        VALUES #{values}
-      SQL
-      @conn.exec(sql)
+      data.each do |row|
+        @conn.exec_params(
+          "INSERT INTO #{@index_name} (#{@content_key}, #{@metadata_key}, #{@vector_key}) VALUES ($1, $2, $3)",
+          [row[@content_key.to_sym], row[@metadata_key.to_sym].to_json, row[@vector_key.to_sym]]
+        )
+      end
     end
 
     def search(query: [], k: 3)

data/lib/llm_memory_pgvector/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmMemoryPgvector
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

data/llm_memory_pgvector.gemspec

@@ -31,6 +31,7 @@ Gem::Specification.new do |spec|
 
   # Uncomment to register a new dependency of your gem
   spec.add_dependency "llm_memory", "~> 0.1.7"
+  spec.add_dependency "pg", "~> 1.5.3"
   spec.add_dependency "pgvector", "~> 0.2.0"
 
   # For more information and examples about making a new gem, check out our

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_memory_pgvector
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Shohei Kameda
@@ -24,6 +24,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.1.7
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.5.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.5.3
 - !ruby/object:Gem::Dependency
   name: pgvector
   requirement: !ruby/object:Gem::Requirement