ruboty-ai_agent 0.1.0

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

@@ -0,0 +1,14 @@
+# Generated from lib/ruboty/ai_agent/actions/remove_ai_memory.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # RemoveAiMemory action for Ruboty::AiAgent
+      class RemoveAiMemory < Base
+        def call: () -> untyped
+
+        def index_param: () -> String
+      end
+    end
+  end
+end

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

@@ -0,0 +1,14 @@
+# Generated from lib/ruboty/ai_agent/actions/remove_mcp.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # RemoveMcp action for Ruboty::AiAgent
+      class RemoveMcp < Base
+        def call: () -> untyped
+
+        def name_param: () -> String
+      end
+    end
+  end
+end

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

@@ -0,0 +1,16 @@
+# Generated from lib/ruboty/ai_agent/actions/set_system_prompt.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # SetSystemPrompt action for Ruboty::AiAgent
+      class SetSystemPrompt < Base
+        def call: () -> untyped
+
+        def prompt_param: () -> String
+
+        def scope_param: () -> String
+      end
+    end
+  end
+end

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

@@ -0,0 +1,12 @@
+# Generated from lib/ruboty/ai_agent/actions/show_system_prompt.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # ShowSystemPrompt action for Ruboty::AiAgent
+      class ShowSystemPrompt < Base
+        def call: () -> untyped
+      end
+    end
+  end
+end

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

@@ -0,0 +1,9 @@
+# Generated from lib/ruboty/ai_agent/actions.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Ruboty AI Agent actions.
+    module Actions
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+# Generated from lib/ruboty/ai_agent/agent.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Agent class to interact with LLM and manage conversations.
+    class Agent
+      attr_reader llm: LLM::OpenAI
+
+      attr_reader messages: Array[ChatMessage]
+
+      attr_reader tools: Array[Tool]
+
+      # @rbs llm: LLM::OpenAI
+      # @rbs messages: Array[ChatMessage]
+      # @rbs tools: Array[Tool]
+      def initialize: (llm: LLM::OpenAI, ?messages: Array[ChatMessage], ?tools: Array[Tool]) -> untyped
+
+      def complete: () ?{ (?) -> untyped } -> untyped
+
+      def on_new_message: (untyped message) ?{ (?) -> untyped } -> untyped
+
+      def on_tool_call: (tool: untyped, tool_arguments: untyped) ?{ (?) -> untyped } -> untyped
+
+      def on_tool_response: (tool_response: untyped, message: untyped) ?{ (?) -> untyped } -> untyped
+
+      def on_response: (untyped response) ?{ (?) -> untyped } -> untyped
+    end
+  end
+end

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

@@ -0,0 +1,28 @@
+# Generated from lib/ruboty/ai_agent/cached_value.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Cache for MCP operations
+    # @rbs generic D < Object
+    class CachedValue[D < Object]
+      include Recordable
+
+      attr_reader expires_at: Time
+
+      attr_reader data: D
+
+      # @rbs data: D
+      # @rbs expires_at: Time
+      def initialize: (data: D, expires_at: Time) -> untyped
+
+      # @rbs return: Hash[Symbol, untyped]
+      def to_h: () -> Hash[Symbol, untyped]
+
+      # @rbs return: bool
+      def expired?: () -> bool
+
+      # @rbs return: bool
+      def valid?: () -> bool
+    end
+  end
+end

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

@@ -0,0 +1,34 @@
+# Generated from lib/ruboty/ai_agent/chat_message.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Save MCP configuration details.
+    class ChatMessage
+      include Recordable
+
+      attr_reader role: Symbol
+
+      attr_reader content: String
+
+      attr_reader tool_call_id: String?
+
+      attr_reader tool_name: String?
+
+      attr_reader tool_arguments: Hash[Symbol | String, untyped]?
+
+      attr_reader token_usage: TokenUsage?
+
+      # @rbs role: Symbol
+      # @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 to_h: () -> Hash[Symbol, untyped]
+
+      def self.from_llm_response: (tool: untyped, tool_call_id: untyped, tool_arguments: untyped, tool_response: untyped) -> ChatMessage
+    end
+  end
+end

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

