diff --git a/packages/server/src/services/model-providers/providers/streaming-engine/streaming-engine.ts b/packages/server/src/services/model-providers/providers/streaming-engine/streaming-engine.ts index 019d6e4..a1f1866 100644 --- a/packages/server/src/services/model-providers/providers/streaming-engine/streaming-engine.ts +++ b/packages/server/src/services/model-providers/providers/streaming-engine/streaming-engine.ts @@ -1,10 +1,11 @@ import { ReadableStream } from "stream/web"; import { ProviderPlugin } from "../provider-plugin"; import { StreamParams } from "./streaming-types"; +import { ChatService } from "../../chat/chat-service"; export async function createSSEStream( plugin: ProviderPlugin, - streamParams: StreamParams + streamParams: StreamParams & { chatService: ChatService } ): Promise> { const { chatService, assistantMessageId } = streamParams; const streamOrReader = await plugin.prepareRequest(streamParams); diff --git a/packages/server/src/services/model-providers/providers/streaming-engine/streaming-types.ts b/packages/server/src/services/model-providers/providers/streaming-engine/streaming-types.ts index 6411d13..5895b58 100644 --- a/packages/server/src/services/model-providers/providers/streaming-engine/streaming-types.ts +++ b/packages/server/src/services/model-providers/providers/streaming-engine/streaming-types.ts @@ -1,11 +1,9 @@ -import { ChatService } from "../../chat/chat-service"; import { ChatCompletionOptions } from "../provider-types"; export type StreamParams = { chatId: string; assistantMessageId: string; userMessage: string; - chatService: ChatService; options: ChatCompletionOptions; tempId?: string; }; \ No newline at end of file