neh 0.0.19 → 0.0.20

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0c98adee45058fca1fd2d8ad4d4b986e9273d053336e380fa9cdc3e745bc2d25
-  data.tar.gz: a2a97a95699a29cde7e0374be428613d34caf49cbba52610917f3e147bf32798
+  metadata.gz: 0567a7b6d4b3447f6101fa3862024751fe912d09c32159c2d60e4d4e59d6ecf1
+  data.tar.gz: 0e25bd53826dd8802f94a0708ea73cb00001c7d1f224acea9d8851f818ec2940
 SHA512:
-  metadata.gz: 02773dbfef6e5f43f48fd77d7f71587a5c3e8b811e4acbeaa47d529504257f714f5c0952e8e515bd8f90c359058fb0c4f591f54cab37ddc5f961a0bb99cafe12
-  data.tar.gz: 48f77eebdb4b7c6b94ea5a449c39db3e56465c916d37cfd7084b46386a767302c7157b9921375742603b224abd454e13fbf0978e315f67ec238321526f6af937
+  metadata.gz: 31e4294dd35df3193358e93b632c2c7746590670a21367efeb60fbfd3fa53553500a3906aa725fb87f8b6b12d8559a164834b4c431f05e145b2d99caf4173a16
+  data.tar.gz: 75779ccf1bdc6e51f25a25348741f130748094abcc71db3199a4ab170f3910dbd9daa423c8f864896be86b5418dceb5b53c138b12e9479db5cff41c4c6d848e2

data/lib/neh/cli/base.rb

@@ -11,7 +11,7 @@ require 'faraday'
 require 'pry' if ENV['NEH_DEBUG']
 require 'active_support/all'
 
-class Neh::Cli::Base
+class Neh::Cli::Base # rubocop:disable Metrics/ClassLength
   def initialize(*args, options:)
     @message = args.join(' ')
 
@@ -19,8 +19,8 @@ class Neh::Cli::Base
     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"
+    @uuid = SecureRandom.uuid
+    url = "ws://#{server_host}:#{server_port}/cable?uuid=#{@uuid}"
     @endpoint = Async::HTTP::Endpoint.parse(url)
   end
 
@@ -56,11 +56,15 @@ class Neh::Cli::Base
   def handle_connection_message(connection, message)
     type = message[:type]
 
+    print "type: #{type}, message: #{message}" if ENV['NEH_DEBUG']
+
     case type
     when 'welcome'
       subscribe(connection)
-    when 'confirm_subscription', 'ping'
+    when 'confirm_subscription'
       on_subscribed
+    when 'ping'
+      # do nothing
     when 'disconnect'
       puts "Connection has been disconnected. Reason: #{message[:reason]}"
       close(connection)
@@ -90,7 +94,8 @@ class Neh::Cli::Base
       {
         command: :subscribe,
         identifier: {
-          channel: @channel
+          channel: @channel,
+          uuid: @uuid
         }.to_json
       }
     connection.write(content.to_json)
@@ -101,6 +106,7 @@ class Neh::Cli::Base
     response = http_connection.post("/api/neh/#{command}") do |req|
       req.body = {
         message: @message,
+        uuid: @uuid,
         token:
       }.to_json
 

data/lib/neh/version.rb

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

metadata

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