luogu 0.1.8 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d415e44469be65939709764b50b5283da141d5e68663acf5e3c376044407970
-  data.tar.gz: 18846bbcd392e9581f6d9becff711b2e551983759460464b169824e14f3c05a1
+  metadata.gz: bf66040bab414493d1be24b2ed42780d2cefae0c79814b48b363bcc2ed094415
+  data.tar.gz: 4e1c53cb018a14ed2ca607d364717fd5f84a8ff7a7246c641e79cd6f17806ae7
 SHA512:
-  metadata.gz: 5eb13f0a6b90843366879a773f816d55dbcdba55f2cc6f06915f9e7f7dbc4a73ceb1916505d482a670755d8197519792d649f0731c12f53071fa87b131bcd62a
-  data.tar.gz: b25c5d0dfa8a2b4d4df66ec9695d226521551ddb77f674722d122ba8e9252546abcfa916d2d379a794f520fedf32247e1dcbeef575739675bdd84ddd705e78f5
+  metadata.gz: 9670b5c6bd270e449b2916313474c4a6ac0eafff47b550a9853e8a6b0a032375eb7d74bc29d68c431fbb8ffe2ba946cc458316aa0046fde514e710b7dc642d06
+  data.tar.gz: 30556b308a94306863f8f2f7e6b1ebf9d53214a3832644cb013dc99fb332e781e519e2dd6f52a99d6da367b9f9973c1b7123b5a5e524aaa4d1d4eb5015f3d0e4

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,11 @@ 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)
+- 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,7 +23,11 @@ module Luogu
       @row_history = []
       @history = HistoryQueue.new @limit_history
 
-      @plugin.setup_proc.call(self) if @plugin.setup_proc
+      @context = OpenStruct.new
+
+      if @plugin.setup_proc
+        @plugin.setup_proc.call(self, OpenStruct.new) 
+      end
     end
 
     def request(messages)
@@ -33,17 +37,27 @@ module Luogu
         temperature: @temperature,
       }
       
-      params = @plugin.before_request_proc.call(self, params) if @plugin.before_request_proc
+      if @plugin.before_request_proc
+        @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, 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)
-      user_message = @plugin.before_input_proc.call(self, user_message) if @plugin.before_input_proc
+      if @plugin.before_input_proc
+        @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}
-      @plugin.after_input_proc.call(self, messages) if @plugin.after_input_proc
+      if @plugin.after_input_proc
+        @context.request_messages = messages
+        messages = @plugin.after_input_proc.call(self, @context).request_messages
+      end
 
       assistant_message = self.request(messages)
       
@@ -53,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, user_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)
@@ -70,7 +86,11 @@ 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
+        @context.user_input = user_message
+        @context.response_message = response_message
+        @plugin.after_save_history_proc.call(self, @context) 
+      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.10"
 end

metadata

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