ruboty-ai_agent 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1782e855675403948a771dd5abe6fdba6a754cfc6e15bebad40f24bf3c7569f6
-  data.tar.gz: 2142af0fb45049c4e1b2397def8009248db8860e43a9fa2ae21f938a31e3af82
+  metadata.gz: 4358837ce9305dcb0965a67c957b75e8320aad2988647a9e6ae751a98e7b7009
+  data.tar.gz: 3b1ad1c43bc443c5d86b1ea80bd8d9a45263056bc91ade861058e600493cb1b7
 SHA512:
-  metadata.gz: 1a1b7fc8e8e41150055e87d2227e9b0ae8570c69482cf72249af38abef0978058419eed73f778a0f4d9c00343a55ce96be0cd62febdb4a20cd8b7976a6852415
-  data.tar.gz: 73aff06e4a6d7fb3601b9c95393b9c1496fd61eddaaabf0fe742e7701faad638a36c3919462e65847dab47d0450579c8047ba492beb4d51c81181b5b66307aef
+  metadata.gz: fb0b30fad316d595d3e088d9262b46b32af67467c23b28494a7214f83093bb2a939953005c19b84e907eae81c4d1ba00a2dd84ca96e8cc85fdffe016d687e036
+  data.tar.gz: 586e425b1c2f806d663e861d18fa57579649b6d29135576a1677f825f637fc017d5cb910e6b1edc1bc2e1a7d5c8eb4e40d6c7795c170e69f418434b84750c0f8

data/.rubocop.yml

@@ -11,9 +11,11 @@
 plugins:
   - rubocop-rake
   - rubocop-rbs_inline
+  - rubocop-rspec
 
 AllCops:
   TargetRubyVersion: 3.1
+  NewCops: enable
 
 Naming/AccessorMethodName:
   Exclude:
@@ -33,6 +35,23 @@ Layout/LeadingCommentSpace:
 Metrics:
   Enabled: false
 
+RSpec/ExampleLength:
+  Max: 7
+  CountAsOne: ["array", "heredoc", "method_call"]
+
+RSpec/MultipleMemoizedHelpers:
+  Enabled: false
+
+RSpec/MultipleExpectations:
+  Enabled: false
+
+RSpec/NestedGroups:
+  Enabled: false
+
+RSpec/SpecFilePathFormat:
+  CustomTransform:
+    OpenAI: openai
+
 Style/CommentedKeyword:
   Enabled: false # because of RBS annotations
 

data/AGENTS.md

@@ -20,3 +20,11 @@
 ## PR instructions
 
 - Always run `rake autocorrect` before committing.
+
+## CHANGELOG instructions
+
+- When you make changes, please update `CHANGELOG.md` accordingly.
+- Add to the `Unreleased` section at the top.
+- Follow the format used in previous entries.
+- Use English for the changelog entries
+

data/CHANGELOG.md

@@ -1,3 +1,15 @@
-# Changelog
+## Unreleased
+## 0.3.0
 
-All notable changes to this project will be documented in this file.
+- Add think tool.
+- Different agent threads are prepared for different slack threads.
+
+## 0.2.0
+
+- Support for user defined commends.
+- Agent uses user defined system prompts and memories.
+- Support automatic prompt compaction.
+
+## 0.1.0
+
+- Initial release with basic features and functionality.

data/Gemfile

@@ -7,8 +7,15 @@ git_source(:github) { |repo_name| "https://github.com/#{repo_name}" }
 # Specify your gem's dependencies in ruboty-ai_agent.gemspec
 gemspec
 
+gem 'bump', '~> 0.10.0'
+gem 'bundler', '~> 2'
+gem 'rake', '~> 13'
+gem 'rspec', '~> 3.0'
+gem 'webmock', '~> 3.0'
+
 gem 'rbs-inline', require: false
