ai-engine 0.4.1 → 0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0702cfaf1ad63d8a98a0b8acb67fdd0b1d3cf46320652f8f86df6b677af981ac
-  data.tar.gz: 1adacb8b8e2ef7da675077a79950ca342df9401a62780b8955c5b785f3d32538
+  metadata.gz: 9f3cd77d6cd96c5e07d9b1e2627abfe022072f69a83819a3191a5bc82c96db84
+  data.tar.gz: d4a2b9d2d26b3c0579e2f109c11801501b1e96d49d37cd789f347d680f6c4c7e
 SHA512:
-  metadata.gz: b1270b59cdbe033e1147cc54ca9124145d3670363ec73b51854832983ae6f1d5dc88ead7ad726d99d641b4c7a9b0d2fe4eb5b236b7c6bdcb792d44c9f175fd77
-  data.tar.gz: 12e4f4c75a87e9f5b3c67312009906f981e01ebe340c2de3f6c8420a625bf2b7bddde7507bd9d0246747434d8ba95c5cd9cf6e4d6d1741cefdd15906d3774d6b
+  metadata.gz: 2b09245610c6a8ce82ba5e73dcee0b5bdc586b26c10ef5af6919894d5d8d58b3edf1b3e5c7340c2046a046bfcbc843fe35dfa447b6417ffc61c1a4b3991bdb0f
+  data.tar.gz: 4384746c755693decb0aef08674eed0222d4b17bb4cc63c8b0a468141242fd42b11d0a03fd719bc4378ff5df0da9dc2995d43b8f4c24d03f1abac3b90ed49120

data/app/models/ai/engine/assistant_thread.rb

@@ -3,10 +3,11 @@ module AI::Engine
     include RemoteIdValidatable
 
     belongs_to :threadable, polymorphic: true
-    has_many :runs, class_name: "AI::Engine::Run", foreign_key: "ai_engine_assistant_thread_id", dependent: :nullify
-    has_many :messages, as: :messageable, class_name: "AI::Engine::Message", foreign_key: "messageable_id", dependent: :nullify
+    has_many :runs, class_name: "AI::Engine::Run", foreign_key: "ai_engine_assistant_thread_id", dependent: :destroy
+    has_many :messages, as: :messageable, class_name: "AI::Engine::Message", foreign_key: "messageable_id", dependent: :destroy
 
     before_create :create_openai_thread
+    before_destroy :delete_openai_thread
 
     def run(assistant_id:, content:)
       # Create the request Message, locally and remotely on OpenAI.
@@ -28,5 +29,12 @@ module AI::Engine
       errors.add(:base, e.message)
       throw(:abort)
     end
+
+    def delete_openai_thread
+      AI::Engine::OpenAI::Threads::Delete.call(remote_id: remote_id)
+    rescue Faraday::Error => e
+      errors.add(:base, e.message)
+      throw(:abort)
+    end
   end
 end

data/app/models/ai/engine/chat.rb

@@ -1,7 +1,7 @@
 module AI::Engine
   class Chat < ApplicationRecord
     belongs_to :chattable, polymorphic: true
-    has_many :messages, as: :messageable, class_name: "AI::Engine::Message", foreign_key: "messageable_id", dependent: :nullify
+    has_many :messages, as: :messageable, class_name: "AI::Engine::Message", foreign_key: "messageable_id", dependent: :destroy
 
     def run(model:)
       # Run the Chat, sending the complete message history to OpenAI.

data/app/models/concerns/ai/engine/assistable.rb

@@ -4,10 +4,11 @@ module AI
       extend ActiveSupport::Concern
 
       included do
-        has_one :assistant, as: :assistable, class_name: "AI::Engine::Assistant", dependent: :nullify
+        has_one :assistant, as: :assistable, class_name: "AI::Engine::Assistant", dependent: :destroy
 
         before_create :create_openai_assistant
         before_update :update_openai_assistant
+        before_destroy :delete_openai_assistant
 
         # Default. Override in including model to customize.
         def ai_engine_assistant
@@ -42,6 +43,13 @@ module AI
           errors.add(:base, e.message)
           throw(:abort)
         end
