diff --git a/apps/simple/summaryIndex.ts b/apps/simple/summaryIndex.ts index d3fe7736b9a64934359b251bad853260aac796e9..cc34e95d392c167d441ccfa4ad115cf0d39bf7b8 100644 --- a/apps/simple/summaryIndex.ts +++ b/apps/simple/summaryIndex.ts @@ -1,8 +1,8 @@ import { Document, - ListRetrieverMode, SimpleNodeParser, SummaryIndex, + SummaryRetrieverMode, serviceContextFromDefaults, } from "llamaindex"; import essay from "./essay"; @@ -18,7 +18,7 @@ async function main() { serviceContext, }); const queryEngine = index.asQueryEngine({ - retriever: index.asRetriever({ mode: ListRetrieverMode.LLM }), + retriever: index.asRetriever({ mode: SummaryRetrieverMode.LLM }), }); const response = await queryEngine.query( "What did the author do growing up?", diff --git a/examples/summaryIndex.ts b/examples/summaryIndex.ts index 9045d8ef517472e0cd74e87cc90d72490d3e8c83..cc34e95d392c167d441ccfa4ad115cf0d39bf7b8 100644 --- a/examples/summaryIndex.ts +++ b/examples/summaryIndex.ts @@ -1,7 +1,8 @@ import { Document, - ListRetrieverMode, SimpleNodeParser, + SummaryIndex, + SummaryRetrieverMode, serviceContextFromDefaults, } from "llamaindex"; import essay from "./essay"; @@ -13,9 +14,11 @@ async function main() { }), }); const document = new Document({ text: essay, id_: "essay" }); - const index = await ListIndex.fromDocuments([document], { serviceContext }); + const index = await SummaryIndex.fromDocuments([document], { + serviceContext, + }); const queryEngine = index.asQueryEngine({ - retriever: index.asRetriever({ mode: ListRetrieverMode.LLM }), + retriever: index.asRetriever({ mode: SummaryRetrieverMode.LLM }), }); const response = await queryEngine.query( "What did the author do growing up?", diff --git a/packages/core/src/indices/summary/SummaryIndex.ts b/packages/core/src/indices/summary/SummaryIndex.ts index fcf4e03260b4944d9b7fafeded5de79ea82a1262..39a8ec5259cf75478d3b72380821664856cbd5a4 100644 --- a/packages/core/src/indices/summary/SummaryIndex.ts +++ b/packages/core/src/indices/summary/SummaryIndex.ts @@ -254,3 +254,4 @@ export class SummaryIndex extends BaseIndex<IndexList> { // Legacy export type ListIndex = SummaryIndex; +export type ListRetrieverMode = SummaryRetrieverMode; diff --git a/packages/core/src/indices/summary/index.ts b/packages/core/src/indices/summary/index.ts index b3f8dba2c7acb7d2feacd47294c33e0b4ac95f7c..b6e0c7d5ec0a635bca54be5d7c4ec38945407717 100644 --- a/packages/core/src/indices/summary/index.ts +++ b/packages/core/src/indices/summary/index.ts @@ -1,8 +1,5 @@ -export { - SummaryRetrieverMode as ListRetrieverMode, - SummaryIndex, -} from "./SummaryIndex"; -export type { ListIndex } from "./SummaryIndex"; +export { SummaryIndex, SummaryRetrieverMode } from "./SummaryIndex"; +export type { ListIndex, ListRetrieverMode } from "./SummaryIndex"; export { SummaryIndexLLMRetriever, SummaryIndexRetriever,