luogu 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c80b1178376383a11b89b3228aadcab81c58446c1cd5b84e9f26d8684c94d541
-  data.tar.gz: a249cbcb501e5a80b2c46ea43cdc85b10cb5b58a0eaba6f73fb79992625e5037
+  metadata.gz: 5dc8912fdf448f5a9cb2f27d32125bf5a82b5231a8c74f36c03c2269d3345a14
+  data.tar.gz: 9039e6a2ebf32390d96625e734e63c49a9fda770945d8240a4a11dc3d8af114b
 SHA512:
-  metadata.gz: 3096c1247c8c1eacc38ef6847f120036d37f753b0a5ede439b279345c473598d85db1a66b9db703dd26b6113612eec98679c2af2de0c00dae71ae345956d5bfa
-  data.tar.gz: 24d27201d1cc31e9e01361448fba1fef1abc1c1459a70908849ba86fe57132f7ffa5eee30cf8eddf60065a82c5590cc178a9d9564eab384a10dbfadd31e4322b
+  metadata.gz: bfc85f6ebf5dc6970243291b277654a86adc93d3a917e4c571f7e2a7b08bbc74fd7d20ee5fbacd4981ba8b983f9b1da05fe97a60da9b8a3b607a4fe5bf8b6b7b
+  data.tar.gz: e5b8db6cb68a539ec09f706fed396dfafccc352817f1072616b66db77843d95fea9c67374fda515407c75cb71273523d0c8ede5793387a4663ab8c8ca9848480

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_file(promtpt_test_file)
+      chatgpt.playload messages
+    end
+
     if subcommands.key?(ARGV.first)
       subcommands[ARGV.first].call(ARGV[1..-1])
     else

data/lib/luogu/init.rb

@@ -2,6 +2,7 @@ require "openai"
 require 'dotenv/load'
 require "tty-prompt"
 require 'json'
+require 'yaml'
 require 'optparse'
 
 require_relative 'history_queue'

data/lib/luogu/version.rb

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

metadata

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