@@ -0,0 +1,22 @@
+# Generated from lib/ruboty/ai_agent/chat_thread.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Manage thread-specific data.
+    class ChatThread
+      attr_reader database: Ruboty::AiAgent::Database
+
+      attr_reader id: String
+
+      def self.find_or_create: (database: untyped, id: untyped) -> ChatThread
+
+      def initialize: (database: untyped, id: untyped) -> untyped
+
+      # @rbs %a{memorized}
+      %a{memorized}
+      def messages: () -> ChatThreadMessages
+
+      def clear: () -> void
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+# Generated from lib/ruboty/ai_agent/chat_thread_associations.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # A set of records for a specific thread.
+    # @rbs generic Record
+    class ChatThreadAssociations[Record] < RecordSet[Record]
+      attr_reader chat_thread_id: String
+
+      def initialize: (database: untyped, chat_thread_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/chat_thread_messages.rbs

@@ -0,0 +1,13 @@
+# Generated from lib/ruboty/ai_agent/chat_thread_messages.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Manage messages for the chat thread.
+    class ChatThreadMessages < ChatThreadAssociations[ChatMessage]
+      # @rbs message: ChatMessage
+      def add: (ChatMessage message) -> void
+
+      alias << add
+    end
+  end
+end

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

@@ -0,0 +1,40 @@
+# Generated from lib/ruboty/ai_agent/commands/base.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module Commands
+      # Base class for commands.
+      # @abstract
+      class 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

@@ -0,0 +1,20 @@
+# Generated from lib/ruboty/ai_agent/commands/clear.rb with RBS::Inline
+
+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
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+# Generated from lib/ruboty/ai_agent/commands/compact.rb with RBS::Inline
+
+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
+      end
+    end
+  end
+end

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

@@ -0,0 +1,26 @@
+# Generated from lib/ruboty/ai_agent/commands/usage.rb with RBS::Inline
+
+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
+
+        private
+
+        # @rbs number: Integer
+        # @rbs return: String
+        def format_number: (Integer number) -> String
+      end
+    end
+  end
+end

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

@@ -0,0 +1,13 @@
+# Generated from lib/ruboty/ai_agent/commands.rb with RBS::Inline
+
+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]
+    end
+  end
+end

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

@@ -0,0 +1,39 @@
+# Generated from lib/ruboty/ai_agent/database/query_methods.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    class Database
+      interface _WithData
+        def data: () -> Hash[keynable, untyped]
+      end
+
+      # @rbs module-self _WithData
+      module QueryMethods : _WithData
+        # @rbs *keys: keynable
+        # @rbs return: untyped
+        def fetch: (*keynable keys) -> untyped
+
+        # @rbs *keys: keynable
+        # @rbs return: Integer
+        def len: (*keynable keys) -> Integer
+
+        # @rbs *keys: keynable
+        # @rbs return: void
+        def delete: (*keynable keys) -> void
+
+        # @rbs *keys: keynable
+        # @rbs return: Array[keynable]
+        def keys: (*keynable keys) -> Array[keynable]
+
+        # @rbs *keys: keynable
+        # @rbs return: boolish
+        def key?: (*keynable keys) -> boolish
+
+        # @rbs at: Array[keynable]
+        # @rbs value: untyped
+        # @rbs return: void
+        def store: (untyped value, at: Array[keynable]) -> void
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+# Generated from lib/ruboty/ai_agent/database.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Memorize and retrieve information using Ruboty's brain.
+    class Database
+      type keynable = Symbol | String | Integer
+
+      include QueryMethods
+
+      NAMESPACE: ::Symbol
+
+      attr_reader brain: Ruboty::Brains::Base
+
+      # @rbs brain: Ruboty::Brains::Base
+      def initialize: (Ruboty::Brains::Base brain) -> untyped
+
+      def data: () -> Hash[keynable, untyped]
+
+      def user: (untyped id) -> User
+
+      def chat_thread: (untyped id) -> ChatThread
+
+      def global_settings: () -> GlobalSettings
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+# Generated from lib/ruboty/ai_agent/global_settings.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Global settings for AI agent.
+    class GlobalSettings
+      attr_reader database: Ruboty::AiAgent::Database
+
+      NAMESPACE_KEYS: Array[Database::keynable]
+
+      def self.find_or_create: (database: untyped) -> GlobalSettings
+
+      # @rbs database: Ruboty::AiAgent::Database
+      def initialize: (database: Ruboty::AiAgent::Database) -> untyped
+
+      def system_prompt: () -> String?
+
+      # @rbs prompt: String?
+      # @rbs return: void
+      def system_prompt=: (String? prompt) -> void
+    end
+  end
+end

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

