chore: account for types moved around

Author: himanshusinghs

tests/accuracy/collection-storage-size.test.ts

@@ -1,7 +1,7 @@
 import { describeAccuracyTests, describeSuite } from "./sdk/describe-accuracy-tests.js";
 import { getAvailableModels } from "./sdk/models.js";
 import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
-import { ExpectedToolCall } from "./sdk/accuracy-scorers.js";
+import { ExpectedToolCall } from "./sdk/accuracy-snapshot-storage/snapshot-storage.js";
 
 function callsCollectionStorageSize(prompt: string, expectedToolCalls: ExpectedToolCall[]): AccuracyTestConfig {
     return {

tests/accuracy/create-collection.test.ts

@@ -1,7 +1,7 @@
 import { describeAccuracyTests, describeSuite } from "./sdk/describe-accuracy-tests.js";
 import { getAvailableModels } from "./sdk/models.js";
 import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
-import { ExpectedToolCall } from "./sdk/accuracy-scorers.js";
+import { ExpectedToolCall } from "./sdk/accuracy-snapshot-storage/snapshot-storage.js";
 
 function callsCreateCollection(prompt: string, database: string, collection: string): AccuracyTestConfig {
     return {

tests/accuracy/drop-collection.test.ts

@@ -1,7 +1,7 @@
 import { describeAccuracyTests, describeSuite } from "./sdk/describe-accuracy-tests.js";
 import { getAvailableModels } from "./sdk/models.js";
 import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
-import { ExpectedToolCall } from "./sdk/accuracy-scorers.js";
+import { ExpectedToolCall } from "./sdk/accuracy-snapshot-storage/snapshot-storage.js";
 
 function onlyCallsDropCollection(prompt: string): AccuracyTestConfig {
     return {

tests/accuracy/drop-database.test.ts

@@ -1,7 +1,7 @@
 import { describeAccuracyTests, describeSuite } from "./sdk/describe-accuracy-tests.js";
 import { getAvailableModels } from "./sdk/models.js";
 import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
-import { ExpectedToolCall } from "./sdk/accuracy-scorers.js";
+import { ExpectedToolCall } from "./sdk/accuracy-snapshot-storage/snapshot-storage.js";
 
 function onlyCallsDropDatabase(prompt: string): AccuracyTestConfig {
     return {

tests/accuracy/logs.test.ts

@@ -1,7 +1,7 @@
 import { describeAccuracyTests, describeSuite } from "./sdk/describe-accuracy-tests.js";
 import { getAvailableModels } from "./sdk/models.js";
 import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
-import { ExpectedToolCall } from "./sdk/accuracy-scorers.js";
+import { ExpectedToolCall } from "./sdk/accuracy-snapshot-storage/snapshot-storage.js";
 
 function callsLogsTool(prompt: string, toolCall: ExpectedToolCall): AccuracyTestConfig {
     return {

tests/accuracy/sdk/accuracy-snapshot-storage/snapshot-storage.ts

@@ -1,12 +1,11 @@
 import z from "zod";
 
 const ExpectedToolCallSchema = z.object({
-    toolCallId: z.string(),
     toolName: z.string(),
     parameters: z.record(z.string(), z.unknown()),
 });
 
-const ActualToolCallSchema = ExpectedToolCallSchema.omit({ toolCallId: undefined });
+const ActualToolCallSchema = ExpectedToolCallSchema.extend({ toolCallId: z.string() });
 
 export type ExpectedToolCall = z.infer<typeof ExpectedToolCallSchema>;
 export type ActualToolCall = z.infer<typeof ActualToolCallSchema>;

tests/accuracy/sdk/accuracy-testing-client.ts

@@ -5,7 +5,7 @@ import { experimental_createMCPClient as createMCPClient, tool as createVercelTo
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
 
-import { ExpectedToolCall } from "./accuracy-snapshot-storage/snapshot-storage.js";
+import { ActualToolCall } from "./accuracy-snapshot-storage/snapshot-storage.js";
 
 const __dirname = fileURLToPath(import.meta.url);
 const distPath = path.join(__dirname, "..", "..", "..", "..", "dist");
@@ -16,7 +16,7 @@ export type MockedTools = Record<string, ToolResultGeneratorFn>;
 
 export class AccuracyTestingClient {
     private mockedTools: MockedTools = {};
-    private recordedToolCalls: ExpectedToolCall[] = [];
+    private recordedToolCalls: ActualToolCall[] = [];
     private constructor(private readonly vercelMCPClient: Awaited<ReturnType<typeof createMCPClient>>) {}
 
     async close() {