fix stream params type

brandon-schabel · Jan 9, 2025 · e93c4c6 · e93c4c6
1 parent 0943968
commit e93c4c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
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
@@ -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<ReadableStream<Uint8Array>> {
     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
@@ -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;
 };