chore: add a few more tests and some more models

Author: himanshusinghs

tests/accuracy/list-collections.test.ts

@@ -0,0 +1,38 @@
+import { describeAccuracyTests } from "./sdk/describe-accuracy-tests.js";
+import { getAvailableModels } from "./sdk/models.js";
+import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
+
+function describeListCollectionsAccuracyTests(prompt: string): AccuracyTestConfig {
+    return {
+        systemPrompt: "Assume that you're already connected.",
+        prompt: prompt,
+        mockedTools: {
+            "list-collections": function listCollections() {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: "Name: coll1",
+                        },
+                        {
+                            type: "text",
+                            text: "Name: coll1",
+                        },
+                    ],
+                };
+            },
+        },
+        expectedToolCalls: [
+            {
+                toolName: "list-collections",
+                parameters: { database: "db1" },
+            },
+        ],
+    };
+}
+
+describeAccuracyTests("list-collections", getAvailableModels(), [
+    describeListCollectionsAccuracyTests("How many collections do I have in database db1?"),
+    describeListCollectionsAccuracyTests("List all the collections in my MongoDB database db1."),
+    describeListCollectionsAccuracyTests("Is there a coll1 collection in my MongoDB database db1?"),
+]);

tests/accuracy/sdk/models.ts

@@ -26,7 +26,7 @@ export class OllamaModel implements Model {
     constructor(readonly modelName: string) {}
 
     isAvailable(): boolean {
-        return false;
+        return true;
     }
 
     getModel() {
@@ -36,8 +36,8 @@ export class OllamaModel implements Model {
 
 const ALL_TESTABLE_MODELS = [
     new GeminiModel("gemini-1.5-flash"),
-    // new GeminiModel("gemini-2.0-flash"),
-    // new OllamaModel("qwen3:latest"),
+    new GeminiModel("gemini-2.0-flash"),
+    new OllamaModel("qwen3:1.7b"),
 ];
 
 export type TestableModels = ReturnType<typeof getAvailableModels>;