luogu 0.1.10 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bf66040bab414493d1be24b2ed42780d2cefae0c79814b48b363bcc2ed094415
-  data.tar.gz: 4e1c53cb018a14ed2ca607d364717fd5f84a8ff7a7246c641e79cd6f17806ae7
+  metadata.gz: 6fe743a8d25cfb2bd189727b99b889fa41d3c4a770b75350dc7da72b11915198
+  data.tar.gz: 951e6cb7340553cdf51a9ef74a160774f354fb66f9eebeaded7e96c326118096
 SHA512:
-  metadata.gz: 9670b5c6bd270e449b2916313474c4a6ac0eafff47b550a9853e8a6b0a032375eb7d74bc29d68c431fbb8ffe2ba946cc458316aa0046fde514e710b7dc642d06
-  data.tar.gz: 30556b308a94306863f8f2f7e6b1ebf9d53214a3832644cb013dc99fb332e781e519e2dd6f52a99d6da367b9f9973c1b7123b5a5e524aaa4d1d4eb5015f3d0e4
+  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,10 +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)
-- OpenStruct.new(response: response)
+在 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, :context
+    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")
@@ -128,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.10"
+  VERSION = "0.1.11"
 end

metadata

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