luogu 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0cd2a93255b9c6ed4fe776dab51414384fea23cad8d73c1e407acb5deb9bb382
-  data.tar.gz: e26cdfa7b3e748a580181f6a916a6fd07667fe6b0a3d3c713c454b9584b09eee
+  metadata.gz: 584995338f86e7c2be1f099920457c424396373ec007e6fa7185face0a0254e9
+  data.tar.gz: 7414294ca9278bc20989b2fb5e58bd5c3b6602a35853ae8f56a4b35423b1974b
 SHA512:
-  metadata.gz: f6ce3327ccd8824a59e5deda151876f1d6fc1b64247c5d5b1140ae4e1fd8455169b8c346f3101304f4b0dd01a5c201b43d5cbb8f322b91ab23a159d6285cf6c3
-  data.tar.gz: 3b6e68d717a10b7ca2feca6cf5b47760e0ab0d672b5cce34883e992a6fffa49caa0a69cd5f36f98cd51c49d1237d76f643f3ca50c241f328f1789ab803ac38ee
+  metadata.gz: ca7945d9ef6ddc5ba18ff6e258f18bf8799049d580b7befbe50b1db43297322171b68852f8783643b16f829c7a7b99faf5fceec6d87c64a20232fd9e7ce04087
+  data.tar.gz: 2982f3f04cb4edec38434db75933921b7574f8b885d8a8265514071724f0ca72eb2bd681377773c97edf5be9ee48a358f729f5a12f507fc753527f1f594300ea

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    luogu (0.1.0)
+    luogu (0.1.1)
       dotenv (~> 2.8, >= 2.8.1)
       ruby-openai (~> 3.7)
       tty-prompt (~> 0.23.1)

data/lib/luogu/chatgpt.rb

@@ -62,21 +62,6 @@ module Luogu
       @_puts_method.call message
     end
 
-    def save(history, file_path)
-      text = ""
-      role_map = {"user" => "@u", "assistant" => "@a", "system" => "@s"}
-      history.each do |item|
-        text += role_map[item[:role]]
-        text += "\n"
-        text += item[:content]
-        text += "\n\n"
-      end
-      File.open(file_path, 'w') do |f|
-        f.write(text)
-      end
-      puts "已经保存文件到 #{file_path}"
-    end
-
     def run
       loop do
         # 从命令行读取输入
@@ -85,8 +70,8 @@ module Luogu
         # 根据用户输入执行相应的操作
         case input
         when "save"
-          self.save @row_history, "./prompt.row_history.md"
-          self.save @history.to_a, "./prompt.history.md"
+          self.class.save @row_history, "./prompt.row_history.md"
+          self.class.save @history.to_a, "./prompt.history.md"
         when "row history"
           p @row_history
         when "history"
@@ -99,5 +84,22 @@ module Luogu
         end
       end
     end
+
+    class << self
+      def save(history, file_path)
+        text = ""
+        role_map = {"user" => "@u", "assistant" => "@a", "system" => "@s"}
+        history.each do |item|
+          text += role_map[item[:role]]
+          text += "\n"
+          text += item[:content]
+          text += "\n\n"
+        end
+        File.open(file_path, 'w') do |f|
+          f.write(text)
+        end
+        puts "已经保存文件到 #{file_path}"
+      end
+    end
   end
 end

data/lib/luogu/cli.rb

@@ -11,6 +11,7 @@ module Luogu
         puts """
         luogu build <file> -> 编译prompt
         luogu run <file> -> 测试 prompt
+        luogu gen <file> <target> -> 根据 json 生成 prompt 文件
         """
         exit
       end
@@ -31,6 +32,11 @@ module Luogu
       chatgpt.run
     end
 
+    subcommands['gen'] = Proc.new do |args|
+      json = JSON.parse File.read(args.first), symbolize_names: true   
+      chatgpt = ChatGPT.save(json, args.last)
+    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.0"
+  VERSION = "0.1.2"
 end

metadata

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