ruboty-ai_agent 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1782e855675403948a771dd5abe6fdba6a754cfc6e15bebad40f24bf3c7569f6
-  data.tar.gz: 2142af0fb45049c4e1b2397def8009248db8860e43a9fa2ae21f938a31e3af82
+  metadata.gz: 6ee903a1f588ede1c0b6e68bea298457b3ee2fbe826494f036f9c6482288475a
+  data.tar.gz: 9acf203be78f86b2eeaf1f6810606a4a5b0f0a05709aba6c52b7194a589ed533
 SHA512:
-  metadata.gz: 1a1b7fc8e8e41150055e87d2227e9b0ae8570c69482cf72249af38abef0978058419eed73f778a0f4d9c00343a55ce96be0cd62febdb4a20cd8b7976a6852415
-  data.tar.gz: 73aff06e4a6d7fb3601b9c95393b9c1496fd61eddaaabf0fe742e7701faad638a36c3919462e65847dab47d0450579c8047ba492beb4d51c81181b5b66307aef
+  metadata.gz: a59e8c688b63d055d59389a95b95a6a5637bac86d917e35a4bd90b7260bc2028b2defd00c6e7bd3a5b26d5cfdb22536de5e4a1428b4e597c29ae573898021922
+  data.tar.gz: 1daed254a4b194c16b3b9dda2180c59e21cb5597d7c965e20905c302523b1f3a9f2a2829528e7746370fa567de487389c5846b392167c7331d8e00422a45a73e

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/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/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)
 
@@ -33,6 +34,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/chat.rb

@@ -9,28 +9,58 @@ 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:, message:, chat_thread:)
+            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(message:, chat_thread:)
+          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
+
+        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
+
           agent = Agent.new(
             llm:,
-            messages: chat_thread.messages.all_values,
+            messages:,
             tools:
           )
 
@@ -39,6 +69,8 @@ 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)
@@ -54,10 +86,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(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/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/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,29 @@ module Ruboty
       # Base class for commands.
       # @abstract
       class Base
-        Matcher = Data.define(:pattern, :description, :name)
+        attr_reader :message #: Ruboty::Message
+        attr_reader :chat_thread #: Ruboty::AiAgent::ChatThread
 
-        class << self
-          def matchers #: Array[Matcher]
-            @matchers ||= []
-          end
+        # @rbs message: Ruboty::Message
+        # @rbs chat_thread: Ruboty::AiAgent::ChatThread
+        def initialize(message:, chat_thread:)
+          @message = message
+          @chat_thread = chat_thread
 
-          def on(pattern, name:, description:)
-            matchers << Matcher.new(pattern:, description:, name:)
-          end
+          super()
         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

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

@@ -4,30 +4,16 @@ module Ruboty
   module AiAgent
     module Commands
       # Compact chat history by summarizing it
-      class Compact < Base
+      class Compact < BuiltinBase
         on(%r{/compact}, name: 'compact', description: 'Compact the chat history by summarizing it.')
 
-        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
-          messages = chat_thread.messages.all_values
-
-          if messages.empty?
+        def call(*) #: void
+          if chat_thread.messages.empty?
             message.reply('No chat history to compact.')
             return
           end
 
-          summary = generate_summary(messages)
+          summary = chat_thread.messages.summarize(llm: LLM::OpenAI.new)
 
           chat_thread.clear
           chat_thread.messages.add(
@@ -45,35 +31,6 @@ module Ruboty
             message.reply("エラーが発生しました: #{e.message}")
           end
         end
-
-        private
-
-        # @rbs messages: Array[ChatMessage]
-        # @rbs return: String
-        def generate_summary(messages)
-          llm = LLM::OpenAI.new(
-            client: OpenAI::Client.new(
-              api_key: ENV.fetch('OPENAI_API_KEY', nil)
-            ),
-            model: ENV.fetch('OPENAI_MODEL', 'gpt-5-nano')
-          )
-
-          summary_prompt = ChatMessage.new(
-            role: :user,
-            content: "Please summarize the following conversation in a concise manner, capturing the key topics, decisions, and context that would be helpful for continuing the conversation:\n\n#{format_messages_for_summary(messages)}"
-          )
-
-          response = llm.complete(messages: [summary_prompt])
-          response.message.content
-        end
-
-        # @rbs messages: Array[ChatMessage]
-        # @rbs return: String
-        def format_messages_for_summary(messages)
-          messages.map do |msg|
-            "#{msg.role}: #{msg.content}"
-          end.join("\n")
-        end
       end
     end
   end

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

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require 'shellwords'
+
+module Ruboty
+  module AiAgent
+    module Commands
+      # Compact chat history by summarizing it
+      class PromptCommand < Base
+        attr_reader :definition #: Ruboty::AiAgent::PromptCommandDefinition
+
+        # @rbs definition: Ruboty::AiAgent::PromptCommandDefinition
+        # @rbs message: Ruboty::Message
+        # @rbs chat_thread: Ruboty::AiAgent::ChatThread
+        def initialize(definition:, message:, chat_thread:)
+          @definition = definition
+
+          super(message:, chat_thread:)
+        end
+
+        # @rbs commandline: String
+        # @rbs return: boolish
+        def match?(commandline)
+          pattern.match?(commandline)
+        end
+
+        def pattern #: Regexp
+          %r{\A\s*/#{definition.name}(?:\s+(?<args>.+))?}
+        end
+
+        # @rbs commandline: String
+        # @rbs return: String
+        def call(commandline:)
+          result = definition.prompt.dup
+
+          match = pattern.match(commandline) || (raise 'Unreachable')
+          args = begin
+            Shellwords.split(match[:args] || '')
+          rescue ArgumentError
+            # If parsing fails, treat the entire string as a single argument
+            [match[:args]].compact
+          end #: Array[String]
+
+          found = {} #: Hash[Integer, boolish]
+          args.each_with_index do |arg, index|
+            placeholder = "$#{index + 1}"
+            result = result.gsub(placeholder) do
+              found[index] = true
+              arg
+            end
+          end
+
+          rest_args = args.each_with_index.reject { |_, index| found[index] }.map(&:first)
+          result += "\n\n#{rest_args.join(' ')}" if rest_args.any?
+
+          result
+        end
+      end
+    end
+  end
+end

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

@@ -4,22 +4,10 @@ module Ruboty
   module AiAgent
     module Commands
       # Show token usage information for the latest AI response
-      class Usage < Base
+      class Usage < BuiltinBase
         on(%r{/usage}, name: 'show_usage', description: 'Show token usage information for the latest AI response')
 
-        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
           latest_message = chat_thread.messages.all_values.find(&:token_usage)
 
           token_usage = latest_message&.token_usage

data/lib/ruboty/ai_agent/commands.rb

@@ -5,13 +5,15 @@ module Ruboty
     # Interaction commands (a.k.a. Slash commands, Prompts, etc)
     module Commands
       autoload :Base, 'ruboty/ai_agent/commands/base'
+      autoload :BuiltinBase, 'ruboty/ai_agent/commands/builtin_base'
       autoload :Clear, 'ruboty/ai_agent/commands/clear'
       autoload :Compact, 'ruboty/ai_agent/commands/compact'
       autoload :Usage, 'ruboty/ai_agent/commands/usage'
+      autoload :PromptCommand, 'ruboty/ai_agent/commands/prompt_command'
 
       # @rbs message: Ruboty::Message
       # @rbs chat_thread: ChatThread
-      # @rbs return: Array[Commands::Base]
+      # @rbs return: Array[Commands::BuiltinBase]
       def self.builtins(message:, chat_thread:)
         [
           Commands::Clear.new(