Skip to content

Commit cc0ce34

Browse files
refactor(xiyan_mcp_server): 重构代码以提高可维护性
- 修改 __init__.py 中的导入方式和 main 函数调用 - 更新 server.py 中的代码结构,定义 main 函数- 在 README.md 中添加 Smithery 徽章链接
1 parent 8b2d253 commit cc0ce34

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
[![smithery badge](https://smithery.ai/badge/@XGenerationLab/xiyan_mcp_server)](https://smithery.ai/server/@XGenerationLab/xiyan_mcp_server)
12

23
# XiYan MCP Server
34

src/xiyan_mcp_server/__init__.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
1-
#from . import server
2-
from server import *
1+
from . import server
32

43
def main():
5-
"""Main entry point for the package."""
6-
mcp.run()
4+
"""Main entry point for the package."""
5+
server.main() # 修改为调用server.py中的main函数
76

87
# Expose important items at package level
98
__all__ = ['main', 'server']

src/xiyan_mcp_server/server.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,11 @@ def get_data_via_natural_language(query: str)-> list[TextContent]:
180180
res=call_xiyan(query)
181181
return [TextContent(type="text", text=res)]
182182

183+
def main():
184+
mcp.run()
183185

184186

185187
if __name__ == "__main__":
186188

187-
mcp.run()
189+
main()
188190

0 commit comments

Comments
 (0)