ruboty-ai_agent 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6ee903a1f588ede1c0b6e68bea298457b3ee2fbe826494f036f9c6482288475a
-  data.tar.gz: 9acf203be78f86b2eeaf1f6810606a4a5b0f0a05709aba6c52b7194a589ed533
+  metadata.gz: 4358837ce9305dcb0965a67c957b75e8320aad2988647a9e6ae751a98e7b7009
+  data.tar.gz: 3b1ad1c43bc443c5d86b1ea80bd8d9a45263056bc91ade861058e600493cb1b7
 SHA512:
-  metadata.gz: a59e8c688b63d055d59389a95b95a6a5637bac86d917e35a4bd90b7260bc2028b2defd00c6e7bd3a5b26d5cfdb22536de5e4a1428b4e597c29ae573898021922
-  data.tar.gz: 1daed254a4b194c16b3b9dda2180c59e21cb5597d7c965e20905c302523b1f3a9f2a2829528e7746370fa567de487389c5846b392167c7331d8e00422a45a73e
+  metadata.gz: fb0b30fad316d595d3e088d9262b46b32af67467c23b28494a7214f83093bb2a939953005c19b84e907eae81c4d1ba00a2dd84ca96e8cc85fdffe016d687e036
+  data.tar.gz: 586e425b1c2f806d663e861d18fa57579649b6d29135576a1677f825f637fc017d5cb910e6b1edc1bc2e1a7d5c8eb4e40d6c7795c170e69f418434b84750c0f8

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/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

@@ -17,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]
 

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
@@ -10,14 +12,14 @@ module Ruboty
         # @rbs override
         def call
           user.prompt_command_definitions.all_values.each do |definition|
-            command = Commands::PromptCommand.new(definition:, message:, chat_thread:)
+            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
 
-          builtin_commands = Commands.builtins(message:, chat_thread:)
+          builtin_commands = Commands.builtins(request:)
           builtin_commands.each do |command|
             return command.call if command.match?(body_param)
           end
@@ -29,6 +31,11 @@ module Ruboty
           message[:body]
         end
 
+        # @rbs %a{memorized}
+        def request #: Request
+          @request ||= Request.new(message:, chat_thread:)
+        end
+
         private
 
         # @rbs body: String
@@ -56,7 +63,10 @@ module Ruboty
           messages += chat_thread.messages.all_values
 
           llm = LLM::OpenAI.new
-          tools = McpClients.new(user.mcp_clients).available_tools
+          tools = [
+            *McpClients.new(user.mcp_clients).available_tools,
+            *ToolDefinitions.builtins(request:).map(&:to_tool)
+          ]
 
           agent = Agent.new(
             llm:,
@@ -72,11 +82,10 @@ module Ruboty
 
               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

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

@@ -7,7 +7,7 @@ module Ruboty
       class ListAiCommands < Base
         # @rbs override
         def call
-          builtin_commands = Commands.builtins(message:, chat_thread:)
+          builtin_commands = Commands.builtins(request: Request.new(message:, chat_thread:))
 
           builtin_list = builtin_commands.flat_map(&:matchers)
                                          .map { |matcher| "#{matcher.pattern.inspect} - #{matcher.description}" }

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/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/commands/base.rb

@@ -6,18 +6,23 @@ module Ruboty
       # Base class for commands.
       # @abstract
       class Base
-        attr_reader :message #: Ruboty::Message
-        attr_reader :chat_thread #: Ruboty::AiAgent::ChatThread
+        attr_reader :request #: Request
 
-        # @rbs message: Ruboty::Message
-        # @rbs chat_thread: Ruboty::AiAgent::ChatThread
-        def initialize(message:, chat_thread:)
-          @message = message
-          @chat_thread = chat_thread
+        # @rbs request: Request
+        def initialize(request:)
+          @request = request
 
           super()
         end
 
+        def message #: Ruboty::Message
+          request.message
+        end
+
+        def chat_thread #: Ruboty::AiAgent::ChatThread
+          request.chat_thread
+        end
+
         # @rbs *args: untyped
         # @rbs return: untyped
         def call(*args)

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

@@ -10,12 +10,11 @@ module Ruboty
         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:)
+        # @rbs request: Ruboty::AiAgent::Request
+        def initialize(definition:, request:)
           @definition = definition
 
-          super(message:, chat_thread:)
+          super(request:)
         end
 
         # @rbs commandline: String

data/lib/ruboty/ai_agent/commands.rb

@@ -11,24 +11,14 @@ module Ruboty
       autoload :Usage, 'ruboty/ai_agent/commands/usage'
       autoload :PromptCommand, 'ruboty/ai_agent/commands/prompt_command'
 
-      # @rbs message: Ruboty::Message
-      # @rbs chat_thread: ChatThread
+      # @rbs request: Request
       # @rbs return: Array[Commands::BuiltinBase]
-      def self.builtins(message:, chat_thread:)
+      def self.builtins(request:)
         [
-          Commands::Clear.new(
-            message:,
-            chat_thread:
-          ),
-          Commands::Compact.new(
-            message:,
-            chat_thread:
-          ),
-          Commands::Usage.new(
-            message:,
-            chat_thread:
-          )
-        ]
+          Commands::Clear,
+          Commands::Compact,
+          Commands::Usage
+        ].map { |cmd_class| cmd_class.new(request: request) }
       end
     end
   end

data/lib/ruboty/ai_agent/database/query_methods.rb

@@ -22,7 +22,7 @@ module Ruboty
             when Integer
               k
             when NilClass
-              nil
+              ''
             else
               raise ArgumentError, "Invalid key type: #{k.class}"
             end

data/lib/ruboty/ai_agent/llm/openai.rb

@@ -158,7 +158,7 @@ module Ruboty
                 tool_call #: OpenAI::Models::Chat::ChatCompletionMessageFunctionToolCall
                 .function.arguments
 
