ibm-watsonx-orchestrate-mcp-server 1.13.0b0__py3-none-any.whl

ibm_watsonx_orchestrate_mcp_server/utils/common.py

@@ -0,0 +1,108 @@
+import io
+import logging
+from typing import Any, Callable
+from contextlib import redirect_stdout, redirect_stderr, ExitStack
+
+
+# Add this at the module level to store original logger states
+_original_logger_states = {}
+
+def __save_all_logger_states():
+    """Save the original state of all loggers."""
+    global _original_logger_states
+    _original_logger_states = {}
+    
+    # Save state for named loggers
+    for name, logger in logging.root.manager.loggerDict.items():
+        if isinstance(logger, logging.Logger):
+            _original_logger_states[name] = {
+                'level': logger.level,
+                'handlers': list(logger.handlers),  # Make a copy of the handlers list
+                'propagate': logger.propagate
+            }
+    
+    # Save state for root logger
+    root_logger = logging.getLogger()
+    _original_logger_states['root'] = {
+        'level': root_logger.level,
+        'handlers': list(root_logger.handlers)  # Root logger doesn't have propagate
+    }
+
+def __reset_all_loggers():
+    """Reset all loggers to their original state."""
+    global _original_logger_states
+    
+    # Reset named loggers
+    for name, logger in logging.root.manager.loggerDict.items():
+        if isinstance(logger, logging.Logger):
+            # Clear current handlers
+            logger.handlers.clear()
+            
+            # Restore original state if we have it
+            if name in _original_logger_states:
+                state = _original_logger_states[name]
+                logger.setLevel(state['level'])
+                for handler in state['handlers']:
+                    logger.addHandler(handler)
+                logger.propagate = state['propagate']
+            else:
+                # Default behavior if we don't have saved state
+                logger.propagate = True
+    
+    # Reset root logger
+    root_logger = logging.getLogger()
+    root_logger.handlers.clear()
+    
+    # Restore original root logger state
+    if 'root' in _original_logger_states:
+        state = _original_logger_states['root']
+        root_logger.setLevel(state['level'])
+        for handler in state['handlers']:
+            root_logger.addHandler(handler)
+
+
+
+def silent_call(fn: Callable, *args, suppress_stdout: bool = True,
+               suppress_stderr: bool = False, suppress_logging: bool = True, **kwargs) -> Any:
+    """
+    Call a function silently, suppressing stdout and/or stderr output.
+    
+    Args:
+        fn: The function to call
+        *args: Positional arguments to pass to the function
+        suppress_stdout: Whether to suppress stdout (default: True)
+        suppress_stderr: Whether to suppress stderr (default: False)
+        suppress_logging: Whether to suppress logging output (default: True)
+        **kwargs: Keyword arguments to pass to the function
+        
+    Returns:
+        The return value of the called function
+    """
+    with ExitStack() as stack:
+        null_stream = io.StringIO()
+        
+        if suppress_stdout:
+            stack.enter_context(redirect_stdout(null_stream))
+        if suppress_stderr:
+            stack.enter_context(redirect_stderr(null_stream))
+        
+        if suppress_logging:
+            # Save original logger states before modifying
+            __save_all_logger_states()
+            
+            stream_handler = logging.StreamHandler(null_stream)
+
+            # Patch all known loggers
+            for name, logger in logging.root.manager.loggerDict.items():
+                if isinstance(logger, logging.Logger):
+                    logger.handlers.clear()
+                    logger.addHandler(stream_handler)
+                    logger.propagate = False            
+
+        try:
+            return fn(*args, **kwargs)
+        except SystemExit:
+            raise Exception(f"{null_stream.getvalue()}")
+        finally:
+            if suppress_logging:
+                __reset_all_loggers()

ibm_watsonx_orchestrate_mcp_server/utils/config/__init__.py

@@ -0,0 +1,2 @@
+from .config import config
+from .types import MCPTransportTypes

ibm_watsonx_orchestrate_mcp_server/utils/config/config.py

