centralize collectEvents

Author: afarntrog

src/models/__tests__/bedrock.test.ts

@@ -2,20 +2,9 @@ import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
 import { BedrockRuntimeClient } from '@aws-sdk/client-bedrock-runtime'
 import { BedrockModel } from '../bedrock'
 import { ContextWindowOverflowError } from '../../errors'
+import { collectEvents } from './test-utils'
 import type { Message } from '../../types/messages'
 import type { StreamOptions } from '../model'
-import type { ModelStreamEvent } from '../streaming'
-
-/**
- * Helper function to collect all events from a stream.
- */
-async function collectEvents(stream: AsyncIterable<ModelStreamEvent>): Promise<ModelStreamEvent[]> {
-  const events: ModelStreamEvent[] = []
-  for await (const event of stream) {
-    events.push(event)
-  }
-  return events
-}
 
 /**
  * Helper function to setup mock send with custom stream generator.

src/models/__tests__/openai.test.ts

@@ -3,19 +3,8 @@ import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
 import OpenAI from 'openai'
 import { OpenAIModel } from '../openai'
 import { ContextWindowOverflowError } from '../../errors'
+import { collectEvents } from './test-utils'
 import type { Message } from '../../types/messages'
-import type { ModelStreamEvent } from '../streaming'
-
-/**
- * Helper function to collect all events from a stream.
- */
-async function collectEvents(stream: AsyncIterable<ModelStreamEvent>): Promise<ModelStreamEvent[]> {
-  const events: ModelStreamEvent[] = []
-  for await (const event of stream) {
-    events.push(event)
-  }
-  return events
-}
 
 /**
  * Helper to create a mock OpenAI client with streaming support

src/models/__tests__/test-utils.ts

@@ -0,0 +1,19 @@
+// ABOUTME: Shared test utilities for model tests
+// ABOUTME: Contains helper functions for collecting stream events and other common test operations
+
+import type { ModelStreamEvent } from '../streaming'
+
+/**
+ * Helper function to collect all events from a stream.
+ * Useful for testing streaming model responses.
+ * 
+ * @param stream - An async iterable of ModelStreamEvent
+ * @returns Promise resolving to an array of all emitted events
+ */
+export async function collectEvents(stream: AsyncIterable<ModelStreamEvent>): Promise<ModelStreamEvent[]> {
+  const events: ModelStreamEvent[] = []
+  for await (const event of stream) {
+    events.push(event)
+  }
+  return events
+}