test: fix openai mock

packages/core/tests/indices/SummaryIndex.test.ts

@@ -7,7 +7,14 @@ import {
   type StorageContext,
 } from "llamaindex";
 import { rmSync } from "node:fs";
-import { afterAll, beforeAll, describe, expect, it } from "vitest";
+import { afterAll, beforeAll, describe, expect, it, vi } from "vitest";
+
+vi.mock("llamaindex/llm/open_ai", () => {
+  return {
+    getOpenAISession: vi.fn().mockImplementation(() => null),
+  };
+});
+
 import { mockServiceContext } from "../utility/mockServiceContext.js";
 
 describe("SummaryIndex", () => {

packages/core/tests/indices/VectorStoreIndex.test.ts

@@ -4,9 +4,15 @@ import {
   VectorStoreIndex,
   storageContextFromDefaults,
 } from "llamaindex";
-import { beforeAll, describe, expect, test } from "vitest";
+import { beforeAll, describe, expect, test, vi } from "vitest";
 import { mockServiceContext } from "../utility/mockServiceContext.js";
 
+vi.mock("llamaindex/llm/open_ai", () => {
+  return {
+    getOpenAISession: vi.fn().mockImplementation(() => null),
+  };
+});
+
 describe.sequential("VectorStoreIndex", () => {
   let serviceContext: ServiceContext;
   let storageContext: StorageContext;

packages/core/tests/objects/ObjectIndex.test.ts

@@ -5,7 +5,14 @@ import {
   SimpleToolNodeMapping,
   VectorStoreIndex,
 } from "llamaindex";
-import { beforeAll, describe, expect, test } from "vitest";
+import { beforeAll, describe, expect, test, vi } from "vitest";
+
+vi.mock("llamaindex/llm/open_ai", () => {
+  return {
+    getOpenAISession: vi.fn().mockImplementation(() => null),
+  };
+});
+
 import { mockServiceContext } from "../utility/mockServiceContext.js";
 
 describe("ObjectIndex", () => {

packages/core/tests/utility/mockServiceContext.ts

@@ -9,15 +9,6 @@ import {
   mockLlmGeneration,
 } from "../utility/mockOpenAI.js";
 
-import { vi } from "vitest";
-
-// Mock the OpenAI getOpenAISession function during testing
-vi.mock("llamaindex/llm/open_ai", () => {
-  return {
-    getOpenAISession: vi.fn().mockImplementation(() => null),
-  };
-});
-
 export function mockServiceContext() {
   const embeddingModel = new OpenAIEmbedding();
   const llm = new OpenAI();