ai_sentinel 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 57b7b717300a40c7538e4a7da84bd574677c9bd38fa55c094ad44725354e07d8
-  data.tar.gz: 1897eb521d5a9b2c71a1b1d54e7aa015172466accb65ef6c0bfa6a3b9ac5879a
+  metadata.gz: 54f5dcacdba120c16ed3a4c09384f8b981aef569b9fb66a350400ec8a3f13b4b
+  data.tar.gz: 0b15c9519a2085dcc7b3a3b920ff4b91e89f282da8346429f6152b886334225f
 SHA512:
-  metadata.gz: 46a5467ba869e34a6b2e4f942c23cdcb719a8b2d8d6583ba29237873bb0145bed9ef736d6949bdf0434bf0f05fce8529b790b89f691594f40545667be12b574d
-  data.tar.gz: 8d05fe5f7477260054a4d9083fa2cbdcecb0effc5c2c701b1768fae3882604e9a8840e0cb5ceba0fb11b61a2fc5a488f1e3eef61d298f3d1e06c6d6e60327e64
+  metadata.gz: 02f23261d5e52c11c03b9c15c7633f5eb12b54daacb49062d3cc8eb5be7168b72c1069e0c1607d3c02d13867af4545455dbd995eb49cc01dfc25007a8ed5d4ff
+  data.tar.gz: 8a9a3159203a5e5e580dfe2801237ff68067a3ef9b30b23389e4e05204851d5135ad6d38093d568bbfd6b328186180d8df56d706363cfc5d5059fb8a110ff067

data/lib/ai_sentinel/actions/ai_prompt.rb

@@ -32,7 +32,8 @@ module AiSentinel
           prompt_template: step.params[:prompt],
           system_template: step.params[:system],
           tool_executor: tool_executor,
-          max_tool_rounds: step.params.fetch(:max_tool_rounds, configuration.max_tool_rounds)
+          max_tool_rounds: step.params.fetch(:max_tool_rounds, configuration.max_tool_rounds),
+          compaction_prompt: step.params[:compaction_prompt]
         )
       end
 

data/lib/ai_sentinel/context_compactor.rb

@@ -10,11 +10,12 @@ module AiSentinel
       Respond with ONLY the summary, no preamble or explanation.
     PROMPT
 
-    attr_reader :context_key, :configuration
+    attr_reader :context_key, :configuration, :custom_compaction_prompt
 
-    def initialize(context_key:, configuration:)
+    def initialize(context_key:, configuration:, compaction_prompt: nil)
       @context_key = context_key
       @configuration = configuration
+      @custom_compaction_prompt = compaction_prompt
     end
 
     def compact_if_needed
@@ -74,7 +75,7 @@ module AiSentinel
 
       result = provider.chat(
         prompt: prompt,
-        system: SUMMARIZATION_PROMPT,
+        system: custom_compaction_prompt || SUMMARIZATION_PROMPT,
         workflow_name: nil,
         step_name: nil,
         remember: false

data/lib/ai_sentinel/providers/anthropic.rb

@@ -9,7 +9,8 @@ module AiSentinel
       API_VERSION = '2023-06-01'
 
       def chat(prompt:, system: nil, model: nil, workflow_name: nil, step_name: nil, remember: true,
-               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10)
+               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10,
+               compaction_prompt: nil)
         model ||= configuration.model
         context_key = "#{workflow_name}:#{step_name}"
 
@@ -26,7 +27,8 @@ module AiSentinel
         assistant_text = extract_text(response_data)
         if remember
           save_context(context_key, prompt, assistant_text,
-                       prompt_template: prompt_template, system_template: system_template)
+                       prompt_template: prompt_template, system_template: system_template,
+                       compaction_prompt: compaction_prompt)
         end
 
         Actions::AiPrompt::Result.new(

data/lib/ai_sentinel/providers/base.rb

@@ -66,7 +66,8 @@ module AiSentinel
         messages
       end
 
-      def save_context(context_key, user_message, assistant_message, prompt_template: nil, system_template: nil)
+      def save_context(context_key, user_message, assistant_message, prompt_template: nil, system_template: nil,
+                       compaction_prompt: nil)
         return unless Persistence::Database.connected?
 
         ctx = Persistence::Database.find_or_create_context(context_key)
@@ -81,7 +82,7 @@ module AiSentinel
 
         save_prompt_hash(context_key, prompt_template, system_template) if prompt_template
         prune_old_messages(ctx[:id])
-        compact_context(context_key)
+        compact_context(context_key, compaction_prompt: compaction_prompt)
       end
 
       def save_prompt_hash(context_key, prompt_template, system_template)
@@ -91,8 +92,9 @@ module AiSentinel
         )
       end
 
-      def compact_context(context_key)
-        ContextCompactor.new(context_key: context_key, configuration: configuration).compact_if_needed
+      def compact_context(context_key, compaction_prompt: nil)
+        ContextCompactor.new(context_key: context_key, configuration: configuration,
+                             compaction_prompt: compaction_prompt).compact_if_needed
       rescue StandardError => e
         AiSentinel.log_error(e, context: "Context compaction failed for '#{context_key}'")
       end

data/lib/ai_sentinel/providers/openai.rb

@@ -7,7 +7,8 @@ module AiSentinel
   module Providers
     class Openai < Base
       def chat(prompt:, system: nil, model: nil, workflow_name: nil, step_name: nil, remember: true,
-               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10)
+               prompt_template: nil, system_template: nil, tool_executor: nil, max_tool_rounds: 10,
+               compaction_prompt: nil)
         model ||= configuration.model
         context_key = "#{workflow_name}:#{step_name}"
 
@@ -24,7 +25,8 @@ module AiSentinel
         assistant_text = extract_text(response_data)
         if remember
           save_context(context_key, prompt, assistant_text,
-                       prompt_template: prompt_template, system_template: system_template)
+                       prompt_template: prompt_template, system_template: system_template,
+                       compaction_prompt: compaction_prompt)
         end
 
         Actions::AiPrompt::Result.new(

data/lib/ai_sentinel/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiSentinel
-  VERSION = '0.3.0'
+  VERSION = '0.4.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai_sentinel
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Mario Celi