-gem 'rubocop'
-gem 'rubocop-rake'
-gem 'rubocop-rbs_inline'
+gem 'rubocop', '~> 1.74.0'
+gem 'rubocop-rake', '~> 0.7.0'
+gem 'rubocop-rbs_inline', '~> 1.4.0'
+gem 'rubocop-rspec', '~> 3.7.0'
 gem 'steep', require: false

data/README.md

@@ -57,9 +57,11 @@ MCP (Model Context Protocol):
 - `add mcp <NAME> <OPTIONS> <URL>` — Add an MCP server
   - Example (HTTP transport with auth header):
     - `add mcp search --transport http --header 'Authorization: Bearer xxx' https://example.com/mcp`
+    - `add mcp search --transport http --bearer-token xxx https://example.com/mcp`
   - Options:
     - `--transport http|sse` (currently only `http` implemented; `sse` is not yet implemented)
     - `--header 'Key: Value'` (repeatable)
+    - `--bearer-token <TOKEN>` (shorthand for `--header 'Authorization: Bearer <TOKEN>'`)
 - `remove mcp <NAME>` — Remove an MCP server
 - `list mcp` / `list mcps` — List configured MCP servers
 

data/Rakefile

@@ -4,6 +4,7 @@ require 'bundler/gem_tasks'
 require 'rspec/core/rake_task'
 require 'rubocop/rake_task'
 require 'steep/rake_task'
+require 'bump/tasks'
 
 RSpec::Core::RakeTask.new(:spec)
 
@@ -16,6 +17,8 @@ Steep::RakeTask.new do |t|
   t.watch.verbose
 end
 
+Bump.changelog = true
+
 task default: %i[rubocop steep spec]
 task autocorrect: %i[rubocop:autocorrect rbs steep spec]
 
@@ -33,6 +36,7 @@ namespace :rbs do
 
   desc 'Run rbs-inline to generate RBS files'
   task :inline do
+    sh('script/clean-orphaned-rbs.rb')
     sh('bundle exec rbs-inline --opt-out --output lib')
   end
 

data/Steepfile

@@ -9,4 +9,5 @@ target :lib do
   library 'json'
   library 'net-http'
   library 'optparse'
+  library 'shellwords'
 end

data/lib/ruboty/ai_agent/actions/add_ai_command.rb

@@ -6,7 +6,14 @@ module Ruboty
       # AddAiCommand action for Ruboty::AiAgent
       class AddAiCommand < Base
         def call
-          message.reply("TODO: Implement AddAiCommand action - name: #{name_param}, prompt: #{prompt_param}")
+          if user.prompt_command_definitions.key?(name_param)
+            message.reply("Command '/#{name_param}' already exists. Use a different name or remove the existing command first.")
+            return
+          end
+
+          new_command = PromptCommandDefinition.new(name: name_param, prompt: undump_string(prompt_param))
+          user.prompt_command_definitions.add(new_command)
+          message.reply("Command '/#{name_param}' has been added successfully!")
         end
 
         def name_param #: String
@@ -16,6 +23,14 @@ module Ruboty
         def prompt_param #: String
           message[:prompt]
         end
+
+        # @rbs str: String
+        # @rbs return: String
+        def undump_string(str)
+          str.undump
+        rescue StandardError
+          str
+        end
       end
     end
   end

data/lib/ruboty/ai_agent/actions/add_mcp.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'optparse'
+require 'shellwords'
 
 module Ruboty
   module AiAgent
@@ -58,11 +59,11 @@ module Ruboty
             args: []
           } #: config
 
-          args = config_param.split(/\s+(?=-)/)
+          args = config_param.shellsplit
 
           parser = OptionParser.new do |opts|
             opts.on('--transport TYPE', %w[http sse], 'Transport type (http or sse)') do |t|
-              options[:transport] = t
+              options[:transport] = t.to_sym
             end
 
             opts.on('--header VALUE', 'Add a header (can be specified multiple times)') do |h|
