From a49337f02d25a8e11cf4748f283045d35dee1e79 Mon Sep 17 00:00:00 2001
From: strawgate <williamseaston@gmail.com>
Date: Sun, 20 Apr 2025 11:27:26 -0500
Subject: [PATCH] Add an init to make mcp_mixin a module

---
 src/contrib/mcp_mixin/README.md   |  2 +-
 src/contrib/mcp_mixin/__init__.py |  8 ++++++++
 src/contrib/mcp_mixin/example.py  |  2 +-
 tests/contrib/test_mcp_mixin.py   | 10 ++++++----
 4 files changed, 16 insertions(+), 6 deletions(-)
 create mode 100644 src/contrib/mcp_mixin/__init__.py

diff --git a/src/contrib/mcp_mixin/README.md b/src/contrib/mcp_mixin/README.md
index 4d1f4d387..3acb66190 100644
--- a/src/contrib/mcp_mixin/README.md
+++ b/src/contrib/mcp_mixin/README.md
@@ -10,7 +10,7 @@ Inherit from `MCPMixin` and use the decorators on the methods you want to regist
 
 ```python
 from fastmcp import FastMCP
-from contrib.mcp_mixin.mcp_mixin import MCPMixin, mcp_tool, mcp_resource
+from contrib.mcp_mixin import MCPMixin, mcp_tool, mcp_resource
 
 class MyComponent(MCPMixin):
     @mcp_tool(name="my_tool", description="Does something cool.")
diff --git a/src/contrib/mcp_mixin/__init__.py b/src/contrib/mcp_mixin/__init__.py
new file mode 100644
index 000000000..8b4cca0e2
--- /dev/null
+++ b/src/contrib/mcp_mixin/__init__.py
@@ -0,0 +1,8 @@
+from .mcp_mixin import MCPMixin, mcp_tool, mcp_resource, mcp_prompt
+
+__all__ = [
+    "MCPMixin",
+    "mcp_tool",
+    "mcp_resource",
+    "mcp_prompt",
+]
diff --git a/src/contrib/mcp_mixin/example.py b/src/contrib/mcp_mixin/example.py
index f3e8aeffc..5e996632e 100644
--- a/src/contrib/mcp_mixin/example.py
+++ b/src/contrib/mcp_mixin/example.py
@@ -2,7 +2,7 @@
 
 import asyncio
 
-from contrib.mcp_mixin.mcp_mixin import (
+from contrib.mcp_mixin import (
     MCPMixin,
     mcp_prompt,
     mcp_resource,
diff --git a/tests/contrib/test_mcp_mixin.py b/tests/contrib/test_mcp_mixin.py
index 6f69f545d..b31919cff 100644
--- a/tests/contrib/test_mcp_mixin.py
+++ b/tests/contrib/test_mcp_mixin.py
@@ -2,15 +2,17 @@
 
 import pytest
 
-from contrib.mcp_mixin.mcp_mixin import (
-    _DEFAULT_SEPARATOR_PROMPT,
-    _DEFAULT_SEPARATOR_RESOURCE,
-    _DEFAULT_SEPARATOR_TOOL,
+from contrib.mcp_mixin import (
     MCPMixin,
     mcp_prompt,
     mcp_resource,
     mcp_tool,
 )
+from contrib.mcp_mixin.mcp_mixin import (
+    _DEFAULT_SEPARATOR_PROMPT,
+    _DEFAULT_SEPARATOR_RESOURCE,
+    _DEFAULT_SEPARATOR_TOOL,
+)
 from fastmcp import FastMCP