neh 0.0.10 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c27c44b5820241a8c0458fbf980dc4e815d1b68d144b1c5f34499e7ac7f82938
-  data.tar.gz: 91bff8dfbee7464c4d88ca05ace0abb862c2224d977fe1dde66800898fa69406
+  metadata.gz: 9ee23f077b2167f86dc40dcf5d0f2c1d8d04f806ebf883cb4b926778a549d37c
+  data.tar.gz: 95cf5614fa37ae4abcbddcfdca91bd364580c2aaf4fe73c99d0fdac13799009e
 SHA512:
-  metadata.gz: dadbf680339bbd0bcb496c4e6b258cbafe64e6729d9742447f4334196dc35d15201cf7dbbf5c8820f7ea7eb08b3aacb36420a855e1e68eab459364f364f0be38
-  data.tar.gz: 48ade2299fad19e7b6bd6b5218de2ce0e26499b281d8acfb43e3334f5211388a48ee9d848738d2488ca966d289fe6ae07ac9d706fd2d6bc943d7b67713e67e96
+  metadata.gz: f0348653539a6f078c28a17544b5a0ca9119458457774937b1e6cbcf5ec59a8038a36992f03885f0fca7773ad1e975a7241c25471ead5055495ff453cebeeb1e
+  data.tar.gz: f4c05d264f7ec7669d846e83eea18680e0178b0830f5ecfdce5cf0a1898c5cb1190e4bcb3f65b7692a7da18d4df2553977359d4a38d55ccf64cca16599bbe05a

data/lib/neh/cli/base.rb

@@ -0,0 +1,126 @@
+# 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::Base
+  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/#{command}") 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 command
+    raise NotImplementedError if instance_of?(Neh::Cli::Base)
+
+    unless self.class.ancestors.include?(Neh::Cli::Base)
+      raise NotImplementedError, 'command method must be implemented in subclasses'
+    end
+
+    self.class.name.split('::').last.downcase
+  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/decache.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module Neh::Cli
+  class Decache < Base
+  end
+end

data/lib/neh/cli/o.rb

@@ -1,116 +1,6 @@
 # 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'
+module Neh::Cli
+  class O < Base
   end
 end

data/lib/neh/cli/version.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module Neh
+  module Cli
+    class Version
+      def execute
+        puts "neh version #{Neh::VERSION}"
+      end
+    end
+  end
+end

data/lib/neh/cli.rb

@@ -5,4 +5,6 @@ require 'neh'
 module Neh::Cli
 end
 
+require_relative 'cli/base'
 require_relative 'cli/o'
+require_relative 'cli/decache'

data/lib/neh/thor.rb

@@ -4,8 +4,21 @@ require 'thor'
 
 require_relative 'cli'
 require_relative 'cli/o'
+require_relative 'cli/version'
 
 class Neh::Thor < Thor
+  map %w[--version -v] => :version
+
+  desc 'version', 'Displays the current version of the application.'
+  def version
+    Neh::Cli::Version.new.execute
+  end
+
+  desc 'decache', 'Deletes the session cache with the LLM AI so far.'
+  def decache(*)
+    Neh::Cli::Decache.new(*, options:).execute
+  end
+
   desc 'o', 'This is the most standard command you can use to query an LLM AI.'
   def o(*)
     Neh::Cli::O.new(*, options:).execute

data/lib/neh/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: neh
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.0.12
 platform: ruby
 authors:
 - Atsushi Ishida
-autorequire:
+autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-01 00:00:00.000000000 Z
+date: 2024-08-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async-io
@@ -84,7 +84,10 @@ files:
 - exe/neh
 - lib/neh.rb
 - lib/neh/cli.rb
+- lib/neh/cli/base.rb
+- lib/neh/cli/decache.rb
 - lib/neh/cli/o.rb
+- lib/neh/cli/version.rb
 - lib/neh/thor.rb
 - lib/neh/version.rb
 - sig/neh.rbs
@@ -95,7 +98,7 @@ metadata:
   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:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -110,8 +113,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.14
-signing_key:
+rubygems_version: 3.4.10
+signing_key: 
 specification_version: 4
 summary: CLI client for AI with large language models
 test_files: []