@@ -74,6 +75,10 @@ module Ruboty
 
               options[:headers][key] = value
             end
+
+            opts.on('--bearer-token TOKEN', 'Set Authorization Bearer token (shorthand for --header "Authorization: Bearer TOKEN")') do |token|
+              options[:headers]['Authorization'] = "Bearer #{token}"
+            end
           end
 
           options[:args] = parser.parse(args)

data/lib/ruboty/ai_agent/actions/base.rb

@@ -35,7 +35,17 @@ module Ruboty
 
         # @rbs %a{memorized}
         def chat_thread #: Ruboty::AiAgent::ChatThread
-          @chat_thread ||= database.chat_thread(message.from || 'default')
+          @chat_thread ||= database.chat_thread(thread_id)
+        end
+
+        private
+
+        def thread_id #: String
+          if message.respond_to?(:original) && message.original&.dig(:thread_ts)
+            message.original[:thread_ts]
+          else
+            message.from || 'default'
+          end
         end
       end
     end

data/lib/ruboty/ai_agent/actions/chat.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'json'
+
 module Ruboty
   module AiAgent
     module Actions
@@ -9,28 +11,66 @@ module Ruboty
 
         # @rbs override
         def call
-          llm = LLM::OpenAI.new(
-            client: OpenAI::Client.new(
-              api_key: ENV.fetch('OPENAI_API_KEY', nil)
-            ),
-            model: ENV.fetch('OPENAI_MODEL', 'gpt-5-nano')
-          )
-
-          commands = Commands.builtins(message:, chat_thread:)
-          tools = McpClients.new(user.mcp_clients).available_tools
+          user.prompt_command_definitions.all_values.each do |definition|
+            command = Commands::PromptCommand.new(definition:, request:)
+            if command.match?(body_param)
+              new_prompt = command.call(commandline: body_param)
+              return complete_chat(new_prompt)
+            end
+          end
 
-          commands.each do |command|
+          builtin_commands = Commands.builtins(request:)
+          builtin_commands.each do |command|
             return command.call if command.match?(body_param)
           end
 
+          complete_chat(body_param)
+        end
+
+        def body_param #: String
+          message[:body]
+        end
+
+        # @rbs %a{memorized}
+        def request #: Request
+          @request ||= Request.new(message:, chat_thread:)
+        end
+
+        private
+
+        # @rbs body: String
+        # @rbs return: void
+        def complete_chat(body)
           chat_thread.messages << ChatMessage.new(
             role: :user,
-            content: body_param
+            content: body
           )
 
+          messages = [] #: Array[ChatMessage]
+
+          global_prompt = database.global_settings.system_prompt
+          messages << ChatMessage.new(role: :system, content: global_prompt) if global_prompt
+
+          user_prompt = user.system_prompt
+          messages << ChatMessage.new(role: :system, content: user_prompt) if user_prompt
+
+          ai_memories = user.ai_memories.all || {}
+          unless ai_memories.empty?
+            memory_content = ai_memories.map { |_idx, memory| memory }.join("\n\n")
+            messages << ChatMessage.new(role: :user, content: "My memories:\n#{memory_content}")
+          end
+
+          messages += chat_thread.messages.all_values
+
+          llm = LLM::OpenAI.new
+          tools = [
+            *McpClients.new(user.mcp_clients).available_tools,
+            *ToolDefinitions.builtins(request:).map(&:to_tool)
+          ]
+
           agent = Agent.new(
             llm:,
-            messages: chat_thread.messages.all_values,
+            messages:,
             tools:
           )
 
@@ -39,12 +79,13 @@ module Ruboty
             when :new_message
               chat_thread.messages << event[:message]
               message.reply(event[:message].content) if event[:message].content.length.positive?
+
+              chat_thread.messages.compact(llm:) if chat_thread.messages.over_auto_compact_threshold?
             when :tool_call
