luogu 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cb3e745f62ebfb4ebf7401ee41225be94c005b911669df30c622d62a44400696
-  data.tar.gz: 181e888e61066c2f99d3102e946b7e35eca9dca07c15fd91e55b7fc8bd83894d
+  metadata.gz: d22ef62531ee8bf0c04bcc4258e2e0287f31b73f03da333b0e4b84aa4438ed3a
+  data.tar.gz: 5d9fffc7274eb10622ece637326b0981aec632714b462b2c3762dc12ca6faa38
 SHA512:
-  metadata.gz: 8c46cd1408e4baba2cf08f7498ba39408a7acf3ac442ed347f7e7ce0b958210a42e042b492dd11d97a768476165fc65f253b3fd138dabd2bd50ec03b3b815880
-  data.tar.gz: d84cf9a0d8ca218452976cadd450f696d294df1d809872dc7775a190a15880268283d72cebe899ccd3e1a77c3a9d9e24befae6a7763d2325310b7470a9b88489
+  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/prompt_parser.rb

@@ -10,6 +10,7 @@ module Luogu
 
     def render
       process_file
+      @messages
     end
 
     def to_json

data/lib/luogu/version.rb

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

metadata

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