girb-ruby_llm 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: a587322ce50087756c3a8e0178062385ea59f83737fb8b9c70816777024b230a
+  data.tar.gz: 165288aad6307d5e658abc4f0385409752bab2f35b2e0cc9fe81f9a0f0fac05d
+SHA512:
+  metadata.gz: 15a8df02dcec847a6c884d7ffc099cb9c9a7bfa484d210b8796102ef60e6a2029828f1efdfe343fefec2196f889fb467eb2d0921cd28468b4d27f317a6a72645
+  data.tar.gz: 2cded93009f453527579f0cf9b1ba9c356760bb3f6dabb63b7aea122172a12d6454294d0a7b9a5f3d1972a8f92f65871e174dbf744d3ed19ffa36984d4308f3c

data/CHANGELOG.md

@@ -0,0 +1,11 @@
+# Changelog
+
+## [0.1.0] - 2025-02-02
+
+### Added
+
+- Initial release
+- RubyLLM provider for girb
+- Support for multiple LLM providers: OpenAI, Anthropic, Gemini, Ollama, DeepSeek, Mistral, and more
+- Auto-configuration via environment variables
+- Function calling support

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Rira
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README.md

@@ -0,0 +1,149 @@
+# girb-ruby_llm
+
+[日本語版 README](README_ja.md)
+
+RubyLLM provider for [girb](https://github.com/rira100000000/girb) (AI-powered IRB assistant).
+
+This gem allows you to use multiple LLM providers (OpenAI, Anthropic, Google Gemini, Ollama, and more) through the [RubyLLM](https://github.com/crmne/ruby_llm) unified API.
+
+## Installation
+
+Add to your Gemfile:
+
+```ruby
+gem 'girb'
+gem 'girb-ruby_llm'
+```
+
+Or install directly:
+
+```bash
+gem install girb girb-ruby_llm
+```
+
+## Setup
+
+Set the provider and your API key:
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export GEMINI_API_KEY=your-api-key  # or other provider's API key
+```
+
+Then start girb:
+
+```bash
+girb
+```
+
+### Using with regular irb
+
+Add to your `~/.irbrc`:
+
+```ruby
+require 'girb-ruby_llm'
+```
+
+Then use regular `irb` command.
+
+## Configuration
+
+Set your API key or endpoint as an environment variable:
+
+### Cloud Providers
+
+| Provider | Environment Variable |
+|----------|---------------------|
+| Anthropic | `ANTHROPIC_API_KEY` |
+| OpenAI | `OPENAI_API_KEY` |
+| Google Gemini | `GEMINI_API_KEY` |
+| DeepSeek | `DEEPSEEK_API_KEY` |
+| Mistral | `MISTRAL_API_KEY` |
+| OpenRouter | `OPENROUTER_API_KEY` |
+| Perplexity | `PERPLEXITY_API_KEY` |
+| xAI | `XAI_API_KEY` |
+
+### Local Providers
+
+| Provider | Environment Variable |
+|----------|---------------------|
+| Ollama | `OLLAMA_API_BASE` |
+| GPUStack | `GPUSTACK_API_BASE` |
+
+### Additional Configuration
+
+| Environment Variable | Description |
+|---------------------|-------------|
+| `OPENAI_API_BASE` | Custom OpenAI-compatible API endpoint |
+| `GEMINI_API_BASE` | Custom Gemini API endpoint |
+| `GPUSTACK_API_KEY` | GPUStack API key |
+| `BEDROCK_API_KEY` | AWS Bedrock access key |
+| `BEDROCK_SECRET_KEY` | AWS Bedrock secret key |
+| `BEDROCK_REGION` | AWS Bedrock region |
+| `VERTEXAI_PROJECT_ID` | Google Vertex AI project ID |
+| `VERTEXAI_LOCATION` | Google Vertex AI location |
+
+## Examples
+
+### Using OpenAI
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export OPENAI_API_KEY="sk-..."
+girb
+```
+
+### Using Anthropic Claude
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export ANTHROPIC_API_KEY="sk-ant-..."
+girb
+```
+
+### Using Ollama (Local)
+
+```bash
+# Start Ollama first
+ollama serve
+
+# Set the provider and API base URL
+export GIRB_PROVIDER=girb-ruby_llm
+export OLLAMA_API_BASE="http://localhost:11434/v1"
+girb
+```
+
+### Using OpenAI-compatible APIs (e.g., LM Studio, vLLM)
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export OPENAI_API_KEY="not-needed"  # Some require any non-empty value
+export OPENAI_API_BASE="http://localhost:1234/v1"
+girb
+```
+
+### Manual Configuration
+
+You can configure the provider manually in your `~/.irbrc`:
+
+```ruby
+# ~/.irbrc
+require 'girb-ruby_llm'
+
+RubyLLM.configure do |config|
+  config.ollama_api_base = "http://localhost:11434/v1"
+end
+RubyLLM::Models.refresh!
+
+Girb.configure do |c|
+  c.provider = Girb::Providers::RubyLlm.new(model: 'llama3.2:latest')
+end
+```
+
+## Supported Models
+
+See [RubyLLM Available Models](https://rubyllm.com/reference/available-models) for the full list of supported models.
+
+## License
+
+MIT License

data/README_ja.md

@@ -0,0 +1,147 @@
+# girb-ruby_llm
+
+[girb](https://github.com/rira100000000/girb)（AI搭載IRBアシスタント）用のRubyLLMプロバイダーです。
+
+このgemを使用すると、[RubyLLM](https://github.com/crmne/ruby_llm)の統一APIを通じて、複数のLLMプロバイダー（OpenAI、Anthropic、Google Gemini、Ollamaなど）を利用できます。
+
+## インストール
+
+Gemfileに追加:
+
+```ruby
+gem 'girb'
+gem 'girb-ruby_llm'
+```
+
+または直接インストール:
+
+```bash
+gem install girb girb-ruby_llm
+```
+
+## セットアップ
+
+プロバイダーとAPIキーを設定:
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export GEMINI_API_KEY=your-api-key  # または他のプロバイダーのAPIキー
+```
+
+girbを起動:
+
+```bash
+girb
+```
+
+### 通常のirbで使用する場合
+
+`~/.irbrc` に追加:
+
+```ruby
+require 'girb-ruby_llm'
+```
+
+通常の `irb` コマンドで使用できます。
+
+## 設定
+
+APIキーまたはエンドポイントを環境変数として設定します:
+
+### クラウドプロバイダー
+
+| プロバイダー | 環境変数 |
+|-------------|---------|
+| Anthropic | `ANTHROPIC_API_KEY` |
+| OpenAI | `OPENAI_API_KEY` |
+| Google Gemini | `GEMINI_API_KEY` |
+| DeepSeek | `DEEPSEEK_API_KEY` |
+| Mistral | `MISTRAL_API_KEY` |
+| OpenRouter | `OPENROUTER_API_KEY` |
+| Perplexity | `PERPLEXITY_API_KEY` |
+| xAI | `XAI_API_KEY` |
+
+### ローカルプロバイダー
+
+| プロバイダー | 環境変数 |
+|-------------|---------|
+| Ollama | `OLLAMA_API_BASE` |
+| GPUStack | `GPUSTACK_API_BASE` |
+
+### 追加設定
+
+| 環境変数 | 説明 |
+|---------|------|
+| `OPENAI_API_BASE` | カスタムOpenAI互換APIエンドポイント |
+| `GEMINI_API_BASE` | カスタムGemini APIエンドポイント |
+| `GPUSTACK_API_KEY` | GPUStack APIキー |
+| `BEDROCK_API_KEY` | AWS Bedrockアクセスキー |
+| `BEDROCK_SECRET_KEY` | AWS Bedrockシークレットキー |
+| `BEDROCK_REGION` | AWS Bedrockリージョン |
+| `VERTEXAI_PROJECT_ID` | Google Vertex AIプロジェクトID |
+| `VERTEXAI_LOCATION` | Google Vertex AIロケーション |
+
+## 使用例
+
+### OpenAIを使用
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export OPENAI_API_KEY="sk-..."
+girb
+```
+
+### Anthropic Claudeを使用
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export ANTHROPIC_API_KEY="sk-ant-..."
+girb
+```
+
+### Ollama（ローカル）を使用
+
+```bash
+# まずOllamaを起動
+ollama serve
+
+# プロバイダーとAPIベースURLを設定
+export GIRB_PROVIDER=girb-ruby_llm
+export OLLAMA_API_BASE="http://localhost:11434/v1"
+girb
+```
+
+### OpenAI互換API（LM Studio、vLLMなど）を使用
+
+```bash
+export GIRB_PROVIDER=girb-ruby_llm
+export OPENAI_API_KEY="not-needed"  # 空でない値が必要な場合
+export OPENAI_API_BASE="http://localhost:1234/v1"
+girb
+```
+
+### 手動設定
+
+`~/.irbrc` でプロバイダーを手動で設定できます:
+
+```ruby
+# ~/.irbrc
+require 'girb-ruby_llm'
+
+RubyLLM.configure do |config|
+  config.ollama_api_base = "http://localhost:11434/v1"
+end
+RubyLLM::Models.refresh!
+
+Girb.configure do |c|
+  c.provider = Girb::Providers::RubyLlm.new(model: 'llama3.2:latest')
+end
+```
+
+## 対応モデル
+
+サポートされているモデルの完全なリストは[RubyLLM Available Models](https://rubyllm.com/reference/available-models)を参照してください。
+
+## ライセンス
+
+MIT License

data/Rakefile

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"

data/girb-ruby_llm.gemspec

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require_relative "lib/girb-ruby_llm/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "girb-ruby_llm"
+  spec.version = GirbRubyLlm::VERSION
+  spec.authors = ["rira100000000"]
+  spec.email = ["101010hayakawa@gmail.com"]
+
+  spec.summary = "RubyLLM provider for girb"
+  spec.description = "RubyLLM provider for girb (AI-powered IRB assistant). " \
+                     "Install this gem to use OpenAI, Anthropic, Gemini and other LLMs via RubyLLM as your LLM backend."
+  spec.homepage = "https://github.com/rira100000000/girb-ruby_llm"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 3.2.0"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = spec.homepage
+  spec.metadata["changelog_uri"] = "#{spec.homepage}/blob/main/CHANGELOG.md"
+
+  spec.files = Dir.chdir(__dir__) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (File.expand_path(f) == __FILE__) ||
+        f.start_with?(*%w[bin/ test/ spec/ features/ .git .github appveyor Gemfile])
+    end
+  end
+  spec.require_paths = ["lib"]
+
+  # Runtime dependencies
+  spec.add_dependency "girb", "~> 0.1"
+  spec.add_dependency "ruby_llm", "~> 1.0"
+end

data/lib/girb/providers/ruby_llm.rb

@@ -0,0 +1,142 @@
+# frozen_string_literal: true
+
+require "ruby_llm"
+require "girb/providers/base"
+
+module Girb
+  module Providers
+    class RubyLlm < Base
+      def initialize(model: nil)
+        @model = model
+      end
+
+      def chat(messages:, system_prompt:, tools:)
+        # Use specified model or RubyLLM's default
+        chat_options = @model ? { model: @model } : {}
+        ruby_llm_chat = ::RubyLLM.chat(**chat_options)
+
+        # Set system prompt
+        ruby_llm_chat.with_instructions(system_prompt) if system_prompt && !system_prompt.empty?
+
+        # Add tools
+        tool_instances = build_tools(tools)
+        tool_instances.each { |tool| ruby_llm_chat.with_tool(tool) }
+
+        # Add messages except the last user message
+        add_messages_to_chat(ruby_llm_chat, messages[0..-2])
+
+        # Get the last user message
+        last_message = messages.last
+        last_content = extract_content(last_message)
+
+        # Send the request
+        response = ruby_llm_chat.ask(last_content)
+
+        parse_response(response)
+      rescue Faraday::BadRequestError => e
+        Response.new(error: "API Error: #{e.message}")
+      rescue StandardError => e
+        Response.new(error: "Error: #{e.message}")
+      end
+
+      private
+
+      def add_messages_to_chat(chat, messages)
+        messages.each do |msg|
+          case msg[:role]
+          when :user
+            chat.add_message(role: :user, content: msg[:content])
+          when :assistant
+            chat.add_message(role: :assistant, content: msg[:content])
+          when :tool_call
+            # Add as assistant message with tool_calls
+            chat.add_message(
+              role: :assistant,
+              content: nil,
+              tool_calls: {
+                msg[:name] => ::RubyLLM::ToolCall.new(
+                  id: msg[:id] || "call_#{SecureRandom.hex(12)}",
+                  name: msg[:name],
+                  arguments: msg[:args]
+                )
+              }
+            )
+          when :tool_result
+            chat.add_message(
+              role: :tool,
+              content: msg[:result].to_s,
+              tool_call_id: msg[:id] || "call_#{SecureRandom.hex(12)}"
+            )
+          end
+        end
+      end
+
+      def extract_content(message)
+        case message[:role]
+        when :user, :assistant
+          message[:content]
+        else
+          message[:content].to_s
+        end
+      end
+
+      def build_tools(tools)
+        return [] if tools.nil? || tools.empty?
+
+        tools.map do |tool|
+          create_dynamic_tool(tool)
+        end
+      end
+
+      def create_dynamic_tool(tool_def)
+        tool_name = tool_def[:name]
+        tool_description = tool_def[:description]
+        tool_parameters = tool_def[:parameters] || {}
+
+        # Create a dynamic tool class
+        tool_class = Class.new(::RubyLLM::Tool) do
+          description tool_description
+
+          # Define parameters
+          properties = tool_parameters[:properties] || {}
+          required_params = tool_parameters[:required] || []
+
+          properties.each do |prop_name, prop_def|
+            param prop_name.to_sym,
+                  type: prop_def[:type] || "string",
+                  desc: prop_def[:description],
+                  required: required_params.include?(prop_name.to_s) || required_params.include?(prop_name)
+          end
+
+          # Override name method to return the custom name
+          define_method(:name) { tool_name }
+
+          # Execute method (never actually called, just for tool definition)
+          define_method(:execute) { |**_args| "" }
+        end
+
+        tool_class.new
+      end
+
+      def parse_response(response)
+        return Response.new(error: "No response") unless response
+
+        text = response.content.is_a?(String) ? response.content : response.content.to_s
+        function_calls = parse_function_calls(response)
+
+        Response.new(text: text, function_calls: function_calls, raw_response: response)
+      end
+
+      def parse_function_calls(response)
+        return [] unless response.tool_call?
+
+        response.tool_calls.map do |_id, tool_call|
+          {
+            name: tool_call.name.to_s,
+            args: tool_call.arguments || {}
+          }
+        end
+      end
+    end
+  end
+end

data/lib/girb-ruby_llm/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module GirbRubyLlm
+  VERSION = "0.1.0"
+end

data/lib/girb-ruby_llm.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require "girb"
+require_relative "girb/providers/ruby_llm"
+
+# Environment variable to RubyLLM config mapping
+RUBY_LLM_CONFIG_MAP = {
+  # Cloud providers (API key based)
+  "ANTHROPIC_API_KEY" => :anthropic_api_key,
+  "OPENAI_API_KEY" => :openai_api_key,
+  "GEMINI_API_KEY" => :gemini_api_key,
+  "DEEPSEEK_API_KEY" => :deepseek_api_key,
+  "MISTRAL_API_KEY" => :mistral_api_key,
+  "OPENROUTER_API_KEY" => :openrouter_api_key,
+  "PERPLEXITY_API_KEY" => :perplexity_api_key,
+  "XAI_API_KEY" => :xai_api_key,
+  # Local providers (URL based)
+  "OLLAMA_API_BASE" => :ollama_api_base,
+  "GPUSTACK_API_BASE" => :gpustack_api_base,
+  "GPUSTACK_API_KEY" => :gpustack_api_key,
+  # Additional configs
+  "OPENAI_API_BASE" => :openai_api_base,
+  "GEMINI_API_BASE" => :gemini_api_base,
+  "BEDROCK_API_KEY" => :bedrock_api_key,
+  "BEDROCK_SECRET_KEY" => :bedrock_secret_key,
+  "BEDROCK_REGION" => :bedrock_region,
+  "BEDROCK_SESSION_TOKEN" => :bedrock_session_token,
+  "VERTEXAI_PROJECT_ID" => :vertexai_project_id,
+  "VERTEXAI_LOCATION" => :vertexai_location
+}.freeze
+
+# Check if any RubyLLM config is available
+has_config = RUBY_LLM_CONFIG_MAP.any? { |env_var, _| ENV[env_var] }
+
+if has_config
+  # Configure RubyLLM with all available environment variables
+  RubyLLM.configure do |config|
+    RUBY_LLM_CONFIG_MAP.each do |env_var, config_key|
+      config.send("#{config_key}=", ENV[env_var]) if ENV[env_var]
+    end
+  end
+
+  Girb.configure do |c|
+    unless c.provider
+      c.provider = Girb::Providers::RubyLlm.new(model: c.model)
+    end
+  end
+end

metadata

@@ -0,0 +1,81 @@
+--- !ruby/object:Gem::Specification
+name: girb-ruby_llm
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- rira100000000
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: girb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: ruby_llm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+description: RubyLLM provider for girb (AI-powered IRB assistant). Install this gem
+  to use OpenAI, Anthropic, Gemini and other LLMs via RubyLLM as your LLM backend.
+email:
+- 101010hayakawa@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE
+- README.md
+- README_ja.md
+- Rakefile
+- girb-ruby_llm.gemspec
+- lib/girb-ruby_llm.rb
+- lib/girb-ruby_llm/version.rb
+- lib/girb/providers/ruby_llm.rb
+homepage: https://github.com/rira100000000/girb-ruby_llm
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/rira100000000/girb-ruby_llm
+  source_code_uri: https://github.com/rira100000000/girb-ruby_llm
+  changelog_uri: https://github.com/rira100000000/girb-ruby_llm/blob/main/CHANGELOG.md
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.7.2
+specification_version: 4
+summary: RubyLLM provider for girb
+test_files: []