-              message.reply("Calling tool #{event[:tool].name} with arguments #{event[:tool_arguments]}",
-                            streaming: true)
+              message.reply("Calling tool #{event[:tool].name} with arguments #{event[:tool_arguments]&.to_json}") unless event[:tool].silent?
             when :tool_response
               chat_thread.messages << event[:message]
-              message.reply("Tool response: #{event[:tool_response].slice(0..100)}")
+              message.reply("Tool response: #{event[:tool_response].slice(0..100)}") unless event[:tool].silent?
             end
           end
         rescue StandardError => e
@@ -54,10 +95,6 @@ module Ruboty
             message.reply("エラーが発生しました: #{e.message}")
           end
         end
-
-        def body_param #: String
-          message[:body]
-        end
       end
     end
   end

data/lib/ruboty/ai_agent/actions/list_ai_commands.rb

@@ -7,11 +7,27 @@ module Ruboty
       class ListAiCommands < Base
         # @rbs override
         def call
-          commands = Commands.builtins(message:, chat_thread:)
+          builtin_commands = Commands.builtins(request: Request.new(message:, chat_thread:))
 
-          message.reply(
-            commands.flat_map(&:matchers).map { |matcher| "#{matcher.pattern.inspect} - #{matcher.description}" }.join("\n")
-          )
+          builtin_list = builtin_commands.flat_map(&:matchers)
+                                         .map { |matcher| "#{matcher.pattern.inspect} - #{matcher.description}" }
+
+          user_defined_list = user.prompt_command_definitions.all_values
+                                  .map { |cmd| "/#{cmd.name} - #{cmd.prompt}" }
+
+          content = <<~TEXT
+            # User-Defined Commands
+            #{user_defined_list.empty? ? 'No user-defined commands available.' : user_defined_list.join("\n")}
+
+            # Built-in Commands
+            #{builtin_list.empty? ? 'No built-in commands available.' : builtin_list.join("\n")}
+          TEXT
+
+          if builtin_commands.empty? && user_defined_list.empty?
+            message.reply('No commands available.')
+          else
+            message.reply(content)
+          end
         end
       end
     end

data/lib/ruboty/ai_agent/actions/list_mcp.rb

@@ -6,11 +6,58 @@ module Ruboty
       # ListMcp action for Ruboty::AiAgent
       class ListMcp < Base
         def call
-          mcp_configurations = (user.mcp_configurations.all || {}).map do |name, mcp_configuration|
-            "#{name}: #{mcp_configuration.to_h.except(:record_type).to_json}"
+          clients = user.mcp_clients
+
+          if clients.empty?
+            message.reply('No MCP servers found.')
+            return
+          end
+
+          show_headers = !message[:with_headers].nil?
+          output = clients.map do |mcp_client|
+            format_mcp_client(mcp_client, show_headers: show_headers)
+          end.join("\n\n")
+
+          message.reply(output)
+        end
+
+        private
+
+        # @rbs client: UserMcpClient
+        # @rbs show_headers: bool
+        # @rbs return: String
+        def format_mcp_client(client, show_headers: false)
+          configuration = client.configuration
+
+          tools_info = format_tools(client)
+          mcp_info = <<~TEXT
+            #{configuration.name}:
+              Transport: #{configuration.transport}
+              URL: #{configuration.url}
+          TEXT
+
+          mcp_info += "  Headers: #{configuration.headers.to_json}\n" if show_headers
+
+          "#{mcp_info}#{tools_info}".chomp
+        end
+
+        # @rbs client: UserMcpClient
+        # @rbs return: String
+        def format_tools(client)
+          tools = client.list_tools
+          return '' if tools.empty?
+
+          tools_output = tools.map do |tool|
+            tool_name = tool['name'] || 'unnamed'
+            description = tool['description'] || 'No description'
+            truncated_description = description.length > 100 ? "#{description[0, 100]}..." : description
+            "  - #{tool_name}: #{truncated_description}"
           end.join("\n")
 
