ruboty-ai_agent 0.1.0

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

@@ -0,0 +1,35 @@
+# Generated from lib/ruboty/ai_agent/mcp_configuration.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    type transports = :http | :websocket
+
+    class McpConfiguration < Data
+      attr_reader name(): String
+
+      attr_reader transport(): transports
+
+      attr_reader headers(): Hash[String, String]
+
+      attr_reader url(): String
+
+      def self.new: (String name, transports transport, Hash[String, String] headers, String url) -> instance
+                  | (name: String, transport: transports, headers: Hash[String, String], url: String) -> instance
+
+      def self.members: () -> [ :name, :transport, :headers, :url ]
+
+      def members: () -> [ :name, :transport, :headers, :url ]
+    end
+
+    # Save MCP configuration details.
+    class McpConfiguration
+      include Recordable
+
+      # @rbs name: String
+      # @rbs transport: transports
+      # @rbs url: String
+      # @rbs headers: Hash[String, String]?
+      def initialize: (name: String, transport: transports, url: String, ?headers: Hash[String, String]?) -> untyped
+    end
+  end
+end

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

@@ -0,0 +1,42 @@
+# Generated from lib/ruboty/ai_agent/record_set.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # A set of records.
+    # @rbs generic Record
+    class RecordSet[Record]
+      attr_reader database: Ruboty::AiAgent::Database
+
+      def initialize: (database: untyped) -> untyped
+
+      def namespace_keys: () -> Array[Database::keynable]
+
+      def length: () -> Integer
+
+      def all: () -> untyped
+
+      def all_values: () -> Array[Record]
+
+      def keys: () -> Array[Database::keynable]
+
+      # @rbs key: Database::keynable
+      # @rbs return: Record | nil
+      def fetch: (Database::keynable key) -> (Record | nil)
+
+      # @rbs key: Database::keynable
+      # @rbs record: Record
+      # @rbs return: void
+      def store: (Record record, key: Database::keynable) -> void
+
+      # @rbs key: Database::keynable
+      # @rbs return: void
+      def remove: (Database::keynable key) -> void
+
+      # @rbs key: Database::keynable
+      # @rbs return: boolish
+      def key?: (Database::keynable key) -> boolish
+
+      def clear: () -> void
+    end
+  end
+end

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

@@ -0,0 +1,56 @@
+# Generated from lib/ruboty/ai_agent/recordable.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    interface _WithToH
+      def to_h: () -> Hash[Database::keynable, untyped]
+    end
+
+    # Convertable between Hash and Recordable bidirectionally.
+    # @rbs module-self _WithToH
+    module Recordable : _WithToH
+      def self.included: (untyped base) -> untyped
+
+      @record_types: Hash[Symbol, Class]
+
+      def self.record_types: () -> Hash[Symbol, Class]
+
+      # @rbs hash: Hash[Symbol, untyped]?
+      # @rbs return: bool
+      def self.convertable?: (Hash[Symbol, untyped]? hash) -> bool
+
+      # @rbs value: untyped
+      # @rbs return: untyped
+      def self.instantiate_recursively: (untyped value) -> untyped
+
+      # @rbs value: untyped
+      # @rbs return: untyped
+      def self.hashify_recursively: (untyped value) -> untyped
+
+      # @rbs record: Recordable
+      # @rbs return: Hash[Database::keynable, untyped]
+      def self.record_to_hash: (Recordable record) -> Hash[Database::keynable, untyped]
+
+      # @rbs hash: Hash[Symbol, untyped]
+      # @rbs return: Recordable
+      def self.record_from_hash: (Hash[Symbol, untyped] hash) -> Recordable
+
+      # @rbs module-self Class
+      module ClassMethods : Class
+        attr_accessor record_type: Symbol
+
+        # @rbs name: Symbol
+        def register_record_type: (Symbol name) -> untyped
+      end
+
+      # @rbs module-self Recordable::ClassMethods.instance
+      module PrependMethods : Recordable::ClassMethods
+        def to_h: () -> Hash[Database::keynable, untyped]
+      end
+
+      # @rbs %a{pure}
+      %a{pure}
+      def record_type: () -> Symbol
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+# Generated from lib/ruboty/ai_agent/token_usage.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Token usage information from LLM responses
+    class TokenUsage
+      include Recordable
+
+      attr_reader prompt_tokens: Integer
+
+      attr_reader completion_tokens: Integer
+
+      attr_reader total_tokens: Integer
+
+      attr_reader token_limit: Integer?
+
+      # @rbs prompt_tokens: Integer
+      # @rbs completion_tokens: Integer
+      # @rbs total_tokens: Integer
+      # @rbs ?token_limit: Integer?
+      def initialize: (prompt_tokens: Integer, completion_tokens: Integer, total_tokens: Integer, ?token_limit: untyped) -> untyped
+
+      # Calculate usage percentage if token limit is available
+      # @rbs return: (Float | nil)
+      def usage_percentage: () -> (Float | nil)
+
+      def to_h: () -> Hash[Symbol, untyped]
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+# Generated from lib/ruboty/ai_agent/tool.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Define a tool that the AI agent can use.
+    class Tool
+      attr_reader name: String
+
+      attr_reader title: String
+
+      attr_reader description: String
+
+      attr_reader input_schema: Hash[untyped, untyped]?
+
+      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
+
+      def call: (untyped params) -> String?
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+# Generated from lib/ruboty/ai_agent/user.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # User class to manage user-specific data.
+    class User
+      attr_reader database: Ruboty::AiAgent::Database
+
+      attr_reader id: String
+
+      def self.find_or_create: (database: untyped, id: untyped) -> User
+
+      def initialize: (database: untyped, id: untyped) -> untyped
+
+      # @rbs %a{memorized}
+      %a{memorized}
+      def mcp_configurations: () -> UserMcpConfigurations
+
+      def mcp_clients: () -> Array[UserMcpClient]
+
+      # @rbs %a{memorized}
+      %a{memorized}
+      def ai_memories: () -> UserAiMemories
+
+      def system_prompt: () -> String?
+
+      # @rbs prompt: String?
+      def system_prompt=: (String? prompt) -> untyped
+
+      # @rbs %a{memorized}
+      %a{memorized}
+      def mcp_tools_caches: () -> UserMcpToolsCaches
+    end
+  end
+end

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

