👌 IMPROVE: generate format

Author: ahmadawais

packages/langbase/src/pipes/pipes.ts

@@ -99,20 +99,31 @@ export class Pipe {
 		this.request = new Request({apiKey: options.apiKey, baseUrl});
 	}
 
-	async generateText(
-		options: GenerateOptions,
-	): Promise<GenerateNonStreamResponse> {
-		return this.request.post<GenerateNonStreamResponse>({
-			endpoint: '/beta/generate',
+	async generateText(options: GenerateOptions): Promise<GenerateResponse> {
+		return this.request.post<GenerateResponse>({
+			endpoint: options.chat ? '/beta/chat' : '/beta/generate',
 			body: {...options, stream: false},
-			stream: false,
 		});
 	}
 
-	async streamText(options: StreamOptions): Promise<StreamResponse> {
-		return this.request.post<StreamResponse>({
-			endpoint: options.chat ? '/beta/chat' : '/beta/generate',
+	async streamText(options: StreamOptions): Promise<GenerateStreamResponse> {
+		return this.request.post<GenerateStreamResponse>({
+			endpoint: '/beta/generate',
 			body: {...options, stream: true},
+			stream: true, // TODO: @ahmadbilaldev - why we need to add here as well?
 		});
 	}
 }
+
+/**
+ * Print stream to standard output (console).
+ * @param stream The stream to print
+ */
+export const printStreamToStdout = async (
+	stream: StreamText,
+): Promise<void> => {
+	for await (const chunk of stream) {
+		const textPart = chunk.choices[0]?.delta?.content || '';
+		process.stdout.write(textPart);
+	}
+};