neh 0.0.18 → 0.0.20

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3a0072dc145f6f3c9465b0ab362284e8524e12714c9fe691e2b3409068053bb2
-  data.tar.gz: dc46224b988d300326cb4bde008d244651e67bf0b723194992217db54d7b2f8c
+  metadata.gz: 0567a7b6d4b3447f6101fa3862024751fe912d09c32159c2d60e4d4e59d6ecf1
+  data.tar.gz: 0e25bd53826dd8802f94a0708ea73cb00001c7d1f224acea9d8851f818ec2940
 SHA512:
-  metadata.gz: 53cec3325e390fc4b6d480139178a97f8fd9b3cf656aabe1b88b033091573fd2b59914ac2ddc69596547588b9fe31eff9c5431d08df55e921d4b0fc971d30418
-  data.tar.gz: 5a6e101c210e48c84a8c29957e46a761c928e00b3a0a50cd62be7b5f803af195d9f77148e3e766e476a906b348d8f1eedacadc283d5a6adb6eeaa9b444ccb786
+  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'
       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.18"
+  VERSION = "0.0.20"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: neh
 version: !ruby/object:Gem::Version
-  version: 0.0.18
+  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