luogu 0.1.10 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bf66040bab414493d1be24b2ed42780d2cefae0c79814b48b363bcc2ed094415
-  data.tar.gz: 4e1c53cb018a14ed2ca607d364717fd5f84a8ff7a7246c641e79cd6f17806ae7
+  metadata.gz: d3ff171b0c4c4e1d93ebec7ffdbb1d9d95103989b4b9f29e0914b0c8f92a5949
+  data.tar.gz: fbb022de66ceaec42953c212e20ecb3a53a21d95664170249dbd7f036898e11b
 SHA512:
-  metadata.gz: 9670b5c6bd270e449b2916313474c4a6ac0eafff47b550a9853e8a6b0a032375eb7d74bc29d68c431fbb8ffe2ba946cc458316aa0046fde514e710b7dc642d06
-  data.tar.gz: 30556b308a94306863f8f2f7e6b1ebf9d53214a3832644cb013dc99fb332e781e519e2dd6f52a99d6da367b9f9973c1b7123b5a5e524aaa4d1d4eb5015f3d0e4
+  metadata.gz: 6abeb1fa0e1bc92d9a744ecd2dbefa109deefd4f8d758de15838be7eb15560885652018662e9d864c201d08339f9e5a6de51b335cfeef0ab8579b2987a00e132
+  data.tar.gz: b9f3a6dfefe8004be775e590ae8ee4763a3bf254a02038ecfae711ba902c97e7468ce060f9308cc1ca1188d088693fd3d8f5c3b70ff78d773569c83b14ddd502

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/prompt_parser.rb

@@ -1,6 +1,6 @@
 module Luogu
   class PromptParser
-    attr_reader :ruby_code, :ruby_code_line, :file_path
+    attr_reader :ruby_code, :ruby_code_line, :file_path, :messages
     def initialize(file_path)
       @file_path = file_path
       @messages = []
@@ -17,7 +17,6 @@ module Luogu
       JSON.pretty_generate self.render
     end
 
-    private
     def process_file
       @messages = []
       file = File.read(@file_path)

data/lib/luogu/version.rb

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

metadata

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