image-search-mcp 0.1.4__tar.gz → 0.1.6__tar.gz

{image_search_mcp-0.1.4 → image_search_mcp-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: image-search-mcp
-Version: 0.1.4
+Version: 0.1.6
 Summary: MCP server for Reverse Image Search using PicImageSearch
 Author-email: Your Name <your.email@example.com>
 License: MIT

{image_search_mcp-0.1.4 → image_search_mcp-0.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "image-search-mcp"
-version = "0.1.4"
+version = "0.1.6"
 description = "MCP server for Reverse Image Search using PicImageSearch"
 readme = "README.md"
 requires-python = ">=3.10"

{image_search_mcp-0.1.4 → image_search_mcp-0.1.6}/src/image_search_mcp/main.py

@@ -43,33 +43,25 @@ def main():
     if args.sse:
         try:
             import uvicorn
-            from starlette.applications import Starlette
-            from starlette.routing import Route, Mount
-            from mcp.server.sse import SseServerTransport
-
-            # 1. Initialize SSE transport
-            sse = SseServerTransport("/messages")
-
-            # 2. Define the SSE handler as a raw ASGI app to get scope, receive, send
-            async def handle_sse(scope, receive, send):
-                async with sse.connect_sse(scope, receive, send) as (read_stream, write_stream):
-                    # Use the underlying server from FastMCP instance
-                    await mcp._server.run(
-                        read_stream,
-                        write_stream,
-                        mcp._server.create_initialization_options()
-                    )
-
-            starlette_app = Starlette(
-                debug=True,
-                routes=[
-                    # Mount handles the path prefix and passes raw ASGI arguments
-                    Mount("/sse", app=handle_sse),
-                    Route("/messages", endpoint=sse.handle_post_message, methods=["POST"]),
-                ]
-            )
+            
+            # The most robust way to get the ASGI app from FastMCP (v2.3.2+)
+            # This app already includes the necessary SSE routes (/sse, /messages)
+            if hasattr(mcp, "http_app"):
+                app = mcp.http_app()
+            elif hasattr(mcp, "as_asgi_app"):
+                app = mcp.as_asgi_app()
+            else:
+                # Fallback to internal app if methods are missing
+                app = getattr(mcp, "_app", mcp)
 
-            app = starlette_app
+            # Check for Auth Token
+            auth_token = os.environ.get("MCP_AUTH_TOKEN")
+            if auth_token:
+                print(f"🔒 Authentication enabled. Require Bearer token.")
+                app = AuthMiddleware(app, auth_token)
+            
+            print(f"Starting SSE server on {args.host}:{args.port}...")
+            uvicorn.run(app, host=args.host, port=args.port)
             
             # 3. Apply Auth Middleware if token is provided
             auth_token = os.environ.get("MCP_AUTH_TOKEN")

{image_search_mcp-0.1.4 → image_search_mcp-0.1.6}/src/image_search_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: image-search-mcp
-Version: 0.1.4
+Version: 0.1.6
 Summary: MCP server for Reverse Image Search using PicImageSearch
 Author-email: Your Name <your.email@example.com>
 License: MIT