count.test.ts•1.33 kB
import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
import { Matcher } from "./sdk/matcher.js";
describeAccuracyTests([
{
prompt: "Count number of documents in 'mflix.movies' namespace.",
expectedToolCalls: [
{
toolName: "count",
parameters: {
database: "mflix",
collection: "movies",
query: Matcher.emptyObjectOrUndefined,
},
},
],
},
{
prompt: "How many documents are there in 'characters' collection in 'comics' database?",
expectedToolCalls: [
{
toolName: "count",
parameters: {
database: "comics",
collection: "characters",
query: Matcher.emptyObjectOrUndefined,
},
},
],
},
{
prompt: "Count all the documents in 'mflix.movies' namespace with runtime less than 100?",
expectedToolCalls: [
{
toolName: "count",
parameters: {
database: "mflix",
collection: "movies",
query: { runtime: { $lt: 100 } },
},
},
],
},
]);