From 1f8950653a0c5475244d0c57e37f2421f483a33d Mon Sep 17 00:00:00 2001
From: ayush chaurasia <ayush.chaurarsia@gmail.com>
Date: Wed, 23 Apr 2025 18:36:46 +0530
Subject: [PATCH] fix

---
 docs/src/guides/mcp.md | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/docs/src/guides/mcp.md b/docs/src/guides/mcp.md
index 2cdd79e0..31bcf261 100644
--- a/docs/src/guides/mcp.md
+++ b/docs/src/guides/mcp.md
@@ -67,6 +67,25 @@ async def ingest_data(content: str) -> str:
 ```
 ### Retreive data tool
 
+```python
+@mcp.tool()
+async def retrieve_data(query: str, limit: int = 5) -> str:
+    """
+    Search db using vector search
+    Args:
+        query: The search query
+        limit: Maximum number of results to return
+    """
+    tbl = db[TABLE_NAME]
+    rs = tbl.search(query).limit(limit).to_list()
+    data = [
+        r["text"] for r in rs
+    ]
+    if not data:
+        return "No relevant data found."
+
+    return "\n\n".join(data)
+```
 This function takes a string and limit as input and searches the LanceDB table for the most relevant memories.