📦 NEW: Helper functions

Author: msaaddev

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

@@ -0,0 +1,57 @@
+import 'dotenv/config';
+import {Langbase, getToolsFromRun, getToolsFromRunStream} from 'langbase';
+
+const langbase = new Langbase({
+	apiKey: process.env.LANGBASE_API_KEY!,
+});
+
+async function main() {
+	const userMsg = "What's the weather in SF";
+
+	const response = await langbase.pipe.run({
+		messages: [
+			{
+				role: 'user',
+				content: userMsg,
+			},
+		],
+		stream: true,
+		name: 'summary',
+		tools: [
+			{
+				type: 'function',
+				function: {
+					name: 'get_current_weather',
+					description: 'Get the current weather of a given location',
+					parameters: {
+						type: 'object',
+						required: ['location'],
+						properties: {
+							unit: {
+								enum: ['celsius', 'fahrenheit'],
+								type: 'string',
+							},
+							location: {
+								type: 'string',
+								description:
+									'The city and state, e.g. San Francisco, CA',
+							},
+						},
+					},
+				},
+			},
+		],
+	});
+
+	const tools = await getToolsFromRunStream(response);
+
+	if (tools.length) {
+		// handle the tool calls
+		console.log('Tools:', tools);
+	} else {
+		// handle the response
+		console.log('Response:', response);
+	}
+}
+
+main();

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

@@ -0,0 +1,57 @@
+import 'dotenv/config';
+import {Langbase, getToolsFromRun} from 'langbase';
+
+const langbase = new Langbase({
+	apiKey: process.env.LANGBASE_API_KEY!,
+});
+
+async function main() {
+	const userMsg = "What's the weather in SF";
+
+	const response = await langbase.pipe.run({
+		messages: [
+			{
+				role: 'user',
+				content: userMsg,
+			},
+		],
+		stream: false,
+		name: 'summary',
+		tools: [
+			{
+				type: 'function',
+				function: {
+					name: 'get_current_weather',
+					description: 'Get the current weather of a given location',
+					parameters: {
+						type: 'object',
+						required: ['location'],
+						properties: {
+							unit: {
+								enum: ['celsius', 'fahrenheit'],
+								type: 'string',
+							},
+							location: {
+								type: 'string',
+								description:
+									'The city and state, e.g. San Francisco, CA',
+							},
+						},
+					},
+				},
+			},
+		],
+	});
+
+	const tools = await getToolsFromRun(response);
+
+	if (tools.length) {
+		// handle the tool calls
+		console.log('Tools:', tools);
+	} else {
+		// handle the response
+		console.log('Response:', response);
+	}
+}
+
+main();

examples/nodejs/package.json

@@ -21,6 +21,8 @@
     "pipe.update": "npx tsx ./examples/pipes/pipe.update.ts",
     "pipe.list": "npx tsx ./examples/pipes/pipe.list.ts",
     "pipe.run": "npx tsx ./examples/pipes/pipe.run.ts",
+    "pipe.tool": "npx tsx ./examples/pipes/pipe.tool.ts",
+    "pipe.tool.stream": "npx tsx ./examples/pipes/pipe.tool.stream.ts",
     "pipe.run.stream": "npx tsx ./examples/pipes/pipe.run.stream.ts",
     "pipe.run.stream.llmkey": "npx tsx ./examples/pipes/pipe.run.stream.llmkey.ts",
     "generate-text-generate-pipe": "npx tsx ./examples/pipes/generate-text-generate-pipe.ts",