44import responses
55
66from stackone_ai import StackOneTool , Tools
7- from stackone_ai .meta_tools import ToolIndex , create_meta_execute_tool , create_meta_filter_tool
7+ from stackone_ai .meta_tools import (
8+ ToolIndex ,
9+ create_meta_search_tools_execute_tool ,
10+ create_meta_search_tools_filter_tool ,
11+ )
812from stackone_ai .models import ExecuteConfig , ToolParameters
913
1014
@@ -134,15 +138,15 @@ class TestMetaFilterTool:
134138 def test_filter_tool_creation (self , sample_tools ):
135139 """Test creating the filter tool"""
136140 index = ToolIndex (sample_tools )
137- filter_tool = create_meta_filter_tool (index )
141+ filter_tool = create_meta_search_tools_filter_tool (index )
138142
139143 assert filter_tool .name == "meta_filter_relevant_tools"
140144 assert "natural language query" in filter_tool .description .lower ()
141145
142146 def test_filter_tool_execute (self , sample_tools ):
143147 """Test executing the filter tool"""
144148 index = ToolIndex (sample_tools )
145- filter_tool = create_meta_filter_tool (index )
149+ filter_tool = create_meta_search_tools_filter_tool (index )
146150
147151 # Execute with a query
148152 result = filter_tool .execute (
@@ -167,7 +171,7 @@ def test_filter_tool_execute(self, sample_tools):
167171 def test_filter_tool_call (self , sample_tools ):
168172 """Test calling the filter tool with call method"""
169173 index = ToolIndex (sample_tools )
170- filter_tool = create_meta_filter_tool (index )
174+ filter_tool = create_meta_search_tools_filter_tool (index )
171175
172176 # Call with kwargs
173177 result = filter_tool .call (query = "candidate" , limit = 2 )
@@ -181,21 +185,21 @@ class TestMetaExecuteTool:
181185
182186 def test_execute_tool_creation (self , tools_collection ):
183187 """Test creating the execute tool"""
184- execute_tool = create_meta_execute_tool (tools_collection )
188+ execute_tool = create_meta_search_tools_execute_tool (tools_collection )
185189
186190 assert execute_tool .name == "meta_execute_tool"
187191 assert "executes a tool" in execute_tool .description .lower ()
188192
189193 def test_execute_tool_missing_name (self , tools_collection ):
190194 """Test execute tool with missing tool name"""
191- execute_tool = create_meta_execute_tool (tools_collection )
195+ execute_tool = create_meta_search_tools_execute_tool (tools_collection )
192196
193197 with pytest .raises (ValueError , match = "toolName is required" ):
194198 execute_tool .execute ({"params" : {}})
195199
196200 def test_execute_tool_invalid_name (self , tools_collection ):
197201 """Test execute tool with invalid tool name"""
198- execute_tool = create_meta_execute_tool (tools_collection )
202+ execute_tool = create_meta_search_tools_execute_tool (tools_collection )
199203
200204 with pytest .raises (ValueError , match = "Tool 'invalid_tool' not found" ):
201205 execute_tool .execute (
@@ -207,7 +211,7 @@ def test_execute_tool_invalid_name(self, tools_collection):
207211
208212 def test_execute_tool_call (self , tools_collection ):
209213 """Test calling the execute tool with call method"""
210- execute_tool = create_meta_execute_tool (tools_collection )
214+ execute_tool = create_meta_search_tools_execute_tool (tools_collection )
211215
212216 # Mock the actual tool execution by patching the requests
213217 with responses .RequestsMock () as rsps :
0 commit comments