-          message.reply(mcp_configurations.empty? ? 'No memories found.' : mcp_configurations)
+          "  Tools:\n#{tools_output}\n"
+        rescue HttpMcpClient::Error => e
+          warn "Failed to list tools for MCP client: #{e.message}"
+          ''
         end
       end
     end

data/lib/ruboty/ai_agent/actions/remove_ai_command.rb

@@ -6,7 +6,13 @@ module Ruboty
       # RemoveAiCommand action for Ruboty::AiAgent
       class RemoveAiCommand < Base
         def call
-          message.reply("TODO: Implement RemoveAiCommand action for name: #{name_param}")
+          unless user.prompt_command_definitions.key?(name_param)
+            message.reply("Command '/#{name_param}' does not exist.")
+            return
+          end
+
+          user.prompt_command_definitions.remove(name_param)
+          message.reply("Command '/#{name_param}' has been removed successfully!")
         end
 
         def name_param #: String

data/lib/ruboty/ai_agent/actions/show_system_prompt.rb

@@ -9,7 +9,7 @@ module Ruboty
           user_prompt = user.system_prompt
           global_prompt = database.global_settings.system_prompt
 
-          reply = []
+          reply = [] #: Array[String]
           reply << if user_prompt
                      "User system prompt: #{user_prompt}"
                    else

data/lib/ruboty/ai_agent/agent.rb

@@ -41,7 +41,7 @@ module Ruboty
               tool_arguments: response.tool_arguments,
               tool_response:
             )
-            on_tool_response(tool_response:, message: tool_response_message, &)
+            on_tool_response(tool: response.tool, tool_response:, message: tool_response_message, &)
             messages << tool_response_message
           else
             messages << response.message
@@ -59,8 +59,8 @@ module Ruboty
         callback&.call({ type: :tool_call, tool:, tool_arguments: })
       end
 
-      def on_tool_response(tool_response:, message:, &callback)
-        callback&.call({ type: :tool_response, tool_response:, message: })
+      def on_tool_response(tool:, tool_response:, message:, &callback)
+        callback&.call({ type: :tool_response, tool:, tool_response:, message: })
       end
 
       def on_response(response, &callback)

data/lib/ruboty/ai_agent/chat_message.rb

@@ -15,14 +15,14 @@ module Ruboty
       attr_reader :tool_arguments #: Hash[Symbol | String, untyped]?
       attr_reader :token_usage #: TokenUsage?
 
-      # @rbs role: Symbol
+      # @rbs role: Symbol | String
       # @rbs content: String
       # @rbs ?tool_call_id: String?
       # @rbs ?tool_name: String?
       # @rbs ?tool_arguments: Hash[Symbol | String, untyped]?
       # @rbs ?token_usage: TokenUsage?
       def initialize(role:, content:, tool_call_id: nil, tool_name: nil, tool_arguments: nil, token_usage: nil)
-        @role = role
+        @role = role.to_sym
         @content = content
         @tool_call_id = tool_call_id
         @tool_name = tool_name
@@ -41,6 +41,11 @@ module Ruboty
         }
       end
 
