🔌 MCP (Model Context Protocol)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

from mcp import Server, Tool, Resource

server = Server("enterprise-tools")

@server.tool()
async def search_documents(
    query: str,
    filters: dict = None,
    limit: int = 10
) -> list[dict]:
    """Search company documents with semantic understanding."""
    results = await vector_store.similarity_search(
        query=query,
        k=limit,
        filter=filters
    )
    return [
        {"title": r.title, "content": r.content, "source": r.source}
        for r in results
    ]

@server.resource("database://customers/{customer_id}")
async def get_customer(customer_id: str) -> dict:
    """Retrieve customer information by ID."""
    customer = await db.customers.find_one({"id": customer_id})
    return customer

# Run with proper authentication
server.run(auth=enterprise_auth_middleware)