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

image_search_mcp/main.py

@@ -43,17 +43,35 @@ def main():
     if args.sse:
         try:
             import uvicorn
-            # FastMCP's Starlette app is usually available via the ._app attribute 
-            # after some initialization, or we can use its built-in server logic.
-            # To apply middleware, we need the actual ASGI app.
-            
-            # This is a more robust way to get the ASGI app from FastMCP
-            from mcp.server.fastmcp import FastMCP
-            
-            # Force initialization of the SSE app if possible
-            app = mcp.as_asgi_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.")

{image_search_mcp-0.1.1.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.1
+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.1.dist-info/RECORD

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