-              JSON.parse(arguments, { symbolize_names: true })
+              JSON.parse(arguments)
             end
 
           Response.new(

data/lib/ruboty/ai_agent/mcp_clients.rb

@@ -16,8 +16,9 @@ module Ruboty
         clients.flat_map do |client|
           tool_defs = client.list_tools
           tool_defs.map do |tool_def|
+            tool_name = "mcp_#{client.mcp_name}__#{tool_def['name']}"
             Tool.new(
-              name: tool_def['name'],
+              name: tool_name,
               title: tool_def['title'] || '',
               description: tool_def['description'] || '',
               input_schema: tool_def['inputSchema']
@@ -31,20 +32,6 @@ module Ruboty
         end
       end
 
-      # @rbs function_name: String
-      # @rbs arguments: Hash[String, untyped]
-      # @rbs return: untyped
-      def execute_tool(function_name, arguments)
-        clients.each do |mcp_client|
-          tools = mcp_client.list_tools
-          return mcp_client.call_tool(function_name, arguments) if tools.any? { |t| t['name'] == function_name }
-        end
-        nil
-      rescue HttpMcpClient::Error => e
-        warn "Failed to execute tool '#{function_name}': #{e.message}"
-        nil
-      end
-
       # @rbs return: bool
       def any?
         @clients.any?

data/lib/ruboty/ai_agent/request.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    Request = Data.define(
+      :message, #: Ruboty::Message
+      :chat_thread #: Ruboty::AiAgent::ChatThread
+    )
+
+    # Request for chat action from user.
+    class Request
+      def message_body #: String
+        message[:body]
+      end
+    end
+  end
+end

data/lib/ruboty/ai_agent/tool.rb

@@ -6,21 +6,29 @@ module Ruboty
     class Tool
       attr_reader :name, :title, :description #: String
       attr_reader :input_schema #: Hash[untyped, untyped]?
-      attr_reader :on_call #: (^(Hash[String, untyped]) -> String )?
+      attr_reader :silent #: boolish
+      attr_reader :on_call #: (^(Hash[String, untyped]) -> String? )?
 
       # @rbs name: String
       # @rbs title: String
       # @rbs description: String
       # @rbs input_schema: Hash[untyped, untyped]?
-      # @rbs &on_call: ? (Hash[String, untyped]) -> String
-      def initialize(name:, title:, description:, input_schema:, &on_call) #: void
+      # @rbs ?silent: boolish?
+      # @rbs &on_call: ? (Hash[String, untyped]) -> String?
+      def initialize(name:, title:, description:, input_schema:, silent: false, &on_call) #: void
         @name = name
         @title = title
         @description = description
         @input_schema = input_schema
+        @silent = silent
         @on_call = on_call
       end
 
+      # Returns true if the tool should be called silently (without notifying the user).
+      def silent? #: boolish
+        silent
+      end
+
       def call(params) #: String?
         on_call&.call(params)
       end

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

@@ -0,0 +1,84 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    module ToolDefinitions
+      # Base class for tool definitions
+      # @abstract
+      class Base
+        # @rbs!
+        #   type input_schema = Hash[String | Symbol, untyped]
+
+        # @rbs!
+        #   def self.tool_name: () -> String?
+        #   def self.tool_name=: (String) -> String
+        #   def self.tool_title: () -> String?
+        #   def self.tool_title=: (String) -> String
+        #   def self.tool_description: () -> String?
+        #   def self.tool_description=: (String) -> String
+        #   def self.tool_input_schema: () -> input_schema?
+        #   def self.tool_input_schema=: (input_schema) -> input_schema
+
+        class << self
+          # @rbs skip
+          attr_accessor :tool_name
+
+          # @rbs skip
+          attr_accessor :tool_title
+
+          # @rbs skip
+          attr_accessor :tool_description
+
+          # @rbs skip
+          attr_accessor :tool_input_schema
+        end
+
+        def tool_name #: String
+          self.class.tool_name || (raise NotImplementedError, "Subclasses must define 'tool_name'")
+        end
+
+        def tool_title #: String
+          self.class.tool_title || ''
+        end
+
+        def tool_description #: String
+          self.class.tool_description || ''
+        end
+
+        def tool_input_schema #: input_schema
+          self.class.tool_input_schema || {} #: input_schema
+        end
+
+        # Return true if you want the tool not to produce call logs in the chat.
+        def silent? #: boolish
+          false
+        end
+
+        attr_reader :request #: Request
+
+        # @rbs request: Request
+        def initialize(request:)
+          @request = request
+        end
+
+        # @abstract
+        # @rbs arguments: Hash[String, untyped]
+        # @rbs return: String?
+        def call(arguments)
+          raise NotImplementedError, "Subclasses must implement the 'call' method"
+        end
+
+        def to_tool #: Tool
+          Tool.new(
+            name: tool_name,
+            title: tool_title,
+            description: tool_description,
+            input_schema: tool_input_schema,
+            silent: silent?,
+            &method(:call) #: ^ (Hash[String, untyped]) -> String?
+          )
+        end
+      end
+    end
+  end
+end

data/lib/ruboty/ai_agent/tool_definitions/think.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    module ToolDefinitions
+      # Tool
+      class Think < Base
+        self.tool_name = 'think'
+        self.tool_title = 'Think'
+
+        self.tool_description = <<~TEXT
+          Use this tool to think abount something. It will not obtain new information or make any changes, but just log the thought.
+          Use it when completx reasoing or brainstorming is needed.
+        TEXT
+
+        self.tool_input_schema = {
+          type: 'object',
+          properties: {
+            thought: { type: 'string', description: 'Your thought.' }
+          },
+          required: ['thought']
+        }
+
+        # @rbs arguments: Hash[String, untyped]
+        # @rbs return: String?
+        def call(arguments)
+          thought = arguments['thought']
+
+          request.message.reply("Thought:\n#{thought}")
+
+          thought
+        end
+
+        # @rbs override
+        def silent?
+          true
+        end
+      end
+    end
+  end
+end

data/lib/ruboty/ai_agent/tool_definitions.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    # Tool Definitions for AI Agent
+    module ToolDefinitions
+      autoload :Base, 'ruboty/ai_agent/tool_definitions/base'
+      autoload :Think, 'ruboty/ai_agent/tool_definitions/think'
+
+      # @rbs request: Request
+      # @rbs return: Array[Base]
+      def self.builtins(request:)
+        [
+          Think
+        ].map { |tool_def| tool_def.new(request:) }
+      end
+    end
+  end
+end

data/lib/ruboty/ai_agent/user_mcp_client.rb

@@ -68,14 +68,15 @@ module Ruboty
         mcp_client.cleanup_session
       end
 
-      private
-
       # @rbs return: McpConfiguration
       def configuration
         user.mcp_configurations.all_values.find { |config| config.name == mcp_name } ||
           raise("MCP configuration not found: #{mcp_name}")
       end
 
+      private
+
+      # @rbs %a{memorized}
       # @rbs return: HttpMcpClient
       def mcp_client
         @mcp_client ||= case configuration.transport

data/lib/ruboty/ai_agent/version.rb

@@ -2,6 +2,6 @@
 
 module Ruboty
   module AiAgent
-    VERSION = '0.2.0'
+    VERSION = '0.3.0'
   end
 end

data/lib/ruboty/ai_agent.rb

@@ -28,8 +28,10 @@ module Ruboty
     autoload :PromptCommandDefinition, 'ruboty/ai_agent/prompt_command_definition'
     autoload :Recordable, 'ruboty/ai_agent/recordable'
     autoload :RecordSet, 'ruboty/ai_agent/record_set'
+    autoload :Request, 'ruboty/ai_agent/request'
     autoload :TokenUsage, 'ruboty/ai_agent/token_usage'
     autoload :Tool, 'ruboty/ai_agent/tool'
+    autoload :ToolDefinitions, 'ruboty/ai_agent/tool_definitions'
     autoload :User, 'ruboty/ai_agent/user'
     autoload :UserAiMemories, 'ruboty/ai_agent/user_ai_memories'
     autoload :UserAssociations, 'ruboty/ai_agent/user_associations'

data/lib/ruboty/handlers/ai_agent.rb

@@ -18,7 +18,7 @@ module Ruboty
 
       on(/add mcp (?<name>\S+)\s+(?<config>.+)\z/, name: 'add_mcp', description: 'Add a new MCP server')
       on(/remove mcp (?<name>\S+)/, name: 'remove_mcp', description: 'Remove the specified MCP server')
-      on(/list mcps?/, name: 'list_mcp', description: 'List configured MCP servers')
+      on(/list mcps?(?<with_headers>\s+with\s+headers)?/, name: 'list_mcp', description: 'List configured MCP servers')
 
       on(/set (?:(?<scope>user|global) )?system prompt "(?<prompt>.+?)"/, name: 'set_system_prompt', description: 'Set system prompt')
       on(/show system prompt/, name: 'show_system_prompt', description: 'Show system prompt')

data/sig/generated/ruboty/ai_agent/actions/base.rbs

@@ -28,6 +28,10 @@ module Ruboty
         # @rbs %a{memorized}
         %a{memorized}
         def chat_thread: () -> Ruboty::AiAgent::ChatThread
+
+        private
+
+        def thread_id: () -> String
       end
     end
   end

data/sig/generated/ruboty/ai_agent/actions/chat.rbs

@@ -12,6 +12,10 @@ module Ruboty
 
         def body_param: () -> String
 
+        # @rbs %a{memorized}
+        %a{memorized}
+        def request: () -> Request
+
         private
 
         # @rbs body: String

data/sig/generated/ruboty/ai_agent/actions/list_mcp.rbs

@@ -6,6 +6,17 @@ module Ruboty
       # ListMcp action for Ruboty::AiAgent
       class ListMcp < Base
         def call: () -> untyped
+
+        private
+
+        # @rbs client: UserMcpClient
+        # @rbs show_headers: bool
+        # @rbs return: String
+        def format_mcp_client: (UserMcpClient client, ?show_headers: bool) -> String
+
+        # @rbs client: UserMcpClient
+        # @rbs return: String
+        def format_tools: (UserMcpClient client) -> String
       end
     end
   end

data/sig/generated/ruboty/ai_agent/agent.rbs

@@ -21,7 +21,7 @@ module Ruboty
 
       def on_tool_call: (tool: untyped, tool_arguments: untyped) ?{ (?) -> untyped } -> untyped
 
-      def on_tool_response: (tool_response: untyped, message: untyped) ?{ (?) -> untyped } -> untyped
+      def on_tool_response: (tool: untyped, tool_response: untyped, message: untyped) ?{ (?) -> untyped } -> untyped
 
       def on_response: (untyped response) ?{ (?) -> untyped } -> untyped
     end

data/sig/generated/ruboty/ai_agent/commands/base.rbs

@@ -6,13 +6,14 @@ module Ruboty
       # Base class for commands.
       # @abstract
       class Base
-        attr_reader message: Ruboty::Message
+        attr_reader request: Request
 
-        attr_reader chat_thread: Ruboty::AiAgent::ChatThread
+        # @rbs request: Request
+        def initialize: (request: Request) -> untyped
 
-        # @rbs message: Ruboty::Message
-        # @rbs chat_thread: Ruboty::AiAgent::ChatThread
-        def initialize: (message: Ruboty::Message, chat_thread: Ruboty::AiAgent::ChatThread) -> untyped
+        def message: () -> Ruboty::Message
+
+        def chat_thread: () -> Ruboty::AiAgent::ChatThread
 
         # @rbs *args: untyped
         # @rbs return: untyped

data/sig/generated/ruboty/ai_agent/commands/prompt_command.rbs

@@ -8,9 +8,8 @@ module Ruboty
         attr_reader definition: Ruboty::AiAgent::PromptCommandDefinition
 
         # @rbs definition: Ruboty::AiAgent::PromptCommandDefinition
-        # @rbs message: Ruboty::Message
-        # @rbs chat_thread: Ruboty::AiAgent::ChatThread
-        def initialize: (definition: Ruboty::AiAgent::PromptCommandDefinition, message: Ruboty::Message, chat_thread: Ruboty::AiAgent::ChatThread) -> untyped
+        # @rbs request: Ruboty::AiAgent::Request
+        def initialize: (definition: Ruboty::AiAgent::PromptCommandDefinition, request: Ruboty::AiAgent::Request) -> untyped
 
         # @rbs commandline: String
         # @rbs return: boolish

data/sig/generated/ruboty/ai_agent/commands.rbs

@@ -4,10 +4,9 @@ module Ruboty
   module AiAgent
     # Interaction commands (a.k.a. Slash commands, Prompts, etc)
     module Commands
-      # @rbs message: Ruboty::Message
-      # @rbs chat_thread: ChatThread
+      # @rbs request: Request
       # @rbs return: Array[Commands::BuiltinBase]
-      def self.builtins: (message: Ruboty::Message, chat_thread: ChatThread) -> Array[Commands::BuiltinBase]
+      def self.builtins: (request: Request) -> Array[Commands::BuiltinBase]
     end
   end
 end

data/sig/generated/ruboty/ai_agent/mcp_clients.rbs

@@ -12,11 +12,6 @@ module Ruboty
       # @rbs return: Array[Tool]
       def available_tools: () -> Array[Tool]
 
-      # @rbs function_name: String
-      # @rbs arguments: Hash[String, untyped]
-      # @rbs return: untyped
-      def execute_tool: (String function_name, Hash[String, untyped] arguments) -> untyped
-
       # @rbs return: bool
       def any?: () -> bool
     end

data/sig/generated/ruboty/ai_agent/request.rbs

@@ -0,0 +1,23 @@
+# Generated from lib/ruboty/ai_agent/request.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    class Request < Data
+      attr_reader message(): Ruboty::Message
+
+      attr_reader chat_thread(): Ruboty::AiAgent::ChatThread
+
+      def self.new: (Ruboty::Message message, Ruboty::AiAgent::ChatThread chat_thread) -> instance
+                  | (message: Ruboty::Message, chat_thread: Ruboty::AiAgent::ChatThread) -> instance
+
+      def self.members: () -> [ :message, :chat_thread ]
+
+      def members: () -> [ :message, :chat_thread ]
+    end
+
+    # Request for chat action from user.
+    class Request
+      def message_body: () -> String
+    end
+  end
+end

data/sig/generated/ruboty/ai_agent/tool.rbs

@@ -12,14 +12,20 @@ module Ruboty
 
       attr_reader input_schema: Hash[untyped, untyped]?
 
-      attr_reader on_call: (^(Hash[String, untyped]) -> String)?
+      attr_reader silent: boolish
+
+      attr_reader on_call: (^(Hash[String, untyped]) -> String?)?
 
       # @rbs name: String
       # @rbs title: String
       # @rbs description: String
       # @rbs input_schema: Hash[untyped, untyped]?
-      # @rbs &on_call: ? (Hash[String, untyped]) -> String
-      def initialize: (name: String, title: String, description: String, input_schema: Hash[untyped, untyped]?) ?{ (Hash[String, untyped]) -> String } -> void
+      # @rbs ?silent: boolish?
+      # @rbs &on_call: ? (Hash[String, untyped]) -> String?
+      def initialize: (name: String, title: String, description: String, input_schema: Hash[untyped, untyped]?, ?silent: untyped) ?{ (Hash[String, untyped]) -> String? } -> void
+
+      # Returns true if the tool should be called silently (without notifying the user).
+      def silent?: () -> boolish
 
       def call: (untyped params) -> String?
     end

data/sig/generated/ruboty/ai_agent/tool_definitions/base.rbs

@@ -0,0 +1,52 @@
+# Generated from lib/ruboty/ai_agent/tool_definitions/base.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module ToolDefinitions
+      # Base class for tool definitions
+      # @abstract
+      class Base
+        type input_schema = Hash[String | Symbol, untyped]
+
+        def self.tool_name: () -> String?
+
+        def self.tool_name=: (String) -> String
+
+        def self.tool_title: () -> String?
+
+        def self.tool_title=: (String) -> String
+
+        def self.tool_description: () -> String?
+
+        def self.tool_description=: (String) -> String
+
+        def self.tool_input_schema: () -> input_schema?
+
+        def self.tool_input_schema=: (input_schema) -> input_schema
+
+        def tool_name: () -> String
+
+        def tool_title: () -> String
+
+        def tool_description: () -> String
+
+        def tool_input_schema: () -> input_schema
+
+        # Return true if you want the tool not to produce call logs in the chat.
+        def silent?: () -> boolish
+
+        attr_reader request: Request
+
+        # @rbs request: Request
+        def initialize: (request: Request) -> untyped
+
+        # @abstract
+        # @rbs arguments: Hash[String, untyped]
+        # @rbs return: String?
+        def call: (Hash[String, untyped] arguments) -> String?
+
+        def to_tool: () -> Tool
+      end
+    end
+  end
+end

data/sig/generated/ruboty/ai_agent/tool_definitions/think.rbs

@@ -0,0 +1,17 @@
+# Generated from lib/ruboty/ai_agent/tool_definitions/think.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module ToolDefinitions
+      # Tool
+      class Think < Base
+        # @rbs arguments: Hash[String, untyped]
+        # @rbs return: String?
+        def call: (Hash[String, untyped] arguments) -> String?
+
+        # @rbs override
+        def silent?: ...
+      end
+    end
+  end
+end

data/sig/generated/ruboty/ai_agent/tool_definitions.rbs

@@ -0,0 +1,12 @@
+# Generated from lib/ruboty/ai_agent/tool_definitions.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Tool Definitions for AI Agent
+    module ToolDefinitions
+      # @rbs request: Request
+      # @rbs return: Array[Base]
+      def self.builtins: (request: Request) -> Array[Base]
+    end
+  end
+end

data/sig/generated/ruboty/ai_agent/user_mcp_client.rbs

@@ -46,12 +46,14 @@ module Ruboty
       # @rbs return: untyped
       def cleanup_session: () -> untyped
 
-      private
-
       # @rbs return: McpConfiguration
       def configuration: () -> McpConfiguration
 
+      private
+
+      # @rbs %a{memorized}
       # @rbs return: HttpMcpClient
+      %a{memorized}
       def mcp_client: () -> HttpMcpClient
     end
   end

data/sig/generated-by-scripts/memorized_ivars.rbs

@@ -13,6 +13,16 @@ module Ruboty
   end
 end
 
+module Ruboty
+  module AiAgent
+    module Actions
+      class Chat
+        @request: Request
+      end
+    end
+  end
+end
+
 module Ruboty
   module AiAgent
     class ChatThread

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruboty-ai_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Tomoya Chiba
@@ -113,8 +113,12 @@ files:
 - lib/ruboty/ai_agent/prompt_command_definition.rb
 - lib/ruboty/ai_agent/record_set.rb
 - lib/ruboty/ai_agent/recordable.rb
+- lib/ruboty/ai_agent/request.rb
 - lib/ruboty/ai_agent/token_usage.rb
 - lib/ruboty/ai_agent/tool.rb
+- lib/ruboty/ai_agent/tool_definitions.rb
+- lib/ruboty/ai_agent/tool_definitions/base.rb
+- lib/ruboty/ai_agent/tool_definitions/think.rb
 - lib/ruboty/ai_agent/user.rb
 - lib/ruboty/ai_agent/user_ai_memories.rb
 - lib/ruboty/ai_agent/user_associations.rb
@@ -177,8 +181,12 @@ files:
 - sig/generated/ruboty/ai_agent/prompt_command_definition.rbs
 - sig/generated/ruboty/ai_agent/record_set.rbs
 - sig/generated/ruboty/ai_agent/recordable.rbs
+- sig/generated/ruboty/ai_agent/request.rbs
 - sig/generated/ruboty/ai_agent/token_usage.rbs
 - sig/generated/ruboty/ai_agent/tool.rbs
+- sig/generated/ruboty/ai_agent/tool_definitions.rbs
+- sig/generated/ruboty/ai_agent/tool_definitions/base.rbs
+- sig/generated/ruboty/ai_agent/tool_definitions/think.rbs
 - sig/generated/ruboty/ai_agent/user.rbs
 - sig/generated/ruboty/ai_agent/user_ai_memories.rbs
 - sig/generated/ruboty/ai_agent/user_associations.rbs