diff --git a/packages/core/src/callbacks/CallbackManager.ts b/packages/core/src/callbacks/CallbackManager.ts index 71e0c3149eb479b48a983d8bc0cb57936b75457c..f79f972fba4c36dc40f7e2d09688276adae5edba 100644 --- a/packages/core/src/callbacks/CallbackManager.ts +++ b/packages/core/src/callbacks/CallbackManager.ts @@ -5,6 +5,7 @@ import { EventCaller, getEventCaller, } from "../internal/context/EventCaller.js"; +import type { LLMEndEvent, LLMStartEvent } from "../llm/types.js"; export class LlamaIndexCustomEvent<T = any> extends CustomEvent<T> { reason: EventCaller | null; @@ -30,21 +31,19 @@ export class LlamaIndexCustomEvent<T = any> extends CustomEvent<T> { } /** - * This type is used to define the event maps for the Llamaindex package. + * This type is used to define the event maps. */ -export interface LlamaIndexEventMaps {} - -declare module "llamaindex" { - interface LlamaIndexEventMaps { - /** - * @deprecated - */ - retrieve: CustomEvent<RetrievalCallbackResponse>; - /** - * @deprecated - */ - stream: CustomEvent<StreamCallbackResponse>; - } +export interface LlamaIndexEventMaps { + /** + * @deprecated + */ + retrieve: CustomEvent<RetrievalCallbackResponse>; + /** + * @deprecated + */ + stream: CustomEvent<StreamCallbackResponse>; + "llm-start": LLMStartEvent; + "llm-end": LLMEndEvent; } //#region @deprecated remove in the next major version diff --git a/packages/core/src/llm/types.ts b/packages/core/src/llm/types.ts index f8ac3f77a9507dc4c20bf3f773baf301b80c31c0..626183b1122d3b40b70e480d56a0c7efdc1318cf 100644 --- a/packages/core/src/llm/types.ts +++ b/packages/core/src/llm/types.ts @@ -21,13 +21,6 @@ export type LLMEndEvent = LLMBaseEvent< } >; -declare module "llamaindex" { - interface LlamaIndexEventMaps { - "llm-start": LLMStartEvent; - "llm-end": LLMEndEvent; - } -} - /** * @internal */