Merge pull request #1 from GollyTicker/main

feat: Add 'limit' of contexts to return in simple search

Author: ernestkoe

openapi.yaml

@@ -1550,6 +1550,13 @@ paths:
           schema:
             default: 100
             type: "number"
+        - description: "How many contexts to return at most"
+          in: "query"
+          name: "limit"
+          required: false
+          schema:
+            default: 100
+            type: "number"
       responses:
         "200":
           content:

src/mcp_obsidian/obsidian.py

@@ -186,9 +186,9 @@ def get_batch_file_contents(self, filepaths: list[str]) -> str:
 
         return "".join(result)
 
-    def search(self, query: str, context_length: int = 100) -> Any:
+    def search(self, query: str, context_length: int = 100, limit: int = 100) -> Any:
         url = f"{self.get_base_url()}/search/simple/"
-        params = {"query": query, "contextLength": context_length}
+        params = {"query": query, "contextLength": context_length, "limit": limit}
 
         def call_fn():
             response = requests.post(

src/mcp_obsidian/tools.py

@@ -166,7 +166,12 @@ def get_tool_description(self):
                     "context_length": {
                         "type": "integer",
                         "description": "How much context to return around the matching string (default: 100)",
-                        "default": 100,
+                        "default": 100
+                    },
+                    "limit": {
+                        "type": "integer",
+                        "description": "How many contexts to return at most (default: 100)",
+                        "default": 100
                     },
                 },
                 "required": ["query"],
@@ -180,14 +185,20 @@ def run_tool(
             raise RuntimeError("query argument missing in arguments")
 
         context_length = args.get("context_length", 100)
-
+        limit = args.get("limit", 100)
+        
         api = obsidian.Obsidian(api_key=api_key, host=obsidian_host)
-        results = api.search(args["query"], context_length)
-
+        results = api.search(args["query"], context_length, limit)
+        
         formatted_results = []
+        total_results = 0
         for result in results:
+            if total_results >= limit:
+                break
             formatted_matches = []
             for match in result.get("matches", []):
+                if total_results >= limit:
+                    break
                 context = match.get("context", "")
                 match_pos = match.get("match", {})
                 start = match_pos.get("start", 0)
@@ -196,6 +207,7 @@ def run_tool(
                 formatted_matches.append(
                     {"context": context, "match_position": {"start": start, "end": end}}
                 )
+                total_results = total_results + 1
 
             formatted_results.append(
                 {