ask-agent 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e875e155f72f3bc773ff218b64243ac918354fac930b0181e02ae1a74d0bc4eb
-  data.tar.gz: 9ec62c98c64b7307f445311cce0f6fd6bf81a9f7fde52daa505cd532d4734520
+  metadata.gz: cc1ea1b84a7139944acf9787cfdd3e5a9fad739b759fc4501f59f68bdd3c0b77
+  data.tar.gz: 37733f0982e0d991e284d46c9202fe5846fff67a433acae0385a703912c7f898
 SHA512:
-  metadata.gz: 14d6490ab16f5e675e05ac3e72f37c113edcfd7e7339c10fe6a73290b88e4cab0ab8535d21860e51d09aac36c7a7066b3dbfe1f19b04d5c873a301094fb610fe
-  data.tar.gz: 729b178fb7f474e2fe5c84ce692169ba184fefe1b761d7ddc2a1dfaf39c57d184155f9f67e186bd91b13faaab05a5e79aac49107055d0089269ca10e9e100f94
+  metadata.gz: 8e4ff593f7856c2fe0da46dfe8091ad38217e29b27aa343e95f6f939ab741b8c249217dbaa397ad609c21b9b4f39e09a781738a556368e69307fa01329fb00d9
+  data.tar.gz: fc45d3ae62bc2e325a22dedab88625f312d3e9b07188d9614c5cb2ddfa9683ed31e2bb8473f10344d73ac18f7cef202fcf53cd68272affbab54c208770017dde

data/lib/ask/agent/session.rb

@@ -14,6 +14,8 @@ module Ask
       end
 
       attr_reader :meta_agent_results
+      # @return [Ask::Skills::Registry, nil] auto-discovered skills registry
+      attr_reader :skills_registry
 
       def initialize(model:, tools: [], max_turns: 25, max_tool_retries: 3,
                      compactor: nil, hooks: {}, persistence: nil,
@@ -39,6 +41,16 @@ module Ask
         @tool_executor = ToolExecutor.new(max_retries: max_tool_retries, parallel: parallel_tools)
         @compactor = compactor ? build_compactor(compactor) : nil
         @hooks = Hooks.new(hooks)
+
+        # Auto-discover skills and inject into system prompt
+        @skills_registry = Ask::Skills.discover rescue nil
+        if @skills_registry && !@skills_registry.names.empty?
+          skill_text = @skills_registry.format_for_prompt
+          if !skill_text.empty? && @chat.messages.any? { |m| m.role == :system }
+            current = @chat.messages.find { |m| m.role == :system }.content.to_s
+            @chat.with_instructions(current + skill_text)
+          end
+        end
         @persistence = persistence
 
         reflector_opts = reflector.is_a?(Hash) ? reflector : {}
@@ -202,6 +214,28 @@ module Ask
 
       def abort_requested? = @abort_requested
 
+      # Load a skill by name or file path.
+      # Injects the skill's full instructions into the conversation as a system message.
+      #
+      # @param name [String] skill name (e.g. "rails.db_debug") or path to a .md file
+      # @raise [Ask::Skills::Error] if the skill is not found
+      def skill(name)
+        if @skills_registry && (s = @skills_registry[name])
+          @chat.add_message(
+            role: :system,
+            content: "## Skill: #{s.name}\n\n#{s.description}\n\n---\n\n#{s.instructions}"
+          )
+        elsif File.exist?(name.to_s)
+          content = File.read(name.to_s)
+          @chat.add_message(
+            role: :system,
+            content: "## Skill: #{name}\n\n---\n\n#{content}"
+          )
+        else
+          raise Ask::Skills::Error, "Skill not found: #{name.inspect}"
+        end
+      end
+
       def reset_messages!
         @chat.reset_messages!
         @messages = []

data/lib/ask/agent/session_backup.rb

@@ -0,0 +1,237 @@
+# frozen_string_literal: true
+
+require "securerandom"
+require "time"
+
+module Ask
+  module Agent
+    class Session
+      attr_reader :id, :chat, :tools, :turn_count, :created_at, :messages
+      attr_reader :tool_calls_made
+
+      def reflection_count
+        @reflector&.reflection_count || 0
+      end
+
+
+      # @return [Ask::Skills::Registry, nil] auto-discovered skills registry
+      attr_reader :skills_registry
+      attr_reader :meta_agent_results
+      # @return [Ask::Skills::Registry, nil] auto-discovered skills registry
+      attr_reader :skills_registry
+
+      def initialize(model:, tools: [], max_turns: 25, max_tool_retries: 3,
+                     compactor: nil, hooks: {}, persistence: nil,
+                     id: nil, system_prompt: nil, parallel_tools: true,
+                     reflector: nil, telemetry: true, meta_agent: nil, **chat_options)
+        @id = id || SecureRandom.uuid
+        @max_turns = max_turns
+        @max_tool_retries = max_tool_retries
+        @parallel_tools = parallel_tools
+        @event_handlers = { all: [] }
+        @running = false
+        @deleted = false
+        @abort_requested = false
+        @turn_count = 0
+        @created_at = Time.now
+        @_no_tools_instructed = false
+
+        @telemetry = telemetry.is_a?(Telemetry) ? telemetry : Telemetry.new(enabled: !!telemetry)
+
+        @chat = build_chat(model, system_prompt, tools, **chat_options)
+        @tools = resolve_tools(tools)
+        @loop = Loop.new(max_turns: max_turns)
+        @tool_executor = ToolExecutor.new(max_retries: max_tool_retries, parallel: parallel_tools)
+
+        # Auto-discover skills and inject into system prompt
+        @skills_registry = Ask::Skills.discover rescue nil
+        if @skills_registry && !@skills_registry.names.empty?
+          skill_text = @skills_registry.format_for_prompt
+          if !skill_text.empty? && @chat.messages.any? { |m| m.role == :system }
+            current = @chat.messages.find { |m| m.role == :system }.content.to_s
+            @chat.with_instructions(current + skill_text)
+          end
+        end
+        @compactor = compactor ? build_compactor(compactor) : nil
+        @hooks = Hooks.new(hooks)
+
+        # Auto-discover skills and inject into system prompt
+        @skills_registry = Ask::Skills.discover rescue nil
+        if @skills_registry && !@skills_registry.names.empty?
+          skill_text = @skills_registry.format_for_prompt
+          if !skill_text.empty? && @chat.messages.any? { |m| m.role == :system }
+            current = @chat.messages.find { |m| m.role == :system }.content.to_s
+            @chat.with_instructions(current + skill_text)
+          end
+        end
+        @persistence = persistence
+
+        reflector_opts = reflector.is_a?(Hash) ? reflector : {}
+        @reflector = if reflector
+          Reflector.new(
+            model: @chat,
+            max_reflections: reflector_opts[:max_reflections] || 1
+          )
+        end
+
+        @meta_agent_config = meta_agent
+        @meta_agent_results = nil
+
+
+        # Auto-discover skills and inject into system prompt
+        @skills_registry = Ask::Skills.discover rescue nil
+        if @skills_registry && !@skills_registry.names.empty?
+          skill_text = @skills_registry.format_for_prompt
+          if !skill_text.empty? && @chat.messages.any? { |m| m.role == :system }
+            current = @chat.messages.find { |m| m.role == :system }.content.to_s
+            @chat.with_instructions(current + skill_text)
+          end
+        end
+        @compactor&.chat = @chat
+      end
+
+      def run(message, tools: nil)
+        raise "Session deleted" if @deleted
+        raise "Session already running" if @running
+
+        @running = true
+        @abort_requested = false
+        @turn_count = 0
+        @loop.reset!
+
+        emit(Events::SessionStart.new)
+
+        active_tools = resolve_tools(tools || [])
+        active_tools = @tools if active_tools.empty?
+
+        if active_tools.empty? && !@_no_tools_instructed
+          @chat.add_message(role: :system, content: "You have no tools available. Do not claim you can look up information or use tools of any kind. Just respond based on your existing knowledge.")
+          @_no_tools_instructed = true
+        end
+
+        begin
+          @tool_executor.telemetry = @telemetry
+
+          response = @loop.run_turn(
+            chat: @chat,
+            message: message,
+            tools: active_tools,
+            tool_executor: @tool_executor,
+            compactor: @compactor,
+            hooks: @hooks,
+            event_emitter: self,
+            session_id: @id
+          )
+        rescue MaxTurnsExceeded => e
+          emit(Events::MaxTurnsExceeded.new(max_turns: @max_turns))
+          @telemetry.log(:max_turns_exceeded, session_id: @id, max_turns: @max_turns)
+          response = last_content
+        rescue LoopDetected => e
+          emit(Events::LoopDetected.new(tool_name: e.message, repeated_count: 3))
+          @telemetry.log(:loop_detected, session_id: @id, tool_name: e.message, repeated_count: 3)
+          response = last_content
+        rescue Ask::ContextLengthExceeded
+          if @compactor && !@compactor.overflow_recovered?
+            @compactor.recover_from_overflow
+            retry
+          end
+          response = "I'm sorry, the conversation has grown too long. Please start a new session."
+        rescue StandardError => e
+          emit(Events::Error.new(error: e.message, recoverable: true))
+          raise
+        ensure
+          @running = false
+          persist! if @persistence
+        end
+
+        @tool_calls_made = @tool_executor.total_executions
+
+        if @reflector && @reflector.reflect?(@tool_calls_made) && !@abort_requested
+          eval_result = @reflector.evaluate(response: response, event_emitter: self)
+          @telemetry.log(:reflection_end, session_id: @id, decision: eval_result[:decision], feedback: eval_result[:feedback])
+
+          if eval_result[:decision] == :improve && !@abort_requested
+            @chat.add_message(
+              role: :system,
+              content: "Improve your last response: #{eval_result[:feedback]}"
+            )
+
+            response = @loop.run_turn(
+              chat: @chat,
+              message: "",
+              tools: active_tools,
+              tool_executor: @tool_executor,
+              compactor: @compactor,
+              hooks: @hooks,
+              event_emitter: self,
+              session_id: @id
+            )
+          end
+        end
+
+        if @meta_agent_config
+          @telemetry.increment_session_count!
+          try_auto_meta_agent
+        end
+
+        emit(Events::SessionEnd.new(result: response, turn_count: @turn_count, tool_calls_made: @tool_calls_made))
+        @messages = @chat.messages.dup
+
+        response
+      end
+
+      def on_event(&block)
+        @event_handlers[:all] << block
+        self
+      end
+
+      def on(type, &block)
+        @event_handlers[type] ||= []
+        @event_handlers[type] << block
+        self
+      end
+
+      def emit(event)
+        @event_handlers[:all].each { |h| h.call(event) }
+        handlers = @event_handlers[event.class]
+        handlers&.each { |h| h.call(event) }
+      end
+
+      def running? = @running
+      def deleted? = @deleted
+
+      def save
+        persist! if @persistence
+      end
+
+      def self.load(id, adapter:)
+        data = adapter.load(id)
+        return nil unless data
+
+        session = new(
+          id: data[:id],
+          model: data.dig(:metadata, :model),
+          tools: data.dig(:metadata, :tools)&.map(&:constantize) || [],
+          persistence: adapter
+        )
+
+        data[:messages].each do |msg|
+          session.chat.add_message(
+            role: msg[:role].to_sym,
+            content: msg[:content],
+            tool_call_id: msg[:tool_call_id]
+          )
+        end
+
+        session
+      end
+
+      def delete
+        @deleted = true
+        @persistence&.delete(@id)
+      end
+
+      def abort
+        @abort_requested = true
+      end
+

data/lib/ask/agent/version.rb

@@ -2,6 +2,6 @@
 
 module Ask
   module Agent
-    VERSION = "0.1.1"
+    VERSION = "0.1.2"
   end
 end

data/lib/ask/agent.rb

@@ -4,6 +4,7 @@ require "fileutils"
 require "json"
 require "securerandom"
 require "time"
+require "ask/skills"
 
 module Ask
   module Agent

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask-agent
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Kaka Ruto
@@ -51,6 +51,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: ask-skills
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
 - !ruby/object:Gem::Dependency
   name: ask-tools-shell
   requirement: !ruby/object:Gem::Requirement
@@ -133,6 +147,7 @@ files:
 - lib/ask/agent/persistence/in_memory.rb
 - lib/ask/agent/reflector.rb
 - lib/ask/agent/session.rb
+- lib/ask/agent/session_backup.rb
 - lib/ask/agent/telemetry.rb
 - lib/ask/agent/tool_abort_controller.rb
 - lib/ask/agent/tool_executor.rb