From 05d111def6fc56c09166b54f047c0f8a84335a7f Mon Sep 17 00:00:00 2001
From: mykonos-ibiza <222371740+mykonos-ibiza@users.noreply.github.com>
Date: Fri, 1 Aug 2025 01:17:32 +0530
Subject: [PATCH] Refactor KortixMCP initialization and tool management

- Updated the constructor of `KortixMCP` to accept a `FastMCPClient` and a list of allowed tools.
- Modified the `initialize` method to retrieve tools from the new client and filter enabled tools based on the allowed list.
- Improved code readability and organization by renaming variables for clarity.
---
 sdk/kortix/tools.py | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/sdk/kortix/tools.py b/sdk/kortix/tools.py
index 2f720b4f..da7171db 100644
--- a/sdk/kortix/tools.py
+++ b/sdk/kortix/tools.py
@@ -1,22 +1,26 @@
 from typing import Union
 from enum import Enum
-from fastmcp import FastMCP
+from fastmcp import Client as FastMCPClient
 
 
 class KortixMCP:
-    def __init__(self, mcp: FastMCP, endpoint: str):
-        self._fastmcp = mcp
+    def __init__(
+        self, mcp_client: FastMCPClient, endpoint: str, allowed_tools: list[str]
+    ):
+        self._mcp_client = mcp_client
         self.url = endpoint
         self._initialized = False
+        self._allowed_tools = allowed_tools
+        self._enabled_tools: list[str] = []
 
     async def initialize(self):
-        self.name = self._fastmcp.name
-        self.type = "http"
+        async with self._mcp_client:
+            tools = await self._mcp_client.list_tools()
+
         self.enabled_tools: list[str] = []
-        tools = await self._fastmcp.get_tools()
-        for tool in tools.values():
-            if tool.enabled:
-                self.enabled_tools.append(tool.name)
+        for tool in tools:
+            if tool.name in self._allowed_tools:
+                self._enabled_tools.append(tool.name)
         self._initialized = True
         return self