neh 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3cf60c744f72e9d315e2f5f05c3d4d5278b0861fa76cee05a0fef9bfb19b8abe
-  data.tar.gz: 6cf8a7ce75f55f56d745e6c980185a5ef9c688ffad6f02483a1b4af54e885bce
+  metadata.gz: c047eb9e56275718ff7a63a6975df93fb0500ec77f894584f44a7c343978dab2
+  data.tar.gz: 2d87c9c0f83b67bc6611bfc192d27ac45b55791e2101cb1acb75c7491db76632
 SHA512:
-  metadata.gz: 87c88a52b19c7fec314f7eecf28b5570574a012fb4f21673587d374d363482da883200bac0ec8315fe65f8ced90657d795aa8431b0b94dbb41e4eb11424ada47
-  data.tar.gz: 8b6dab176eff9508e8a9d64424badc473165c08fb93a1b351a8b9426759a75cb61aaff4353873345f0f6820a434a4e6ecc995c6e1d2f8bb472500a7c3261c0a6
+  metadata.gz: 4c5b963cc9a2b129c030dfb840d8916abfb06b61599734e9677318fb4c79d02acff16c2abe31d05edc0f6fd8111d9e8e16f5992bda88a4a0bc7fd5be9faa86e1
+  data.tar.gz: 0d319c601228111049f649741b5b7a80c0598b0841c990a4bb5aad0c4ca9a738943178b23e9db44b5cc9ee7f595707f368745bc41a62995f3357d71617fa1cd8

data/.rubocop.yml

@@ -1,8 +1,14 @@
 AllCops:
-  TargetRubyVersion: 3.0
+  TargetRubyVersion: <%= File.read(File.expand_path('../.ruby-version', __FILE__)).strip %>
+  NewCops: enable
 
+Metrics/MethodLength:
+  Max: 20
+Style/ClassAndModuleChildren:
+  Enabled: false
+Style/Documentation:
+  Enabled: false
 Style/StringLiterals:
-  EnforcedStyle: double_quotes
-
+  Enabled: false
 Style/StringLiteralsInInterpolation:
-  EnforcedStyle: double_quotes
+  Enabled: false

data/.ruby-version

@@ -0,0 +1 @@
+3.2.2

data/README.md

@@ -1,31 +1,6 @@
-# Neh
+# neh
 
