Merge pull request #3 from skydeckai/features/memorize-API

Add memorize tool to server

Author: ThangDPham

src/mcp_server_rememberizer/server.py

@@ -13,6 +13,7 @@
     APP_NAME,
     LIST_DOCUMENTS_PATH,
     LIST_INTEGRATIONS_PATH,
+    MEMORIZE_PATH,
     RETRIEVE_DOCUMENT_PATH,
     RETRIEVE_SLACK_PATH,
     SEARCH_PATH,
@@ -192,6 +193,29 @@ async def list_tools() -> list[types.Tool]:
                     },
                 },
             ),
+            types.Tool(
+                name=RememberizerTools.MEMORIZE.value,
+                description=(
+                    "Save a piece of text information in your Rememberizer account "
+                    "to help you discover semantically similar knowledge in the future."
+                ),
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "name": {
+                            "type": "string",
+                            "description": (
+                                "Name of the information. "
+                                "This is used to identify the information in the future."
+                            ),
+                        },
+                        "content": {
+                            "type": "string",
+                            "description": "The information you wish to memorize.",
+                        },
+                    },
+                },
+            ),
         ]
 
     @server.call_tool()
@@ -231,6 +255,13 @@ async def call_tool(
                 params = {"page": page, "page_size": page_size}
                 data = await client.get(LIST_DOCUMENTS_PATH, params=params)
                 return [types.TextContent(type="text", text=str(data))]
+            case RememberizerTools.MEMORIZE.value:
+                params = {
+                    "name": arguments["name"],
+                    "content": arguments["content"],
+                }
+                data = await client.post(MEMORIZE_PATH, data=params)
+                return [types.TextContent(type="text", text=str(data))]
             case _:
                 raise ValueError(f"Unknown tool: {name}")
 

src/mcp_server_rememberizer/utils.py

@@ -19,6 +19,7 @@
 SEARCH_PATH = "documents/search/"
 AGENTIC_SEARCH_PATH = "documents/agentic_search/"
 LIST_INTEGRATIONS_PATH = "integrations/"
+MEMORIZE_PATH = "documents/memorize/"
 
 
 class RememberizerTools(Enum):
@@ -27,6 +28,7 @@ class RememberizerTools(Enum):
     LIST_INTEGRATIONS = "rememberizer_list_integrations"
     ACCOUNT_INFORMATION = "rememberizer_account_information"
     LIST_DOCUMENTS = "rememberizer_list_documents"
+    MEMORIZE = "rememberizer_memorize"
 
 
 class APIClient: