ruboty-ai_agent 0.1.0 → 0.3.0

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

@@ -18,16 +18,19 @@ module Ruboty
 
       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: Symbol, content: String, ?tool_call_id: untyped, ?tool_name: untyped, ?tool_arguments: untyped, ?token_usage: untyped) -> untyped
+      def initialize: (role: Symbol | String, content: String, ?tool_call_id: untyped, ?tool_name: untyped, ?tool_arguments: untyped, ?token_usage: untyped) -> untyped
 
       def to_h: () -> Hash[Symbol, untyped]
 
+      # @rbs return: bool
+      def tool_call?: () -> bool
+
       def self.from_llm_response: (tool: untyped, tool_call_id: untyped, tool_arguments: untyped, tool_response: untyped) -> ChatMessage
     end
   end

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

@@ -8,6 +8,20 @@ module Ruboty
       def add: (ChatMessage message) -> void
 
       alias << add
+
+      # Check if any message's token usage exceeds auto compact threshold
+      # @rbs return: bool
+      def over_auto_compact_threshold?: () -> bool
+
+      # Compact chat messages by summarizing them
+      # @rbs llm: LLM::OpenAI
+      # @rbs return: void
+      def compact: (llm: LLM::OpenAI) -> void
+
+      # Generate summary of chat messages
+      # @rbs llm: LLM::OpenAI
+      # @rbs return: String
+      def summarize: (llm: LLM::OpenAI) -> String
     end
   end
 end

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

@@ -6,34 +6,23 @@ module Ruboty
       # Base class for commands.
       # @abstract
       class Base
-        class Matcher < Data
-          attr_reader pattern(): untyped
+        attr_reader request: Request
 
-          attr_reader description(): untyped
+        # @rbs request: Request
+        def initialize: (request: Request) -> untyped
 
-          attr_reader name(): untyped
+        def message: () -> Ruboty::Message
 
-          def self.new: (untyped pattern, untyped description, untyped name) -> instance
-                      | (pattern: untyped, description: untyped, name: untyped) -> instance
-
-          def self.members: () -> [ :pattern, :description, :name ]
-
-          def members: () -> [ :pattern, :description, :name ]
-        end
-
-        def self.matchers: () -> Array[Matcher]
-
-        def self.on: (untyped pattern, name: untyped, description: untyped) -> untyped
+        def chat_thread: () -> Ruboty::AiAgent::ChatThread
 
         # @rbs *args: untyped
-        # @rbs return: void
-        def call: (*untyped args) -> void
+        # @rbs return: untyped
+        def call: (*untyped args) -> untyped
 
         # @rbs commandline: String
         # @rbs return: boolish
+        # @abstract
         def match?: (String commandline) -> boolish
-
-        def matchers: () -> Array[Matcher]
       end
     end
   end

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

@@ -0,0 +1,40 @@
+# Generated from lib/ruboty/ai_agent/commands/builtin_base.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module Commands
+      # Base class for commands.
+      # @abstract
+      class BuiltinBase < Base
+        class Matcher < Data
+          attr_reader pattern(): untyped
+
+          attr_reader description(): untyped
+
+          attr_reader name(): untyped
+
+          def self.new: (untyped pattern, untyped description, untyped name) -> instance
+                      | (pattern: untyped, description: untyped, name: untyped) -> instance
+
+          def self.members: () -> [ :pattern, :description, :name ]
+
+          def members: () -> [ :pattern, :description, :name ]
+        end
+
+        def self.matchers: () -> Array[Matcher]
+
+        def self.on: (untyped pattern, name: untyped, description: untyped) -> untyped
+
+        # @rbs *args: untyped
+        # @rbs return: void
+        def call: (*untyped args) -> void
+
+        # @rbs commandline: String
+        # @rbs return: boolish
+        def match?: (String commandline) -> boolish
+
+        def matchers: () -> Array[Matcher]
+      end
+    end
+  end
+end

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

@@ -4,16 +4,8 @@ module Ruboty
   module AiAgent
     module Commands
       # Clear histories of the chat thread.
-      class Clear < Base
-        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: Ruboty::Message, chat_thread: Ruboty::AiAgent::ChatThread) -> untyped
-
-        def call: () -> void
+      class Clear < BuiltinBase
+        def call: (*untyped) -> void
       end
     end
   end

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

@@ -4,26 +4,8 @@ module Ruboty
   module AiAgent
     module Commands
       # Compact chat history by summarizing it
-      class Compact < Base
-        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: Ruboty::Message, chat_thread: Ruboty::AiAgent::ChatThread) -> untyped
-
-        def call: () -> void
-
-        private
-
-        # @rbs messages: Array[ChatMessage]
-        # @rbs return: String
-        def generate_summary: (Array[ChatMessage] messages) -> String
-
-        # @rbs messages: Array[ChatMessage]
-        # @rbs return: String
-        def format_messages_for_summary: (Array[ChatMessage] messages) -> String
+      class Compact < BuiltinBase
+        def call: (*untyped) -> void
       end
     end
   end

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

@@ -0,0 +1,26 @@
+# Generated from lib/ruboty/ai_agent/commands/prompt_command.rb with RBS::Inline
+
+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 request: Ruboty::AiAgent::Request
+        def initialize: (definition: Ruboty::AiAgent::PromptCommandDefinition, request: Ruboty::AiAgent::Request) -> untyped
+
+        # @rbs commandline: String
+        # @rbs return: boolish
+        def match?: (String commandline) -> boolish
+
+        def pattern: () -> Regexp
+
+        # @rbs commandline: String
+        # @rbs return: String
+        def call: (commandline: String) -> String
+      end
+    end
+  end
+end

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

@@ -4,16 +4,8 @@ module Ruboty
   module AiAgent
     module Commands
       # Show token usage information for the latest AI response
-      class Usage < Base
-        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: Ruboty::Message, chat_thread: Ruboty::AiAgent::ChatThread) -> untyped
-
-        def call: () -> void
+      class Usage < BuiltinBase
+        def call: (*untyped) -> void
 
         private
 

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 return: Array[Commands::Base]
-      def self.builtins: (message: Ruboty::Message, chat_thread: ChatThread) -> Array[Commands::Base]
+      # @rbs request: Request
+      # @rbs return: Array[Commands::BuiltinBase]
+      def self.builtins: (request: Request) -> Array[Commands::BuiltinBase]
     end
   end
 end

data/sig/generated/ruboty/ai_agent/database/query_methods.rbs

@@ -7,32 +7,38 @@ module Ruboty
         def data: () -> Hash[keynable, untyped]
       end
 
+      type query_key = keynable | Symbol | nil
+
       # @rbs module-self _WithData
       module QueryMethods : _WithData
-        # @rbs *keys: keynable
+        # @rbs keys: Array[query_key]
+        # @rbs return: Array[keynable]
+        def self.keys_to_keynable: (Array[query_key] keys) -> Array[keynable]
+
+        # @rbs *keys: query_key
         # @rbs return: untyped
-        def fetch: (*keynable keys) -> untyped
+        def fetch: (*query_key keys) -> untyped
 
-        # @rbs *keys: keynable
+        # @rbs *keys: query_key
         # @rbs return: Integer
-        def len: (*keynable keys) -> Integer
+        def len: (*query_key keys) -> Integer
 
-        # @rbs *keys: keynable
+        # @rbs *keys: query_key
         # @rbs return: void
-        def delete: (*keynable keys) -> void
+        def delete: (*query_key keys) -> void
 
-        # @rbs *keys: keynable
+        # @rbs *keys: query_key
         # @rbs return: Array[keynable]
-        def keys: (*keynable keys) -> Array[keynable]
+        def keys: (*query_key keys) -> Array[keynable]
 
-        # @rbs *keys: keynable
+        # @rbs *keys: query_key
         # @rbs return: boolish
-        def key?: (*keynable keys) -> boolish
+        def key?: (*query_key keys) -> boolish
 
-        # @rbs at: Array[keynable]
+        # @rbs at: Array[query_key]
         # @rbs value: untyped
         # @rbs return: void
-        def store: (untyped value, at: Array[keynable]) -> void
+        def store: (untyped value, at: Array[query_key]) -> void
       end
     end
   end

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

@@ -4,7 +4,7 @@ module Ruboty
   module AiAgent
     # Memorize and retrieve information using Ruboty's brain.
     class Database
-      type keynable = Symbol | String | Integer
+      type keynable = String | Integer
 
       include QueryMethods
 
@@ -21,6 +21,8 @@ module Ruboty
 
       def chat_thread: (untyped id) -> ChatThread
 
+      # @rbs %a{memorized}
+      %a{memorized}
       def global_settings: () -> GlobalSettings
     end
   end

data/sig/generated/ruboty/ai_agent/llm/openai.rbs

@@ -9,9 +9,9 @@ module Ruboty
 
         attr_reader model: String
 
-        # @rbs client: OpenAI::Client
-        # @rbs model: String
-        def initialize: (client: OpenAI::Client, model: String) -> untyped
+        # @rbs ?client: OpenAI::Client
+        # @rbs ?model: String
+        def initialize: (?client: untyped, ?model: untyped) -> untyped
 
         # @rbs %a{memorized}
         %a{memorized}

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/mcp_configuration.rbs

@@ -4,6 +4,8 @@ module Ruboty
   module AiAgent
     type transports = :http | :websocket
 
+    type transports_str = "http" | "websocket"
+
     class McpConfiguration < Data
       attr_reader name(): String
 
@@ -26,10 +28,10 @@ module Ruboty
       include Recordable
 
       # @rbs name: String
-      # @rbs transport: transports
+      # @rbs transport: transports | transports_str
       # @rbs url: String
       # @rbs headers: Hash[String, String]?
-      def initialize: (name: String, transport: transports, url: String, ?headers: Hash[String, String]?) -> untyped
+      def initialize: (name: String, transport: transports | transports_str, url: String, ?headers: Hash[String, String]?) -> untyped
     end
   end
 end

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

