Merge pull request #1 from LinkupPlatform/feat/clean-for-launch

BruceWouaigne · web-flow · commit dad1363c02ab · 2024-12-10T14:37:24.000+01:00
Remove resources and force standard/searchResults
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,20 +1,34 @@
 [project]
 name = "mcp-search-linkup"
-version = "0.1.1"
+version = "0.1.2"
 description = "A MCP server to search web with Linkup"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
  "linkup-sdk>=0.1.8",
  "mcp>=1.0.0",
 ]
-[[project.authors]]
-name = "Guillaume Larcher"
-email = "glm.larcher@gmail.com"
 
 [build-system]
 requires = [ "hatchling",]
 build-backend = "hatchling.build"
 
 [project.scripts]
 mcp-search-linkup = "mcp_search_linkup:main"
+
+[tool.ruff]
+line-length = 100
+target-version = "py38"
+
+[tool.ruff.lint]
+select = [
+  "E", # pycodestyle
+  "F", # pyflakes
+  "I", # isort
+  "S", # flake8-bandit
+]
+
+[dependency-groups]
+dev = [
+    "ruff>=0.8.2",
+]
diff --git a/src/mcp_search_linkup/__init__.py b/src/mcp_search_linkup/__init__.py
@@ -1,9 +1,11 @@
 from . import server
 import asyncio
 
+
 def main():
     """Main entry point for the package."""
     asyncio.run(server.main())
 
+
 # Optionally expose other important items at package level
-__all__ = ['main', 'server']
+__all__ = ["main", "server"]
diff --git a/src/mcp_search_linkup/server.py b/src/mcp_search_linkup/server.py
@@ -7,43 +7,32 @@
 from pydantic import AnyUrl
 import logging
 
-import os
-# Initialize the LinkupClient
-
 server = Server("mcp-search-linkup")
 logger = logging.getLogger("mcp-search-linkup")
 logger.setLevel(logging.INFO)
 
+
 ## Logging
 @server.set_logging_level()
 async def set_logging_level(level: types.LoggingLevel) -> types.EmptyResult:
     logger.setLevel(level.upper())
     await server.request_context.session.send_log_message(
-        level="info",
-        data=f"Log level set to {level}",
-        logger="mcp-search-linkup"
+        level="info", data=f"Log level set to {level}", logger="mcp-search-linkup"
     )
     return types.EmptyResult()
 
+
 ## Resources
 @server.list_resources()
 async def list_resources() -> list[types.Resource]:
     return [
         types.Resource(
             uri=AnyUrl("https://www.thebridgechronicle.com/media"),
             name="The Bridge Chronicle",
-            mimeType="text/html"
+            mimeType="text/html",
         )
     ]
 
-@server.read_resource()
-async def read_resource(uri: AnyUrl) -> str:
-    if str(uri) == "https://www.thebridgechronicle.com/media":
-        page = client.content(url="https://www.thebridgechronicle.com/news/capgemini-employees-walk-together-in-celebration-of-indias-independence")
-        return page.content
-
-
-    raise ValueError("Resource not found")
 
 ## Tools
 @server.list_tools()
@@ -60,26 +49,15 @@ async def handle_list_tools() -> list[types.Tool]:
                 "properties": {
                     "query": {
                         "type": "string",
-                        "description": "The query to search the web with. This should be a question, no need to write in keywords."
+                        "description": "The query to search the web with. This should be a question, no need to write in keywords.",
                     },
-                    "depth": {
-                        "type": "string",
-                        "enum": ["standard", "deep"],
-                        "default": "standard",
-                        "description": "The depth of the search. Standard is a basic search while deep takes more time to complete but can answer more complex questions."
-                    },
-                    "output_type": {
-                        "type": "string",
-                        "enum": ["searchResults", "sourcedAnswer", "structured"],
-                        "default": "searchResults",
-                        "description": "The type of output to return"
-                    }
                 },
                 "required": ["query"],
             },
         )
     ]
 
+
 @server.call_tool()
 async def handle_call_tool(
     name: str, arguments: dict | None
@@ -94,8 +72,6 @@ async def handle_call_tool(
         raise ValueError("Missing arguments")
 
     query = arguments.get("query")
-    depth = arguments.get("depth", "standard")
-    output_type = arguments.get("output_type", "searchResults")
 
     if not query:
         raise ValueError("Missing query")
@@ -104,8 +80,8 @@ async def handle_call_tool(
     # Perform the search using LinkupClient
     search_response = client.search(
         query=query,
-        depth=depth,
-        output_type=output_type,
+        depth="standard",
+        output_type="searchResults",
     )
 
     return [
@@ -115,6 +91,7 @@ async def handle_call_tool(
         )
     ]
 
+
 async def main():
     # Run the server using stdin/stdout streams
     async with mcp.server.stdio.stdio_server() as (read_stream, write_stream):
@@ -131,5 +108,6 @@ async def main():
             ),
         )
 
+
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())
diff --git a/uv.lock b/uv.lock