@@ -0,0 +1,91 @@
+import os
+import tomllib
+from typing import Any
+from importlib.metadata import version
+
+DEFAULT_SERVER_NAME: str = "WXO-MCP"
+DEFAULT_VERSION: str = "Unknown"
+DEFAULT_DESCRIPTION: str = "MCP Server used to interact with watsonx Orchestrate"
+
+DEFAULT_HOST = "127.0.0.1"  # Secure default - localhost only
+DEFAULT_PORT = 8080
+DEFAULT_TRANSPORT = "stdio"  # Options: "stdio", "http", "sse" TODO: Make enum
+
+def _find_pyproject_toml() -> str | None:
+    """
+    Locate the pyproject.toml file by traversing the directory tree upwards until found.
+
+    Returns:
+        The path to the pyproject.toml file if found, otherwise None.
+    """
+
+    current: str = os.path.abspath(os.path.dirname(__file__))
+    while True:
+        current_toml = os.path.join(current, "pyproject.toml")
+        if os.path.isfile(current_toml):
+            return current_toml
+        parent = os.path.dirname(current)
+        if parent == current:
+            raise FileNotFoundError("pyproject.toml not found")
+        current = parent
+
+def _get_pyproject_configuration() -> dict:
+    """
+    Load configuration from the pyproject.toml file.
+
+    Returns:
+        A dictionary containing the configuration values from the pyproject.toml file.
+        If no configuration is found, an empty dictionary is returned.
+    """
+    toml_path: str | None = _find_pyproject_toml()
+    if not toml_path:
+        return {}
+    try:
+        with open(toml_path, "rb") as f:
+            return tomllib.load(f).get("project", {})
+    except FileNotFoundError:
+        return {}
+
+class Config:
+    """Centralized configuration manager."""
+    
+    def __init__(self):
+        pyproject_config: dict[Any, Any] = _get_pyproject_configuration()
+
+        self.server_name: str = pyproject_config.get("name", DEFAULT_SERVER_NAME)
+        try:
+            self.version: str = version("ibm-watsonx-orchestrate-mcp-server")
+        except:
+            self.version: str = DEFAULT_VERSION
+        self.description: str = pyproject_config.get("description", DEFAULT_DESCRIPTION)
+        
+        # Network
+        self.host: str = os.getenv("WXO_MCP_HOST", DEFAULT_HOST)
+        self.port: int = int(os.getenv("WXO_MCP_PORT", str(DEFAULT_PORT)))
+        self.transport: str = os.getenv("WXO_MCP_TRANSPORT", DEFAULT_TRANSPORT).lower()
+        
+        # Development
+        self.debug:bool = os.getenv("WXO_MCP_DEBUG", "false").lower() == "true"
+
+        # Application
+        self.working_directory = os.getenv("WXO_MCP_WORKING_DIRECTORY")
+    
+    def __repr__(self) -> str:
+        return f"""
+        ========================================================
+        Server Name: {self.server_name}
+        Version: {self.version}
+        Description: {self.description}
+        ========================================================
+        Transport: {self.transport}
+        Network:
+            Host: {self.host}
+            Port: {self.port}
+        ========================================================
+        Debug: {self.debug}
+        ========================================================
+        DEFAULT_WORKING_DIRECTORY: {self.working_directory}
+        ========================================================
+        """
+
+config: Config = Config()

ibm_watsonx_orchestrate_mcp_server/utils/config/types.py

@@ -0,0 +1,6 @@
+from enum import Enum
+
+class MCPTransportTypes(str, Enum):
+    STDIO = "stdio"
+    HTTP_STREAMABLE = "http"
+    SSE = "sse"

ibm_watsonx_orchestrate_mcp_server/utils/files/files.py

@@ -0,0 +1,14 @@
+from pathlib import Path
+from ibm_watsonx_orchestrate_mcp_server.utils.config.config import config
+
+def get_working_directory_path(path_string: str) -> str:
+    if config.working_directory is None:
+        raise Exception("No working directory defined. Access to file system is blocked. To configure working directory expose the environment variable 'WXO_MCP_WORKING_DIRECTORY'")
+    
+    path: Path = Path(path_string).resolve()
+    working_directory: Path = Path(config.working_directory).resolve()
+
+    try:
+        return str(path.relative_to(working_directory))
+    except ValueError:
+        raise Exception("Attempting to access resources outside the working directory is forbidden.")

ibm_watsonx_orchestrate_mcp_server/utils/logging/__init__.py

@@ -0,0 +1 @@
+from .logger import setup_logging

ibm_watsonx_orchestrate_mcp_server/utils/logging/log_config.yaml

@@ -0,0 +1,19 @@
+version: 1
+disable_existing_loggers: false
+
+formatters:
+  standard:
+    format: "%(levelname)s - %(message)s"
+
+handlers:
+  console:
+    class: logging.StreamHandler
+    level: DEBUG
+    formatter: standard
+    stream: ext://sys.stderr
+
+loggers:
+  wxo-mcp:
+    level: DEBUG
+    handlers: [console]
+    propagate: false

ibm_watsonx_orchestrate_mcp_server/utils/logging/logger.py

