AnatomyLite / textbook_server.py
gladguy's picture
Fresh
4c6a071
Raw
History Blame Contribute Delete
534 Bytes
# Concept Code for a new MCP Tool
from llama_index.core import VectorStoreIndex, SimpleDirectoryReader
from mcp.server.fastmcp import FastMCP
mcp = FastMCP("Textbook Knowledge")
documents = SimpleDirectoryReader("./textbooks").load_data()
index = VectorStoreIndex.from_documents(documents)
query_engine = index.as_query_engine()
@mcp.tool()
def consult_textbook(question: str) -> str:
"""Consults the uploaded Gray's Anatomy textbook for exact definitions."""
response = query_engine.query(question)
return str(response)