-TODO: Delete this and the text below, and describe your gem
+Check connectivity from the client to the server.
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/neh`. To experiment with that code, run `bin/console` for an interactive prompt.
-
-## Installation
-
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
-
-Install the gem and add to the application's Gemfile by executing:
-
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
-
-If bundler is not being used to manage dependencies, install the gem by executing:
-
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
-
-## Usage
-
-TODO: Write usage instructions here
-
-## Development
-
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
-
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
-
-## Contributing
-
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/neh.
+    $ bundle exec neh o
+    Worker Progress ====================== 5/5 100.00%

data/Rakefile

@@ -1,11 +1,11 @@
 # frozen_string_literal: true
 
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
 
 RSpec::Core::RakeTask.new(:spec)
 
-require "rubocop/rake_task"
+require 'rubocop/rake_task'
 
 RuboCop::RakeTask.new
 

data/exe/neh

@@ -1,6 +1,6 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-require "neh"
+require "neh/thor"
 
-puts Neh.greet
+Neh::Thor.start(ARGV)

data/lib/neh/cli/o.rb

@@ -0,0 +1,116 @@
+# frozen_string_literal: true
+
+require 'securerandom'
+require 'net/http'
+require 'async'
+require 'async/io/stream'
+require 'async/http/endpoint'
+require 'async/websocket/client'
+require 'ruby-progressbar'
+require 'faraday'
+require 'pry'
+
+class Neh::Cli::O
+  def initialize(*args, options:)
+    @message = args.join(' ')
+
+    @options = options
+    server_host = ENV.fetch('NEH_SERVER_HOST', 'yoryo.gipcompany.com')
+    server_port = ENV.fetch('NEH_SERVER_PORT', 443)
+    @channel = 'LargeLanguageModelQueryChannel'
+
+    url = "ws://#{server_host}:#{server_port}/cable"
+    @endpoint = Async::HTTP::Endpoint.parse(url)
+  end
+
+  def execute
+    Async do |_task|
+      Async::WebSocket::Client.connect(@endpoint, headers: { 'Authorization' => "Bearer #{token}" }) do |connection|
+        while (message = connection.read)
+          parsed_message =
+            JSON.parse(message, symbolize_names: true)
+
+          on_receive(connection, parsed_message)
+        end
+      end
+    end
+  end
+
+  private
+
+  def token
+    ENV.fetch('NEH_PERSONAL_ACCESS_TOKEN')
+  end
+
+  def on_receive(connection, message)
+    if message[:type]
+      handle_connection_message(connection, message)
+    else
+      handle_channel_message(connection, message[:message])
+    end
+  end
+
+  def handle_connection_message(connection, message)
+    type = message[:type]
+
+    case type
+    when 'welcome'
+      subscribe(connection)
+    when 'confirm_subscription'
+      on_subscribed
+    end
+  end
+
+  def handle_channel_message(connection, message)
+    type = message[:type]
+
+    case type
+    when 'output'
+      print message[:message]
+    when 'worker_done'
+      puts ''
+      connection.close
+    end
+  end
+
+  def subscribe(connection)
+    content =
+      {
+        command: :subscribe,
+        identifier: {
+          channel: @channel
+        }.to_json
+      }
+    connection.write(content.to_json)
+    connection.flush
+  end
+
+  def on_subscribed
+    response = http_connection.post("/api/neh/o") do |req|
+      req.body = {
+        message: @message,
+        token:
+      }.to_json
+
+      req.headers['Content-Type'] = 'application/json'
+    end
+
+    body = JSON.parse response.body
+    puts body['message'] if debug_mode?
+  end
+
+  def http_connection
+    Faraday.new(url: "#{protocol}://#{@endpoint.hostname}:#{@endpoint.port}") do |faraday|
+      faraday.adapter Faraday.default_adapter
+      faraday.headers['Authorization'] = "Bearer #{token}"
+    end
+  end
+
+  def protocol
+    @endpoint.secure? ? 'https' : 'http'
+  end
+
+  def debug_mode?
+    ENV['NEH_DEBUG'] == 'true'
+  end
+end

data/lib/neh/cli.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+require 'neh'
+
+module Neh::Cli
+end
+
+require_relative 'cli/o'

data/lib/neh/thor.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require 'thor'
+
+require_relative 'cli'
+require_relative 'cli/o'
+
+class Neh::Thor < Thor
+  desc 'o', 'This is the most standard command you can use to query an LLM AI.'
+  def o(*)
+    Neh::Cli::O.new(*, options:).execute
+  end
+end

data/lib/neh/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Neh
-  VERSION = "0.0.2"
+  VERSION = "0.0.4"
 end

data/lib/neh.rb

@@ -2,15 +2,5 @@
 
 require_relative "neh/version"
 
-# Neh module provides a sample greet method.
 module Neh
-  # Error class is a standard error class used within the Neh module.
-  class Error < StandardError; end
-
-  # The `greet` method returns a greeting message from the Neh module.
-  #
-  # @return [String] Greeting message
-  def self.greet
-    "Hello from the neh gem!"
-  end
 end

metadata

@@ -1,15 +1,71 @@
 --- !ruby/object:Gem::Specification
 name: neh
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Atsushi Ishida
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-05-27 00:00:00.000000000 Z
-dependencies: []
+date: 2024-07-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: async-io
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.43.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.43.2
+- !ruby/object:Gem::Dependency
+  name: async-websocket
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.26.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.26.2
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.9.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.9.0
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
 description: Lightweight CLI client for interacting with AI using large language models
   (LLM)
 email:
@@ -21,11 +77,15 @@ extra_rdoc_files: []
 files:
 - ".rspec"
 - ".rubocop.yml"
+- ".ruby-version"
 - CHANGELOG.md
 - README.md
 - Rakefile
 - exe/neh
 - lib/neh.rb
+- lib/neh/cli.rb
+- lib/neh/cli/o.rb
+- lib/neh/thor.rb
 - lib/neh/version.rb
 - sig/neh.rbs
 homepage: https://github.com/gipcompany/neh
@@ -34,22 +94,23 @@ metadata:
   homepage_uri: https://github.com/gipcompany/neh
   source_code_uri: https://github.com/gipcompany/neh
   changelog_uri: https://github.com/gipcompany/neh/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">="
+  - - "~>"
     - !ruby/object:Gem::Version
-      version: 3.0.0
+      version: 3.2.2
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.10
+rubygems_version: 3.5.14
 signing_key:
 specification_version: 4
 summary: CLI client for AI with large language models