active_pinecone 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 810ca56ec86165e1c43932c6f26a7e0909f75ba68e3242d16e48d443de91a69a
-  data.tar.gz: 35e89c03e5e0d13b68e0aa0c43082669dc7c094fdbb8a1471eebee3c16df4772
+  metadata.gz: ac83a40bff569558fe83e8b92eef9f9f81fc152a92f69a101dfde763f633c108
+  data.tar.gz: 0cf15188e17fae9e367024c4197168f3620205fe24719ab095c41a150cca6176
 SHA512:
-  metadata.gz: 905d43792ce720898bde25b3f599ecb18286b250d53ae50354aae86c1ddbba6d07ca7210a6ee2a3719bb8b8b00f84c0b6d35f5a26de5a5f42781c5a56e569c5c
-  data.tar.gz: d1357320f74503960717be02a51e1aae21ceaf9d7326d7b9cc41f920a76234683b25a63a20f60a0623cb158ba06eb83b5839b7839fddec76c37f37c26d40ae08
+  metadata.gz: efafb2f1c7a68b2e4fa3a9f08668038595f126cbf41f191e929f2645cb413968e0c6786681fe5d65298dcf5b3efda20b876875fd6ce889a51c546c71f6ddf8e2
+  data.tar.gz: c01271f4b41521e3a259666c8937c60dd16a204ea663fd3fa4a73b0fc87a1fa90d75dad941ff251e5fd84097b3a7597b168b6c99555499d302f7de321f5c65d0

data/CHANGELOG.md

@@ -3,3 +3,8 @@
 ## [0.1.0] - 2023-06-24
 
 - Initial release
+
+## [0.1.1] - 2023-05-24
+
+- Bug fix
+- Add example

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    active_pinecone (0.1.0)
+    active_pinecone (0.1.1)
       activesupport (>= 5.0.0)
       pinecone (~> 0.1.5)
       ruby-openai (~> 4.1.0)

data/README.md

@@ -37,6 +37,15 @@ class Recipe < ActivePinecone::Base
 end
 ```
 
+### Initialize
+
+Create Pinecone index. Index initialization takes a few minutes.
+
+```ruby
+Post.init
+```
+
+
 ### Create
 
 Pinecone index `Recipe` is created automatically.

data/examples/main.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require_relative 'post'
+
+Post.init
+
+post = Post.create(title: 'This is An Example', body: 'This is an example', author: 'Kevin')
+
+p post.title #=> 'This is An Example'
+
+p post.author #=> 'Kevin'
+
+post.update(body: 'This will be changed')
+
+posts = Post.search('example')
+
+posts.each do |pt|
+  p pt.title #=> 'This is An Example'
+end
+
+assistant = Post.assistant
+
+p assistant.reply('hoge')
+
+p assistant.messages

data/examples/post.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require 'active_pinecone'
+
+ActivePinecone.configure do |config|
+  config.openai_access_token = ENV.fetch('OPENAI_ACCESS_TOKEN')
+  config.pinecone_api_key = ENV.fetch('PINECONE_API_KEY')
+  config.pinecone_environment = ENV.fetch('PINECONE_ENVIRONMENT')
+end
+
+class Post < ActivePinecone::Base
+  vectorizes :title, :body
+end

data/lib/active_pinecone/base.rb

@@ -17,7 +17,7 @@ module ActivePinecone
     end
 
     def self.index_name
-      to_s
+      to_s.downcase.pluralize
     end
 
     def self.pinecone
@@ -33,7 +33,8 @@ module ActivePinecone
     end
 
     def self.embed(input)
-      openai.embeddings(parameters: { model: ActivePinecone.configuration.embedding_model, input: input })
+      embeded = openai.embeddings(parameters: { model: ActivePinecone.configuration.embedding_model, input: input })
+      embeded['data'][0]['embedding']
     end
 
     def self.vectorized_attributes
@@ -49,6 +50,10 @@ module ActivePinecone
       @@vectorized_attributes += attr_names
     end
 
+    def self.init
+      pinecone.create_index({ name: index_name, dimension: ActivePinecone.configuration.dimension, metrics: ActivePinecone.configuration.metrics })
+    end
+
     def self.create(**attributes)
       id = attributes[:id] || SecureRandom.uuid
       index.upsert(vectors: [{ id: id, metadata: attributes, values: embed(attributes.slice(vectorized_attributes).to_json) }])
@@ -57,15 +62,11 @@ module ActivePinecone
 
     def self.delete(id)
       index.delete(ids: [id])
-    rescue ::Pinecone::IndexNotFoundError
-      pinecone.create_index({ name: index_name, dimension: ActivePinecone.configuration.dimension, metrics: ActivePinecone.configuration.metrics })
     end
 
     def self.find(id)
       fetched = index.fetch(ids: [id])
       new(fetched['vectors'][id]['metadata'].merge(id: id).symbolize_keys)
-    rescue ::Pinecone::IndexNotFoundError
-      pinecone.create_index({ name: index_name, dimension: ActivePinecone.configuration.dimension, metrics: ActivePinecone.configuration.metrics })
     end
 
     def self.search(text, filter: {})
@@ -73,15 +74,11 @@ module ActivePinecone
       searched['matches'].map do |match|
         new(match['metadata'].merge(id: match['id']).symbolize_keys)
       end
-    rescue ::Pinecone::IndexNotFoundError
-      pinecone.create_index({ name: index_name, dimension: ActivePinecone.configuration.dimension, metrics: ActivePinecone.configuration.metrics })
     end
 
     def update(**attributes)
       self.class.index.upsert(vectors: [{ id: id, metadata: attributes, values: self.class.embed(attributes.slice(self.class.vectorized_attributes).to_json) }])
       @attributes = attributes
-    rescue ::Pinecone::IndexNotFoundError
-      pinecone.create_index({ name: index_name, dimension: ActivePinecone.configuration.dimension, metrics: ActivePinecone.configuration.metrics })
     end
   end
 end

data/lib/active_pinecone/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActivePinecone
-  VERSION = '0.1.0'
+  VERSION = '0.1.1'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_pinecone
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Moeki Kawakami
@@ -68,6 +68,8 @@ files:
 - README.md
 - Rakefile
 - active_pinecone.gemspec
+- examples/main.rb
+- examples/post.rb
 - lib/active_pinecone.rb
 - lib/active_pinecone/assistant.rb
 - lib/active_pinecone/base.rb