fix up

Signed-off-by: SamYuan1990 <[email protected]>

Author: SamYuan1990

examples/clients/homebrewupload-client/main.py

@@ -10,9 +10,7 @@
 from mcp.client.session import ClientSession
 from mcp.client.sse import sse_client
 
-logging.basicConfig(
-    format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO
-)
+logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO)
 
 load_dotenv()  # load environment variables from .env
 
@@ -45,7 +43,7 @@ async def connect_to_server(self):
             response = await self.session.list_tools()
             tools = response.tools
             print("\nConnected to server with tools:", [tool.name for tool in tools])
-            
+
             return True
         except Exception as e:
             logging.error(f"Failed to connect to server: {e}")
@@ -56,23 +54,23 @@ async def process_chat(
         self,
         file_path: str | None = None,
     ) -> str:
-        """ Porcess a chat"""
+        """Porcess a chat"""
         messages = []
         user_content = """please help make file into markdown format, file path file:///tmp/test.pdf, 
                 you are free to use convert_to_markdown tool, 
                 the file will upload to MCP server in secure."""
 
         try:
-            with open(file_path,"rb") as f:
-                    file_content = base64.b64encode(f.read()).decode("utf-8")
-                # 发送请求
+            with open(file_path, "rb") as f:
+                file_content = base64.b64encode(f.read()).decode("utf-8")
+            # 发送请求
             response = requests.post(
-                    "http://localhost:3001/upload",
-                    json={"filename": "test.pdf", "file_content_base64": file_content},
+                "http://localhost:3001/upload",
+                json={"filename": "test.pdf", "file_content_base64": file_content},
             )
         except Exception as e:
-                logging.info(f"file handle error: {str(e)}")
-                return f"file handle error: {str(e)}"    
+            logging.info(f"file handle error: {str(e)}")
+            return f"file handle error: {str(e)}"
         messages.append({"role": "user", "content": user_content})
         response = await self.session.list_tools()
         available_tools = [
@@ -95,7 +93,7 @@ async def process_chat(
                 final_text.append(content.text)
             elif content.type == "tool_use":
                 tool_name = content.name
-                tool_args = "file:///tmp/test.pdf"#content.input
+                tool_args = "file:///tmp/test.pdf"  # content.input
 
                 # 执行工具调用
                 try:
@@ -135,16 +133,15 @@ async def process_chat(
                     for next_content in next_response.content:
                         if next_content.type == "text":
                             final_text.extend(
-                            next_content.text
-                            for next_content in next_response.content
-                            if next_content.type == "text"
-                        )
+                                next_content.text
+                                for next_content in next_response.content
+                                if next_content.type == "text"
+                            )
 
                 except Exception as e:
                     final_text.append(f"tool invoke {tool_name} error: {str(e)}")
 
-        return "\n".join(final_text)        
-
+        return "\n".join(final_text)
 
     async def close(self):
         """Properly close all connections"""

examples/servers/homebrewupload/main.py

@@ -22,12 +22,14 @@
 # to different with markitdown official
 mcp = FastMCP("markitdown_homebrew")
 
+
 # Same with markitdown official
 @mcp.tool()
 async def convert_to_markdown(uri: str) -> str:
     """Convert a resource described by an http:, https:, file: or data: URI to markdown"""
     return MarkItDown(enable_plugins=check_plugins_enabled()).convert_uri(uri).markdown
 
+
 # Same with markitdown official
 def check_plugins_enabled() -> bool:
     return os.getenv("MARKITDOWN_ENABLE_PLUGINS", "false").strip().lower() in (
@@ -36,6 +38,7 @@ def check_plugins_enabled() -> bool:
         "yes",
     )
 
+
 # Same with markitdown official
 def create_starlette_app(mcp_server: Server, *, debug: bool = False) -> Starlette:
     sse = SseServerTransport("/messages/")
@@ -58,9 +61,7 @@ async def handle_sse(request: Request) -> None:
                 mcp_server.create_initialization_options(),
             )
 
-    async def handle_streamable_http(
-        scope: Scope, receive: Receive, send: Send
-    ) -> None:
+    async def handle_streamable_http(scope: Scope, receive: Receive, send: Send) -> None:
         await session_manager.handle_request(scope, receive, send)
 
     @contextlib.asynccontextmanager
@@ -80,13 +81,12 @@ async def lifespan(app: Starlette) -> AsyncIterator[None]:
             Mount("/mcp", app=handle_streamable_http),
             Mount("/messages/", app=sse.handle_post_message),
             # adding this as URL path
-            Route(
-                "/upload", endpoint=handle_file_upload, methods=["POST"]
-            ),
+            Route("/upload", endpoint=handle_file_upload, methods=["POST"]),
         ],
         lifespan=lifespan,
     )
 
+
 # added function for handle file upload
 async def handle_file_upload(request: Request) -> JSONResponse:
     """
@@ -103,29 +103,23 @@ async def handle_file_upload(request: Request) -> JSONResponse:
             return JSONResponse({"error": "Filename is required"}, status_code=400)
 
         if not file_content_base64:
-            return JSONResponse(
-                {"error": "file_content_base64 is required"}, status_code=400
-            )
+            return JSONResponse({"error": "file_content_base64 is required"}, status_code=400)
         filename = os.path.basename(filename)
         file_path = os.path.join("/tmp/", filename)
         logging.info("start processing")
         try:
             file_data = base64.b64decode(file_content_base64)
         except Exception as e:
             logging.info(e)
-            return JSONResponse(
-                {"error": f"Invalid base64 data: {str(e)}"}, status_code=400
-            )
+            return JSONResponse({"error": f"Invalid base64 data: {str(e)}"}, status_code=400)
         os.makedirs("/tmp", exist_ok=True)
 
         with open(file_path, "wb") as f:
             f.write(file_data)
 
         if os.path.exists(file_path):
             file_size = os.path.getsize(file_path)
-            logging.info(
-                f"File saved successfully: {file_path}, size: {file_size} bytes"
-            )
+            logging.info(f"File saved successfully: {file_path}, size: {file_size} bytes")
 
             return JSONResponse(
                 {
@@ -144,20 +138,20 @@ async def handle_file_upload(request: Request) -> JSONResponse:
         return JSONResponse({"error": "Invalid JSON in request body"}, status_code=400)
     except Exception as e:
         logging.error(f"Error handling file upload: {str(e)}")
-        return JSONResponse(
-            {"error": f"Internal server error: {str(e)}"}, status_code=500
-        )
+        return JSONResponse({"error": f"Internal server error: {str(e)}"}, status_code=500)
+
 
 # Main entry point
 def main():
     mcp_server = mcp._mcp_server
     logging.info("start MCP server at 0.0.0.0:3001")
     starlette_app = create_starlette_app(mcp_server, debug=True)
     uvicorn.run(
-            starlette_app,
-            host="0.0.0.0",
-            port=3001,
-        )
+        starlette_app,
+        host="0.0.0.0",
+        port=3001,
+    )
+
 
 if __name__ == "__main__":
     main()