mle-kit-mcp 0.2.6__tar.gz → 1.0.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (27) hide show
  1. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/PKG-INFO +1 -1
  2. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/server.py +18 -7
  3. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp.egg-info/PKG-INFO +1 -1
  4. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/pyproject.toml +1 -1
  5. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/LICENSE +0 -0
  6. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/README.md +0 -0
  7. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/__init__.py +0 -0
  8. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/__main__.py +0 -0
  9. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/files.py +0 -0
  10. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/llm_proxy.py +0 -0
  11. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/py.typed +0 -0
  12. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/tools/__init__.py +0 -0
  13. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/tools/bash.py +0 -0
  14. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/tools/llm_proxy.py +0 -0
  15. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/tools/remote_gpu.py +0 -0
  16. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/tools/text_editor.py +0 -0
  17. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp/utils.py +0 -0
  18. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp.egg-info/SOURCES.txt +0 -0
  19. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp.egg-info/dependency_links.txt +0 -0
  20. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp.egg-info/entry_points.txt +0 -0
  21. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp.egg-info/requires.txt +0 -0
  22. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/mle_kit_mcp.egg-info/top_level.txt +0 -0
  23. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/setup.cfg +0 -0
  24. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/tests/test_bash.py +0 -0
  25. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/tests/test_llm_proxy.py +0 -0
  26. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/tests/test_text_editor.py +0 -0
  27. {mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.0}/tests/test_truncate_context.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mle-kit-mcp
3
- Version: 0.2.6
3
+ Version: 1.0.0
4
4
  Summary: MCP server that provides different tools for MLE
5
5
  Author-email: Ilya Gusev <phoenixilya@gmail.com>
6
6
  Project-URL: Homepage, https://github.com/IlyaGusev/mle_kit_mcp
@@ -1,9 +1,8 @@
1
1
  import os
2
2
  from pathlib import Path
3
- from typing import Optional
3
+ from typing import Optional, Literal
4
4
 
5
5
  import fire # type: ignore
6
- import uvicorn
7
6
  from mcp.server.fastmcp import FastMCP
8
7
  from dotenv import load_dotenv
9
8
 
@@ -21,14 +20,26 @@ from .tools.llm_proxy import (
21
20
  from .files import get_workspace_dir, WorkspaceDirectory
22
21
 
23
22
 
24
- def run(host: str = "0.0.0.0", port: int = 5050, workspace: Optional[str] = None) -> None:
23
+ def run(
24
+ host: str = "0.0.0.0",
25
+ port: int = 5050,
26
+ mount_path: str = "/",
27
+ streamable_http_path: str = "/mcp",
28
+ workspace: Optional[str] = None,
29
+ transport: Literal["stdio", "sse", "streamable-http"] = "streamable-http",
30
+ ) -> None:
25
31
  load_dotenv()
26
32
  if workspace:
27
33
  WorkspaceDirectory.set_dir(Path(workspace))
28
34
  workspace_path = get_workspace_dir()
29
35
  workspace_path.mkdir(parents=True, exist_ok=True)
30
36
 
31
- server = FastMCP("MLE kit MCP", stateless_http=True)
37
+ server = FastMCP(
38
+ "MLE kit MCP",
39
+ stateless_http=True,
40
+ streamable_http_path=streamable_http_path,
41
+ mount_path=mount_path,
42
+ )
32
43
 
33
44
  remote_text_editor = create_remote_text_editor(text_editor)
34
45
 
@@ -41,9 +52,9 @@ def run(host: str = "0.0.0.0", port: int = 5050, workspace: Optional[str] = None
41
52
  server.add_tool(llm_proxy_local)
42
53
  server.add_tool(llm_proxy_remote)
43
54
 
44
- http_app = server.streamable_http_app()
45
-
46
- uvicorn.run(http_app, host=host, port=port)
55
+ server.settings.port = port
56
+ server.settings.host = host
57
+ server.run(transport=transport)
47
58
 
48
59
 
49
60
  if __name__ == "__main__":
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mle-kit-mcp
3
- Version: 0.2.6
3
+ Version: 1.0.0
4
4
  Summary: MCP server that provides different tools for MLE
5
5
  Author-email: Ilya Gusev <phoenixilya@gmail.com>
6
6
  Project-URL: Homepage, https://github.com/IlyaGusev/mle_kit_mcp
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "mle-kit-mcp"
7
- version = "0.2.6"
7
+ version = "1.0.0"
8
8
  description = "MCP server that provides different tools for MLE"
9
9
  readme = "README.md"
10
10
  authors = [
File without changes
File without changes
File without changes