ruboty-ai_agent 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 1782e855675403948a771dd5abe6fdba6a754cfc6e15bebad40f24bf3c7569f6
+  data.tar.gz: 2142af0fb45049c4e1b2397def8009248db8860e43a9fa2ae21f938a31e3af82
+SHA512:
+  metadata.gz: 1a1b7fc8e8e41150055e87d2227e9b0ae8570c69482cf72249af38abef0978058419eed73f778a0f4d9c00343a55ce96be0cd62febdb4a20cd8b7976a6852415
+  data.tar.gz: 73aff06e4a6d7fb3601b9c95393b9c1496fd61eddaaabf0fe742e7701faad638a36c3919462e65847dab47d0450579c8047ba492beb4d51c81181b5b66307aef

data/.gitignore

@@ -0,0 +1,19 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+
+.env
+
+# rspec failure tracking
+.rspec_status
+
+Gemfile.lock
+
+# rbs collection
+.gem_rbs_collection/
+rbs_collection.lock.yaml

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.rubocop.yml

@@ -0,0 +1,45 @@
+# The behavior of RuboCop can be controlled via the .rubocop.yml
+# configuration file. It makes it possible to enable/disable
+# certain cops (checks) and to alter their behavior if they accept
+# any parameters. The file can be placed either in your home
+# directory or in some project directory.
+#
+# RuboCop will start looking for the configuration file in the directory
+# where the inspected file is and continue its way up to the root directory.
+#
+# See https://docs.rubocop.org/rubocop/configuration
+plugins:
+  - rubocop-rake
+  - rubocop-rbs_inline
+
+AllCops:
+  TargetRubyVersion: 3.1
+
+Naming/AccessorMethodName:
+  Exclude:
+    - lib/ruboty/handlers/ai_agent.rb # we name handler methods from its word pattern
+
+Naming/FileName:
+  Exclude:
+    - Steepfile
+
+Layout/LineLength:
+  Enabled: false
+
+Layout/LeadingCommentSpace:
+  AllowRBSInlineAnnotation: true
+  AllowSteepAnnotation: true
+
+Metrics:
+  Enabled: false
+
+Style/CommentedKeyword:
+  Enabled: false # because of RBS annotations
+
+Style/Documentation:
+  AllowedConstants:
+    - ClassMethods
+    - PrependFeatures
+
+Style/RedundantBegin:
+  Enabled: false # because of Steep annotations

data/AGENTS.md

@@ -0,0 +1,22 @@
+# Development Guide for AI
+
+## Dev instructions
+- Use English for codes, commit messages and PR descriptions.
+
+## Dev environment tips
+- Our RBS types in `sig/` are generated by rbs-inline and our scripts in `script/`.
+  - To update them, run `rake rbs`.
+
+## Testing instructions
+
+- In this repository, we use the following tools:
+  - RSpec (for testing)
+  - RuboCop (for linting)
+  - Steep (for type checking)
+- To run these all tools, use `bundle exec rake`.
+- When you add or modify code, please follow TDD workflow.
+- Fix any test or type errors until the whole suite is green.
+
+## PR instructions
+
+- Always run `rake autocorrect` before committing.

data/CHANGELOG.md

@@ -0,0 +1,3 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.

data/CLAUDE.md

