📦 NEW: Pipe create and update support (#73)

* 📦 NEW: Pipe create and update support

* 👌 IMPROVE: Type

Author: msaaddev

examples/nodejs/examples/pipes/pipe.create.ts

@@ -0,0 +1,17 @@
+import 'dotenv/config';
+import {Pipe} from 'langbase';
+
+const pipe = new Pipe({
+	apiKey: process.env.LANGBASE_API_KEY!,
+});
+
+async function main() {
+	const response = await pipe.create({
+		name: 'summary-pipe',
+		status: 'private',
+	});
+
+	console.log(response);
+}
+
+main();

examples/nodejs/examples/pipes/pipe.update.ts

@@ -0,0 +1,18 @@
+import 'dotenv/config';
+import {Pipe} from 'langbase';
+
+const pipe = new Pipe({
+	apiKey: process.env.LANGBASE_API_KEY!,
+});
+
+async function main() {
+	const response = await pipe.update({
+		name: 'summary-pipe',
+		description: 'This is a pipe updated with the SDK',
+		model: 'google:gemini-1.5-flash-8b-latest',
+	});
+
+	console.log(response);
+}
+
+main();

examples/nodejs/package.json

@@ -7,6 +7,8 @@
 	"main": "index.js",
 	"scripts": {
 		"generate-text": "npx tsx ./examples/pipes/generate-text.ts",
+		"pipe.create": "npx tsx ./examples/pipes/pipe.create.ts",
+		"pipe.update": "npx tsx ./examples/pipes/pipe.update.ts",
 		"pipe.run": "npx tsx ./examples/pipes/pipe.run.ts",
 		"pipe.run.stream": "npx tsx ./examples/pipes/pipe.run.stream.ts",
 		"generate-text-generate-pipe": "npx tsx ./examples/pipes/generate-text-generate-pipe.ts",

package.json

@@ -4,6 +4,7 @@
 	"license": "Apache-2.0",
 	"scripts": {
 		"dev": "turbo dev",
+		"dev:pkgs": "turbo run dev --filter=./packages/*",
 		"prepare": "husky",
 		"lint": "turbo lint",
 		"test": "turbo test",

packages/langbase/src/pipes/pipes.ts

@@ -24,7 +24,7 @@ export interface ToolCall {
 export interface Message {
 	role: Role;
 	content: string | null;
-	name?: string;
+	name?: 'json' | 'safety' | 'opening' | 'rag';
 	tool_call_id?: string;
 	tool_calls?: ToolCall[];
 }
@@ -107,6 +107,57 @@ export interface PipeOptions {
 	name?: string;
 }
 
+interface ToolChoice {
+	type: 'function';
+	function: {name: string};
+}
+
+interface BaseOptions {
+	name: string;
+	description?: string;
+	status?: 'public' | 'private';
+	upsert?: boolean;
+	model?: string;
+	stream?: boolean;
+	json?: boolean;
+	store?: boolean;
+	moderate?: boolean;
+	top_p?: number;
+	max_tokens?: number;
+	temperature?: number;
+	presence_penalty?: number;
+	frequency_penalty?: number;
+	stop?: string[];
+	tools?: {
+		type: 'function';
+		function: {
+			name: string;
+			description?: string;
+			parameters?: Record<string, any>;
+		};
+	}[];
+	tool_choice?: 'auto' | 'required' | ToolChoice;
+	parallel_tool_calls?: boolean;
+	messages?: Message[];
+	variables?: Variable[];
+	memory?: {
+		name: string;
+	}[];
+}
+
+interface BaseResponse {
+	name: string;
+	description: string;
+	status: 'public' | 'private';
+	owner_login: string;
+	url: string;
+}
+
+export interface CreateOptions extends BaseOptions {}
+export interface UpdateOptions extends BaseOptions {}
+export interface CreateResponse extends BaseResponse {}
+export interface UpdateResponse extends BaseResponse {}
+
 export class Pipe {
 	private request: Request;
 	private pipe: PipeBaseAI;
@@ -152,6 +203,20 @@ export class Pipe {
 			body: {...options, stream: true},
 		});
 	}
+
+	async create(options: CreateOptions): Promise<CreateResponse> {
+		return this.request.post({
+			endpoint: '/v1/pipes',
+			body: options,
+		});
+	}
+
+	async update(options: UpdateOptions): Promise<UpdateResponse> {
+		return this.request.post({
+			endpoint: `/v1/pipes/${options.name}`,
+			body: options,
+		});
+	}
 }
 
 /**