insert-one.ts•1.4 kB
import { db } from "../../mongodb/client.js";
import { BaseTool, ToolParams } from "../base/tool.js";
interface InsertOneParams extends ToolParams {
collection: string;
document: Record<string, unknown>;
[key: string]: unknown;
}
export class InsertOneTool extends BaseTool<InsertOneParams> {
name = "insertOne";
description = "Insert a single document into a collection";
inputSchema = {
type: "object" as const,
properties: {
collection: {
type: "string",
description: "Name of the collection",
},
document: {
type: "object",
description: "Document to insert",
},
},
required: ["collection", "document"],
};
async execute(params: InsertOneParams) {
try {
const collection = this.validateCollection(params.collection);
const document = this.validateObject(params.document, "Document");
const result = await db.collection(collection).insertOne(document);
return {
content: [
{
type: "text" as const,
text: JSON.stringify(
{
acknowledged: result.acknowledged,
insertedId: result.insertedId,
},
null,
2
),
},
],
isError: false,
};
} catch (error) {
return this.handleError(error);
}
}
}