👌 IMPROVE: Tool stream code

Author: msaaddev

examples/nodejs/examples/pipes/pipe.tool.stream.ts

@@ -1,5 +1,5 @@
 import 'dotenv/config';
-import {Langbase, getToolsFromRun, getToolsFromRunStream} from 'langbase';
+import {Langbase, getToolsFromRunStream} from 'langbase';
 
 const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
@@ -43,14 +43,15 @@ async function main() {
 		],
 	});
 
-	const tools = await getToolsFromRunStream(response);
+	const [streamForResponse, streamForToolCall] = response.stream.tee();
+
+	const tools = await getToolsFromRunStream(streamForToolCall);
 
 	if (tools.length) {
 		// handle the tool calls
 		console.log('Tools:', tools);
 	} else {
-		// handle the response
-		console.log('Response:', response);
+		// handle the response stream
 	}
 }
 

packages/langbase/src/lib/helpers/index.ts

@@ -129,8 +129,8 @@ export async function getToolsFromStream(
  * @param stream - The readable stream to extract tools from
  * @returns A promise that resolves with the tools extracted from the stream
  */
-export async function getToolsFromRunStream(response: RunResponseStream) {
-	return getToolsFromStream(response.stream);
+export async function getToolsFromRunStream(stream: ReadableStream<any>) {
+	return getToolsFromStream(stream);
 }
 
 /**