luogu 0.2.5 → 0.2.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0e121e0329cf0f92c56754426facf7c5142941bdb859a19b3125cc9a53e30221
-  data.tar.gz: 0a8d2f4ae3b4599bbe60960dfaf6d5cfc8c6d21c5e9e69bd7e7432d30ec9c36f
+  metadata.gz: b43c4e8273fcaae684832481f227a4dc1ad3e453b3a7f97681a5c5251b4a0349
+  data.tar.gz: 4283068e62756e7f0c0c1bc3d2e8f08681e6dcfa4cdeec10c99aef32019b3e0f
 SHA512:
-  metadata.gz: 22d8dd554b0620536b9ac858ad6e5009da3bf5eb3bba502b10710f4f783ca0b8f392895805cb9ad6db3c2f567d7c95891dc2c26149e1be76f9493622427bf74e
-  data.tar.gz: 89be4a1d4d4525885fcdb77889c07365b1dd22f223f2de225c79ebb99b80126bd110cec51721c03110321ce239a0bbef4e262011f9b8822c0135383c27256a81
+  metadata.gz: 99c634daff06d449330d33fe506e0a8afdbd7b1d24cf8cb57d889835e19cd083ecdf25eb3faaea35da234fa4890e5a415bf5e19cdeff6f8248ee0741de6d29eb
+  data.tar.gz: 1dcb79abfdccec8b0dff6e5af95fec56e23a7b52302c267c04330173fe53d69d16fe0866948217a2c2cb5d82e1fadce21b8cffdb00acd8d54d18821d6dbd19bb

data/lib/luogu/agent_runner.rb

@@ -9,6 +9,8 @@ module Luogu
     end
 
     setting :run_agent_retries, default: Application.config.run_agent_retries
+    setting :mode, default: :agent # route 路由模式 agent 代理模式
+    setting :route_mode_response_handle, default: ->(tool_responses) { tool_responses&.first.fetch(:response, nil) }
 
     setting :provider do
       setting :parameter_model, default: ->() {
@@ -117,7 +119,11 @@ module Luogu
         { role: "assistant", content: agents.to_json },
         { role: "user", content: templates.tool.result(binding) }
       ]
-      request messages, run_agent_retries: run_agent_retries
+      if config.mode == :agent
+        request messages, run_agent_retries: run_agent_retries
+      else
+        config.route_mode_response_handle.call @tools_response
+      end
     end
 
   end

data/lib/luogu/openai.rb

@@ -1,6 +1,10 @@
 # frozen_string_literal: true
 
 module Luogu::OpenAI
+  extend Dry::Configurable
+
+  setting :after_parse_json, default: ->(data) { data }
+
   class ChatRequestParams < Struct.new(:model, :messages, :temperature,
                                      :top_p, :n, :stream, :stop,
                                      :max_tokens, :presence_penalty,
@@ -45,12 +49,14 @@ module Luogu::OpenAI
     end
 
     if result_json.nil?
-      JSON.parse markdown
+      result_json = JSON.parse markdown
     else
       result_json
     end
+    Luogu::OpenAI.config.after_parse_json.call(result_json)
   rescue => e
-    raise e, "Input must be JSON and contain action and action_input."
+    Luogu::Application.logger.error "parse json error: #{markdown}"
+    Luogu::OpenAI.config.after_parse_json.call(markdown)
   end
 
   def chat_response_handle(response)

data/lib/luogu/version.rb

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

metadata

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