@@ -0,0 +1,38 @@
+import logging
+import logging.config
+import os
+import yaml
+from enum import Enum
+
+class LogColors(str, Enum):
+    INFO = "\033[0;36m" #cyan
+    DEBUG = "\033[0;35m" #magenta
+    WARNING = "\033[0;33m" #yellow
+    ERROR = "\033[0;31m" #red
+    RESET = "\033[0;0m"
+
+
+def setup_logging(debug=False):
+    """Setup logging configuration
+    
+    Args:
+        debug (bool, optional): Whether to enable debug logging. Defaults to False.
+    """
+    config_path = os.path.join(os.path.dirname(__file__), "log_config.yaml")
+    with open(config_path, "r") as f:
+        config = yaml.safe_load(f)
+    
+    # Adjust root logger level based on debug parameter
+    if not debug:
+        # Set console handler to INFO level if debug is False
+        for _, handler in config.get('handlers', {}).items():
+            if handler.get('class') == 'logging.StreamHandler':
+                handler['level'] = 'INFO'
+    
+    logging.config.dictConfig(config)
+
+     # Add log colors
+    logging.addLevelName( logging.INFO, LogColors.INFO + f"[{logging.getLevelName(logging.INFO)}]" + LogColors.RESET)
+    logging.addLevelName( logging.DEBUG, LogColors.DEBUG + f"[{logging.getLevelName(logging.DEBUG)}]" + LogColors.RESET)
+    logging.addLevelName( logging.WARNING, LogColors.WARNING + f"[{logging.getLevelName(logging.WARNING)}]" + LogColors.RESET)
+    logging.addLevelName( logging.ERROR, LogColors.ERROR + f"[{logging.getLevelName(logging.ERROR)}]" + LogColors.RESET)

ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: ibm-watsonx-orchestrate-mcp-server
+Version: 1.13.0b0
+Summary: IBM watsonx Orchestrate ADK MCP Server
+Author-email: IBM <support@ibm.com>
+License: MIT License
+Requires-Python: <3.14,>=3.11
+Requires-Dist: authlib>=1.6.3
+Requires-Dist: fastmcp~=2.10
+Requires-Dist: ibm-watsonx-orchestrate
+Requires-Dist: pyyaml<7.0.0,>=6.0.2
+Requires-Dist: typing-extensions>=4.4.0

ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/RECORD

