mcp_server.py

from pydantic import Field
from mcp.server.fastmcp import FastMCP
from mcp.server.fastmcp.prompts import base


mcp = FastMCP("DocumentMCP", log_level="ERROR")


docs = {
    "deposition.md": "This deposition covers the testimony of Angela Smith, P.E.",
    "report.pdf": "The report details the state of a 20m condenser tower.",
    "financials.docx": "These financials outline the project's budget and expenditures.",
    "outlook.pdf": "This document presents the projected future performance of the system.",
    "plan.md": "The plan outlines the steps for the project's implementation.",
    "spec.txt": "These specifications define the technical requirements for the equipment.",
}

@mcp.tool (
    name="read_doc_contents",
    description="Read the contents of a document and return it as a string.",
)
def read_documents(
    doc_id: str = Field(description="Id of the document to read"),
):
    if doc_id not in docs:
        raise ValueError(f"document with Id {doc_id} is not in docs.")
    
    return docs[doc_id]


@mcp.tool (
    name="edit_document",
    description="Edit a document by replacing a string in the content by a new string.",
)
def edit_doc(
    doc_id: str = Field(description="Id of the document to edit"),
    old_string: str = Field(description="the text to replace must match exactly including white space"),
    new_string: str = Field(description="the new text to input instead of the old text"),   
):
    if doc_id not in docs:
        raise ValueError(f"document with Id {doc_id} is not in docs.")
    
    docs[doc_id] = docs[doc_id].replace(old_string,new_string)



@mcp.resource ("docs://documents", mime_type="application/json")
def list_docs() -> list[str]:
    return list(docs.keys())


@mcp.resource ("docs://documents/{doc_id}", mime_type="text/plain")
def fetch_doc(doc_id: str) -> str:
    if doc_id not in docs:
        raise ValueError(f"document with Id {doc_id} is not in docs.")
    return docs[doc_id]


@mcp.prompt ( 
    name = "format",
    description = "rewrites the content of the document in the markdown format"
)
def format_document(doc_id: str = Field(description="Id of the document to read")) -> list[base.Message]:
    prompt = f"""
Your goal is to reformat a document to be written with markdown syntax.

The id of the document you need to reformat is:
<document_id>
{doc_id}
</document_id>

Add in headers, bullet points, tables, etc as necessary. Feel free to add in structure.
Use the 'edit_document' tool to edit the document. After the document has been reformatted...
"""
    
    return [
        base.UserMessage(prompt)
    ]

@mcp.prompt ( 
    name = "summarize",
    description = "summarize the content of the document"
)
def summarize_document(doc_id: str = Field(description="Id of the document to read")) -> list[base.Message]:
    prompt = f"""
Your goal is to summarize a document.

The id of the document you need to reformat is:
<document_id>
{doc_id}
</document_id>

Feel free to add in structure.
Use the 'edit_document' tool to edit the document. After the document has been reformatted...
"""
    
    return [
        base.UserMessage(prompt)
    ]

if __name__ == "__main__":
    mcp.run(transport="stdio")