luogu 0.1.9 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8b6f8d477025b929343fbd70605905c557d33550d3f0f3921d256b295f993a96
-  data.tar.gz: 49aaf6803e6a8428bf143a743dd07ad840ddd0655a3228fad59543842af65466
+  metadata.gz: bf66040bab414493d1be24b2ed42780d2cefae0c79814b48b363bcc2ed094415
+  data.tar.gz: 4e1c53cb018a14ed2ca607d364717fd5f84a8ff7a7246c641e79cd6f17806ae7
 SHA512:
-  metadata.gz: 6de98f0d6be3266c9b4d873fb0e97bd1cf94f63231411961af39671a1be9b4ce0040d3efd1476fe37a309bee3722e1a7e724fdbd37a0dc949e4982fb8e8425d4
-  data.tar.gz: 0cc7c4f3f9fa709c7fd1c7ca1a3b10ede905e5be36b90701edd0f100b404e64144deb20de6f081d3f6b89c757932f2932651fb0b6c6f60ec938419dfde2d5855
+  metadata.gz: 9670b5c6bd270e449b2916313474c4a6ac0eafff47b550a9853e8a6b0a032375eb7d74bc29d68c431fbb8ffe2ba946cc458316aa0046fde514e710b7dc642d06
+  data.tar.gz: 30556b308a94306863f8f2f7e6b1ebf9d53214a3832644cb013dc99fb332e781e519e2dd6f52a99d6da367b9f9973c1b7123b5a5e524aaa4d1d4eb5015f3d0e4

data/README.md

@@ -62,9 +62,10 @@ end
 - 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)
+- 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)
+- OpenStruct.new(response: response)
 
 ## MIT 协议

data/lib/luogu/chatgpt.rb

@@ -1,7 +1,7 @@
 module Luogu
   class ChatGPT
 
-    attr_accessor :template, :limit_history, :prompt, :row_history, :history, :temperature, :model_name
+    attr_accessor :template, :limit_history, :prompt, :row_history, :history, :temperature, :model_name, :context
 
     def initialize(file, history_path='.', plugin_file_path=nil)
       @plugin_file_path = plugin_file_path || file.sub(File.extname(file), ".plugin.rb")
@@ -23,6 +23,8 @@ module Luogu
       @row_history = []
       @history = HistoryQueue.new @limit_history
 
+      @context = OpenStruct.new
+
       if @plugin.setup_proc
         @plugin.setup_proc.call(self, OpenStruct.new) 
       end
@@ -36,21 +38,25 @@ module Luogu
       }
       
       if @plugin.before_request_proc
-        params = @plugin.before_request_proc.call(self, OpenStruct.new(request_params: params)).request_params
+        @context.request_params = params
+        params = @plugin.before_request_proc.call(self, @context).request_params
       end
       response = client.chat(parameters: params)
-      @plugin.after_request_proc.call(self, OpenStruct.new(response: response)) if @plugin.after_request_proc
+      @context.response = response
+      @plugin.after_request_proc.call(self, @context) if @plugin.after_request_proc
 
       response.dig("choices", 0, "message", "content")
     end
 
     def chat(user_message)
       if @plugin.before_input_proc
-        user_message = @plugin.before_input_proc.call(self, OpenStruct.new(user_input: user_message)).user_input
+        @context.user_input = user_message
+        user_message = @plugin.before_input_proc.call(self, @context).user_input
       end
       messages = (@prompt.render + @history.to_a) << {role: "user", content: user_message}
       if @plugin.after_input_proc
-        messages = @plugin.after_input_proc.call(self, OpenStruct.new(request_messages: messages)).request_messages
+        @context.request_messages = messages
+        messages = @plugin.after_input_proc.call(self, @context).request_messages
       end
 
       assistant_message = self.request(messages)
@@ -61,7 +67,9 @@ 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, OpenStruct.new(user_input: user_message, response_message: assistant_message))
+        @context.user_input = user_message
+        @context.response_message = assistant_message
+        @plugin.before_save_history_proc.call(self, @context)
       else
         puts "执行默认的历史记录"
         self.push_history(user_message, assistant_message)
@@ -79,7 +87,9 @@ module Luogu
       @history.enqueue({role: "user", content: user_message})
       @history.enqueue({role: "assistant", content: assistant_message})
       if @plugin.after_save_history_proc
-        @plugin.after_save_history_proc.call(self, OpenStruct.new(user_input: user_message, response_message: assistant_message)) 
+        @context.user_input = user_message
+        @context.response_message = response_message
+        @plugin.after_save_history_proc.call(self, @context) 
       end
     end
 

data/lib/luogu/version.rb

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

metadata

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