...

Author: cboettig

README.md

@@ -17,9 +17,7 @@ This server is based on [MotherDuck's DuckDB MCP Server](https://github.com/moth
 
 ### Prompts
 
-The server provides specialized prompts for wetlands data:
-
-`src/mcp_data_server/wetlands-data.md`
+The server accepts custom prompts with --custom-prompt argument.  
 
 ### Tools
 

example-custom-prompt-config.json

@@ -5,9 +5,9 @@
             "args": [
                 "mcp-data-server",
                 "--db-path",
-                "md:",
+                ":memory:",
                 "--custom-prompt",
-                "/home/cboettig/Documents/github/boettiger-lab/mcp-data-server/wetlands-data.md"
+                "custom-data-prompt.md"
             ]
         }
     }

k8s/deployment.yaml

@@ -52,7 +52,7 @@ spec:
             apt-get update && apt-get install -y git && \
             git clone https://github.com/boettiger-lab/mcp-data-server.git /app && \
             cd /app && \
-            uv run mcp-data-server --port 8001 --host 127.0.0.1 --db-path :memory: --transport stream --custom-prompt wetlands-data.md
+            uv run mcp-data-server --port 8001 --host 127.0.0.1 --db-path :memory: --transport stream --custom-prompt custom-data-prompt.md
         resources:
           requests:
             cpu: "4"

src/mcp_data_server/__init__.py

@@ -73,7 +73,7 @@
 @click.option(
     "--custom-prompt",
     default=None,
-    help="Path to a custom prompt markdown file to use instead of the built-in wetlands-data.md. The file will be combined with duckdb-prompt.md to form the system prompt.",
+    help="Path to a custom prompt markdown file to use. The file will be combined with duckdb-prompt.md to form the system prompt.",
 )
 def main(
     port,

src/mcp_data_server/prompt.py

@@ -16,10 +16,9 @@ def get_full_prompt(custom_prompt_path: str | None = None) -> str:
     
     Args:
         custom_prompt_path: Optional path to a custom prompt markdown file.
-                          If provided, this will be used instead of the built-in wetlands-data.md.
     
     Returns:
-        Combined prompt text from custom/wetlands context and DuckDB context.
+        Combined prompt text from custom context and DuckDB context.
     """
     # Load custom prompt if provided, otherwise use built-in wetlands context
     if custom_prompt_path:
@@ -34,6 +33,3 @@ def get_full_prompt(custom_prompt_path: str | None = None) -> str:
     duckdb_context = load_prompt_from_file('duckdb-prompt.md')
 
     return f"{context}\n\n{duckdb_context}"
-
-# For backwards compatibility, export PROMPT_TEMPLATE
-PROMPT_TEMPLATE = get_full_prompt()

src/mcp_data_server/server.py

@@ -30,6 +30,16 @@ def build_application(
 
     # Generate the prompt template with custom prompt if provided
     prompt_template = get_full_prompt(custom_prompt_path)
+    
+    # Set prompt metadata based on whether custom prompt is used
+    if custom_prompt_path:
+        prompt_name = "data-analyst-prompt"
+        prompt_description = "A prompt for analyzing data using DuckDB SQL queries"
+        prompt_result_description = "Data analyst prompt for querying databases"
+    else:
+        prompt_name = "data-descriptions-prompt"
+        prompt_description = "A prompt for analyzing global wetlands data including GLWD, protected areas, carbon storage, species ranges, and more using DuckDB"
+        prompt_result_description = "Wetlands data analyst prompt for querying global wetlands datasets"
     db_client = DatabaseClient(
         db_path=db_path,
         motherduck_token=motherduck_token,
@@ -70,8 +80,8 @@ async def handle_list_prompts() -> list[types.Prompt]:
         logger.info("Listing prompts")
         return [
             types.Prompt(
-                name="wetlands-data-analyst-prompt",
-                description="A prompt for analyzing global wetlands data including GLWD, protected areas, carbon storage, species ranges, and more using DuckDB",
+                name=prompt_name,
+                description=prompt_description,
             )
         ]
 
@@ -81,14 +91,14 @@ async def handle_get_prompt(
     ) -> types.GetPromptResult:
         """
         Generate a prompt by combining arguments with server state.
-        The prompt provides comprehensive wetlands data analysis capabilities.
+        The prompt provides data analysis capabilities using DuckDB.
         """
         logger.info(f"Getting prompt: {name}::{arguments}")
-        if name != "wetlands-data-analyst-prompt":
+        if name != prompt_name:
             raise ValueError(f"Unknown prompt: {name}")
 
         return types.GetPromptResult(
-            description="Wetlands data analyst prompt for querying global wetlands datasets",
+            description=prompt_result_description,
             messages=[
                 types.PromptMessage(
                     role="user",