Merge pull request #181 from Teradata/embed-config-files-in-the-build

Embed config files in the build

Author: remi-td

.gitignore

@@ -10,5 +10,6 @@ working/
 env
 CLAUDE.md
 var/
-profiles.yml
-*_objects.yml
+/# Only ignore root-level overrides
+/profiles.yml
+/*_objects.yml

MANIFEST.in

@@ -0,0 +1,3 @@
+include README.md
+include LICENSE
+recursive-include src/teradata_mcp_server *.py *.yml

docs/server_guide/QUICK_START.md

@@ -31,8 +31,7 @@ uvx teradata-mcp-server --help
 If that works, test with your database:
 
 ```bash
-export DATABASE_URI="teradata://username:password@host:1025/database"
-uvx teradata-mcp-server
+uvx teradata-mcp-server --database_uri "teradata://username:password@host:1025/database"
 ```
 
 You should see "NFO     Starting MCP server 'teradata-mcp-server' with transport 'stdio'" messages. Press `Ctrl+C` to stop.
@@ -91,6 +90,7 @@ You should see Claude connect to your Teradata database and return results!
 **Server won't start?**
 - Check your `DATABASE_URI` format
 - Verify database connectivity: `ping your-host`
+- Rollback to a prior version changing your config file: `"args": ["teradata-mcp-server==0.1.3", "--profile", "all"],`
 - See [Installation Guide](INSTALLATION.md) for alternative methods
 
 **Claude can't see tools?**

examples/app-voice-agent/profiles.yml

@@ -33,10 +33,23 @@ profiles:
       You will help the user analyze sales data, retrieve customer information, and provide insights based on the tools available.
       Your responses should be clear and concise and always grounded in the data retrieved.
       When reading order numbers, please read digits by two or individually, separated by pauses. For example, order #1234567 should be read as 'twelve, thirty-four, fifty-six, seven'.
+      When you fetch detailed data, DO NOT share contact information unless explicitely asked for: for example when you get a customer profile containing an email address and phone number, give the customer name, key metrics, and say "I also have their customer email address and phone number if you need it", and wait for the user confirmation to share the details.
       Do not share technical details of your tool interactions and do not use technical jargon or spell out technical attribute names, only the results. For example if you are using a tool to track an customer and get a <customer_key>, 'I identified the customer, it is the customer ' and then read the <customer_key> number or simply the name if you have it.
     language: "fr"
     voice_id: "ambre"
-    
+
+  sales:
+    mcp_server_url: "http://127.0.0.1:8001/mcp"
+    system_prompt: |
+      You are a friendly analyst supporting sales operations. The user and you will engage in a spoken dialog exchanging the transcripts of a natural real-time conversation.
+      You will help the user analyze sales data, retrieve customer information, and provide insights based on the tools available.
+      Your responses should be clear and concise and always grounded in the data retrieved.
+      When reading numbers longer than 4 digits, please read digits individually, separated by pauses, otherwise read the number normally. For example, customer_key #36 should be read as 'customer number thirty-six'; order_nb #34543 should be read as 'order number three, four, five, four, three'.
+      Do not share technical details of your tool interactions and do not use technical jargon or spell out technical attribute names, only the results. For example if you are using a tool to track an customer and get a <customer_key>, 'I identified the customer, it is the customer ' and then read the <customer_key> number or simply the name if you have it.
+      When you fetch detailed data, DO NOT share contact information unless explicitely asked for: for example when you get a customer profile containing an email address and phone number, give the customer name, key metrics, and say "I also have their customer email address and phone number if you need it", and wait for the user confirmation to share the details.
+    language: "en"
+    voice_id: "matthew"
+
   # Business analyst profile with custom MCP prompt
   business_analyst:
     mcp_server_url: "http://127.0.0.1:8001/mcp"

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "teradata-mcp-server"
-version = "0.1.5"
+version = "0.1.5.post1"
 description = "Model Context Protocol (MCP) server for Teradata, Community edition"
 readme = {file = "README.md", content-type = "text/markdown"}
 license = {text = "MIT"}
@@ -23,7 +23,7 @@ classifiers = [
 ]
 dependencies = [
     "pydantic>=2.8.0,<3.0.0",
-    "fastmcp==2.11.3",
+    "fastmcp>=2.12.0,<3.0.0",
     "mcp[cli]>=1.0.0,<2.0.0",
     "teradatasqlalchemy>=20.0.0.0",
     "python-dotenv>=1.0.0",

src/teradata_mcp_server/tools/module_loader.py

@@ -151,26 +151,31 @@ def get_all_functions(self) -> dict[str, Any]:
 
         return all_functions
 
-    def get_required_yaml_paths(self) -> list[str]:
+    def get_required_yaml_paths(self) -> list:
         """
         Get the paths to YAML files for only the required modules.
         
         Returns:
-            List of file paths for YAML files that should be loaded
+            List of file paths/resources for YAML files that should be loaded
         """
-        import glob
-        import os
+        from importlib.resources import files as pkg_files
 
         yaml_paths = []
-        base_path = os.path.dirname(__file__)
-
-        for module_name in self._required_modules:
-            if module_name in self.MODULE_MAP:
-                # Get YAML files for this specific module
-                module_dir = os.path.join(base_path, module_name)
-                if os.path.exists(module_dir):
-                    pattern = os.path.join(module_dir, "*.yml")
-                    yaml_paths.extend(glob.glob(pattern))
+        
+        try:
+            tools_pkg_root = pkg_files("teradata_mcp_server").joinpath("tools")
+            if tools_pkg_root.is_dir():
+                for module_name in self._required_modules:
+                    if module_name in self.MODULE_MAP:
+                        module_dir = tools_pkg_root.joinpath(module_name)
+                        if module_dir.is_dir():
+                            for entry in module_dir.iterdir():
+                                if entry.is_file() and entry.name.endswith('.yml'):
+                                    yaml_paths.append(entry)
+        except Exception as e:
+            import logging
+            logger = logging.getLogger("teradata_mcp_server")
+            logger.error(f"Failed to load packaged YAML files: {e}")
 
         return yaml_paths
 

src/teradata_mcp_server/utils.py

@@ -161,22 +161,46 @@ def load_profiles(working_dir: Optional[Path] = None) -> Dict[str, Any]:
 
     # Load packaged profiles.yml
     try:
-        config_files = pkg_files("teradata_mcp_server.config")
-        profiles_file = config_files / "profiles.yml"
-        if profiles_file.is_file():
-            profiles.update(yaml.safe_load(profiles_file.read_text(encoding='utf-8')) or {})
+        import importlib.resources
+        try:
+            # Try the new importlib.resources API (Python 3.9+)
+            config_files = importlib.resources.files("teradata_mcp_server.config")
+            profiles_file = config_files / "profiles.yml"
+            logger.debug(f"Looking for packaged profiles at: {profiles_file}")
+            if profiles_file.is_file():
+                packaged_profiles = yaml.safe_load(profiles_file.read_text(encoding='utf-8')) or {}
+                profiles.update(packaged_profiles)
+                logger.info(f"Loaded {len(packaged_profiles)} packaged profiles: {list(packaged_profiles.keys())}")
+            else:
+                logger.warning(f"Packaged profiles.yml not found at: {profiles_file}")
+        except AttributeError:
+            # Fallback for older Python versions
+            import importlib.resources as resources
+            with resources.path("teradata_mcp_server.config", "profiles.yml") as profiles_path:
+                logger.debug(f"Looking for packaged profiles at: {profiles_path}")
+                if profiles_path.exists():
+                    packaged_profiles = yaml.safe_load(profiles_path.read_text(encoding='utf-8')) or {}
+                    profiles.update(packaged_profiles)
+                    logger.info(f"Loaded {len(packaged_profiles)} packaged profiles: {list(packaged_profiles.keys())}")
+                else:
+                    logger.warning(f"Packaged profiles.yml not found at: {profiles_path}")
     except Exception as e:
-        logger.error(f"Failed to load packaged profiles: {e}")
+        logger.error(f"Failed to load packaged profiles: {e}", exc_info=True)
 
     # Load working directory profiles.yml (overrides packaged)
     profiles_path = working_dir / "profiles.yml"
     if profiles_path.exists():
         try:
             with open(profiles_path, encoding='utf-8') as f:
-                profiles.update(yaml.safe_load(f) or {})
+                working_dir_profiles = yaml.safe_load(f) or {}
+                profiles.update(working_dir_profiles)
+                logger.info(f"Loaded {len(working_dir_profiles)} working directory profiles: {list(working_dir_profiles.keys())}")
         except Exception as e:
             logger.error(f"Failed to load external profiles: {e}")
+    else:
+        logger.debug(f"No working directory profiles.yml found at: {profiles_path}")
 
+    logger.info(f"Total profiles loaded: {list(profiles.keys())}")
     return profiles