ask-agent 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bb3f240855fe04ef37b3d474cd4f9ddae5db28256673aa1d1804bdfbc1a8836c
-  data.tar.gz: 84684f83697510a261de607ede4a6aafd2a396a2151d169290769148214bc4a0
+  metadata.gz: f9b59871c0216de559acd011a9553c650ca4f3714150c7e71811b375a92f2a48
+  data.tar.gz: edb6f7137010f1c448da22ca99a7cf2c90dda29f70d210db4c2e18d5f4079568
 SHA512:
-  metadata.gz: 41ac967bf49f11bea0d4a65d9984ddfc0838007096b153c2294691abd8ce3bdd60aae0890e10f7bfed39ce2ed0b7f71f577e33d10a029f1660c6b779b4117f13
-  data.tar.gz: e097d8c0bbd5bf5f78cccc49fd167d951119d91ee473915803c92dfbbae245167b95050ad321ba2ccfbed71fc3c224effe30252c2badeff8c6fbb07f624c801e
+  metadata.gz: c3c7ddc41ed53e2978b9860814ed802b9bae305f81fe442dc3b123b8f34786f4bd16a1f9fa90ff6069cae4570aba8033f135a4187e39806745b16f33fef976b4
+  data.tar.gz: dafe52d28098ac72a45557547818bcea55c609d1db0856d6c67ea7fd330a837ed80ded9145cc6d0d3cc3f45818f268db5c4faca7cb38b9220a63ccc3a84457b9

data/lib/ask/agent/chat.rb

@@ -66,7 +66,7 @@ module Ask
         # Accumulator for tool calls during streaming (keyed by index)
         calls_acc = {}
 
-        result = provider.chat(
+        result = provider.chat(@extra_params || {}, 
           @messages.map(&:to_h),
           model: @model_id,
           tools: tool_defs,
@@ -134,6 +134,15 @@ module Ask
       #
       # @param schema [Ask::Schema, Hash] structured output schema
       # @return [self]
+  # Set additional parameters for the provider call and return self.
+  #
+  # @param params [Hash] extra parameters passed to the provider
+  # @return [self]
+  def with_params(**params)
+    @extra_params = (@extra_params || {}).merge(params)
+    self
+  end
+
       def with_schema(schema)
         @schema = schema.respond_to?(:to_json_schema) ? schema.to_json_schema : schema
         self

data/lib/ask/agent/version.rb

@@ -2,6 +2,6 @@
 
 module Ask
   module Agent
-    VERSION = "0.1.3"
+    VERSION = "0.1.4"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask-agent
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Kaka Ruto
@@ -147,7 +147,6 @@ 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

data/lib/ask/agent/session_backup.rb

@@ -1,237 +0,0 @@
-# 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
-