image-search-mcp 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

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.5.dist-info → image_search_mcp-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: image-search-mcp
-Version: 0.1.5
+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.6.dist-info/RECORD

@@ -0,0 +1,8 @@
+image_search_mcp/__init__.py,sha256=h6DgLFO4TMUk7_wCJ2jn2Y6IkFmfzb-Z7jX-G5UCYVc,43
+image_search_mcp/main.py,sha256=OzbpjQPQ0JD8C43hVsIplcmAaKNIbjKgD2tcg4wDhvQ,3222
+image_search_mcp/server.py,sha256=zDfg1BAB1yPFdvBBkGbBxsGpUEnOAMORnA7LN-4g5Xg,13445
+image_search_mcp-0.1.6.dist-info/METADATA,sha256=h0W96KtxprNmxM6jvrKjT2mAL4oKclPwFGS67_615Wc,5434
+image_search_mcp-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+image_search_mcp-0.1.6.dist-info/entry_points.txt,sha256=BD8buPHSQ9yLqGM6w_UGTKjYkVRu_IjH_R736MsDQXM,64
+image_search_mcp-0.1.6.dist-info/top_level.txt,sha256=toz9LX_-JYwJIcGbvfVSH0ap99Nsp2mdzvitLWHyYWo,17
+image_search_mcp-0.1.6.dist-info/RECORD,,

image_search_mcp-0.1.5.dist-info/RECORD

@@ -1,8 +0,0 @@
-image_search_mcp/__init__.py,sha256=h6DgLFO4TMUk7_wCJ2jn2Y6IkFmfzb-Z7jX-G5UCYVc,43
-image_search_mcp/main.py,sha256=qlKRdiHH7rZIkfSFyb-sW1Vm4FQ2C7PDExGnDOdDaLo,3517
-image_search_mcp/server.py,sha256=zDfg1BAB1yPFdvBBkGbBxsGpUEnOAMORnA7LN-4g5Xg,13445
-image_search_mcp-0.1.5.dist-info/METADATA,sha256=eTZQCd1hD4_bc3lSkATn1Br1LaXEBQd0Z-DiW35iLe0,5434
-image_search_mcp-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-image_search_mcp-0.1.5.dist-info/entry_points.txt,sha256=BD8buPHSQ9yLqGM6w_UGTKjYkVRu_IjH_R736MsDQXM,64
-image_search_mcp-0.1.5.dist-info/top_level.txt,sha256=toz9LX_-JYwJIcGbvfVSH0ap99Nsp2mdzvitLWHyYWo,17
-image_search_mcp-0.1.5.dist-info/RECORD,,