image-search-mcp 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

image_search_mcp/main.py

@@ -43,14 +43,35 @@ def main():
     if args.sse:
         try:
             import uvicorn
-            # Attempt to find the ASGI app from FastMCP instance
-            app = mcp
-            if hasattr(mcp, "_sse_app"):
-                app = mcp._sse_app
-            elif hasattr(mcp, "app"):
-                app = mcp.app
+            from starlette.applications import Starlette
+            from starlette.routing import Route
+            from mcp.server.sse import SseServerTransport
+
+            # 1. Initialize SSE transport
+            # The transport handles the /sse and /messages endpoints
+            sse = SseServerTransport("/messages")
+
+            # 2. Define the ASGI app routes manually to match FastMCP internal logic
+            async def handle_sse(request):
+                async with sse.connect_sse(request.scope, request.receive, request.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=[
+                    Route("/sse", endpoint=handle_sse),
+                    Route("/messages", endpoint=sse.handle_messages, methods=["POST"]),
+                ]
+            )
+
+            app = starlette_app
             
-            # Check for Auth Token
+            # 3. Apply Auth Middleware if token is provided
             auth_token = os.environ.get("MCP_AUTH_TOKEN")
             if auth_token:
                 print(f"🔒 Authentication enabled. Require Bearer token.")
@@ -58,11 +79,10 @@ def main():
             
             print(f"Starting SSE server on {args.host}:{args.port}...")
             uvicorn.run(app, host=args.host, port=args.port)
-        except ImportError:
-            print("Error: 'uvicorn' is required for SSE mode. Please install it: pip install uvicorn")
-            sys.exit(1)
         except Exception as e:
             print(f"Error starting SSE server: {e}")
+            import traceback
+            traceback.print_exc()
             sys.exit(1)
     else:
         # Stdio mode (Default)

{image_search_mcp-0.1.0.dist-info → image_search_mcp-0.1.2.dist-info}/METADATA

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

image_search_mcp-0.1.2.dist-info/RECORD

@@ -0,0 +1,8 @@
+image_search_mcp/__init__.py,sha256=h6DgLFO4TMUk7_wCJ2jn2Y6IkFmfzb-Z7jX-G5UCYVc,43
+image_search_mcp/main.py,sha256=7QbxEYk0FRbvZ5mR5Saxtw5wUsdJrEo_AYId-tC--cA,3510
+image_search_mcp/server.py,sha256=zDfg1BAB1yPFdvBBkGbBxsGpUEnOAMORnA7LN-4g5Xg,13445
+image_search_mcp-0.1.2.dist-info/METADATA,sha256=iqYyvMEdIQR5K1VEhom7_Ml0-N8cXYkEiJ1Znqj17bI,5434
+image_search_mcp-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+image_search_mcp-0.1.2.dist-info/entry_points.txt,sha256=BD8buPHSQ9yLqGM6w_UGTKjYkVRu_IjH_R736MsDQXM,64
+image_search_mcp-0.1.2.dist-info/top_level.txt,sha256=toz9LX_-JYwJIcGbvfVSH0ap99Nsp2mdzvitLWHyYWo,17
+image_search_mcp-0.1.2.dist-info/RECORD,,

image_search_mcp-0.1.0.dist-info/RECORD

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