+      # @rbs return: bool
+      def tool_call?
+        !!(tool_call_id && !tool_call_id.empty?)
+      end
+
       def self.from_llm_response(
         tool:,
         tool_call_id:,

data/lib/ruboty/ai_agent/chat_thread_messages.rb

@@ -12,6 +12,46 @@ module Ruboty
       end
 
       alias << add
+
+      # Check if any message's token usage exceeds auto compact threshold
+      # @rbs return: bool
+      def over_auto_compact_threshold?
+        all_values.any? { |message| message.token_usage&.over_auto_compact_threshold? }
+      end
+
+      # Compact chat messages by summarizing them
+      # @rbs llm: LLM::OpenAI
+      # @rbs return: void
+      def compact(llm:)
+        messages = all_values
+        return if messages.empty?
+
+        last_assistant_message = messages.reverse.find { |msg| msg.role == :assistant }
+
+        summary = summarize(llm:)
+
+        clear
+        add(ChatMessage.new(
+              role: :system,
+              content: "Previous conversation summary: #{summary}"
+            ))
+        add(last_assistant_message) if last_assistant_message&.tool_call?
+      end
+
+      # Generate summary of chat messages
+      # @rbs llm: LLM::OpenAI
+      # @rbs return: String
+      def summarize(llm:)
+        summary_prompt = ChatMessage.new(
+          role: :system,
+          content: <<~TEXT
+            Please summarize the following conversation in a concise manner, capturing the key topics, decisions, and context that would be helpful for continuing the conversation:
+          TEXT
+        )
+
+        response = llm.complete(messages: [summary_prompt, *all_values])
+        response.message.content
+      end
     end
   end
 end

data/lib/ruboty/ai_agent/commands/base.rb

@@ -6,32 +6,34 @@ module Ruboty
       # Base class for commands.
       # @abstract
       class Base
-        Matcher = Data.define(:pattern, :description, :name)
+        attr_reader :request #: Request
 
-        class << self
-          def matchers #: Array[Matcher]
-            @matchers ||= []
-          end
+        # @rbs request: Request
+        def initialize(request:)
+          @request = request
 
-          def on(pattern, name:, description:)
-            matchers << Matcher.new(pattern:, description:, name:)
-          end
+          super()
+        end
+
+        def message #: Ruboty::Message
+          request.message
+        end
+
+        def chat_thread #: Ruboty::AiAgent::ChatThread
+          request.chat_thread
         end
 
         # @rbs *args: untyped
-        # @rbs return: void
+        # @rbs return: untyped
         def call(*args)
           raise NotImplementedError
         end
 
         # @rbs commandline: String
         # @rbs return: boolish
+        # @abstract
         def match?(commandline)
-          matchers.any? { |matcher| /\A\s*#{matcher.pattern}/.match?(commandline) }
-        end
-
-        def matchers #: Array[Matcher]
-          self.class.matchers
+          raise NotImplementedError
         end
       end
     end

data/lib/ruboty/ai_agent/commands/builtin_base.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    module Commands
+      # Base class for commands.
+      # @abstract
+      class BuiltinBase < Base
+        Matcher = Data.define(:pattern, :description, :name)
+
+        class << self
+          def matchers #: Array[Matcher]
+            @matchers ||= []
+          end
+
+          def on(pattern, name:, description:)
+            matchers << Matcher.new(pattern:, description:, name:)
+          end
+        end
+
+        # @rbs *args: untyped
+        # @rbs return: void
+        def call(*args)
+          raise NotImplementedError
+        end
+
+        # @rbs commandline: String
+        # @rbs return: boolish
+        def match?(commandline)
+          matchers.any? { |matcher| /\A\s*#{matcher.pattern}/.match?(commandline) }
+        end
+
+        def matchers #: Array[Matcher]
+          self.class.matchers
+        end
+      end
+    end
+  end
+end

data/lib/ruboty/ai_agent/commands/clear.rb

@@ -4,22 +4,10 @@ module Ruboty
   module AiAgent
     module Commands
       # Clear histories of the chat thread.
-      class Clear < Base
+      class Clear < BuiltinBase
         on(%r{/clear}, name: 'clear', description: 'Clear the chat history.')
 
-        attr_reader :message #: Ruboty::Message
-        attr_reader :chat_thread #: Ruboty::AiAgent::ChatThread
-
-        # @rbs message: Ruboty::Message
-        # @rbs chat_thread: Ruboty::AiAgent::ChatThread
-        def initialize(message:, chat_thread:)
-          @message = message
-          @chat_thread = chat_thread
-
-          super()
-        end
-
-        def call #: void
+        def call(*) #: void
           chat_thread.clear
           message.reply('Cleared the chat history.')
         end