@@ -0,0 +1,23 @@
+# Generated from lib/ruboty/ai_agent/prompt_command_definition.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    class PromptCommandDefinition < Data
+      attr_reader name(): String
+
+      attr_reader prompt(): String
+
+      def self.new: (String name, String prompt) -> instance
+                  | (name: String, prompt: String) -> instance
+
+      def self.members: () -> [ :name, :prompt ]
+
+      def members: () -> [ :name, :prompt ]
+    end
+
+    # User-defined command model
+    class PromptCommandDefinition
+      include Recordable
+    end
+  end
+end

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

@@ -9,7 +9,9 @@ module Ruboty
 
       def initialize: (database: untyped) -> untyped
 
-      def namespace_keys: () -> Array[Database::keynable]
+      def namespace_keys: () -> Array[Database::query_key]
+
+      def empty?: () -> boolish
 
       def length: () -> Integer
 
@@ -19,22 +21,22 @@ module Ruboty
 
       def keys: () -> Array[Database::keynable]
 
-      # @rbs key: Database::keynable
+      # @rbs key: Database::query_key
       # @rbs return: Record | nil
-      def fetch: (Database::keynable key) -> (Record | nil)
+      def fetch: (Database::query_key key) -> (Record | nil)
 
-      # @rbs key: Database::keynable
+      # @rbs key: Database::query_key
       # @rbs record: Record
       # @rbs return: void
-      def store: (Record record, key: Database::keynable) -> void
+      def store: (Record record, key: Database::query_key) -> void
 
-      # @rbs key: Database::keynable
+      # @rbs key: Database::query_key
       # @rbs return: void
-      def remove: (Database::keynable key) -> void
+      def remove: (Database::query_key key) -> void
 
-      # @rbs key: Database::keynable
+      # @rbs key: Database::query_key
       # @rbs return: boolish
-      def key?: (Database::keynable key) -> boolish
+      def key?: (Database::query_key key) -> boolish
 
       def clear: () -> void
     end

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

@@ -3,7 +3,7 @@
 module Ruboty
   module AiAgent
     interface _WithToH
-      def to_h: () -> Hash[Database::keynable, untyped]
+      def to_h: () -> Hash[Symbol, untyped]
     end
 
     # Convertable between Hash and Recordable bidirectionally.
@@ -15,9 +15,9 @@ module Ruboty
 
       def self.record_types: () -> Hash[Symbol, Class]
 
-      # @rbs hash: Hash[Symbol, untyped]?
+      # @rbs hash: Hash[Symbol | String, untyped]?
       # @rbs return: bool
-      def self.convertable?: (Hash[Symbol, untyped]? hash) -> bool
+      def self.convertable?: (Hash[Symbol | String, untyped]? hash) -> bool
 
       # @rbs value: untyped
       # @rbs return: untyped
@@ -31,9 +31,9 @@ module Ruboty
       # @rbs return: Hash[Database::keynable, untyped]
       def self.record_to_hash: (Recordable record) -> Hash[Database::keynable, untyped]
 
-      # @rbs hash: Hash[Symbol, untyped]
+      # @rbs hash: Hash[Symbol | String, untyped]
       # @rbs return: Recordable
-      def self.record_from_hash: (Hash[Symbol, untyped] hash) -> Recordable
+      def self.record_from_hash: (Hash[Symbol | String, untyped] hash) -> Recordable
 
       # @rbs module-self Class
       module ClassMethods : Class
@@ -45,7 +45,7 @@ module Ruboty
 
       # @rbs module-self Recordable::ClassMethods.instance
       module PrependMethods : Recordable::ClassMethods
-        def to_h: () -> Hash[Database::keynable, untyped]
+        def to_h: () -> Hash[Symbol, untyped]
       end
 
       # @rbs %a{pure}

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/token_usage.rbs

@@ -24,6 +24,10 @@ module Ruboty
       # @rbs return: (Float | nil)
       def usage_percentage: () -> (Float | nil)
 
+      # Check if usage percentage exceeds auto compact threshold
+      # @rbs return: bool
+      def over_auto_compact_threshold?: () -> bool
+
       def to_h: () -> Hash[Symbol, untyped]
     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.rbs

@@ -30,6 +30,10 @@ module Ruboty
       # @rbs %a{memorized}
       %a{memorized}
       def mcp_tools_caches: () -> UserMcpToolsCaches
+
+      # @rbs %a{memorized}
+      %a{memorized}
+      def prompt_command_definitions: () -> UserPromptCommandDefinitions
     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/ruboty/ai_agent/user_prompt_command_definitions.rbs

@@ -0,0 +1,12 @@
+# Generated from lib/ruboty/ai_agent/user_prompt_command_definitions.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # User-defined commands record set
+    class UserPromptCommandDefinitions < UserAssociations[PromptCommandDefinition]
+      # @rbs command: PromptCommandDefinition
+      # @rbs return: PromptCommandDefinition
+      def add: (PromptCommandDefinition command) -> PromptCommandDefinition
+    end
+  end
+end