smart_agent 0.2.1 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 92f316229977853df082c594f3dfe42677dcb1d92e6e0ffe7e8c12dbf09db3cd
-  data.tar.gz: 42890f035e21b320b205b62f9de845cd7d2d0f492c13967e9147f52822af4702
+  metadata.gz: d6400f1562f197a6506e7d11644b703dba1a247f72b264bc0d4fdf20fbda9156
+  data.tar.gz: 19057cba182dcd003525eb44abc98f3e445e40749f0df8a584054ec3ec24a022
 SHA512:
-  metadata.gz: 0a837125205d17f4b9c3c3a3417c014f718c406a874fbc88a7339d59ee2dee1d5fd01fd4eb13654a28349bbbda97f1acc71b8b03b7a3ef14f21afe5f1dbbdf11
-  data.tar.gz: e721b313294d3f149cf986e9d4a3140a83fb68368542b527f5c2205fc1ff850c3a4f8982f08185bb0ac28fe681b21e95c604b557d81f19cd7a4e87b7eccc9111
+  metadata.gz: 4476b0feeac3175ed2ff681ab41731f4b3c0f590b9318a193836b646b4028564c99791a38181e8021ec67d4986ec024355f9d15121ab239cf2fff4174c71be75
+  data.tar.gz: a29fdfeee975a91f7b58ee96552d92d80040c3aa0829f16022ce604f5d7e81f60ed14704aedc192d2c0b977cbc81889ec290580ea251ba9d02a774d11e73ae60

data/lib/smart_agent/agent.rb

@@ -26,6 +26,10 @@ module SmartAgent
       @tool_call_proc = block
     end
 
+    def on_logging(&block)
+      @log_proc = block
+    end
+
     def on_event
       if @reasoning_event_proc || @content_event_proc
         return true
@@ -42,6 +46,8 @@ module SmartAgent
         return @content_event_proc
       when :tool
         return @tool_call_proc
+      when :logging
+        return @log_proc
       else
         return nil
       end
@@ -69,6 +75,10 @@ module SmartAgent
       @agent = agent
     end
 
+    def show_log(msg)
+      @agent.processor(:logging).call(msg) if @agent.processor(:logging)
+    end
+
     def call_worker(name, params, with_tools: true, with_history: false)
       SmartAgent.logger.info("Call Worker name is: #{name}")
       SmartAgent.logger.info("Call Worker params is: #{params}")
@@ -151,19 +161,16 @@ module SmartAgent
         tool_name = tool["function"]["name"].to_sym
         params = safe_parse(tool["function"]["arguments"])
         if Tool.find_tool(tool_name)
-          @agent.processor(:tool).call({ :content => "ToolName is `#{tool_name}`\n" }) if @agent.processor(:tool)
-          @agent.processor(:tool).call({ :content => "params is `#{params}`\n" }) if @agent.processor(:tool)
-          tool_result = Tool.find_tool(tool_name).call(params)
+          tool_result = Tool.find_tool(tool_name).call(params, @agent)
           if tool_result
+            @agent.processor(:tool).call({ :content => tool_result })
             SmartAgent.prompt_engine.history_messages << { "role" => "assistant", "content" => "", "tool_calls" => [tool] } #result.response.dig("choices", 0, "message")
             SmartAgent.prompt_engine.history_messages << { "role" => "tool", "tool_call_id" => tool_call_id, "content" => tool_result.to_s.force_encoding("UTF-8") }
             results << tool_result
           end
         end
         if server_name = MCPClient.find_server_by_tool_name(tool_name)
-          @agent.processor(:tool).call({ :content => "MCP Server is `#{server_name}`, ToolName is `#{tool_name}`\n" }) if @agent.processor(:tool)
-          @agent.processor(:tool).call({ :content => "params is `#{params}`\n" }) if @agent.processor(:tool)
-          tool_result = MCPClient.new(server_name).call(tool_name, params)
+          tool_result = MCPClient.new(server_name).call(tool_name, params, @agent)
           if tool_result
             @agent.processor(:tool).call({ :content => tool_result })
             SmartAgent.prompt_engine.history_messages << { "role" => "assistant", "content" => "", "tool_calls" => [tool] } # result.response.dig("choices", 0, "message")
@@ -177,6 +184,15 @@ module SmartAgent
       return results
     end
 
+    def call_tool(name, params = {})
+      if Tool.find_tool(name)
+        return Tool.find_tool(name).call(params, @agent)
+      end
+      if server_name = MCPClient.find_server_by_tool_name(name)
+        return MCPClient.new(server_name).call(name, params, @agent)
+      end
+    end
+
     def params
       @params ||= {}
     end

data/lib/smart_agent/mcp_client.rb

@@ -25,7 +25,11 @@ module SmartAgent
       convertFormat(mcp_server_json)
     end
 
-    def call(tool_name, params)
+    def call(tool_name, params, agent = nil)
+      if agent
+        agent.processor(:tool).call({ :content => "MCP Server is `#{@name}`, ToolName is `#{tool_name}`\n" }) if agent.processor(:tool)
+        agent.processor(:tool).call({ :content => "params is `#{params}`\n" }) if agent.processor(:tool)
+      end
       @client.call_method(
         {
           "name": tool_name.to_s,

data/lib/smart_agent/tool.rb

@@ -8,7 +8,11 @@ module SmartAgent
       @context = ToolContext.new(self)
     end
 
-    def call(params)
+    def call(params, agent = nil)
+      if agent
+        agent.processor(:tool).call({ :content => "ToolName is `#{@name}`\n" }) if agent.processor(:tool)
+        agent.processor(:tool).call({ :content => "params is `#{params}`\n" }) if agent.processor(:tool)
+      end
       @context.input_params = params
       @context.instance_eval(&@context.proc)
     end
@@ -90,7 +94,7 @@ module SmartAgent
       SmartAgent.prompt_engine.call_worker(name, params)
     end
 
-    def call_tool(name, params)
+    def call_tool(name, params = {})
       if Tool.find_tool(name)
         return Tool.find_tool(name).call(params)
       end

data/lib/smart_agent/version.rb

@@ -1,3 +1,3 @@
 module SmartAgent
-  VERSION = "0.2.1"
+  VERSION = "0.2.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: smart_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
 platform: ruby
 authors:
 - Zhuang Biaowei