Move to official FastMCP, add lengthy prompt to improve SQL generation (#8)

* update mcp, fastmcp and tp connect version

* replace fastmcp with mcp

* adding 2k lines of prompt

* Update prompt_template.py

* 0.1.4 and update README

* Update uv.lock

Author: Jove Zhong

README.md

@@ -7,6 +7,10 @@ An MCP server for Timeplus.
 
 ## Features
 
+### Prompts
+
+* `generate_sql` to give LLM more knowledge about how to query Timeplus via SQL
+
 ### Tools
 
 * `run_sql`
@@ -92,7 +96,7 @@ TIMEPLUS_KAFKA_CONFIG={"bootstrap.servers":"a.aivencloud.com:28864", "sasl.mecha
 
 3. Run `uv sync` to install the dependencies. Then do `source .venv/bin/activate`.
 
-4. For easy testing, you can run `fastmcp dev mcp_timeplus/mcp_server.py` to start the MCP server. Click the "Connect" button to connect the UI with the MCP server, then switch to the "Tools" tab to run the available tools.
+4. For easy testing, you can run `mcp dev mcp_timeplus/mcp_server.py` to start the MCP server. Click the "Connect" button to connect the UI with the MCP server, then switch to the "Tools" tab to run the available tools.
 
 ### Environment Variables
 

mcp_timeplus/__init__.py

@@ -6,6 +6,7 @@
     list_kafka_topics,
     explore_kafka_topic,
     create_kafka_stream,
+    generate_sql,
 )
 
 __all__ = [
@@ -16,4 +17,5 @@
     "list_kafka_topics",
     "explore_kafka_topic",
     "create_kafka_stream",
+    "generate_sql",
 ]

mcp_timeplus/mcp_server.py

@@ -4,10 +4,11 @@
 import timeplus_connect
 from timeplus_connect.driver.binding import quote_identifier, format_query_value
 from dotenv import load_dotenv
-from fastmcp import FastMCP
+from mcp.server.fastmcp import FastMCP
 
 MCP_SERVER_NAME = "mcp-timeplus"
 from mcp_timeplus.mcp_env import config
+from mcp_timeplus.prompt_template import TEMPLATE
 
 import json, os, time
 from confluent_kafka.admin import (AdminClient)
@@ -128,6 +129,10 @@ def run_sql(query: str):
         logger.error(f"Error executing query: {err}")
         return f"error running query: {err}"
 
+@mcp.prompt()
+def generate_sql(requirements: str) -> str:
+    return f"Please generate Timeplus SQL for the requirement:\n\n{requirements}\n\nMake sure following the guide {TEMPLATE}"
+
 @mcp.tool()
 def list_kafka_topics():
     logger.info("Listing all topics in the Kafka cluster")