Merge pull request #28 from keboola/KAB-923-fix-server-start

Kab 923 fix server start

Author: vita-stejskal

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "keboola-mcp-server"
 
-version = "0.2.2"
+version = "0.2.3"
 description = "MCP server for interacting with Keboola Connection"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -14,22 +14,20 @@ authors = [
     { name = "Your Name", email = "[email protected]" }
 ]
 dependencies = [
-    "mcp[cli]",
-    "kbcstorage",
-    "httpx",
-    "pandas"
+    "mcp[cli] ~= 1.6",
+    "kbcstorage ~= 0.9",
+    "httpx ~= 0.28",
 ]
 
 [project.optional-dependencies]
 dev = [
-    "pytest",
-    "pytest-asyncio",
-    "pytest-cov",
-    "pytest-mock",
-    "black",
-    "isort",
-    "mypy",
-    "pandas-stubs"
+    "pytest ~= 8.3",
+    "pytest-asyncio ~= 0.25",
+    "pytest-cov ~= 6.0",
+    "pytest-mock ~= 3.14",
+    "black ~= 25.1",
+    "isort ~= 6.0",
+    "mypy ~= 1.5",
 ]
 
 [project.scripts]

src/keboola_mcp_server/cli.py

@@ -65,7 +65,7 @@ def main(args: Optional[List[str]] = None) -> None:
         mcp = create_server(config)
         mcp.run(transport=parsed_args.transport)
     except Exception as e:
-        logger.error(f"Server failed: {e}")
+        logger.exception(f"Server failed: {e}")
         sys.exit(1)
 
 

src/keboola_mcp_server/server.py

@@ -64,14 +64,7 @@ def create_server(config: Optional[Config] = None) -> FastMCP:
     """
     # Initialize FastMCP server with system instructions
     mcp = KeboolaMcpServer(
-        "Keboola Explorer",
-        session_state_factory=_create_session_state_factory(config),
-        dependencies=[
-            "keboola.storage-api-client",
-            "httpx",
-            "pandas",
-            "snowflake-connector-python",
-        ],
+        "Keboola Explorer", session_state_factory=_create_session_state_factory(config)
     )
 
     add_storage_tools(mcp)

src/keboola_mcp_server/sql_tools.py

@@ -9,7 +9,7 @@
 
 from keboola_mcp_server.client import KeboolaClient
 
-LOG = logging.getLogger(__name__)
+logger = logging.getLogger(__name__)
 
 
 def add_sql_tools(mcp: FastMCP) -> None:
@@ -120,7 +120,7 @@ async def get_table_fqn(self, table: dict[str, Any]) -> Optional[TableFqn]:
             if result.is_ok and result.data and result.data.rows:
                 db_name = result.data.rows[0]["DATABASE_NAME"]
             else:
-                LOG.error(f"Failed to run SQL: {sql}, SAPI response: {result}")
+                logger.error(f"Failed to run SQL: {sql}, SAPI response: {result}")
 
         else:
             sql = f'select CURRENT_DATABASE() as "current_database", CURRENT_SCHEMA() as "current_schema";'
@@ -138,7 +138,7 @@ async def get_table_fqn(self, table: dict[str, Any]) -> Optional[TableFqn]:
                     schema_name = row["current_schema"]
                     table_name = table["name"]
             else:
-                LOG.error(f"Failed to run SQL: {sql}, SAPI response: {result}")
+                logger.error(f"Failed to run SQL: {sql}, SAPI response: {result}")
 
         if db_name and schema_name and table_name:
             fqn = TableFqn(db_name, schema_name, table_name)

tests/test_server.py

@@ -0,0 +1,32 @@
+import pytest
+
+from keboola_mcp_server.server import create_server
+
+
+class TestServer:
+    @pytest.mark.asyncio
+    async def test_list_tools(self):
+        server = create_server()
+        tools = await server.list_tools()
+        assert sorted(t.name for t in tools) == [
+            "get_bucket_metadata",
+            "get_table_metadata",
+            "list_bucket_info",
+            "list_bucket_tables",
+            "list_component_configs",
+            "list_components",
+            "query_table",
+        ]
+
+    @pytest.mark.asyncio
+    async def test_tools_have_descriptions(self):
+        server = create_server()
+        tools = await server.list_tools()
+
+        missing_descriptions: list[str] = []
+        for t in tools:
+            if not t.description:
+                missing_descriptions.append(t.name)
+
+        missing_descriptions.sort()
+        assert not missing_descriptions, f"These tools have no description: {missing_descriptions}"