AI4Semantics-MCP-server/server.py at main · SEMICeu/AI4Semantics-MCP-server · GitHub

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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from dotenv import load_dotenv
load_dotenv()


from fastmcp import FastMCP
from fastmcp.tools import Tool
from fastmcp.server.event_store import EventStore
import resources
import tools


# CHANGE FOR DEPLOYMENT/DEVELOPMENT!
mcp = FastMCP(
    name="...",
)

mcp.add_tool(
    Tool.from_function(
        tools.plan_workflow_with_tools,
        name="plan_workflow_with_tools",
    ),
)

mcp.add_tool(
    Tool.from_function(
        tools.get_style_guide,
        name="get_style_guide",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.retrieve_documents,
        name="retrieve_documents",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.upload_model,
        name="upload_model",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.add_class,
        name="add_class",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.add_attribute,
        name="add_attribute",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.add_connector,
        name="add_connector",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.metadata_checker,
        name="metadata_checker",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.reuse_check,
        name="reuse_check",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.style_guide_check,
        name="style_guide_check",
    )
)

mcp.add_tool(
    Tool.from_function(
        tools.validator_check,
        name="validator_check",
    )
)


mcp.resource(
    "resource://model/{user}/{session_name}",
    mime_type="application/json"
)(
    resources.get_model
)

mcp.resource(
    "resource://Style_Guide}",
    mime_type="text/plain"
)(
    resources.get_style_guide
)


event_store = EventStore()
app = mcp.http_app(
    event_store=event_store,
    retry_interval=2000,
)

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8001)