ai-agents 0.1.2 → 0.2.0

data/lib/agents/runner.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "message_extractor"
+
 module Agents
   # The execution engine that orchestrates conversations between users and agents.
   # Runner manages the conversation flow, handles tool execution through RubyLLM,
@@ -77,14 +79,15 @@ module Agents
     # @param context [Hash] Shared context data accessible to all tools
     # @param registry [Hash] Registry of agents for handoff resolution
     # @param max_turns [Integer] Maximum conversation turns before stopping
+    # @param callbacks [Hash] Optional callbacks for real-time event notifications
     # @return [RunResult] The result containing output, messages, and usage
-    def run(starting_agent, input, context: {}, registry: {}, max_turns: DEFAULT_MAX_TURNS)
+    def run(starting_agent, input, context: {}, registry: {}, max_turns: DEFAULT_MAX_TURNS, callbacks: {})
       # The starting_agent is already determined by AgentRunner based on conversation history
       current_agent = starting_agent
 
       # Create context wrapper with deep copy for thread safety
       context_copy = deep_copy_context(context)
-      context_wrapper = RunContext.new(context_copy)
+      context_wrapper = RunContext.new(context_copy, callbacks: callbacks)
       current_turn = 0
 
       # Create chat and restore conversation history
@@ -97,8 +100,12 @@ module Agents
 
         # Get response from LLM (Extended Chat handles tool execution with handoff detection)
         result = if current_turn == 1
+                   # Emit agent thinking event for initial message
+                   context_wrapper.callback_manager.emit_agent_thinking(current_agent.name, input)
                    chat.ask(input)
                  else
+                   # Emit agent thinking event for continuation
+                   context_wrapper.callback_manager.emit_agent_thinking(current_agent.name, "(continuing conversation)")
                    chat.complete
                  end
         response = result
@@ -119,6 +126,9 @@ module Agents
           # Save current conversation state before switching
           save_conversation_state(chat, context_wrapper, current_agent)
 
+          # Emit agent handoff event
+          context_wrapper.callback_manager.emit_agent_handoff(current_agent.name, next_agent.name, "handoff")
+
           # Switch to new agent - store agent name for persistence
           current_agent = next_agent
           context_wrapper.context[:current_agent] = next_agent.name
@@ -143,7 +153,7 @@ module Agents
 
         return RunResult.new(
           output: response.content,
-          messages: extract_messages(chat, current_agent),
+          messages: MessageExtractor.extract_messages(chat, current_agent),
           usage: context_wrapper.usage,
           context: context_wrapper.context
         )
