some logging update

Author: daimor

docker-compose.yml

@@ -0,0 +1,10 @@
+services:
+  database:
+    image: containers.intersystems.com/intersystems/iris-community:2025.1
+    restart: always
+    ports:
+      - "1972:1972"
+      - "52773:52773"
+    command:
+      - -a
+      - iris session iris -U%SYS '##class(Security.Users).UnExpireUserPasswords("*")'

example.py

@@ -0,0 +1,74 @@
+import sys
+import asyncio
+from typing import Optional
+from contextlib import AsyncExitStack
+
+from mcp import ClientSession, StdioServerParameters
+from mcp.client.stdio import stdio_client
+
+from dotenv import load_dotenv
+
+load_dotenv()  # load environment variables from .env
+
+
+class MCPClient:
+    def __init__(self):
+        # Initialize session and client objects
+        self.session: Optional[ClientSession] = None
+        self.exit_stack = AsyncExitStack()
+
+    async def connect_to_server(self):
+        """Connect to an MCP server"""
+        server_params = StdioServerParameters(
+            command=sys.executable, args=["-m", "mcp_server_iris"], env=None
+        )
+        # server_params = StdioServerParameters(
+        #     command="uvx", args=["mcp-server-iris"], env=None
+        # )
+
+        stdio_transport = await self.exit_stack.enter_async_context(
+            stdio_client(server_params)
+        )
+        self.stdio, self.write = stdio_transport
+        self.session = await self.exit_stack.enter_async_context(
+            ClientSession(self.stdio, self.write)
+        )
+
+        await self.session.initialize()
+
+        # List available tools
+        response = await self.session.list_tools()
+        tools = response.tools
+        print(
+            "\nConnected to server with tools:",
+            "\n\t".join([""] + [tool.name for tool in tools]),
+        )
+
+    async def process_query(self, query: str) -> str:
+        response = await self.session.call_tool(
+            "execute_sql",
+            {
+                "query": query,
+                "params": [],
+            },
+        )
+        print("Response from execute_sql:", response.content)
+        return
+
+    async def cleanup(self):
+        """Clean up resources"""
+        await self.exit_stack.aclose()
+
+
+async def main():
+    client = MCPClient()
+    try:
+        await client.connect_to_server()
+        await client.session.set_logging_level("debug")
+        await client.process_query("select $namespace, $zversion")
+    finally:
+        await client.cleanup()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

src/mcp_server_iris/mcpserver.py

@@ -98,6 +98,8 @@ def __init__(
         )
 
         self.settings = Settings(**settings)
+        configure_logging(self.settings.log_level)
+        logger.setLevel(self.settings.log_level.upper())
 
         self._tool_manager = ToolManager(
             warn_on_duplicate_tools=self.settings.warn_on_duplicate_tools
@@ -110,8 +112,13 @@ def __init__(
         )
 
         self._setup_handlers()
+        self._mcp_server.set_logging_level()(self.set_logging_level)
 
-        configure_logging(self.settings.log_level)
+    async def set_logging_level(self, level) -> None:
+        """Set the logging level for the server."""
+        logger.info(f"Logging level set to {level}")
+        logger.setLevel(level.upper())
+        configure_logging(level=level)
 
     @property
     def name(self) -> str:
@@ -138,7 +145,7 @@ def run(self, transport: Literal["stdio", "sse"] = "stdio") -> None:
         Args:
             transport: Transport protocol to use ("stdio" or "sse")
         """
-
+        logger.info(f"Running server with transport: {transport}")
         TRANSPORTS = Literal["stdio", "sse"]
         if transport not in TRANSPORTS.__args__:  # type: ignore
             raise ValueError(f"Unknown transport: {transport}")

src/mcp_server_iris/server.py

@@ -1,14 +1,12 @@
 import os
-import logging
 from importlib.metadata import version
 from contextlib import asynccontextmanager
 from collections.abc import AsyncIterator
 import mcp.types as types
 import iris as irisnative
-from mcp_server_iris.mcpserver import MCPServer, Context
+from mcp_server_iris.mcpserver import MCPServer, Context, logger
 from mcp_server_iris.interoperability import init as interoperability
 
-logger = logging.getLogger("mcp_server_iris")
 logger.info("Starting InterSystems IRIS MCP Server")
 
 
@@ -22,6 +20,7 @@ def get_db_config():
         "password": os.getenv("IRIS_PASSWORD", "SYS"),
     }
 
+    logger.info("Server configuration: iris://" + config["hostname"] + ":" + str(config["port"]) + "/" + config["namespace"])
     if not all([config["username"], config["password"], config["namespace"]]):
         raise ValueError("Missing required database configuration")