smart_agent 0.2.3 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d6400f1562f197a6506e7d11644b703dba1a247f72b264bc0d4fdf20fbda9156
-  data.tar.gz: 19057cba182dcd003525eb44abc98f3e445e40749f0df8a584054ec3ec24a022
+  metadata.gz: 1a92b61bd5f0241c702a83b4d0c598341c263516965d0dffa2ef9adff0bb21cc
+  data.tar.gz: d2208e003c420c9aa638e3553cab1ab3826024efe1e8a83334a27fe2bf71130f
 SHA512:
-  metadata.gz: 4476b0feeac3175ed2ff681ab41731f4b3c0f590b9318a193836b646b4028564c99791a38181e8021ec67d4986ec024355f9d15121ab239cf2fff4174c71be75
-  data.tar.gz: a29fdfeee975a91f7b58ee96552d92d80040c3aa0829f16022ce604f5d7e81f60ed14704aedc192d2c0b977cbc81889ec290580ea251ba9d02a774d11e73ae60
+  metadata.gz: 6e1ec7cf916db65f2b3ea5c3b6fc67ab1848679178e5277a1de0dd20162ddc40e6f4de38f4a836c2b7fd4a86d9179dcddf4e5a5d4194eb7f93a194d4e607a05e
+  data.tar.gz: 6fbf7faf9afb9c2925ae50bed76e98a453f1a2c221d08d755db49ae39ee826ec7dabf2b3f8c4752706824a07f10e48548948d4fa900e5ed2df199529b960bd6d

data/README.md

@@ -2,7 +2,7 @@
 
 [![Ruby Version](https://img.shields.io/badge/Ruby-3.2%2B-red)](https://www.ruby-lang.org/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](./LICENSE)
-[![Version](https://img.shields.io/badge/Version-0.1.8-green.svg)](./lib/smart_agent/version.rb)
+[![Version](https://img.shields.io/badge/Version-0.2.3-green.svg)](./lib/smart_agent/version.rb)
 
 **An intelligent agent framework for Ruby with MCP support, tool calling, and multi-LLM integration**
 
@@ -167,10 +167,18 @@ SmartAgent::MCPClient.define :web_service do
   url "https://api.example.com/mcp/sse"
 end
 
+SmartAgent::MCPClient.define :limited_web_service do
+  type :sse
+  url "https://api.example.com/mcp/sse"
+  functions [:search, :fetch]
+end
+
 # Use with agent
 agent = engine.build_agent(:research_bot, mcp_servers: [:opendigger, :postgres])
 ```
 
+If `functions` is configured, only the listed MCP tools are exposed to the agent. If `functions` is omitted, all tools from that MCP server are exposed.
+
 ### Advanced Features
 
 #### Stream Processing with Events
@@ -300,4 +308,4 @@ This project is licensed under the MIT License - see the [LICENSE](LICENSE) file
 
 ---
 
-**⭐ Star this repository if you find it useful!**
+**⭐ Star this repository if you find it useful!**

data/lib/smart_agent/agent.rb

@@ -26,10 +26,6 @@ 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
@@ -46,8 +42,6 @@ module SmartAgent
         return @content_event_proc
       when :tool
         return @tool_call_proc
-      when :logging
-        return @log_proc
       else
         return nil
       end
@@ -75,10 +69,6 @@ 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}")

data/lib/smart_agent/mcp_client.rb

@@ -18,6 +18,8 @@ module SmartAgent
     def to_json
       mcp_server_json = @client.list_tools
       if mcp_server_json
+        mcp_server_json["tools"] = filter_tools(mcp_server_json["tools"])
+        MCPClient.clear_server_tools(@name)
         mcp_server_json["tools"].each do |tool|
           MCPClient.set_server(tool["name"].to_sym, @name)
         end
@@ -42,6 +44,18 @@ module SmartAgent
       @client.stop
     end
 
+    private
+
+    def filter_tools(tools)
+      return [] unless tools
+
+      allowed_functions = @context.functions
+      return tools unless allowed_functions
+
+      allowed_names = allowed_functions.map(&:to_s)
+      tools.select { |tool| allowed_names.include?(tool["name"].to_s) }
+    end
+
     class << self
       def servers
         @servers ||= {}
@@ -51,6 +65,10 @@ module SmartAgent
         @tool_to_server ||= {}
       end
 
+      def server_to_tools
+        @server_to_tools ||= Hash.new { |hash, key| hash[key] = [] }
+      end
+
       def define(name, &block)
         servers[name] = block
         client = MCPClient.new(name)
@@ -59,6 +77,14 @@ module SmartAgent
 
       def set_server(tool_name, server_name)
         tool_to_server[tool_name] = server_name
+        server_to_tools[server_name] << tool_name unless server_to_tools[server_name].include?(tool_name)
+      end
+
+      def clear_server_tools(server_name)
+        server_to_tools[server_name].each do |tool_name|
+          tool_to_server.delete(tool_name)
+        end
+        server_to_tools[server_name] = []
       end
 
       def find_server_by_tool_name(tool_name)
@@ -68,6 +94,10 @@ module SmartAgent
   end
 
   class MCPContext
+    def initialize
+      @functions = nil
+    end
+
     def type(mcp_type)
       @mcp_type = mcp_type
     end
@@ -87,5 +117,11 @@ module SmartAgent
     def url(url)
       @command_path = url
     end
+
+    def functions(names = :__smart_agent_not_provided__)
+      return @functions if names == :__smart_agent_not_provided__
+
+      @functions = names
+    end
   end
 end

data/lib/smart_agent/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: smart_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.2.4
 platform: ruby
 authors:
 - Zhuang Biaowei
@@ -13,16 +13,16 @@ dependencies:
   name: smart_prompt
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.3.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.3.5
 - !ruby/object:Gem::Dependency
   name: mcp-sdk.rb
   requirement: !ruby/object:Gem::Requirement
@@ -85,7 +85,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.1
+rubygems_version: 4.0.6
 specification_version: 4
 summary: Intelligent agent framework with DSL and MCP integration
 test_files: []