Update server.py

server.py

@@ -1,9 +1,93 @@
+# import os
+# import uvicorn
+
+# from mcp.server.fastmcp import FastMCP
+# from starlette.requests import Request
+# from starlette.responses import PlainTextResponse, Response
+
+# from langchain_community.utilities import SQLDatabase
+# from langchain_community.tools.sql_database.tool import QuerySQLCheckerTool
+# from langchain_openai import ChatOpenAI
+
+# llm = ChatOpenAI(
+#     api_key=os.environ.get('OPENAI_API_KEY', None),
+#     base_url=os.environ['OPENAI_BASE_URL'],
+#     model='gpt-4o-mini',
+#     temperature=0
+# )
+
+# # Create an MCP server
+# mcp = FastMCP("Credit Card Database Server")
+
+# credit_card_db = SQLDatabase.from_uri(r"sqlite:///data/ccms.db")
+# query_checker_tool = QuerySQLCheckerTool(db=credit_card_db, llm=llm)
+
+
+# @mcp.custom_route("/", methods=["GET"])
+# async def home(request: Request) -> PlainTextResponse:
+#     return PlainTextResponse(
+#         """
+#         Credit Card Database MCP Server
+#         ----
+#         This server gives you access to the following tools:
+#         1. sql_db_list_tables: This tool can be used to list the tables in the database.
+#         2. sql_db_schema: This tool can be used to get the schema of a table.
+#         3. sql_db_query_checker: This tool can be used to check if a query is valid.
+#         4. sql_db_query: This tool can be used to execute a query.
+#         """
+#     )
+
+
+# @mcp.tool()
+# def sql_db_list_tables():
+#     """
+#     Returns a comma-separated list of table names in the database.
+#     """
+#     return credit_card_db.get_usable_table_names()
+
+
+# @mcp.tool()
+# def sql_db_schema(table_names: list[str]) -> str:
+#     """
+#     Input 'table_names_str' is a comma-separated string of table names.
+#     Returns the DDL SQL schema for these tables.
+#     """
+#     return credit_card_db.get_table_info(table_names)
+
+
+# @mcp.tool()
+# def sql_db_query_checker(query: str) -> str:
+#     """
+#     Input 'query' is a SQL query string.
+#     Checks if the query is valid.
+#     If the query is valid, it returns the original query.
+#     If the query is not valid, it returns the corrected query.
+#     This tool is used to ensure the query is valid before executing it.
+#     """
+
+#     return query_checker_tool.run(query)
+
+# @mcp.tool()
+# def sql_db_query(query: str) -> str:
+#     """
+#     Input 'query' is a SQL query string.
+#     Executes the query (SELECT only) and returns the result.
+#     """
+#     return credit_card_db.run(query)
+
+
+# if __name__ == "__main__":
+#     uvicorn.run(mcp.streamable_http_app, host="0.0.0.0", port=8000)
+
+
+
 import os
 import uvicorn
+import inspect
 
 from mcp.server.fastmcp import FastMCP
 from starlette.requests import Request
-from starlette.responses import PlainTextResponse, Response
+from starlette.responses import PlainTextResponse, JSONResponse
 
 from langchain_community.utilities import SQLDatabase
 from langchain_community.tools.sql_database.tool import QuerySQLCheckerTool
@@ -16,37 +100,52 @@ llm = ChatOpenAI(
     temperature=0
 )
 
-# Create an MCP server
+# Create an MCP server and the tool registry
 mcp = FastMCP("Credit Card Database Server")
+tool_registry = []
+
+def register_tool(fn):
+    """Decorator to register tool metadata and with MCP."""
+    # Register with MCP
+    mcp.tool()(fn)
+    # Save metadata
+    sig = inspect.signature(fn)
+    params = [
+        {
+            "name": param.name,
+            "type": str(param.annotation) if param.annotation is not inspect._empty else "Any",
+            "default": param.default if param.default is not inspect._empty else None,
+        }
+        for param in sig.parameters.values()
+    ]
+    tool_registry.append({
+        "name": fn.__name__,
+        "description": fn.__doc__.strip() if fn.__doc__ else "",
+        "parameters": params,
+    })
+    return fn
 
 credit_card_db = SQLDatabase.from_uri(r"sqlite:///data/ccms.db")
 query_checker_tool = QuerySQLCheckerTool(db=credit_card_db, llm=llm)
 
-
 @mcp.custom_route("/", methods=["GET"])
 async def home(request: Request) -> PlainTextResponse:
     return PlainTextResponse(
         """
         Credit Card Database MCP Server
         ----
-        This server gives you access to the following tools:
-        1. sql_db_list_tables: This tool can be used to list the tables in the database.
-        2. sql_db_schema: This tool can be used to get the schema of a table.
-        3. sql_db_query_checker: This tool can be used to check if a query is valid.
-        4. sql_db_query: This tool can be used to execute a query.
+        See /tools for a list of tools and their documentation.
         """
     )
 
-
-@mcp.tool()
+@register_tool
 def sql_db_list_tables():
     """
     Returns a comma-separated list of table names in the database.
     """
     return credit_card_db.get_usable_table_names()
 
-
-@mcp.tool()
+@register_tool
 def sql_db_schema(table_names: list[str]) -> str:
     """
     Input 'table_names_str' is a comma-separated string of table names.
@@ -54,8 +153,7 @@ def sql_db_schema(table_names: list[str]) -> str:
     """
     return credit_card_db.get_table_info(table_names)
 
-
-@mcp.tool()
+@register_tool
 def sql_db_query_checker(query: str) -> str:
     """
     Input 'query' is a SQL query string.
@@ -64,10 +162,9 @@ def sql_db_query_checker(query: str) -> str:
     If the query is not valid, it returns the corrected query.
     This tool is used to ensure the query is valid before executing it.
     """
-
     return query_checker_tool.run(query)
 
-@mcp.tool()
+@register_tool
 def sql_db_query(query: str) -> str:
     """
     Input 'query' is a SQL query string.
@@ -75,6 +172,10 @@ def sql_db_query(query: str) -> str:
     """
     return credit_card_db.run(query)
 
+@mcp.custom_route("/tools", methods=["GET"])
+async def list_tools(request: Request) -> JSONResponse:
+    """Return all registered tool metadata as JSON."""
+    return JSONResponse(tool_registry)
 
 if __name__ == "__main__":
     uvicorn.run(mcp.streamable_http_app, host="0.0.0.0", port=8000)