From 1b6263e08d842ec4e0b9e0214fd39bfff64df4ab Mon Sep 17 00:00:00 2001
From: Alex Yang <himself65@outlook.com>
Date: Fri, 9 Aug 2024 10:10:12 -0700
Subject: [PATCH] fix: export schema in top level (#1112)
---
packages/llamaindex/src/index.edge.ts | 2 ++
packages/llamaindex/tests/import.test.ts | 11 +++++++++++
2 files changed, 13 insertions(+)
create mode 100644 packages/llamaindex/tests/import.test.ts
diff --git a/packages/llamaindex/src/index.edge.ts b/packages/llamaindex/src/index.edge.ts
index ebc8807fd..cf1162d73 100644
--- a/packages/llamaindex/src/index.edge.ts
+++ b/packages/llamaindex/src/index.edge.ts
@@ -1,6 +1,8 @@
import type { AgentEndEvent, AgentStartEvent } from "./agent/types.js";
import type { RetrievalEndEvent, RetrievalStartEvent } from "./llm/types.js";
+export * from "@llamaindex/core/schema";
+
declare module "@llamaindex/core/global" {
export interface LlamaIndexEventMaps {
"retrieve-start": RetrievalStartEvent;
diff --git a/packages/llamaindex/tests/import.test.ts b/packages/llamaindex/tests/import.test.ts
new file mode 100644
index 000000000..9bcdd4474
--- /dev/null
+++ b/packages/llamaindex/tests/import.test.ts
@@ -0,0 +1,11 @@
+import { expect, test } from "vitest";
+
+test("Node classes should be included in the top level", async () => {
+ const { Document, IndexNode, TextNode, BaseNode } = await import(
+ "llamaindex"
+ );
+ expect(Document).toBeDefined();
+ expect(IndexNode).toBeDefined();
+ expect(TextNode).toBeDefined();
+ expect(BaseNode).toBeDefined();
+});
--
GitLab