llm_memory_pgvector 0.1.1 → 0.1.2

Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: c5e26736ec1dec1732102c6f88b099115aa73adba6f57ed1ea3caf4d6b907829
4
- data.tar.gz: 87d1b4970086e025dd1184c5d3e91cbad0a61a59109839ab7dd16bec658b7f58
3
+ metadata.gz: 02a8d906cb751a0dffa89839cc7f293c970c618285c917ca5521ed9091f4cad0
4
+ data.tar.gz: 7eeced97f49923746900caf68269041f93621f588cf743195aa4acdcdb725d8f
5
5
  SHA512:
6
- metadata.gz: d26030470a22a37b6e7c83e4a877d92bc5895cc18c78b3ab56a9309748165b4fe5d3c0b204430e5c2b8a0029e8bc46e7ea9ddafd0c17cc6b7e962326c686a921
7
- data.tar.gz: e9c446f18393602154483ea6a2a18335ce48e820632be177212beeb6a5ea9ffe396b4f139c355c578748acb5d50bb18dd7a28ab77af82ae301ae7db947b7ba13
6
+ metadata.gz: c92ec2de76562dca9b5b3a8d0c0414858d58a73fc1cf43a575e96dd446016eb780f42ef4ae04354eec3bb63e615bef7f2efb4b6aae3a4faffc507e0a78efa781
7
+ data.tar.gz: 8989283b6f74ffb336b479114aa3b5a43790442924f51582c5ab3e6402e7460fe3cabd466591877dfc6c9aaf0f95037a926c24fa8b506aa5ecfd2ce9b2ae3a35
@@ -43,14 +43,12 @@ module LlmMemoryPgvector
43
43
 
44
44
  # data = [{ content: "", vector: [], metadata: {} },,]
45
45
  def add(data: [])
46
- values = data.map { |row|
47
- "('#{row[@content_key.to_sym]}', '#{row[@metadata_key.to_sym].to_json}', '#{row[@vector_key.to_sym]}')"
48
- }.join(",")
49
- sql = <<~SQL
50
- INSERT INTO #{@index_name} (#{@content_key}, #{@metadata_key}, #{@vector_key})
51
- VALUES #{values}
52
- SQL
53
- @conn.exec(sql)
46
+ data.each do |row|
47
+ @conn.exec_params(
48
+ "INSERT INTO #{@index_name} (#{@content_key}, #{@metadata_key}, #{@vector_key}) VALUES ($1, $2, $3)",
49
+ [row[@content_key.to_sym], row[@metadata_key.to_sym].to_json, row[@vector_key.to_sym]]
50
+ )
51
+ end
54
52
  end
55
53
 
56
54
  def search(query: [], k: 3)
@@ -1,5 +1,5 @@
1
1
  # frozen_string_literal: true
2
2
 
3
3
  module LlmMemoryPgvector
4
- VERSION = "0.1.1"
4
+ VERSION = "0.1.2"
5
5
  end
metadata CHANGED
@@ -1,7 +1,7 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: llm_memory_pgvector
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.1.1
4
+ version: 0.1.2
5
5
  platform: ruby
6
6
  authors:
7
7
  - Shohei Kameda