luogu 0.1.9 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8b6f8d477025b929343fbd70605905c557d33550d3f0f3921d256b295f993a96
-  data.tar.gz: 49aaf6803e6a8428bf143a743dd07ad840ddd0655a3228fad59543842af65466
+  metadata.gz: 6fe743a8d25cfb2bd189727b99b889fa41d3c4a770b75350dc7da72b11915198
+  data.tar.gz: 951e6cb7340553cdf51a9ef74a160774f354fb66f9eebeaded7e96c326118096
 SHA512:
-  metadata.gz: 6de98f0d6be3266c9b4d873fb0e97bd1cf94f63231411961af39671a1be9b4ce0040d3efd1476fe37a309bee3722e1a7e724fdbd37a0dc949e4982fb8e8425d4
-  data.tar.gz: 0cc7c4f3f9fa709c7fd1c7ca1a3b10ede905e5be36b90701edd0f100b404e64144deb20de6f081d3f6b89c757932f2932651fb0b6c6f60ec938419dfde2d5855
+  metadata.gz: 501920c17b82fc2de7678c16c9688f95a112d61d633025af0a9fcbbf9ab52dd47a71e1b666bee0e3d3c725c41d67f1262c5a2d8f2833e94c21e58b9d38956a15
+  data.tar.gz: afdbe6c180a29db054c2d818ab71bcf441cd2c41e65310d6df5d6c4575fb16a9d57e89c718f7b8c41a35ac5c7ac96bbb5a8fcd0e37f67092ba135243a08dcc9f

data/Gemfile.lock

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

data/README.md

@@ -62,9 +62,69 @@ 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)
+在 run 和 test 中可以使用，可以使用 --plugin=<file>.plugin.rb
+默认情况下你可以使用 <文件名>.plugin.rb 来实现一个prompt.md的插件
+在插件中有两个对象
+
+```ruby
+#gpt
+gpt.row_history # 访问裸的历史记录 可读写
+gpt.history # 处理过的历史记录 可读写
+gpt.temperature # 设置请求的 temperature
+gpt.model_name # 获取模型名称
+
+#context
+OpenStruct.new(
+  request_params: params, # 请求参数
+  user_input: user_message, # 用户输入
+  request_messages: messages, #请求时的messages
+  response_message: assistant_message, # 回复的message
+  response: response # 回复的response
+)
+# 如果需要在方法中使用使用变量传递，必须使用context包括你要的变量名，比如 context.name = "luogu"
+```
+
+支持的回调
+```ruby
+# 所有方法都必须返回context
+# 可以使用
+# require 'irb'
+# binding.irb
+# 来调试断点
+
+setup do |gpt, context|
+  puts gpt
+  context
+end
+
+before_input do |gpt, context|
+  puts "用户输入了: #{context.user_input}"
+  context
+end
+
+after_input do |gpt, context|
+  context
+end
+
+before_request do |gpt, context|
+  puts "request params: "
+  puts context
+  context
+end
+
+after_request do |gpt, context|
+  context
+end
+
+before_save_history do |gpt, context|
+  context
+end
+
+after_save_historydo |gpt, context|
+  context
+end
+
+
+```
 
 ## 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 :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
 
@@ -118,15 +128,21 @@ module Luogu
           puts "再见！"
           break
         else
-          self.puts self.chat(input)
+          time = Benchmark.measure do
+            self.puts self.chat(input)
+          end
+          puts "input: #{input} 执行时间为 #{time.real} 秒"
         end
       end
     end
 
     def playload(messages)
       messages.each do |message|
-        puts "test: #{message}"
-        self.puts self.chat(message)
+        time = Benchmark.measure do
+          puts "test: #{message}"
+          self.puts self.chat(message)
+        end
+        puts "test: #{message} 执行时间为 #{time.real} 秒"
       end
       now = Time.now.to_i
       file_name = File.basename(@prompt_file, ".*")

data/lib/luogu/init.rb

@@ -6,6 +6,7 @@ require 'yaml'
 require "dry/cli"
 require 'fileutils'
 require 'ostruct'
+require 'benchmark'
 
 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.9"
+  VERSION = "0.1.11"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: luogu
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.11
 platform: ruby
 authors:
 - MJ
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-11 00:00:00.000000000 Z
+date: 2023-04-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai