44from fastmcp .tools import Tool
55
66from fenic import SystemTool , SystemToolConfig
7- from fenic .api .mcp ._tool_generation_utils import auto_generate_system_tools_from_tables
87from fenic .api .functions import col , tool_param
8+ from fenic .api .mcp ._tool_generation_utils import auto_generate_system_tools_from_tables
99from fenic .api .mcp .server import create_mcp_server
1010from fenic .api .session .session import Session
1111from fenic .core ._utils .misc import to_snake_case
1616
1717
1818def test_server_generation_with_parameterized_tools (local_session : Session ):
19- pytest .importorskip ("fastmcp" )
2019 df = local_session .create_dataframe ({"city" : ["SF" ], "age" : [10 ], "user_name" : ["Alice" ]})
2120 query = df .filter (
2221 (col ("city" ) == tool_param ("city_name" , StringType ))
@@ -36,7 +35,7 @@ def test_server_generation_with_parameterized_tools(local_session: Session):
3635 query = query ,
3736 )
3837
39- server = create_mcp_server (local_session , "Test Server" , parameterized_tools = [parameterized_tool ])
38+ server = create_mcp_server (local_session , "Test Server" , user_defined_tools = [parameterized_tool ])
4039 server_tools = asyncio .run (server .mcp .get_tools ())
4140 assert len (server_tools ) == 1
4241 parameter_schema = server_tools ["tool_x" ].parameters ['properties' ]
@@ -57,8 +56,7 @@ def test_server_generation_with_parameterized_tools(local_session: Session):
5756 assert user_names_param ['minItems' ] == 1
5857 assert user_names_param ['description' ] == "User names"
5958
60- def test_server_generation (local_session : Session ):
61- pytest .importorskip ("fastmcp" )
59+ def test_server_generation_with_system_tools (local_session : Session ):
6260 create_table_with_rows (local_session , "t1" , [1 , 2 , 3 ], description = "table one" )
6361 create_table_with_rows (local_session , "t2" , [10 , 20 ], description = "table two" )
6462 tools = auto_generate_system_tools_from_tables (["t1" , "t2" ], local_session , tool_namespace = "Auto" )
0 commit comments