fix: 🐛 Restore files

Author: chezsmithy

core/llm/llms/Bedrock.test.ts

@@ -7,21 +7,13 @@ import {
 import Bedrock from "./Bedrock.js";
 
 // Mock AWS SDK
-const mockSend = jest.fn();
-const mockMiddlewareStackAdd = jest.fn();
-
 jest.mock("@aws-sdk/client-bedrock-runtime", () => ({
   BedrockRuntimeClient: jest.fn().mockImplementation(() => ({
-    send: mockSend,
-    middlewareStack: { add: mockMiddlewareStackAdd },
+    send: jest.fn(),
+    middlewareStack: { add: jest.fn() },
   })),
   ConverseStreamCommand: jest.fn(),
-  ConverseCommand: jest.fn(),
   InvokeModelCommand: jest.fn(),
-  ConversationRole: {
-    USER: "user",
-    ASSISTANT: "assistant",
-  },
 }));
 
 // Mock credential provider
@@ -98,8 +90,6 @@ class TestBedrock extends Bedrock {
 describe("Bedrock", () => {
   beforeEach(() => {
     jest.clearAllMocks();
-    mockSend.mockReset();
-    mockMiddlewareStackAdd.mockReset();
   });
 
   describe("constructor", () => {
@@ -314,142 +304,6 @@ describe("Bedrock", () => {
     });
   });
 
-  describe("streaming vs non-streaming behavior", () => {
-    let bedrock: Bedrock;
-    const mockAbortSignal = new AbortController().signal;
-
-    beforeEach(() => {
-      bedrock = new Bedrock({
-        apiKey: "test-key",
-        model: "anthropic.claude-3-sonnet-20240229-v1:0",
-        region: "us-east-1",
-      });
-    });
-
-    it("should route to streaming path when stream is not false", async () => {
-      const messages: ChatMessage[] = [
-        { role: "user", content: "Hello" } as UserChatMessage,
-      ];
-
-      // Mock the private methods by spying on them
-      const streamingMethodSpy = jest
-        .spyOn(bedrock as any, "_streamChatStreaming")
-        .mockImplementation(async function* () {
-          yield { role: "assistant", content: "Streaming response" };
-        });
-
-      const nonStreamingMethodSpy = jest
-        .spyOn(bedrock as any, "_streamChatNonStreaming")
-        .mockImplementation(async function* () {
-          yield { role: "assistant", content: "Non-streaming response" };
-        });
-
-      const options: CompletionOptions = {
-        model: "anthropic.claude-3-sonnet-20240229-v1:0",
-        stream: true, // Explicitly set streaming
-      };
-
-      const results = [];
-      for await (const message of bedrock["_streamChat"](
-        messages,
-        mockAbortSignal,
-        options,
-      )) {
-        results.push(message);
-      }
-
-      expect(streamingMethodSpy).toHaveBeenCalled();
-      expect(nonStreamingMethodSpy).not.toHaveBeenCalled();
-      expect(results).toEqual([
-        { role: "assistant", content: "Streaming response" },
-      ]);
-
-      streamingMethodSpy.mockRestore();
-      nonStreamingMethodSpy.mockRestore();
-    });
-
-    it("should route to non-streaming path when stream is false", async () => {
-      const messages: ChatMessage[] = [
-        { role: "user", content: "Hello" } as UserChatMessage,
-      ];
-
-      // Mock the private methods by spying on them
-      const streamingMethodSpy = jest
-        .spyOn(bedrock as any, "_streamChatStreaming")
-        .mockImplementation(async function* () {
-          yield { role: "assistant", content: "Streaming response" };
-        });
-
-      const nonStreamingMethodSpy = jest
-        .spyOn(bedrock as any, "_streamChatNonStreaming")
-        .mockImplementation(async function* () {
-          yield { role: "assistant", content: "Non-streaming response" };
-        });
-
-      const options: CompletionOptions = {
-        model: "anthropic.claude-3-sonnet-20240229-v1:0",
-        stream: false, // Explicitly set non-streaming
-      };
-
-      const results = [];
-      for await (const message of bedrock["_streamChat"](
-        messages,
-        mockAbortSignal,
-        options,
-      )) {
-        results.push(message);
-      }
-
-      expect(streamingMethodSpy).not.toHaveBeenCalled();
-      expect(nonStreamingMethodSpy).toHaveBeenCalled();
-      expect(results).toEqual([
-        { role: "assistant", content: "Non-streaming response" },
-      ]);
-
-      streamingMethodSpy.mockRestore();
-      nonStreamingMethodSpy.mockRestore();
-    });
-
-    it("should default to streaming path when stream option is undefined", async () => {
-      const messages: ChatMessage[] = [
-        { role: "user", content: "Hello" } as UserChatMessage,
-      ];
-
-      // Mock the private methods by spying on them
-      const streamingMethodSpy = jest
-        .spyOn(bedrock as any, "_streamChatStreaming")
-        .mockImplementation(async function* () {
-          yield { role: "assistant", content: "Streaming response" };
-        });
-
-      const nonStreamingMethodSpy = jest
-        .spyOn(bedrock as any, "_streamChatNonStreaming")
-        .mockImplementation(async function* () {
-          yield { role: "assistant", content: "Non-streaming response" };
-        });
-
-      const options: CompletionOptions = {
-        model: "anthropic.claude-3-sonnet-20240229-v1:0",
-        // stream option is undefined, should default to streaming
-      };
-
-      const results = [];
-      for await (const message of bedrock["_streamChat"](
-        messages,
-        mockAbortSignal,
-        options,
-      )) {
-        results.push(message);
-      }
-
-      expect(streamingMethodSpy).toHaveBeenCalled();
-      expect(nonStreamingMethodSpy).not.toHaveBeenCalled();
-
-      streamingMethodSpy.mockRestore();
-      nonStreamingMethodSpy.mockRestore();
-    });
-  });
-
   describe("message conversion", () => {
     let bedrock: TestBedrock;