ruboty-ai_agent 0.1.0 → 0.2.0

data/script/generate-concern-rbs.rb

@@ -303,7 +303,7 @@ class ConcernRbsGenerator
 
       # Open namespaces
       namespace_parts.each do |part|
-        content << ('  ' * indent_level) + "module #{part}"
+        content << (('  ' * indent_level) + "module #{part}")
         indent_level += 1
       end
 
@@ -315,17 +315,17 @@ class ConcernRbsGenerator
         # Add type parameters if present
         if includer[:type_params] && !includer[:type_params].empty?
           type_params_str = "[#{includer[:type_params].join(', ')}]"
-          content << ('  ' * indent_level) + "#{keyword} #{simple_name}#{type_params_str}"
+          content << (('  ' * indent_level) + "#{keyword} #{simple_name}#{type_params_str}")
         else
-          content << ('  ' * indent_level) + "#{keyword} #{simple_name}"
+          content << (('  ' * indent_level) + "#{keyword} #{simple_name}")
         end
 
         concern = includer[:concern]
         concern_simple_name = concern[:name].sub(/^::/, '').split('::').last
 
-        content << ('  ' * (indent_level + 1)) + "extend #{concern_simple_name}::ClassMethods" if concern[:has_class_methods]
+        content << (('  ' * (indent_level + 1)) + "extend #{concern_simple_name}::ClassMethods") if concern[:has_class_methods]
 
-        content << ('  ' * (indent_level + 1)) + "prepend #{concern_simple_name}::PrependMethods" if concern[:has_prepend_methods]
+        content << (('  ' * (indent_level + 1)) + "prepend #{concern_simple_name}::PrependMethods") if concern[:has_prepend_methods]
 
         content << "#{'  ' * indent_level}end"
         content << ''

data/script/generate-data-rbs.rb

@@ -31,7 +31,7 @@ class Processor
     puts "Scanning Ruby files in #{@lib_path}..."
 
     @lib_path.glob('**/*.rb').map do |file|
-      puts "Processing #{file}..."
+      puts "Processing #{file}..." if ENV['DEBUG']
       content = file.read
 
       parsed = Prism.parse(content)
@@ -127,9 +127,7 @@ class DataClassRbsGenerator
 
     def extract_constant_name(constant_path)
       case constant_path
-      when Prism::ConstantReadNode
-        constant_path.name.to_s
-      when Prism::ConstantPathNode
+      when Prism::ConstantReadNode, Prism::ConstantPathNode
         constant_path.name.to_s
       end
     end
@@ -242,7 +240,7 @@ class DataClassRbsGenerator
     end
 
     def indent(string, level:)
-      '  ' * level + string
+      ('  ' * level) + string
     end
   end
 end

data/script/generate-memorized-ivar-rbs.rb

@@ -101,15 +101,15 @@ class MemorizedIvarRbsGenerator
 
       # Open namespaces
       parts[0...-1].each do |part|
-        content << ('  ' * indent_level) + "module #{part}"
+        content << (('  ' * indent_level) + "module #{part}")
         indent_level += 1
       end
 
       # Class declaration with instance variables
-      content << ('  ' * indent_level) + "class #{parts.last}"
+      content << (('  ' * indent_level) + "class #{parts.last}")
 
       methods.each do |method|
-        content << ('  ' * (indent_level + 1)) + "@#{method.ivar_name}: #{method.return_type}"
+        content << (('  ' * (indent_level + 1)) + "@#{method.ivar_name}: #{method.return_type}")
       end
 
       content << "#{'  ' * indent_level}end"
@@ -188,9 +188,7 @@ class MemorizedIvarRbsGenerator
 
     def extract_constant_name(constant_path)
       case constant_path
-      when Prism::ConstantReadNode
-        constant_path.name.to_s
-      when Prism::ConstantPathNode
+      when Prism::ConstantReadNode, Prism::ConstantPathNode
         constant_path.name.to_s
       end
     end
@@ -248,11 +246,8 @@ class MemorizedIvarRbsGenerator
           return result if result
         end
         nil
-      when Prism::InstanceVariableOperatorWriteNode, Prism::InstanceVariableOrWriteNode
-        # This handles @ivar ||= ...
-        node.name.to_s.delete('@')
-      when Prism::InstanceVariableWriteNode
-        # This handles @ivar = ...
+      when Prism::InstanceVariableOperatorWriteNode, Prism::InstanceVariableOrWriteNode, Prism::InstanceVariableWriteNode
+        # This handles @ivar ||= ... or @ivar = ...
         node.name.to_s.delete('@')
       else
         # Recursively search in child nodes

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

@@ -10,6 +10,10 @@ module Ruboty
         def name_param: () -> String
 
         def prompt_param: () -> String
+
+        # @rbs str: String
+        # @rbs return: String
+        def undump_string: (String str) -> String
       end
     end
   end

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

@@ -11,6 +11,12 @@ module Ruboty
         def call: ...
 
         def body_param: () -> String
+
+        private
+
+        # @rbs body: String
+        # @rbs return: void
+        def complete_chat: (String body) -> void
       end
     end
   end

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,22 @@ module Ruboty
       # Base class for commands.
       # @abstract
       class Base
-        class Matcher < Data
-          attr_reader pattern(): untyped
+        attr_reader message: Ruboty::Message
 
-          attr_reader description(): untyped
+        attr_reader chat_thread: Ruboty::AiAgent::ChatThread
 
-          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 message: Ruboty::Message
+        # @rbs chat_thread: Ruboty::AiAgent::ChatThread
+        def initialize: (message: Ruboty::Message, chat_thread: Ruboty::AiAgent::ChatThread) -> untyped
 
         # @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,27 @@
+# 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 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 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

@@ -6,8 +6,8 @@ module Ruboty
     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 return: Array[Commands::BuiltinBase]
+      def self.builtins: (message: Ruboty::Message, chat_thread: ChatThread) -> 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_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/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/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_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

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

@@ -4,29 +4,29 @@ module Ruboty
   module Handlers
     # Handler for AI agent commands
     class AiAgent < Base
-      def chat: (untyped message) -> untyped
+      def chat: (untyped message) -> true
 
-      def add_mcp: (untyped message) -> untyped
+      def add_mcp: (untyped message) -> true
 
-      def remove_mcp: (untyped message) -> untyped
+      def remove_mcp: (untyped message) -> true
 
-      def list_mcp: (untyped message) -> untyped
+      def list_mcp: (untyped message) -> true
 
-      def set_system_prompt: (untyped message) -> untyped
+      def set_system_prompt: (untyped message) -> true
 
-      def show_system_prompt: (untyped message) -> untyped
+      def show_system_prompt: (untyped message) -> true
 
-      def add_ai_memory: (untyped message) -> untyped
+      def add_ai_memory: (untyped message) -> true
 
-      def remove_ai_memory: (untyped message) -> untyped
+      def remove_ai_memory: (untyped message) -> true
 
-      def list_ai_memories: (untyped message) -> untyped
+      def list_ai_memories: (untyped message) -> true
 
-      def add_ai_command: (untyped message) -> untyped
+      def add_ai_command: (untyped message) -> true
 
-      def remove_ai_command: (untyped message) -> untyped
+      def remove_ai_command: (untyped message) -> true
 
-      def list_ai_commands: (untyped message) -> untyped
+      def list_ai_commands: (untyped message) -> true
     end
   end
 end