👌 IMPROVE: SDK structure (#77)

LangbaseInc · Nov 26, 2024 · fcfb0db · fcfb0db
1 parent 8b84733
commit fcfb0db
Show file tree

Hide file tree

Showing 18 changed files with 321 additions and 253 deletions.
diff --git a/examples/nextjs/app/langbase/pipe/run-stream/route.ts b/examples/nextjs/app/langbase/pipe/run-stream/route.ts
@@ -1,16 +1,16 @@
-import {Pipe} from 'langbase';
+import {Langbase} from 'langbase';
 import {NextRequest} from 'next/server';
 
 export async function POST(req: NextRequest) {
 	const {prompt} = await req.json();
 
 	// 1. Initiate the Pipe.
-	const myPipe = new Pipe({
+	const langbase = new Langbase({
 		apiKey: process.env.LANGBASE_API_KEY!,
 	});
 
 	// 2. Generate a stream by asking a question
-	const {stream, threadId} = await myPipe.run({
+	const {stream, threadId} = await langbase.pipe.run({
 		messages: [{role: 'user', content: prompt}],
 		stream: true,
 		name: 'summary',

diff --git a/examples/nextjs/app/langbase/pipe/run/route.ts b/examples/nextjs/app/langbase/pipe/run/route.ts
@@ -1,16 +1,16 @@
-import {Pipe} from 'langbase';
+import {Langbase} from 'langbase';
 import {NextRequest} from 'next/server';
 
 export async function POST(req: NextRequest) {
 	const {prompt} = await req.json();
 
 	// 1. Initiate the Pipe.
-	const myPipe = new Pipe({
+	const langbase = new Langbase({
 		apiKey: process.env.LANGBASE_API_KEY!,
 	});
 
 	// 2. Generate a stream by asking a question
-	const result = await myPipe.run({
+	const result = await langbase.pipe.run({
 		messages: [{role: 'user', content: prompt}],
 		name: 'summary',
 	});

diff --git a/examples/nodejs/examples/memory/memory.create.ts b/examples/nodejs/examples/memory/memory.create.ts
@@ -1,13 +1,13 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await memory.create({
-		name: 'sdk-memory-2',
+	const response = await langbase.memory.create({
+		name: 'sdk-memory',
 		embedding_model: 'cohere:embed-multilingual-v3.0'
 	});
 

diff --git a/examples/nodejs/examples/memory/memory.delete.ts b/examples/nodejs/examples/memory/memory.delete.ts
@@ -1,12 +1,12 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await memory.delete({
+	const response = await langbase.memory.delete({
 		name: 'memory-sdk',
 	});
 

diff --git a/examples/nodejs/examples/memory/memory.docs.delete.ts b/examples/nodejs/examples/memory/memory.docs.delete.ts
@@ -1,12 +1,12 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await memory.deleteDoc({
+	const response = await langbase.memory.documents.delete({
 		memoryName: 'memory-sdk',
 		documentName: 'readme.md',
 	});

diff --git a/examples/nodejs/examples/memory/memory.docs.list.ts b/examples/nodejs/examples/memory/memory.docs.list.ts
@@ -1,13 +1,13 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await memory.listDocs({
-		memoryName: 'memory-sdk'
+	const response = await langbase.memory.documents.list({
+		memoryName: 'memory-sdk',
 	});
 
 	console.log(response);

diff --git a/examples/nodejs/examples/memory/memory.docs.retry-embed.ts b/examples/nodejs/examples/memory/memory.docs.retry-embed.ts
@@ -1,12 +1,12 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await memory.retryDocEmbed({
+	const response = await langbase.memory.documents.embedding.retry({
 		memoryName: 'memory-sdk',
 		documentName: 'memory.upload.doc.ts',
 	});

diff --git a/examples/nodejs/examples/memory/memory.docs.upload.ts b/examples/nodejs/examples/memory/memory.docs.upload.ts
@@ -1,9 +1,9 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 import fs from 'fs';
 import path from 'path';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
@@ -16,7 +16,7 @@ async function main() {
 	);
 	const file = fs.readFileSync(src);
 
-	const response = await memory.uploadDoc({
+	const response = await langbase.memory.documents.upload({
 		file,
 		memoryName: 'memory-sdk',
 		fileName: 'memory.upload.doc.ts',

diff --git a/examples/nodejs/examples/memory/memory.list.ts b/examples/nodejs/examples/memory/memory.list.ts
@@ -1,12 +1,12 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await memory.list();
+	const response = await langbase.memory.list();
 	console.log(response);
 }
 

diff --git a/examples/nodejs/examples/memory/memory.retrieve.ts b/examples/nodejs/examples/memory/memory.retrieve.ts
@@ -1,12 +1,12 @@
 import 'dotenv/config';
-import {Memory} from 'langbase';
+import {Langbase} from 'langbase';
 
-const memory = new Memory({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await memory.retrieve({
+	const response = await langbase.memory.retrieve({
 		memory: [
 			{
 				name: 'langbase-docs',

diff --git a/examples/nodejs/examples/pipes/pipe.create.ts b/examples/nodejs/examples/pipes/pipe.create.ts
@@ -1,13 +1,13 @@
 import 'dotenv/config';
-import {Pipe} from 'langbase';
+import {Langbase} from 'langbase';
 
-const pipe = new Pipe({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await pipe.create({
-		name: 'summary-pipe',
+	const response = await langbase.pipe.create({
+		name: 'summary-pipe2',
 		status: 'private',
 	});
 

diff --git a/examples/nodejs/examples/pipes/pipe.list.ts b/examples/nodejs/examples/pipes/pipe.list.ts
@@ -1,12 +1,12 @@
 import 'dotenv/config';
-import {Pipe} from 'langbase';
+import {Langbase} from 'langbase';
 
-const pipe = new Pipe({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
-	const response = await pipe.list();
+	const response = await langbase.pipe.list();
 	console.log(response);
 }
 

diff --git a/examples/nodejs/examples/pipes/pipe.run.stream.ts b/examples/nodejs/examples/pipes/pipe.run.stream.ts
@@ -1,15 +1,15 @@
 import 'dotenv/config';
-import {getRunner, Pipe} from 'langbase';
+import {getRunner, Langbase} from 'langbase';
 
-const pipe = new Pipe({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
 	const userMsg = 'Who is an AI Engineer?';
 
 	// Get readable stream
-	const {stream, threadId, rawResponse} = await pipe.run({
+	const {stream, threadId, rawResponse} = await langbase.pipe.run({
 		messages: [{role: 'user', content: userMsg}],
 		stream: true,
 		rawResponse: true,

diff --git a/examples/nodejs/examples/pipes/pipe.run.ts b/examples/nodejs/examples/pipes/pipe.run.ts
@@ -1,14 +1,14 @@
 import 'dotenv/config';
-import {Pipe} from 'langbase';
+import {Langbase} from 'langbase';
 
-const pipe = new Pipe({
+const langbase = new Langbase({
 	apiKey: process.env.LANGBASE_API_KEY!,
 });
 
 async function main() {
 	const userMsg = 'Who is an AI Engineer?';
 
-	const response = await pipe.run({
+	const response = await langbase.pipe.run({
 		messages: [
 			{
 				role: 'user',

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

diff --git a/packages/langbase/src/index.ts b/packages/langbase/src/index.ts
@@ -1,6 +1,6 @@
 export {fromReadableStream} from './lib/browser/stream';
+export * from './langbase/langbase';
 export * from './pipes/pipes';
-export * from './memory/memory';
 export * from '@baseai/core/helpers';
 export type {
 	RunOptions,