luogu 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d415e44469be65939709764b50b5283da141d5e68663acf5e3c376044407970
-  data.tar.gz: 18846bbcd392e9581f6d9becff711b2e551983759460464b169824e14f3c05a1
+  metadata.gz: 8b6f8d477025b929343fbd70605905c557d33550d3f0f3921d256b295f993a96
+  data.tar.gz: 49aaf6803e6a8428bf143a743dd07ad840ddd0655a3228fad59543842af65466
 SHA512:
-  metadata.gz: 5eb13f0a6b90843366879a773f816d55dbcdba55f2cc6f06915f9e7f7dbc4a73ceb1916505d482a670755d8197519792d649f0731c12f53071fa87b131bcd62a
-  data.tar.gz: b25c5d0dfa8a2b4d4df66ec9695d226521551ddb77f674722d122ba8e9252546abcfa916d2d379a794f520fedf32247e1dcbeef575739675bdd84ddd705e78f5
+  metadata.gz: 6de98f0d6be3266c9b4d873fb0e97bd1cf94f63231411961af39671a1be9b4ce0040d3efd1476fe37a309bee3722e1a7e724fdbd37a0dc949e4982fb8e8425d4
+  data.tar.gz: 0cc7c4f3f9fa709c7fd1c7ca1a3b10ede905e5be36b90701edd0f100b404e64144deb20de6f081d3f6b89c757932f2932651fb0b6c6f60ec938419dfde2d5855

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    luogu (0.1.7)
+    luogu (0.1.9)
       dotenv (~> 2.8, >= 2.8.1)
       dry-cli (~> 1.0)
       ruby-openai (~> 3.7)

data/README.md

@@ -61,4 +61,10 @@ end
 - history 查看当前上下文
 - exit 退出
 
+## 插件模式
+OpenStruct.new(request_params: params)
+OpenStruct.new(user_input: user_message)
+OpenStruct.new(request_messages: messages)
+OpenStruct.new(user_input: user_message, response_message: assistant_message)
+
 ## MIT 协议

data/lib/luogu/chatgpt.rb

@@ -23,7 +23,9 @@ module Luogu
       @row_history = []
       @history = HistoryQueue.new @limit_history
 
-      @plugin.setup_proc.call(self) if @plugin.setup_proc
+      if @plugin.setup_proc
+        @plugin.setup_proc.call(self, OpenStruct.new) 
+      end
     end
 
     def request(messages)
@@ -33,17 +35,23 @@ module Luogu
         temperature: @temperature,
       }
       
-      params = @plugin.before_request_proc.call(self, params) if @plugin.before_request_proc
+      if @plugin.before_request_proc
+        params = @plugin.before_request_proc.call(self, OpenStruct.new(request_params: params)).request_params
+      end
       response = client.chat(parameters: params)
-      @plugin.after_request_proc.call(self, response) if @plugin.after_request_proc
+      @plugin.after_request_proc.call(self, OpenStruct.new(response: response)) if @plugin.after_request_proc
 
       response.dig("choices", 0, "message", "content")
     end
 
     def chat(user_message)
-      user_message = @plugin.before_input_proc.call(self, user_message) if @plugin.before_input_proc
+      if @plugin.before_input_proc
+        user_message = @plugin.before_input_proc.call(self, OpenStruct.new(user_input: user_message)).user_input
+      end
       messages = (@prompt.render + @history.to_a) << {role: "user", content: user_message}
-      @plugin.after_input_proc.call(self, messages) if @plugin.after_input_proc
+      if @plugin.after_input_proc
+        messages = @plugin.after_input_proc.call(self, OpenStruct.new(request_messages: messages)).request_messages
+      end
 
       assistant_message = self.request(messages)
       
@@ -53,7 +61,7 @@ module Luogu
         puts "执行文档中的callback"
         instance_eval @prompt.ruby_code, @prompt.file_path, @prompt.ruby_code_line
       elsif @plugin.before_save_history_proc
-        @plugin.before_save_history_proc.call(self, user_message, assistant_message)
+        @plugin.before_save_history_proc.call(self, OpenStruct.new(user_input: user_message, response_message: assistant_message))
       else
         puts "执行默认的历史记录"
         self.push_history(user_message, assistant_message)
@@ -70,7 +78,9 @@ module Luogu
     def push_history(user_message, assistant_message)
       @history.enqueue({role: "user", content: user_message})
       @history.enqueue({role: "assistant", content: assistant_message})
-      @plugin.after_save_history_proc.call(self, user_message, assistant_message) if @plugin.after_save_history_proc
+      if @plugin.after_save_history_proc
+        @plugin.after_save_history_proc.call(self, OpenStruct.new(user_input: user_message, response_message: assistant_message)) 
+      end
     end
 
     def ask(message)

data/lib/luogu/init.rb

@@ -5,6 +5,7 @@ require 'json'
 require 'yaml'
 require "dry/cli"
 require 'fileutils'
+require 'ostruct'
 
 require_relative 'plugin'
 require_relative 'history_queue'

data/lib/luogu/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Luogu
-  VERSION = "0.1.8"
+  VERSION = "0.1.9"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: luogu
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - MJ