@@ -0,0 +1,62 @@
+# Generated from lib/ruboty/ai_agent/http_mcp_client.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Mcp client with HTTP transport.
+    class HttpMcpClient
+      attr_reader base_url: untyped
+
+      attr_reader headers: untyped
+
+      attr_reader session_id: untyped
+
+      def initialize: (url: untyped, ?headers: untyped, ?session_id: untyped) -> untyped
+
+      def initialize_session: () -> untyped
+
+      def ping: () -> untyped
+
+      def list_tools: () -> untyped
+
+      # @rbs name: String
+      # @rbs &block: ? (Hash[String, untyped]) -> void
+      def call_tool: (String name, ?untyped arguments) ?{ (Hash[String, untyped]) -> void } -> untyped
+
+      def list_prompts: () -> untyped
+
+      def get_prompt: (untyped name, ?untyped arguments) -> untyped
+
+      def list_resources: () -> untyped
+
+      def read_resource: (untyped uri) -> untyped
+
+      def cleanup_session: () -> untyped
+
+      private
+
+      @initialize_called: bool
+
+      def ensure_initialized: () -> untyped
+
+      # @rbs method: String
+      # @rbs ?params: Hash[String | Symbol, untyped]?
+      # @rbs ?id: String?
+      # @rbs &block: ? (Hash[String, untyped]) -> void
+      # @rbs return: Array[Hash[String, untyped]]
+      def send_request: (method: String, ?params: untyped, ?id: untyped) ?{ (Hash[String, untyped]) -> void } -> Array[Hash[String, untyped]]
+
+      # @rbs response: Net::HTTPResponse
+      # @rbs &block: ? (Hash[String, untyped]) -> void
+      # @rbs return: Array[Hash[String, untyped]]
+      def handle_response: (Net::HTTPResponse response) ?{ (Hash[String, untyped]) -> void } -> Array[Hash[String, untyped]]
+
+      # @rbs response: Net::HTTPResponse
+      # @rbs &block: ? (Hash[String, untyped]) -> void
+      # @rbs return: Array[Hash[String, untyped]]
+      def handle_streaming_response: (Net::HTTPResponse response) ?{ (Hash[String, untyped]) -> void } -> Array[Hash[String, untyped]]
+
+      class Error < StandardError
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+# Generated from lib/ruboty/ai_agent/llm/openai/model.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module LLM
+      class OpenAI
+        # Model information and token limit estimation
+        class Model
+          attr_reader name: String
+
+          # @rbs name: String
+          def initialize: (String name) -> untyped
+
+          # Estimate token limit based on model name
+          # @rbs return: Integer
+          def token_limit: () -> Integer
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,54 @@
+# Generated from lib/ruboty/ai_agent/llm/openai.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module LLM
+      # LLM interface for OpenAI's chat completion API.
+      class OpenAI
+        attr_reader client: OpenAI::Client
+
+        attr_reader model: String
+
+        # @rbs client: OpenAI::Client
+        # @rbs model: String
+        def initialize: (client: OpenAI::Client, model: String) -> untyped
+
+        # @rbs %a{memorized}
+        %a{memorized}
+        def model_info: () -> Model
+
+        # @rbs messages: Array[ChatMessage]
+        # @rbs tools: Array[Tool]
+        # @rbs return: Response
+        def complete: (messages: Array[ChatMessage], ?tools: Array[Tool]) -> Response
+
+        private
+
+        type tool_call = { id: String, type: "function", function: { name: String, arguments: String } }
+
+        type system_message = { role: "system", content: String }
+
+        type user_message = { role: "user", content: String }
+
+        type assistant_message = { role: "assistant", content: String, tool_calls: Array[tool_call]? }
+
+        type tool_message = { role: "tool", tool_call_id: String, content: String }
+
+        type message = system_message | user_message | assistant_message | tool_message
+
+        # @rbs messages: Array[ChatMessage]
+        # @rbs return: Array[OpenAI::Models::Chat::chat_completion_message_param]
+        def openai_messages_from_messages: (Array[ChatMessage] messages) -> Array[OpenAI::Models::Chat::chat_completion_message_param]
+
+        # @rbs tools: Array[Tool]
+        # @rbs return: Array[OpenAI::Models::Chat::chat_completion_tool]
+        def openai_tools_from_tools: (Array[Tool] tools) -> Array[OpenAI::Models::Chat::chat_completion_tool]
+
+        # @rbs openai_response: OpenAI::Models::Chat::ChatCompletion
+        # @rbs tools: Array[Tool]
+        # @rbs return: Response
+        def to_response: (openai_response: OpenAI::Models::Chat::ChatCompletion, tools: Array[Tool]) -> Response
+      end
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+# Generated from lib/ruboty/ai_agent/llm/response.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    module LLM
+      class Response < Data
+        attr_reader message(): ChatMessage
+
+        attr_reader tool(): Ruboty::AiAgent::Tool?
+
+        attr_reader tool_call_id(): String?
+
+        attr_reader tool_arguments(): Hash[String, String]?
+
+        def self.new: (ChatMessage message, Ruboty::AiAgent::Tool? tool, String? tool_call_id, Hash[String, String]? tool_arguments) -> instance
+                    | (message: ChatMessage, tool: Ruboty::AiAgent::Tool?, tool_call_id: String?, tool_arguments: Hash[String, String]?) -> instance
+
+        def self.members: () -> [ :message, :tool, :tool_call_id, :tool_arguments ]
+
+        def members: () -> [ :message, :tool, :tool_call_id, :tool_arguments ]
+      end
+
+      # General response class for LLM interactions.
+      class Response
+        def call_tool: () -> String?
+      end
+    end
+  end
+end

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

@@ -0,0 +1,9 @@
+# Generated from lib/ruboty/ai_agent/llm.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # LLM-related backends and its utilities.
+    module LLM
+    end
+  end
+end

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

@@ -0,0 +1,24 @@
+# Generated from lib/ruboty/ai_agent/mcp_clients.rb with RBS::Inline
+
+module Ruboty
+  module AiAgent
+    # Manages multiple MCP (Model Context Protocol) clients
+    class McpClients
+      attr_reader clients: Array[UserMcpClient]
+
+      # @rbs clients: Array[UserMcpClient]
+      def initialize: (Array[UserMcpClient] clients) -> untyped
+
+      # @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
+  end
+end