@@ -0,0 +1 @@
+AGENTS.md

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,74 @@
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+In the interest of fostering an open and welcoming environment, we as
+contributors and maintainers pledge to making participation in our project and
+our community a harassment-free experience for everyone, regardless of age, body
+size, disability, ethnicity, gender identity and expression, level of experience,
+nationality, personal appearance, race, religion, or sexual identity and
+orientation.
+
+## Our Standards
+
+Examples of behavior that contributes to creating a positive environment
+include:
+
+* Using welcoming and inclusive language
+* Being respectful of differing viewpoints and experiences
+* Gracefully accepting constructive criticism
+* Focusing on what is best for the community
+* Showing empathy towards other community members
+
+Examples of unacceptable behavior by participants include:
+
+* The use of sexualized language or imagery and unwelcome sexual attention or
+advances
+* Trolling, insulting/derogatory comments, and personal or political attacks
+* Public or private harassment
+* Publishing others' private information, such as a physical or electronic
+  address, without explicit permission
+* Other conduct which could reasonably be considered inappropriate in a
+  professional setting
+
+## Our Responsibilities
+
+Project maintainers are responsible for clarifying the standards of acceptable
+behavior and are expected to take appropriate and fair corrective action in
+response to any instances of unacceptable behavior.
+
+Project maintainers have the right and responsibility to remove, edit, or
+reject comments, commits, code, wiki edits, issues, and other contributions
+that are not aligned to this Code of Conduct, or to ban temporarily or
+permanently any contributor for other behaviors that they deem inappropriate,
+threatening, offensive, or harmful.
+
+## Scope
+
+This Code of Conduct applies both within project spaces and in public spaces
+when an individual is representing the project or its community. Examples of
+representing a project or community include using an official project e-mail
+address, posting via an official social media account, or acting as an appointed
+representative at an online or offline event. Representation of a project may be
+further defined and clarified by project maintainers.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported by contacting the project team at tomo.asleep@gmail.com. All
+complaints will be reviewed and investigated and will result in a response that
+is deemed necessary and appropriate to the circumstances. The project team is
+obligated to maintain confidentiality with regard to the reporter of an incident.
+Further details of specific enforcement policies may be posted separately.
+
+Project maintainers who do not follow or enforce the Code of Conduct in good
+faith may face temporary or permanent repercussions as determined by other
+members of the project's leadership.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 1.4,
+available at [http://contributor-covenant.org/version/1/4][version]
+
+[homepage]: http://contributor-covenant.org
+[version]: http://contributor-covenant.org/version/1/4/

data/Gemfile

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+
+git_source(:github) { |repo_name| "https://github.com/#{repo_name}" }
+
+# Specify your gem's dependencies in ruboty-ai_agent.gemspec
+gemspec
+
+gem 'rbs-inline', require: false
+gem 'rubocop'
+gem 'rubocop-rake'
+gem 'rubocop-rbs_inline'
+gem 'steep', require: false

data/README.md

@@ -0,0 +1,118 @@
+# Ruboty::AiAgent
+
+A Ruboty handler that uses LLMs (e.g., OpenAI) and MCP (Model Context Protocol) tools to generate AI-assisted replies. When an incoming message does not match any other handler, this plugin responds via an LLM and can call MCP tools when available.
+
+## Installation
+
+Add to your Gemfile:
+
+```ruby
+gem 'ruboty'
+gem 'ruboty-ai_agent'
+```
+
+Then install:
+
+```
+bundle install
+```
+
+## Setup
+
+### OpenAI
+
+Configure environment variables (e.g., via `.env`):
+
+- `OPENAI_API_KEY`: Your LLM provider API key (required)
+- `OPENAI_MODEL`: Model name (optional)
+
+ 
+
+## Usage
+
+- Any message that does not match other handlers is sent to the AI and replied to.
+- Conversations are tracked per thread, and tools exposed via MCP servers can be invoked through function calls when the model requests them.
+
+## Commands
+
+Conversation management:
+- `/clear` — Clear the current thread’s conversation history
+- `/compact` — Summarize and compact the conversation history
+
+System prompt (work in progress):
+- `set system prompt "<PROMPT>"` — Set the system prompt (WIP)
+- `show system prompt` — Show the current system prompt (WIP)
+
+AI memory (lightweight profile):
+- `add ai memory "<TEXT>"` — Add a memory entry
+- `remove ai memory <INDEX>` — Remove a memory by index
+- `list ai memories` — List all memory entries
+
+AI custom commands (work in progress):
+- `add ai command /<NAME> "<PROMPT>"` — Add a custom AI command (WIP)
+- `remove ai command /<NAME>` — Remove a custom AI command (WIP)
+- `list ai commands` — List available built-in commands
+
+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`
+  - Options:
+    - `--transport http|sse` (currently only `http` implemented; `sse` is not yet implemented)
+    - `--header 'Key: Value'` (repeatable)
+- `remove mcp <NAME>` — Remove an MCP server
+- `list mcp` / `list mcps` — List configured MCP servers
+
+## How It Works
+
+- Unmatched messages are handled by `Ruboty::Handlers::AiAgent#chat` and sent to the configured LLM.
+- If MCP servers are configured, their tool definitions are exposed to the model; when the model requests a function call, the corresponding MCP tool is invoked and its response is fed back into the conversation.
+- During tool calls, the bot streams short status updates and returns a brief summary of the tool’s response.
+- Conversation data is stored in Ruboty’s brain under the `:ai_agent` namespace.
+
+
+## Development
+
+Install dependencies:
+
+```
+bin/setup
+```
+
+Local run with dotenv (for development):
+
+```
+bundle exec ruboty --dotenv
+
+# or using the included launcher
+bin/ruboty --dotenv
+```
+
+Run tests:
+
+```
+rake spec
+```
+
+Interactive console:
+
+```
+bin/console
+```
+
+Install locally:
+
+```
+bundle exec rake install
+```
+
+Release:
+Update `version.rb` and run `bundle exec rake release` to tag, push, and publish to RubyGems.
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/tomoasleep/ruboty-ai_agent.
+
+## Code of Conduct
+
+Everyone interacting in the Ruboty::AiAgent project’s codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/tomoasleep/ruboty-ai_agent/blob/main/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+require 'rubocop/rake_task'
+require 'steep/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+RuboCop::RakeTask.new do |task|
+  task.plugins << 'rubocop-rake'
+end
+
+Steep::RakeTask.new do |t|
+  t.check.severity_level = :error
+  t.watch.verbose
+end
+
+task default: %i[rubocop steep spec]
+task autocorrect: %i[rubocop:autocorrect rbs steep spec]
+
+namespace :rbs do
+  desc 'Clean generated RBS files'
+  task :clean do
+    sh('rm -rf sig/generate')
+    sh('rm -rf sig/generated-by-scripts')
+  end
+
+  desc 'Install rbs collection'
+  task :collection do
+    sh('bundle exec rbs collection install')
+  end
+
+  desc 'Run rbs-inline to generate RBS files'
+  task :inline do
+    sh('bundle exec rbs-inline --opt-out --output lib')
+  end
+
+  desc 'Generate RBS definitions by script/generate-rbs.rb'
+  task :script do
+    sh('script/generate-data-rbs.rb')
+    sh('script/generate-concern-rbs.rb')
+    sh('script/generate-memorized-ivar-rbs.rb')
+  end
+end
+
+task rbs: %i[rbs:inline rbs:script]

data/Steepfile

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+target :lib do
+  check 'lib'
+
+  signature 'sig'
+  signature 'sig-lib'
+
+  library 'json'
+  library 'net-http'
+  library 'optparse'
+end

data/bin/console

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require 'ruboty/ai_agent'
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require 'irb'
+IRB.start(__FILE__)

data/bin/ruboty

@@ -0,0 +1,34 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Run Ruboty with this gem for testing.
+#
+# Usage:
+#   bin/ruboty [options]
+#
+# Example:
+#   bin/ruboty --dotenv
+
+require 'bundler/inline'
+
+gemfile do
+  source 'https://rubygems.org'
+
+  gem 'ruboty'
+  gem 'ruboty-ai_agent', path: '../'
+
+  # You can add other ruboty plugins here for testing.
+  # gem 'ruboty-echo'
+  # gem 'ruboty-alias'
+  # gem 'ruboty-slack_events'
+end
+
+module Ruboty
+  class Robot # rubocop:disable Style/Documentation
+    def bundle
+      # noop
+    end
+  end
+end
+
+Ruboty::CommandBuilder.new(ARGV).build.call

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

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

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # AddAiCommand action for Ruboty::AiAgent
+      class AddAiCommand < Base
+        def call
+          message.reply("TODO: Implement AddAiCommand action - name: #{name_param}, prompt: #{prompt_param}")
+        end
+
+        def name_param #: String
+          message[:name]
+        end
+
+        def prompt_param #: String
+          message[:prompt]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # AddAiMemory action for Ruboty::AiAgent
+      class AddAiMemory < Base
+        def call
+          idx = user.ai_memories.add(prompt_param)
+
+          message.reply("Added memory #{idx}: #{prompt_param}")
+        end
+
+        def prompt_param #: String
+          message[:prompt]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+require 'optparse'
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # AddMcp action for Ruboty::AiAgent
+      class AddMcp < Base
+        def call
+          options = parse_config
+
+          case options[:transport]
+          when :http
+            url = options[:args].first #: String?
+            if url
+              new_mcp_configuration = McpConfiguration.new(
+                transport: :http,
+                name: name_param,
+                headers: options[:headers],
+                url: url
+              )
+              user.mcp_configurations.add(
+                new_mcp_configuration
+              )
+
+              message.reply("Added MCP configuration #{name_param}: #{new_mcp_configuration.to_h.except(:record_type).to_json}")
+            else
+              message.reply('Error: URL is required for HTTP transport. Please specify the URL as an argument.')
+              nil
+            end
+
+          when :sse
+            message.reply('Error: SSE transport is not yet implemented.')
+          else # steep:ignore UnreachableValueBranch
+            message.reply('Error: Invalid or missing transport type. Please specify --transport http or --transport sse.')
+          end
+        rescue OptionParser::InvalidOption, OptionParser::InvalidArgument => e
+          message.reply("Error parsing options: #{e.message}")
+        end
+
+        def name_param #: String
+          message[:name]
+        end
+
+        def config_param #: String
+          message[:config]
+        end
+
+        private
+
+        # @rbs! type config = { transport: :http | :sse, headers: Hash[String, String], args: Array[String] }
+
+        def parse_config #: config
+          options = {
+            transport: :http,
+            headers: {},
+            args: []
+          } #: config
+
+          args = config_param.split(/\s+(?=-)/)
+
+          parser = OptionParser.new do |opts|
+            opts.on('--transport TYPE', %w[http sse], 'Transport type (http or sse)') do |t|
+              options[:transport] = t
+            end
+
+            opts.on('--header VALUE', 'Add a header (can be specified multiple times)') do |h|
+              key, value = undump_string(h).split(':', 2).map!(&:strip)
+              unless key && value
+                message.reply("Warning: Invalid format for --header '#{h}'. Expected format is 'Key: Value'.")
+                next
+              end
+
+              options[:headers][key] = value
+            end
+          end
+
+          options[:args] = parser.parse(args)
+
+          options
+        end
+
+        # @rbs str: String
+        # @rbs return: String
+        def undump_string(str)
+          str.undump
+        rescue StandardError
+          str
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # Base class for actions.
+      # @abstract
+      class Base
+        attr_reader :message #: Ruboty::Message
+
+        # @rbs message: Ruboty::Message
+        # @rbs return: void
+        def self.call(message)
+          new(message).call
+        end
+
+        # @rbs return: void
+        def call
+          raise NotImplementedError
+        end
+
+        def initialize(message)
+          @message = message
+        end
+
+        # @rbs %a{memorized}
+        def database #: Ruboty::AiAgent::Database
+          @database ||= Ruboty::AiAgent::Database.new(message.robot.brain)
+        end
+
+        # @rbs %a{memorized}
+        def user #: Ruboty::AiAgent::User
+          @user ||= database.user(message.from_name)
+        end
+
+        # @rbs %a{memorized}
+        def chat_thread #: Ruboty::AiAgent::ChatThread
+          @chat_thread ||= database.chat_thread(message.from || 'default')
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+module Ruboty
+  module AiAgent
+    module Actions
+      # Chat action for Ruboty::AiAgent
+      class Chat < Base
+        CONVERSATION_KEY = :conversations
+
+        # @rbs override
+        def call
+          llm = LLM::OpenAI.new(
+            client: OpenAI::Client.new(
+              api_key: ENV.fetch('OPENAI_API_KEY', nil)
+            ),
+            model: ENV.fetch('OPENAI_MODEL', 'gpt-5-nano')
+          )
+
+          commands = Commands.builtins(message:, chat_thread:)
+          tools = McpClients.new(user.mcp_clients).available_tools
+
+          commands.each do |command|
+            return command.call if command.match?(body_param)
+          end
+
+          chat_thread.messages << ChatMessage.new(
+            role: :user,
+            content: body_param
+          )
+
+          agent = Agent.new(
+            llm:,
+            messages: chat_thread.messages.all_values,
+            tools:
+          )
+
+          agent.complete do |event|
+            case event[:type]
+            when :new_message
+              chat_thread.messages << event[:message]
+              message.reply(event[:message].content) if event[:message].content.length.positive?
+            when :tool_call
+              message.reply("Calling tool #{event[:tool].name} with arguments #{event[:tool_arguments]}",
+                            streaming: true)
+            when :tool_response
+              chat_thread.messages << event[:message]
+              message.reply("Tool response: #{event[:tool_response].slice(0..100)}")
+            end
+          end
+        rescue StandardError => e
+          if ENV['DEBUG']
+            message.reply("エラーが発生しました: #{e.full_message}")
+          else
+            message.reply("エラーが発生しました: #{e.message}")
+          end
+        end
+
+        def body_param #: String
+          message[:body]
+        end
+      end
+    end
+  end
+end