@@ -0,0 +1,31 @@
+./.gitignore,sha256=cwcrsXHa5M1CLvBt7W3FLclADzVFBR_4wXfO9c6jRCM,146
+./__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+./pyproject.toml,sha256=LptPuPmDSYvrg0iUsb0pIhcEwXQWzy-e__Y4iDWAakI,927
+./ibm_watsonx_orchestrate_mcp_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+./ibm_watsonx_orchestrate_mcp_server/server.py,sha256=cjmVZ8hUGHJUW9O8NpALzQm9eGSzrtTNK2OLzaGqsI4,1547
+./ibm_watsonx_orchestrate_mcp_server/src/__init__.py,sha256=VoM9wg6T_ggOhX7VsKJip7H-ghaXYubu43GCeCP-qak,645
+./ibm_watsonx_orchestrate_mcp_server/src/version_checker.py,sha256=XNsG5xp_Gu-JsPdWkjYMNC0Qxew0JR52hd2MBmLrRaU,216
+./ibm_watsonx_orchestrate_mcp_server/src/agents/mcp_tools.py,sha256=jBV3tkoApnffWY9d6sk96MRiHOBEnO1yyX1gU98SeAg,6862
+./ibm_watsonx_orchestrate_mcp_server/src/agents/types.py,sha256=G3c4iKM8lZ11AU-w1VKz3s8xtnjRTp6Vp7Db9Hb5noA,5269
+./ibm_watsonx_orchestrate_mcp_server/src/connections/helpers.py,sha256=3pedZYlEc3WueUBTlWqeS6qtWp19_UIZw6utcAfEhrE,1465
+./ibm_watsonx_orchestrate_mcp_server/src/connections/mcp_tools.py,sha256=CBAKq0fKHSi7ZiTgupGJLIy2IdVzo2jjkkkSjOf-xqI,5903
+./ibm_watsonx_orchestrate_mcp_server/src/connections/types.py,sha256=mjD8dNWVeK2giTi1h0cXd9V0zoJtbRUcTxd5Bd4AW7s,6727
+./ibm_watsonx_orchestrate_mcp_server/src/knowledge_bases/mcp_tools.py,sha256=TVzTPkw6Ixg-9lmEGPsxDgOq4YnK6fyGdLueIFggDPY,3362
+./ibm_watsonx_orchestrate_mcp_server/src/models/mcp_tools.py,sha256=J5fBCSyt1QpVDGkV7c8ZWX81QvwgPceRnXbX6Ia7UAY,8734
+./ibm_watsonx_orchestrate_mcp_server/src/models/types.py,sha256=Us4uiqxlWaabJu50O8I1VVOIw7fNjvsKjlYcciqE9vg,2686
+./ibm_watsonx_orchestrate_mcp_server/src/toolkits/mcp_tools.py,sha256=5mXzwkQ-SPBYlD3OpD2Vn2-BcEGqOs4YcwQs6LOLZIY,2864
+./ibm_watsonx_orchestrate_mcp_server/src/toolkits/types.py,sha256=44_W7KTGzs-8f9wjLrhkRU09NUxO8TG7IvGz5gn5XFk,2425
+./ibm_watsonx_orchestrate_mcp_server/src/tools/mcp_tools.py,sha256=YJ4a3Mfv_9FhHa9C3IRScALW_cAp0HgITT-B5l9Tz6A,6873
+./ibm_watsonx_orchestrate_mcp_server/src/voice_configurations/mcp_tools.py,sha256=gMdTHuwtjdK8HcZaGQNRwViL3uQNF6RgQ3UaJkPIFWw,2667
+./ibm_watsonx_orchestrate_mcp_server/utils/common.py,sha256=naRTpvgufoX8e5AOO7p6D43O8kaOOZLiMePEU8_Hg3E,3902
+./ibm_watsonx_orchestrate_mcp_server/utils/config/__init__.py,sha256=kPWW3exDC8gxfpExOMgnELu04Y-7rm9Zk-DIrkZqQLs,63
+./ibm_watsonx_orchestrate_mcp_server/utils/config/config.py,sha256=sATsXhzCqNyCF70YaqG-s5MhXu2Ohsdiltg_awmNrFg,3222
+./ibm_watsonx_orchestrate_mcp_server/utils/config/types.py,sha256=r9c6NGYf8rbUc4F0A9_ywEHMxGuWfiA5hzhbZ3YQXaw,123
+./ibm_watsonx_orchestrate_mcp_server/utils/files/files.py,sha256=wCjWachGS_haNw7ZtCcotjDbD56sXChG0RCml5x2lZU,692
+./ibm_watsonx_orchestrate_mcp_server/utils/logging/__init__.py,sha256=ZLPI0klEH2WFlrZ4BaPYLHRpWgOFRoVf5edR5eJIisQ,33
+./ibm_watsonx_orchestrate_mcp_server/utils/logging/log_config.yaml,sha256=nKkUdI0y-4kkoK7K7hSaiksYUdkTEj_yKT8ksNhsPLI,318
+./ibm_watsonx_orchestrate_mcp_server/utils/logging/logger.py,sha256=UWOoLCH1Acl641Z_Q2f949C2JvdKDutG8hKULniN4Iw,1450
+ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/METADATA,sha256=Y5YvhBiX0QEjLfTSk5Go1ppGynFpY5rcxA-3TnCyZbk,390
+ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/entry_points.txt,sha256=z9lzFFCOEZ9lX00T37YHJKnPfkKtaSA80iKe8LS0OPw,110
+ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/RECORD,,

ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

ibm_watsonx_orchestrate_mcp_server-1.13.0b0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ibm-watsonx-orchestrate-mcp-server = ibm_watsonx_orchestrate_mcp_server.server:start_server

pyproject.toml

@@ -0,0 +1,39 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "ibm-watsonx-orchestrate-mcp-server"
+dynamic = ["version"]
+description = "IBM watsonx Orchestrate ADK MCP Server"
+authors = [
+    { name = "IBM", email = "support@ibm.com" }
+]
+#readme = { file = "README.md", content-type = 'text/markdown'}
+license = {text = "MIT License"}
+license-files = ["LICENSE"]
+requires-python = ">=3.11, <3.14"
+dependencies = [
+    "typing-extensions>=4.4.0",
+    "pyyaml>=6.0.2,<7.0.0",
+    "fastmcp~=2.10",
+    "ibm-watsonx-orchestrate",
+    "authlib>=1.6.3",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["."]
+package-data = true
+
+[tool.hatch.build]
+include = [
+    "./*"
+]
+directory="../../dist"
+
+[tool.hatch.version]
+path = "../../src/ibm_watsonx_orchestrate/__init__.py"
+validate-bump=false
+
+[project.scripts]
+ibm-watsonx-orchestrate-mcp-server = "ibm_watsonx_orchestrate_mcp_server.server:start_server"