@@ -154,7 +164,7 @@ module Agents
 
       RunResult.new(
         output: "Conversation ended: #{e.message}",
-        messages: chat ? extract_messages(chat, current_agent) : [],
+        messages: chat ? MessageExtractor.extract_messages(chat, current_agent) : [],
         usage: context_wrapper.usage,
         error: e,
         context: context_wrapper.context
@@ -165,7 +175,7 @@ module Agents
 
       RunResult.new(
         output: nil,
-        messages: chat ? extract_messages(chat, current_agent) : [],
+        messages: chat ? MessageExtractor.extract_messages(chat, current_agent) : [],
         usage: context_wrapper.usage,
         error: e,
         context: context_wrapper.context
@@ -208,7 +218,7 @@ module Agents
 
     def save_conversation_state(chat, context_wrapper, current_agent)
       # Extract messages from chat
-      messages = extract_messages(chat, current_agent)
+      messages = MessageExtractor.extract_messages(chat, current_agent)
 
       # Update context with latest state
       context_wrapper.context[:conversation_history] = messages
@@ -244,26 +254,5 @@ module Agents
       chat.with_tools(*wrapped_regular_tools) if wrapped_regular_tools.any?
       chat
     end
-
-    def extract_messages(chat, current_agent)
-      return [] unless chat.respond_to?(:messages)
-
-      chat.messages.filter_map do |msg|
-        # Only include user and assistant messages with content
-        next unless %i[user assistant].include?(msg.role)
-        next unless msg.content && !msg.content.strip.empty?
-
-        message = {
-          role: msg.role,
-          content: msg.content
-        }
-
-        # Add agent attribution for assistant messages to enable conversation continuity
-        # This allows AgentRunner to determine which agent should continue the conversation
-        message[:agent_name] = current_agent.name if msg.role == :assistant && current_agent
-
-        message
-      end
-    end
   end
 end

data/lib/agents/tool_wrapper.rb

@@ -46,7 +46,17 @@ module Agents
     # RubyLLM calls this method (follows RubyLLM::Tool pattern)
     def call(args)
       tool_context = ToolContext.new(run_context: @context_wrapper)
-      @tool.execute(tool_context, **args.transform_keys(&:to_sym))
+
+      @context_wrapper.callback_manager.emit_tool_start(@tool.name, args)
+
+      begin
+        result = @tool.execute(tool_context, **args.transform_keys(&:to_sym))
+        @context_wrapper.callback_manager.emit_tool_complete(@tool.name, result)
+        result
+      rescue StandardError => e
+        @context_wrapper.callback_manager.emit_tool_complete(@tool.name, "ERROR: #{e.message}")
+        raise
+      end
     end
 
     # Delegate metadata methods to the tool

data/lib/agents/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Agents
-  VERSION = "0.1.2"
+  VERSION = "0.2.0"
 end

data/lib/agents.rb

@@ -80,5 +80,8 @@ require_relative "agents/agent"
 # Execution components
 require_relative "agents/chat"
 require_relative "agents/tool_wrapper"
+require_relative "agents/message_extractor"
+require_relative "agents/callback_manager"
 require_relative "agents/agent_runner"
 require_relative "agents/runner"
+require_relative "agents/agent_tool"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai-agents
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Shivam Mishra
@@ -31,13 +31,57 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".claude/commands/bump-version.md"
 - ".rspec"
 - ".rubocop.yml"
+- CHANGELOG.md
 - CLAUDE.md
 - LICENSE
 - README.md
 - Rakefile
+- docs/Gemfile
+- docs/Gemfile.lock
+- docs/_config.yml
+- docs/_sass/color_schemes/ruby.scss
+- docs/_sass/custom/custom.scss
+- docs/architecture.md
+- docs/assets/fonts/InterVariable.woff2
+- docs/concepts.md
+- docs/concepts/agent-tool.md
+- docs/concepts/agents.md
+- docs/concepts/callbacks.md
+- docs/concepts/context.md
+- docs/concepts/handoffs.md
+- docs/concepts/runner.md
+- docs/concepts/tools.md
+- docs/guides.md
+- docs/guides/agent-as-tool-pattern.md
+- docs/guides/multi-agent-systems.md
+- docs/guides/rails-integration.md
+- docs/guides/state-persistence.md
+- docs/index.md
 - examples/README.md
+- examples/collaborative-copilot/README.md
+- examples/collaborative-copilot/agents/analysis_agent.rb
+- examples/collaborative-copilot/agents/answer_suggestion_agent.rb
+- examples/collaborative-copilot/agents/copilot_orchestrator.rb
+- examples/collaborative-copilot/agents/integrations_agent.rb
+- examples/collaborative-copilot/agents/research_agent.rb
+- examples/collaborative-copilot/data/contacts.json
+- examples/collaborative-copilot/data/conversations.json
+- examples/collaborative-copilot/data/knowledge_base.json
+- examples/collaborative-copilot/data/linear_issues.json
+- examples/collaborative-copilot/data/stripe_billing.json
+- examples/collaborative-copilot/interactive.rb
+- examples/collaborative-copilot/tools/create_linear_ticket_tool.rb
+- examples/collaborative-copilot/tools/get_article_tool.rb
+- examples/collaborative-copilot/tools/get_contact_tool.rb
+- examples/collaborative-copilot/tools/get_conversation_tool.rb
+- examples/collaborative-copilot/tools/get_stripe_billing_tool.rb
+- examples/collaborative-copilot/tools/search_contacts_tool.rb
+- examples/collaborative-copilot/tools/search_conversations_tool.rb
+- examples/collaborative-copilot/tools/search_knowledge_base_tool.rb
+- examples/collaborative-copilot/tools/search_linear_issues_tool.rb
 - examples/isp-support/README.md
 - examples/isp-support/agents_factory.rb
 - examples/isp-support/data/customers.json
@@ -52,8 +96,11 @@ files:
 - lib/agents.rb
 - lib/agents/agent.rb
 - lib/agents/agent_runner.rb
+- lib/agents/agent_tool.rb
+- lib/agents/callback_manager.rb
 - lib/agents/chat.rb
 - lib/agents/handoff.rb
+- lib/agents/message_extractor.rb
 - lib/agents/result.rb
 - lib/agents/run_context.rb
 - lib/agents/runner.rb