updated init logic

Author: daimor

example.py

@@ -20,10 +20,16 @@ def __init__(self):
     async def connect_to_server(self):
         """Connect to an MCP server"""
         server_params = StdioServerParameters(
-            command=sys.executable, args=["-m", "mcp_server_iris"], env=None
+            command=sys.executable, args=["-m", "mcp_server_iris"], env={
+                "IRIS_HOSTNAME": "localhost",
+                "IRIS_PORT": "1972",
+                "IRIS_NAMESPACE": "USER",
+                "IRIS_USERNAME": "_SYSTEM",
+                "IRIS_PASSWORD": "SYS",
+            }
         )
         # server_params = StdioServerParameters(
-        #     command="uvx", args=["mcp-server-iris"], env=None
+        #     command="uvx", args=["."],
         # )
 
         stdio_transport = await self.exit_stack.enter_async_context(

src/mcp_server_iris/server.py

@@ -13,25 +13,30 @@
 def get_db_config():
     """Get database configuration from environment variables."""
     config = {
-        "hostname": os.getenv("IRIS_HOSTNAME", "localhost"),
+        "hostname": os.getenv("IRIS_HOSTNAME"),
         "port": int(os.getenv("IRIS_PORT", 1972)),
-        "namespace": os.getenv("IRIS_NAMESPACE", "USER"),
-        "username": os.getenv("IRIS_USERNAME", "_SYSTEM"),
-        "password": os.getenv("IRIS_PASSWORD", "SYS"),
+        "namespace": os.getenv("IRIS_NAMESPACE"),
+        "username": os.getenv("IRIS_USERNAME"),
+        "password": os.getenv("IRIS_PASSWORD"),
     }
 
-    logger.info("Server configuration: iris://" + config["hostname"] + ":" + str(config["port"]) + "/" + config["namespace"])
-    if not all([config["username"], config["password"], config["namespace"]]):
+    if not all([config["hostname"], config["username"], config["password"], config["namespace"]]):
         raise ValueError("Missing required database configuration")
+    logger.info(f"Server configuration: iris://{config["username"]}:{"x"*8}@{config["hostname"]}:{config["port"]}/{config["namespace"]}")
 
     return config
 
 
 @asynccontextmanager
 async def server_lifespan(server: MCPServer) -> AsyncIterator[dict]:
     """Manage server startup and shutdown lifecycle."""
-    config = get_db_config()
     try:
+        config = get_db_config()
+    except ValueError:
+        yield {"db": None, "iris": None}
+        return
+    try:
+
         db = irisnative.connect(**config)
         iris = irisnative.createIRIS(db)
         yield {"db": db, "iris": iris}