iflow-mcp_yugabyte-yugabytedb-mcp-server 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

{iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info → iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iflow-mcp_yugabyte-yugabytedb-mcp-server
-Version: 1.0.3
+Version: 1.0.4
 Summary: Add your description here
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info/RECORD

@@ -0,0 +1,7 @@
+server.py,sha256=ZDllYa-YCwuY6GUFIksIu6vx6Up1vnTQy-mn1Ig8qDo,9512
+iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info/METADATA,sha256=uNufOUEnt__eFhRmn7XTEaHPZlTEJEFVi7F4g-LpoKo,9655
+iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info/entry_points.txt,sha256=0qR8G3C-z5Jb99iC45ysBZsnVGYgPefWGinP2Bhq9hE,54
+iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info/top_level.txt,sha256=StKOSmRhvWS5IPcvhsDRbtxUTEofJgYFGOu5AAJdSWo,7
+iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.4.dist-info/RECORD,,

server.py

@@ -13,6 +13,9 @@ from fastapi import FastAPI
 from starlette.responses import JSONResponse
 import boto3
 
+# Global config variable
+CONFIG: ServerConfig | None = None
+
 @dataclass
 class AppContext:
     conn: psycopg2.extensions.connection | None = None
@@ -42,13 +45,13 @@ def normalize_pem(pem: str) -> str:
     return pem + "\n"
 
 
-def write_root_cert(config: ServerConfig):
-    if not config.ssl_root_cert_secret_arn:
+def write_root_cert():
+    if not CONFIG or not CONFIG.ssl_root_cert_secret_arn:
         return None
 
     try:
-        sm = boto3.client("secretsmanager", region_name=config.ssl_root_cert_secret_region)
-        resp = sm.get_secret_value(SecretId=config.ssl_root_cert_secret_arn)
+        sm = boto3.client("secretsmanager", region_name=CONFIG.ssl_root_cert_secret_region)
+        resp = sm.get_secret_value(SecretId=CONFIG.ssl_root_cert_secret_arn)
         secret_string = resp["SecretString"]
 
         # If raw PEM, just use it
@@ -57,10 +60,10 @@ def write_root_cert(config: ServerConfig):
         else:
             data = json.loads(secret_string)
 
-            if config.ssl_root_cert_key:
-                if config.ssl_root_cert_key not in data:
-                    raise RuntimeError(f"Certificate key '{config.ssl_root_cert_key}' not found in secret")
-                pem = data[config.ssl_root_cert_key]
+            if CONFIG.ssl_root_cert_key:
+                if CONFIG.ssl_root_cert_key not in data:
+                    raise RuntimeError(f"Certificate key '{CONFIG.ssl_root_cert_key}' not found in secret")
+                pem = data[CONFIG.ssl_root_cert_key]
             else:
                 # Backward-compatible: allow exactly one entry
                 if len(data) != 1:
@@ -70,10 +73,10 @@ def write_root_cert(config: ServerConfig):
                 pem = next(iter(data.values()))
         
         pem = normalize_pem(pem)
-        with open(config.ssl_root_cert_path, "w") as f:
+        with open(CONFIG.ssl_root_cert_path, "w") as f:
             f.write(pem.strip() + "\n")
 
-        return config.ssl_root_cert_path
+        return CONFIG.ssl_root_cert_path
 
     except Exception as e:
         print(f"Failed to load root cert from Secrets Manager: {e}", file=sys.stderr)
@@ -81,16 +84,17 @@ def write_root_cert(config: ServerConfig):
 
 
 @asynccontextmanager
-async def app_lifespan(server: FastMCP, config: ServerConfig) -> AsyncIterator[AppContext]:
+async def app_lifespan(server: FastMCP) -> AsyncIterator[AppContext]:
+    global CONFIG
     # Skip database connection for testing if YUGABYTEDB_URL is not set
-    if not config.yugabytedb_url:
+    if not CONFIG or not CONFIG.yugabytedb_url:
         print("YUGABYTEDB_URL is not set, running in demo mode without database connection", file=sys.stderr)
         yield AppContext(conn=None)
         return
 
     print("Connecting to database...", file=sys.stderr)
-    database_url = config.yugabytedb_url
-    cert_path = write_root_cert(config)
+    database_url = CONFIG.yugabytedb_url
+    cert_path = write_root_cert()
     if cert_path and "sslrootcert" not in database_url:
         database_url += f" sslrootcert={cert_path}"
     conn = psycopg2.connect(database_url)
@@ -106,8 +110,6 @@ def summarize_database(ctx: Context, schema: str = "public") -> List[Dict[str, A
     Summarize the database: list tables with schema and row counts.
     """
     summary = []
-    # Get config from context
-    config = ctx.request_context.lifespan_context.config
     conn = ctx.request_context.lifespan_context.conn
     
     # If no database connection, return demo data
@@ -227,13 +229,13 @@ def parse_config() -> ServerConfig:
 
 
 class YugabyteDBMCPServer:
-    def __init__(self, config: ServerConfig):
-        self.config = config
+    def __init__(self):
+        global CONFIG
         self.mcp = FastMCP(
             "yugabytedb-mcp",
-            lifespan=lambda server: app_lifespan(server, config),
+            lifespan=app_lifespan,
             json_response=True,
-            stateless_http=config.stateless_http,
+            stateless_http=CONFIG.stateless_http if CONFIG else False,
         )
 
         self._register_tools()
@@ -243,7 +245,7 @@ class YugabyteDBMCPServer:
         self.mcp.add_tool(run_read_only_query)
 
     def run(self, host="0.0.0.0", port=8000):
-        if self.config.transport == "http":
+        if CONFIG and CONFIG.transport == "http":
             self._run_http(host, port)
         else:
             self.mcp.run(transport="stdio")
@@ -268,11 +270,12 @@ class YugabyteDBMCPServer:
 
 if __name__ == "__main__":
     CONFIG = parse_config()
-    server = YugabyteDBMCPServer(CONFIG)
+    server = YugabyteDBMCPServer()
     server.run()
 
 
 def main():
+    global CONFIG
     CONFIG = parse_config()
-    server = YugabyteDBMCPServer(CONFIG)
+    server = YugabyteDBMCPServer()
     server.run()

iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info/RECORD

@@ -1,7 +0,0 @@
-server.py,sha256=bc2juN3YT-P6yla0kglOyxeEAvpM4mcAqz_ySgUzwDc,9567
-iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info/METADATA,sha256=RHm9yfD5pPEv1VWSW4RinmmfCxSY9hy-vVxCbDI7bqY,9655
-iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info/entry_points.txt,sha256=0qR8G3C-z5Jb99iC45ysBZsnVGYgPefWGinP2Bhq9hE,54
-iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info/top_level.txt,sha256=StKOSmRhvWS5IPcvhsDRbtxUTEofJgYFGOu5AAJdSWo,7
-iflow_mcp_yugabyte_yugabytedb_mcp_server-1.0.3.dist-info/RECORD,,