@@ -0,0 +1,11 @@
+# Generated from lib/ruboty/ai_agent/user_ai_memories.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Manage Ai Memories for a specific user.
+    class UserAiMemories < UserAssociations[String]
+      # @rbs memory: String
+      def add: (String memory) -> void
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+# Generated from lib/ruboty/ai_agent/user_associations.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # A set of records for a specific user.
+    # @rbs generic Record
+    class UserAssociations[Record] < RecordSet[Record]
+      attr_reader user_id: String
+
+      def initialize: (database: untyped, user_id: untyped) -> untyped
+
+      def self.association_key: () -> Symbol
+
+      def self.association_key=: (Symbol) -> Symbol
+
+      def association_key: () -> Symbol
+
+      def namespace_keys: () -> untyped
+    end
+  end
+end

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

@@ -0,0 +1,44 @@
+# Generated from lib/ruboty/ai_agent/user_mcp_caches.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Manage MCP caches for a specific user.
+    # @abstract
+    # @rbs generic D < Object
+    class UserMcpCaches[D < Object] < UserAssociations[CachedValue[D]]
+      def self.cache_type_key: () -> Symbol?
+
+      def self.cache_type_key=: (Symbol?) -> Symbol?
+
+      def self.cache_duration: () -> Integer?
+
+      def self.cache_duration=: (Integer?) -> Integer?
+
+      def self.association_key: () -> Symbol
+
+      def cache_type_key: () -> Symbol
+
+      def cache_duration: () -> Integer
+
+      # @rbs key: String)
+      # @rbs &block: () -> (D | CachedValue[D])
+      # @rbs return: D
+      def fetch_or_store_data: (String key) { () -> (D | CachedValue[D]) } -> D
+
+      # @rbs key: String
+      # @rbs return: D?
+      def fetch_data: (String key) -> D?
+
+      # @rbs override
+      def all: ...
+
+      # @rbs override
+      def fetch: ...
+
+      # @rbs key: String
+      # @rbs data: D
+      # @rbs return: void
+      def store_data: (D data, key: String) -> void
+    end
+  end
+end

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

@@ -0,0 +1,58 @@
+# Generated from lib/ruboty/ai_agent/user_mcp_client.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Wrapper for MCP client with caching support for a specific user
+    class UserMcpClient
+      attr_reader user: User
+
+      attr_reader mcp_name: String
+
+      # @rbs user: User
+      # @rbs mcp_name: String
+      # @rbs return: void
+      def initialize: (user: User, mcp_name: String) -> void
+
+      # @rbs return: Array[tool_def]
+      def list_tools: () -> Array[tool_def]
+
+      # @rbs name: String
+      # @rbs arguments: Hash[String, untyped]
+      # @rbs &block: ? (Hash[String, untyped]) -> void
+      # @rbs return: untyped
+      def call_tool: (String name, ?Hash[String, untyped] arguments) ?{ (Hash[String, untyped]) -> void } -> untyped
+
+      # @rbs return: untyped
+      def list_prompts: () -> untyped
+
+      # @rbs name: String
+      # @rbs arguments: Hash[String, untyped]
+      # @rbs return: untyped
+      def get_prompt: (String name, ?Hash[String, untyped] arguments) -> untyped
+
+      # @rbs return: untyped
+      def list_resources: () -> untyped
+
+      # @rbs uri: String
+      # @rbs return: untyped
+      def read_resource: (String uri) -> untyped
+
+      # @rbs return: untyped
+      def ping: () -> untyped
+
+      # @rbs return: untyped
+      def initialize_session: () -> untyped
+
+      # @rbs return: untyped
+      def cleanup_session: () -> untyped
+
+      private
+
+      # @rbs return: McpConfiguration
+      def configuration: () -> McpConfiguration
+
+      # @rbs return: HttpMcpClient
+      def mcp_client: () -> HttpMcpClient
+    end
+  end
+end

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

@@ -0,0 +1,11 @@
+# Generated from lib/ruboty/ai_agent/user_mcp_configurations.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Manage MCP configurations for a specific user.
+    class UserMcpConfigurations < UserAssociations[McpConfiguration]
+      # @rbs mcp_configuration: McpConfiguration:
+      def add: (McpConfiguration mcp_configuration) -> void
+    end
+  end
+end

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

@@ -0,0 +1,11 @@
+# Generated from lib/ruboty/ai_agent/user_mcp_tools_caches.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    type tool_def = Hash[Symbol | String, untyped]
+
+    # Manage tools cache for a specific user and server.
+    class UserMcpToolsCaches < UserMcpCaches[Array[tool_def]]
+    end
+  end
+end

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

@@ -0,0 +1,7 @@
+# Generated from lib/ruboty/ai_agent/version.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    VERSION: ::String
+  end
+end

data/sig/generated/ruboty/ai_agent.rbs

@@ -0,0 +1,9 @@
+# Generated from lib/ruboty/ai_agent.rb with RBS::Inline
+
+module Ruboty
+  # Add AI Agent feature to Ruboty.
+  module AiAgent
+    class Error < StandardError
+    end
+  end
+end

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

@@ -0,0 +1,32 @@
+# Generated from lib/ruboty/handlers/ai_agent.rb with RBS::Inline
+
+module Ruboty
+  module Handlers
+    # Handler for AI agent commands
+    class AiAgent < Base
+      def chat: (untyped message) -> untyped
+
+      def add_mcp: (untyped message) -> untyped
+
+      def remove_mcp: (untyped message) -> untyped
+
+      def list_mcp: (untyped message) -> untyped
+
+      def set_system_prompt: (untyped message) -> untyped
+
+      def show_system_prompt: (untyped message) -> untyped
+
+      def add_ai_memory: (untyped message) -> untyped
+
+      def remove_ai_memory: (untyped message) -> untyped
+
+      def list_ai_memories: (untyped message) -> untyped
+
+      def add_ai_command: (untyped message) -> untyped
+
+      def remove_ai_command: (untyped message) -> untyped
+
+      def list_ai_commands: (untyped message) -> untyped
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+# Generated RBS definitions for concern modules
+# This file is automatically generated by script/generate-concern-rbs.rb
+
+module Ruboty
+  module AiAgent
+    class CachedValue[D < Object]
+      extend Recordable::ClassMethods
+      prepend Recordable::PrependMethods
+    end
+
+    class ChatMessage
+      extend Recordable::ClassMethods
+      prepend Recordable::PrependMethods
+    end
+
+    class McpConfiguration
+      extend Recordable::ClassMethods
+      prepend Recordable::PrependMethods
+    end
+
+    class TokenUsage
+      extend Recordable::ClassMethods
+      prepend Recordable::PrependMethods
+    end
+
+  end
+end

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

@@ -0,0 +1,42 @@
+# Generated RBS definitions for memorized instance variables
+# This file is automatically generated by script/generate-memorized-ivar-rbs.rb
+
+module Ruboty
+  module AiAgent
+    module Actions
+      class Base
+        @database: Ruboty::AiAgent::Database
+        @user: Ruboty::AiAgent::User
+        @chat_thread: Ruboty::AiAgent::ChatThread
+      end
+    end
+  end
+end
+
+module Ruboty
+  module AiAgent
+    class ChatThread
+      @messages: ChatThreadMessages
+    end
+  end
+end
+
+module Ruboty
+  module AiAgent
+    module LLM
+      class OpenAI
+        @model_info: Model
+      end
+    end
+  end
+end
+
+module Ruboty
+  module AiAgent
+    class User
+      @mcp_configurations: UserMcpConfigurations
+      @ai_memories: UserAiMemories
+      @mcp_tools_caches: UserMcpToolsCaches
+    end
+  end
+end

data/sig-lib/event_stream_parser/event_stream_parser.rbs

@@ -0,0 +1,21 @@
+module EventStreamParser
+  VERSION: ::String
+
+  class Parser
+    def feed: (untyped chunk) { (*untyped) -> untyped } -> untyped
+
+    def stream: () -> untyped
+
+    private
+
+    def dispatch_event: () -> untyped
+
+    def ignore: () -> untyped
+
+    def initialize: () -> void
+
+    def process_field: (untyped field, untyped value) -> untyped
+
+    def process_line: (untyped line) { (*untyped) -> untyped } -> untyped
+  end
+end

data/sig-lib/mem/mem.rbs

@@ -0,0 +1,19 @@
+module Mem
+  def self.included: (untyped base) -> untyped
+
+  def has_memoized?: (untyped key) -> untyped
+
+  def memoize: (untyped key, untyped value) -> untyped
+
+  def memoized: (untyped key) -> untyped
+
+  def memoized_table: () -> untyped
+
+  def unmemoize: (untyped key) -> untyped
+
+  VERSION: ::String
+
+  module ClassMethods
+    def memoize: (untyped method_name) -> untyped
+  end
+end