From 74bfbc04ca572380e5520a9cbf82833c28bcce98 Mon Sep 17 00:00:00 2001
From: Kamil Potrec <kamil@keycard.ai>
Date: Mon, 15 Sep 2025 17:01:15 +0100
Subject: [PATCH] feat(keycardai-mcp-fastmcp): unify client arguments

---
 .../src/keycardai/mcp/integrations/fastmcp/middleware.py       | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/packages/mcp-fastmcp/src/keycardai/mcp/integrations/fastmcp/middleware.py b/packages/mcp-fastmcp/src/keycardai/mcp/integrations/fastmcp/middleware.py
index 55e0f3a2..67e63f1c 100644
--- a/packages/mcp-fastmcp/src/keycardai/mcp/integrations/fastmcp/middleware.py
+++ b/packages/mcp-fastmcp/src/keycardai/mcp/integrations/fastmcp/middleware.py
@@ -85,6 +85,9 @@ def __init__(
             zone_url: OAuth server zone URL (from environment if not provided)
             client_name: OAuth client name for registration
         """
+        if zone_url is None and client is not None:
+            zone_url = client.base_url
+
         settings = AccessMiddlewareSettings.model_validate({
             "zone_url": zone_url,
             "client_name": client_name,