+
+        def delete_openai_assistant
+          AI::Engine::OpenAI::Assistants::Delete.call(remote_id: assistant&.remote_id)
+        rescue Faraday::Error => e
+          errors.add(:base, e.message)
+          throw(:abort)
+        end
       end
     end
   end

data/app/models/concerns/ai/engine/chattable.rb

@@ -4,7 +4,7 @@ module AI
       extend ActiveSupport::Concern
 
       included do
-        has_many :chats, as: :chattable, class_name: "AI::Engine::Chat"
+        has_many :chats, as: :chattable, class_name: "AI::Engine::Chat", dependent: :destroy
 
         def ai_engine_on_message_create(message:)
           # This is a hook for the AI Engine to notify the Chattable that a Message has been updated.

data/app/models/concerns/ai/engine/threadable.rb

@@ -4,7 +4,7 @@ module AI
       extend ActiveSupport::Concern
 
       included do
-        has_many :assistant_threads, as: :threadable, class_name: "AI::Engine::AssistantThread", dependent: :nullify
+        has_many :assistant_threads, as: :threadable, class_name: "AI::Engine::AssistantThread", dependent: :destroy
 
         def ai_engine_on_message_create(message:)
           # This is a hook for the AI Engine to notify the AssistantThreadtable that a Message has been updated.

data/app/services/ai/engine/openai/assistants/delete.rb

@@ -0,0 +1,17 @@
+class AI::Engine::OpenAI::Assistants::Delete
+  # Deletes an OpenAI Assistant by its ID.
+  def self.call(remote_id:)
+    client.assistants.delete(id: remote_id)
+  rescue Faraday::ResourceNotFound => e
+    Rails.logger.error("#{self} - #{e.response.dig(:body, "error", "message")}")
+  end
+
+  private_class_method def self.client
+    @client ||= OpenAI::Client.new(
+      access_token: AI::Engine::Engine.config.openai_access_token,
+      organization_id: AI::Engine::Engine.config.openai_organization_id,
+      log_errors: Rails.env.development? || Rails.env.test?,
+      request_timeout: 2.minutes.to_i
+    )
+  end
+end

data/app/services/ai/engine/openai/threads/delete.rb

@@ -0,0 +1,19 @@
+class AI::Engine::OpenAI::Threads::Delete
+  # Deletes an OpenAI Thread by its ID.
+  def self.call(remote_id:)
+    response = client.threads.delete(id: remote_id)
+
+    response["id"]
+  rescue Faraday::ResourceNotFound => e
+    Rails.logger.error("#{self} - #{e.response.dig(:body, "error", "message")}")
+  end
+
+  private_class_method def self.client
+    @client ||= OpenAI::Client.new(
+      access_token: AI::Engine::Engine.config.openai_access_token,
+      organization_id: AI::Engine::Engine.config.openai_organization_id,
+      log_errors: Rails.env.development? || Rails.env.test?,
+      request_timeout: 2.minutes.to_i
+    )
+  end
+end

data/lib/ai/engine/version.rb

@@ -1,5 +1,5 @@
 module AI
   module Engine
-    VERSION = "0.4.1"
+    VERSION = "0.4.2"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai-engine
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.2
 platform: ruby
 authors:
 - Alex Rudall
@@ -44,6 +44,7 @@ files:
 - app/models/concerns/ai/engine/remote_id_validatable.rb
 - app/models/concerns/ai/engine/threadable.rb
 - app/services/ai/engine/openai/assistants/create.rb
+- app/services/ai/engine/openai/assistants/delete.rb
 - app/services/ai/engine/openai/assistants/retrieve.rb
 - app/services/ai/engine/openai/assistants/update.rb
 - app/services/ai/engine/openai/chats/stream.rb
@@ -51,6 +52,7 @@ files:
 - app/services/ai/engine/openai/runs/create.rb
 - app/services/ai/engine/openai/runs/retrieve.rb
 - app/services/ai/engine/openai/threads/create.rb
+- app/services/ai/engine/openai/threads/delete.rb
 - config/initializers/inflections.rb
 - config/initializers/openai.rb
 - db/migrate/20240528153439_create_ai_engine_assistants.rb