strawgate commited on
Commit
066be76
·
1 Parent(s): cca9a30

Add an init to make mcp_mixin a module

Browse files
src/contrib/mcp_mixin/README.md CHANGED
@@ -10,7 +10,7 @@ Inherit from `MCPMixin` and use the decorators on the methods you want to regist
10
 
11
  ```python
12
  from fastmcp import FastMCP
13
- from contrib.mcp_mixin.mcp_mixin import MCPMixin, mcp_tool, mcp_resource
14
 
15
  class MyComponent(MCPMixin):
16
  @mcp_tool(name="my_tool", description="Does something cool.")
 
10
 
11
  ```python
12
  from fastmcp import FastMCP
13
+ from contrib.mcp_mixin import MCPMixin, mcp_tool, mcp_resource
14
 
15
  class MyComponent(MCPMixin):
16
  @mcp_tool(name="my_tool", description="Does something cool.")
src/contrib/mcp_mixin/__init__.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from .mcp_mixin import MCPMixin, mcp_tool, mcp_resource, mcp_prompt
2
+
3
+ __all__ = [
4
+ "MCPMixin",
5
+ "mcp_tool",
6
+ "mcp_resource",
7
+ "mcp_prompt",
8
+ ]
src/contrib/mcp_mixin/example.py CHANGED
@@ -2,7 +2,7 @@
2
 
3
  import asyncio
4
 
5
- from contrib.mcp_mixin.mcp_mixin import (
6
  MCPMixin,
7
  mcp_prompt,
8
  mcp_resource,
 
2
 
3
  import asyncio
4
 
5
+ from contrib.mcp_mixin import (
6
  MCPMixin,
7
  mcp_prompt,
8
  mcp_resource,
tests/contrib/test_mcp_mixin.py CHANGED
@@ -2,15 +2,17 @@
2
 
3
  import pytest
4
 
5
- from contrib.mcp_mixin.mcp_mixin import (
6
- _DEFAULT_SEPARATOR_PROMPT,
7
- _DEFAULT_SEPARATOR_RESOURCE,
8
- _DEFAULT_SEPARATOR_TOOL,
9
  MCPMixin,
10
  mcp_prompt,
11
  mcp_resource,
12
  mcp_tool,
13
  )
 
 
 
 
 
14
  from fastmcp import FastMCP
15
 
16
 
 
2
 
3
  import pytest
4
 
5
+ from contrib.mcp_mixin import (
 
 
 
6
  MCPMixin,
7
  mcp_prompt,
8
  mcp_resource,
9
  mcp_tool,
10
  )
11
+ from contrib.mcp_mixin.mcp_mixin import (
12
+ _DEFAULT_SEPARATOR_PROMPT,
13
+ _DEFAULT_SEPARATOR_RESOURCE,
14
+ _DEFAULT_SEPARATOR_TOOL,
15
+ )
16
  from fastmcp import FastMCP
17
 
18