llm_memory_pgvector 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c5e26736ec1dec1732102c6f88b099115aa73adba6f57ed1ea3caf4d6b907829
-  data.tar.gz: 87d1b4970086e025dd1184c5d3e91cbad0a61a59109839ab7dd16bec658b7f58
+  metadata.gz: 02a8d906cb751a0dffa89839cc7f293c970c618285c917ca5521ed9091f4cad0
+  data.tar.gz: 7eeced97f49923746900caf68269041f93621f588cf743195aa4acdcdb725d8f
 SHA512:
-  metadata.gz: d26030470a22a37b6e7c83e4a877d92bc5895cc18c78b3ab56a9309748165b4fe5d3c0b204430e5c2b8a0029e8bc46e7ea9ddafd0c17cc6b7e962326c686a921
-  data.tar.gz: e9c446f18393602154483ea6a2a18335ce48e820632be177212beeb6a5ea9ffe396b4f139c355c578748acb5d50bb18dd7a28ab77af82ae301ae7db947b7ba13
+  metadata.gz: c92ec2de76562dca9b5b3a8d0c0414858d58a73fc1cf43a575e96dd446016eb780f42ef4ae04354eec3bb63e615bef7f2efb4b6aae3a4faffc507e0a78efa781
+  data.tar.gz: 8989283b6f74ffb336b479114aa3b5a43790442924f51582c5ab3e6402e7460fe3cabd466591877dfc6c9aaf0f95037a926c24fa8b506aa5ecfd2ce9b2ae3a35

data/lib/llm_memory_pgvector/pgvector_store.rb

@@ -43,14 +43,12 @@ module LlmMemoryPgvector
 
     # 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.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_memory_pgvector
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Shohei Kameda