luogu 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c80b1178376383a11b89b3228aadcab81c58446c1cd5b84e9f26d8684c94d541
-  data.tar.gz: a249cbcb501e5a80b2c46ea43cdc85b10cb5b58a0eaba6f73fb79992625e5037
+  metadata.gz: d22ef62531ee8bf0c04bcc4258e2e0287f31b73f03da333b0e4b84aa4438ed3a
+  data.tar.gz: 5d9fffc7274eb10622ece637326b0981aec632714b462b2c3762dc12ca6faa38
 SHA512:
-  metadata.gz: 3096c1247c8c1eacc38ef6847f120036d37f753b0a5ede439b279345c473598d85db1a66b9db703dd26b6113612eec98679c2af2de0c00dae71ae345956d5bfa
-  data.tar.gz: 24d27201d1cc31e9e01361448fba1fef1abc1c1459a70908849ba86fe57132f7ffa5eee30cf8eddf60065a82c5590cc178a9d9564eab384a10dbfadd31e4322b
+  metadata.gz: 395e2b399e098e1c566aa88cf90ecd197b8866528ce7a97c95e11c64bc829778bd1be17a1258cf0f2144c64ad073f18cead2d174881cfa6f01a6fa5b7794831a
+  data.tar.gz: 05f9abbddd765a5b0265415fc0afa448446870958cb99b8d07d71930c748c8bb97e80a2c45122c00de3cc0feb906c88cec51168b6b137c3b3ee85975fe1d48b2

data/README.md

@@ -13,6 +13,7 @@
 - luogu build <file> 编译成对应的json
 - luogu run <file> 测试prompt
 - luogu gen <prompt.json file> <target.md> 用来逆向生成md文件的命令
+- luogu test <file> <yml> 用来跑自动化测试 yaml 一行一句话
 
 你可以在项目目录的.env中设置下面的环境变量，或者直接系统设置
 ```

data/lib/luogu/chatgpt.rb

@@ -85,6 +85,16 @@ module Luogu
       end
     end
 
+    def playload(messages)
+      messages.each do |message|
+        puts "test: #{message}"
+        self.puts self.chat(message)
+      end
+      now = Time.now.to_i
+      self.class.save @row_history, "./prompt.row_history.test-#{now}.md"
+      self.class.save @history.to_a, "./prompt.history.test-#{now}.md"
+    end
+
     class << self
       def save(history, file_path)
         text = ""

data/lib/luogu/cli.rb

@@ -12,6 +12,7 @@ module Luogu
         luogu build <file> -> 编译prompt
         luogu run <file> -> 测试 prompt
         luogu gen <file> <target> -> 根据 json 生成 prompt 文件
+        luogu test <file> <test_file.yml> -> 根据 yaml 来对 prompt 进行测试
         """
         exit
       end
@@ -37,6 +38,16 @@ module Luogu
       chatgpt = ChatGPT.save(json, args.last)
     end
 
+    subcommands['test'] = Proc.new do |args|
+      promtpt_file = args.first
+      promtpt_test_file = args.last
+
+      chatgpt = ChatGPT.new(args.first)
+
+      messages = YAML.load(promtpt_test_file)
+      chatgpt.playload messages
+    end
+
     if subcommands.key?(ARGV.first)
       subcommands[ARGV.first].call(ARGV[1..-1])
     else

data/lib/luogu/version.rb

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

metadata

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