ask-agent 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e875e155f72f3bc773ff218b64243ac918354fac930b0181e02ae1a74d0bc4eb
-  data.tar.gz: 9ec62c98c64b7307f445311cce0f6fd6bf81a9f7fde52daa505cd532d4734520
+  metadata.gz: bb3f240855fe04ef37b3d474cd4f9ddae5db28256673aa1d1804bdfbc1a8836c
+  data.tar.gz: 84684f83697510a261de607ede4a6aafd2a396a2151d169290769148214bc4a0
 SHA512:
-  metadata.gz: 14d6490ab16f5e675e05ac3e72f37c113edcfd7e7339c10fe6a73290b88e4cab0ab8535d21860e51d09aac36c7a7066b3dbfe1f19b04d5c873a301094fb610fe
-  data.tar.gz: 729b178fb7f474e2fe5c84ce692169ba184fefe1b761d7ddc2a1dfaf39c57d184155f9f67e186bd91b13faaab05a5e79aac49107055d0089269ca10e9e100f94
+  metadata.gz: 41ac967bf49f11bea0d4a65d9984ddfc0838007096b153c2294691abd8ce3bdd60aae0890e10f7bfed39ce2ed0b7f71f577e33d10a029f1660c6b779b4117f13
+  data.tar.gz: e097d8c0bbd5bf5f78cccc49fd167d951119d91ee473915803c92dfbbae245167b95050ad321ba2ccfbed71fc3c224effe30252c2badeff8c6fbb07f624c801e

data/lib/ask/agent/chat.rb

@@ -26,8 +26,11 @@ module Ask
     # internal usage (Hash of { id => ToolCallInfo }).
     class Chat
       # @return [String] model ID (e.g. "gpt-4o")
-      attr_reader :model
-
+        attr_reader :model_id
+      # @return [String] model ID (e.g. "gpt-4o")
+      def model
+        @model_id
+      end
       # @return [Array<Ask::Message>] all messages in the conversation
       attr_reader :messages
 
@@ -35,13 +38,17 @@ module Ask
       # @param tools [Array<Ask::Tool>] tool instances available to the chat
       # @param temperature [Float, nil] sampling temperature
       # @param schema [Ask::Schema, Hash, nil] structured output schema
-      def initialize(model:, tools: [], temperature: nil, schema: nil, **)
+      # @param assume_model_exists [Boolean] whether to skip model catalog lookup
+      # @param provider [String, Symbol, nil] provider slug to use (overrides model catalog)
+      def initialize(model:, tools: [], temperature: nil, schema: nil,
+                     assume_model_exists: false, provider: nil, **)
         @model_id = model.respond_to?(:id) ? model.id : model.to_s
-        @model_info = resolve_model(@model_id)
+        @model_info = resolve_model(@model_id) unless assume_model_exists
         @tools = tools
         @temperature = temperature
         @schema = schema
         @messages = []
+        @provider_override = provider
         @provider = nil
       end
 
@@ -123,6 +130,15 @@ module Ask
         self
       end
 
+      # Set the structured output schema and return self.
+      #
+      # @param schema [Ask::Schema, Hash] structured output schema
+      # @return [self]
+      def with_schema(schema)
+        @schema = schema.respond_to?(:to_json_schema) ? schema.to_json_schema : schema
+        self
+      end
+
       # Clear all messages from the conversation.
       def reset_messages!
         @messages.clear
@@ -143,19 +159,19 @@ module Ask
       end
 
       def build_provider
-        slug = @model_info&.provider || "openai"
+        slug = @provider_override&.to_s || @model_info&.provider || "openai"
         klass = Ask::Provider.resolve(slug)
         klass.new(provider_config(slug))
       end
 
       def provider_config(slug, extra_keys: {})
-          env_key = "#{slug.upcase}_API_KEY"
-          key = ENV[env_key] || ENV["OPENCODE_API_KEY"] || ENV["OPENAI_API_KEY"]
-          base = ENV["#{slug.upcase}_API_BASE"] || ENV["OPENCODE_API_BASE"]
-          config = { api_key: key }
-          config[:"#{slug}_api_key"] = key
-          config[:"#{slug}_api_base"] = base if base
-          Ask::LLM::Config.new(config)
+        env_key = "#{slug.upcase}_API_KEY"
+        key = ENV[env_key] || ENV["OPENCODE_API_KEY"] || ENV["OPENAI_API_KEY"]
+        base = ENV["#{slug.upcase}_API_BASE"] || ENV["OPENCODE_API_BASE"]
+        config = { api_key: key }
+        config[:"#{slug}_api_key"] = key
+        config[:"#{slug}_api_base"] = base if base
+        Ask::LLM::Config.new(config)
       end
 
       # Accumulate partial tool calls from streaming chunks.

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.3"
   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.3
 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