neh 0.0.22 → 0.0.24

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a348fb883a6cb05766862daf8761222a2899ab9f6487296d490cdfb1dafee8ad
-  data.tar.gz: '04319db4bc44ab64d7efc0425ccaef06a08baeb960a4613abba4e558f24ee54d'
+  metadata.gz: acfcef0ee91e2a3a87f1618116449112811a1db27d457f4b822618e73823935c
+  data.tar.gz: 8573e2ac999dd18e5a5529793f548420d69ab70d8c98f8e29195354011c43222
 SHA512:
-  metadata.gz: 34e7279aa665e9782e2ae78626f36b4f7e94b4163e979a08746cc6fc8ca25af8845256b51739bb4d951a059e82dcbd53c326fe91ce3c51bd8a356d5eb80c729e
-  data.tar.gz: 10b7893ee7fca52b57c3a0872a9136d83258f3602e2fe2742bc22d792c482b2b5b8649dee26e841c67b6009b803bb4ebf2ade8e7dc0212c12149f47a98f73b9d
+  metadata.gz: d90e5b71167430ef4a4eb85c3a3024f207757a28593c71bfe398230e724cbf378755deea51921f73136031d1bbca6b7b03f7629baca83bc4fe711ca301fd51b3
+  data.tar.gz: 7918d5ffaa197c6c36d0e38d95efb1cf906bf4d8fed5c00e218cfac5e66c47dc55a690b088808ab3b10a0bc301783594f766216ba1fabd9ab5091455d315ea7a

data/lib/neh/cli/base.rb

@@ -6,28 +6,23 @@ require 'async'
 require 'async/io/stream'
 require 'async/http/endpoint'
 require 'async/websocket/client'
-require 'ruby-progressbar'
 require 'faraday'
 require 'pry' if ENV['NEH_DEBUG']
 require 'active_support/all'
 
 class Neh::Cli::Base # rubocop:disable Metrics/ClassLength
+  attr_reader :uuid
+
   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'
-    @uuid = SecureRandom.uuid
-    url = "ws://#{server_host}:#{server_port}/cable?uuid=#{@uuid}"
-    @endpoint = Async::HTTP::Endpoint.parse(url)
     @message_pool = {}
     @expected_sequence_number = 1
+    @uuid = SecureRandom.uuid
   end
 
   def execute
-    Async::WebSocket::Client.connect(@endpoint, headers: { 'Authorization' => "Bearer #{token}" }) do |connection|
+    Async::WebSocket::Client.connect(endpoint, headers: { 'Authorization' => "Bearer #{token}" }) do |connection|
       while (message = connection.read)
         parsed_message =
           JSON.parse(message, symbolize_names: true)
@@ -107,19 +102,23 @@ class Neh::Cli::Base # rubocop:disable Metrics/ClassLength
       {
         command: :subscribe,
         identifier: {
-          channel: @channel,
-          uuid: @uuid
+          channel:,
+          uuid:
         }.to_json
       }
     connection.write(content.to_json)
     connection.flush
   end
 
+  def channel
+    'LargeLanguageModelQueryChannel'
+  end
+
   def on_subscribed
     response = http_connection.post("/api/neh/#{command}") do |req|
       req.body = {
         message: @message,
-        uuid: @uuid,
+        uuid:,
         token:
       }.to_json
 
@@ -142,13 +141,20 @@ class Neh::Cli::Base # rubocop:disable Metrics/ClassLength
   end
 
   def http_connection
-    Faraday.new(url: "#{protocol}://#{@endpoint.hostname}:#{@endpoint.port}") do |faraday|
+    Faraday.new(url:) do |faraday|
       faraday.adapter Faraday.default_adapter
       faraday.headers['Authorization'] = "Bearer #{token}"
     end
   end
 
-  def protocol
-    @endpoint.secure? ? 'https' : 'http'
+  def endpoint
+    Async::HTTP::Endpoint.parse(
+      url,
+      alpn_protocols: Async::HTTP::Protocol::HTTP11.names #=> ["http/1.1"]
+    )
+  end
+
+  def url
+    ENV['WORKING_ON_LOCALHOST'].present? ? "http://localhost:6060/cable?uuid=#{uuid}" : "https://yoryo-app.onrender.com/cable?uuid=#{uuid}"
   end
 end

data/lib/neh/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: neh
 version: !ruby/object:Gem::Version
-  version: 0.0.22
+  version: 0.0.24
 platform: ruby
 authors:
 - Atsushi Ishida
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-19 00:00:00.000000000 Z
+date: 2024-08-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport