luogu 0.1.7 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fc5e0c2faef39054b2d1b084393c5af185fa52660a46b552af6fe26def302c8b
-  data.tar.gz: d20f25098e3fe03226375d3a0471acdbbd207523db7b101c431007b82cc49fa7
+  metadata.gz: 8b6f8d477025b929343fbd70605905c557d33550d3f0f3921d256b295f993a96
+  data.tar.gz: 49aaf6803e6a8428bf143a743dd07ad840ddd0655a3228fad59543842af65466
 SHA512:
-  metadata.gz: 5d493d41fd71f846549921c4ee5d2ac9aa5d398f8c07010dd0d0ca4ae5b33a9c30d2625325161cf0eb4427e9297b9b103783dadef2e08d321e63aa6b0f1255cb
-  data.tar.gz: 44c77c4a4a4be4f2eaaef0bada464279c26c3bb90c9a56017211ae1c6f656299d521328b4235a53dd28db0706c8e7fc6616b1cea1429f6e54442aab3b0e031e1
+  metadata.gz: 6de98f0d6be3266c9b4d873fb0e97bd1cf94f63231411961af39671a1be9b4ce0040d3efd1476fe37a309bee3722e1a7e724fdbd37a0dc949e4982fb8e8425d4
+  data.tar.gz: 0cc7c4f3f9fa709c7fd1c7ca1a3b10ede905e5be36b90701edd0f100b404e64144deb20de6f081d3f6b89c757932f2932651fb0b6c6f60ec938419dfde2d5855

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    luogu (0.1.6)
+    luogu (0.1.9)
       dotenv (~> 2.8, >= 2.8.1)
       dry-cli (~> 1.0)
       ruby-openai (~> 3.7)

data/README.md

@@ -61,4 +61,10 @@ end
 - history 查看当前上下文
 - 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)
+
 ## MIT 协议

data/lib/luogu/chatgpt.rb

@@ -1,8 +1,20 @@
 module Luogu
   class ChatGPT
-    def initialize(file, history_path='.')
+
+    attr_accessor :template, :limit_history, :prompt, :row_history, :history, :temperature, :model_name
+
+    def initialize(file, history_path='.', plugin_file_path=nil)
+      @plugin_file_path = plugin_file_path || file.sub(File.extname(file), ".plugin.rb")
+
+      if File.exist?(@plugin_file_path)
+        @plugin = Plugin.new(@plugin_file_path).load()
+      else
+        @plugin = Plugin.new(@plugin_file_path)
+      end
+
       @temperature = ENV.fetch('OPENAI_TEMPERATURE', '0.7').to_f
       @limit_history = ENV.fetch('OPENAI_LIMIT_HISTORY', '6').to_i * 2
+      @model_name = "gpt-3.5-turbo"
       
       @history_path = history_path
       @prompt_file = file
@@ -10,20 +22,37 @@ module Luogu
       @prompt = PromptParser.new(file)
       @row_history = []
       @history = HistoryQueue.new @limit_history
+
+      if @plugin.setup_proc
+        @plugin.setup_proc.call(self, OpenStruct.new) 
+      end
     end
 
     def request(messages)
-      response = client.chat(
-        parameters: {
-            model: "gpt-3.5-turbo",
-            messages: messages,
-            temperature: 0.7,
-        })
+      params = {
+        model: @model_name,
+        messages: messages,
+        temperature: @temperature,
+      }
+      
+      if @plugin.before_request_proc
+        params = @plugin.before_request_proc.call(self, OpenStruct.new(request_params: params)).request_params
+      end
+      response = client.chat(parameters: params)
+      @plugin.after_request_proc.call(self, OpenStruct.new(response: response)) if @plugin.after_request_proc
+
       response.dig("choices", 0, "message", "content")
     end
 
     def chat(user_message)
+      if @plugin.before_input_proc
+        user_message = @plugin.before_input_proc.call(self, OpenStruct.new(user_input: user_message)).user_input
+      end
       messages = (@prompt.render + @history.to_a) << {role: "user", content: user_message}
+      if @plugin.after_input_proc
+        messages = @plugin.after_input_proc.call(self, OpenStruct.new(request_messages: messages)).request_messages
+      end
+
       assistant_message = self.request(messages)
       
       self.push_row_history(user_message, assistant_message)
@@ -31,6 +60,8 @@ module Luogu
       if @prompt.ruby_code
         puts "执行文档中的callback"
         instance_eval @prompt.ruby_code, @prompt.file_path, @prompt.ruby_code_line
+      elsif @plugin.before_save_history_proc
+        @plugin.before_save_history_proc.call(self, OpenStruct.new(user_input: user_message, response_message: assistant_message))
       else
         puts "执行默认的历史记录"
         self.push_history(user_message, assistant_message)
@@ -47,6 +78,9 @@ module Luogu
     def push_history(user_message, assistant_message)
       @history.enqueue({role: "user", content: user_message})
       @history.enqueue({role: "assistant", content: assistant_message})
+      if @plugin.after_save_history_proc
+        @plugin.after_save_history_proc.call(self, OpenStruct.new(user_input: user_message, response_message: assistant_message)) 
+      end
     end
 
     def ask(message)

data/lib/luogu/cli.rb

@@ -32,9 +32,10 @@ module Luogu
         desc "编译 Prompt.md 成能够提交给 ChatGPT API 的 messages. 默认输出为 <同文件名>.json"
         argument :prompt_file, type: :string, required: true, desc: "Prompt文件, 使用markdown书写"
         option :out, type: :string, default: ".", desc: "保存历史时存放的目录，默认为当前目录"
+        option :plugin, type: :string, desc: "运行的时候载入对应的插件"
 
         def call(prompt_file: nil, **options)
-          chatgpt = ChatGPT.new(prompt_file, options.fetch(:out))
+          chatgpt = ChatGPT.new(prompt_file, options.fetch(:out), options.fetch(:plugin, nil))
           chatgpt.run
         end
 
@@ -61,11 +62,12 @@ module Luogu
         argument :prompt_file, type: :string, require: true, desc: "输出 Prompt 文件"
         argument :test_file, type: :string, require: false, desc: "测试文件, 使用 YAML 文件, 一个字符串数组。默认为 同名.test.yml"
         option :out, type: :string, default: ".", desc: "保存测试历史时存放的目录，默认为当前目录"
+        option :plugin, type: :string, desc: "运行的时候载入对应的插件"
 
         def call(prompt_file: nil, test_file:nil, **options)
           test_file ||= prompt_file.sub(File.extname(prompt_file), ".test.yml")
 
-          chatgpt = ChatGPT.new(prompt_file, options.fetch(:out))
+          chatgpt = ChatGPT.new(prompt_file, options.fetch(:out), options.fetch(:plugin, nil))
           messages = YAML.load_file(test_file)
           chatgpt.playload messages
         end

data/lib/luogu/init.rb

@@ -5,7 +5,9 @@ require 'json'
 require 'yaml'
 require "dry/cli"
 require 'fileutils'
+require 'ostruct'
 
+require_relative 'plugin'
 require_relative 'history_queue'
 require_relative "prompt_parser"
 require_relative "chatgpt"

data/lib/luogu/plugin.rb

@@ -0,0 +1,52 @@
+module Luogu
+  class Plugin
+    attr_reader :before_input_proc, :before_save_history_proc, :after_input_proc, :after_save_history_proc,
+     :setup_proc, :before_request_proc, :after_request_proc
+
+    def initialize(plugin_file_path)
+      @plugin_file_path = plugin_file_path
+
+      @before_input_proc = nil
+      @before_save_history_proc = nil
+
+      @after_input_proc = nil
+      @after_save_history_proc = nil
+
+      @setup_proc = nil
+    end
+
+    def before_input(&block)
+      @before_input_proc = block
+    end
+
+    def before_save_history(&block)
+      @before_save_history_proc = block
+    end
+
+    def after_input(&block)
+      @after_input_proc = block
+    end
+
+    def after_save_history(&block)
+      @after_save_history_proc = block
+    end
+
+    def setup(&block)
+      @setup_proc = block
+    end
+
+    def before_request(&block)
+      @before_request_proc = block
+    end
+
+    def after_request(&block)
+      @after_request_proc = block
+    end
+
+    def load()
+      self.instance_eval File.read(@plugin_file_path)
+      self
+    end
+    
+  end
+end

data/lib/luogu/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: luogu
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.9
 platform: ruby
 authors:
 - MJ
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-09 00:00:00.000000000 Z
+date: 2023-04-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -90,6 +90,7 @@ files:
 - lib/luogu/cli.rb
 - lib/luogu/history_queue.rb
 - lib/luogu/init.rb
+- lib/luogu/plugin.rb
 - lib/luogu/prompt_parser.rb
 - lib/luogu/version.rb
 - luogu.gemspec