import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";
import {
createErrorResponse,
debugLog,
getSearchcraftConfig,
makeSearchcraftRequest,
} from "../../../helpers.js";
export const registerAddDocuments = (server: McpServer) => {
/**
* Tool: add_documents
* POST /index/:index/documents - Add one or multiple documents to an index
*/
server.tool(
"add_documents",
"Add one or multiple documents to an index. Documents should be provided as an array of JSON objects. Each document should have the schema specified by the corresponding index's schema.",
{
index_name: z
.string()
.describe("The name of the index to add documents to"),
documents: z
.array(z.record(z.any()))
.describe("Array of document objects to add to the index"),
},
async ({ index_name, documents }) => {
debugLog("[Tool Call] add_documents");
try {
const config = getSearchcraftConfig();
if (config.error) {
return config.error;
}
const { endpointUrl, apiKey } = config;
const endpoint = `${endpointUrl.replace(/\/$/, "")}/index/${index_name}/documents`;
const response = await makeSearchcraftRequest(
endpoint,
"POST",
apiKey,
documents,
);
const commitEndpoint = `${endpointUrl.replace(/\/$/, "")}/index/${index_name}/commit`;
const commitResponse = await makeSearchcraftRequest(
commitEndpoint,
"POST",
apiKey,
documents,
);
return {
content: [
{
type: "resource",
resource: {
uri: `searchcraft://documents-added/${index_name}/${Date.now()}`,
mimeType: "application/json",
text: JSON.stringify(commitResponse, null, 2),
},
},
],
};
} catch (error) {
const errorMessage =
error instanceof Error
? error.message
: "Unknown error occurred";
return createErrorResponse(
`Failed to add documents: ${errorMessage}`,
);
}
},
);
};
