📦 NEW: Docs MCP Server

packages/cli/README.md

@@ -1,7 +1,63 @@
-## Langbase CLI
+# Langbase CLI
 
-Langbase CLI is a command line interface for Langbase. It allows you to interact with Langbase from the command line.
+The Langbase CLI is a command-line interface for Langbase. It provides a set of commands to interact with the Langbase SDK and perform various tasks related to AI development.
 
-## Documentation
+## Usage
 
-Please follow the [Langbase documentation](https://langbase.com/docs) for the latest information.
+### Langbase Docs MCP Server
+
+Integrate the Langbase Docs MCP server into your IDEs and Claude Desktop.
+
+#### Cursor
+- Open Cursor settings
+- Navigate to the MCP settings
+- Click on the `+` button to add a new global MCP server
+- Paste the following configuration in the `mcp.json` file:
+```json
+{
+    "mcpServers": {
+        "Langbase": {
+        "command": "npx",
+        "args": ["@langbase/cli","docs-mcp-server"]
+        }
+    }
+}
+```
+
+#### Windsurf
+- Navigate to Windsurf - Settings > Advanced Settings
+- You will find the option to Add Server
+- Click “Add custom server +”
+- Paste the following configuration in the `mcp_config.json` file:
+```json
+{
+    "mcpServers": {
+        "Langbase": {
+            "command": "npx",
+            "args": ["@langbase/cli", "docs-mcp-server"]
+        }
+    }
+}
+```
+
+#### Claude Desktop
+- Open Claude Desktop File Menu
+- Navigate to the settings
+- Go to Developer Tab
+- Click on the Edit Config button
+- Paste the following configuration in the `claude_desktop_config.json` file:
+```json
+{
+    "mcpServers": {
+        "Langbase": {
+            "command": "npx",
+            "args": ["@langbase/cli", "docs-mcp-server"]
+        }
+    }
+}
+```
+
+## Next steps
+
+- Read the [Langbase SDK documentation](https://langbase.com/docs/sdk) for more details
+- Join our [Discord](https://langbase.com/discord) community for feedback, requests, and support

packages/cli/package.json

@@ -51,6 +51,7 @@
 		"@clack/prompts": "^0.7.0",
 		"@hono/node-server": "^1.13.1",
 		"@hono/zod-openapi": "^0.16.0",
+		"@modelcontextprotocol/sdk": "^1.8.0",
 		"@sindresorhus/slugify": "^2.2.1",
 		"camelcase": "^8.0.0",
 		"chalk": "^5.3.0",
@@ -71,6 +72,7 @@
 		"get-package-json-file": "^2.0.0",
 		"hono": "^4.5.11",
 		"js-tiktoken": "^1.0.14",
+		"jsdom": "^24.1.0",
 		"log-symbols": "^7.0.0",
 		"lowdb": "^7.0.1",
 		"meow": "^13.2.0",
@@ -90,6 +92,7 @@
 	"devDependencies": {
 		"@langbase/eslint-config": "workspace:*",
 		"@langbase/tsconfig": "workspace:*",
+		"@types/jsdom": "^21.1.7",
 		"@types/node": "^22.6.1",
 		"tsup": "^8.3.0",
 		"tsx": "^4.19.1",

packages/cli/src/docs-mcp-server/docs.ts

@@ -0,0 +1,58 @@
+import { JSDOM } from 'jsdom';
+
+/**
+ * Fetches a list of all the docs on the langbase website
+ *
+ *
+ * @returns {Promise<string>} A promise that resolves to a string of all the docs on the langbase website
+ */
+export async function fetchDocsList() {
+	try {
+		const response = await fetch('https://langbase.com/docs/llms.txt');
+		if (!response.ok) {
+			throw new Error('Failed to fetch docs');
+		}
+
+		const text = await response.text();
+		return text;
+	} catch (error) {
+		throw new Error('Failed to fetch docs ' + JSON.stringify(error));
+	}
+}
+
+/**
+ * Fetches and converts a blog post to markdown
+ *
+ *
+ * @param {string} url - The URL of the blog post to fetch
+ * @returns {Promise<string>} A promise that resolves to a string of the blog post in markdown format
+ */
+export async function fetchDocsPost(url: string): Promise<string> {
+	try {
+		const response = await fetch(url);
+		if (!response.ok) {
+			throw new Error('Failed to fetch blog post');
+		}
+
+		const html = await response.text();
+
+		const dom = new JSDOM(html);
+		const document = dom.window.document;
+
+		// Remove Next.js initialization code
+		const scripts = document.querySelectorAll('script');
+		scripts.forEach(script => script.remove());
+
+		// Get the main content
+		const content = document.body.textContent?.trim() || '';
+		if (!content) {
+			throw new Error('No content found in docs');
+		}
+
+		return content;
+	} catch (error) {
+		throw new Error(
+			`Failed to fetch docs: ${error instanceof Error ? error.message : 'Something went wrong. Please try again.'}`
+		);
+	}
+}

packages/cli/src/docs-mcp-server/index.ts

@@ -0,0 +1,181 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { z } from 'zod';
+import { fetchDocsList, fetchDocsPost } from './docs';
+import { getRelevanceScore } from '@/utils/get-score';
+
+export async function docsMcpServer() {
+	const server = new McpServer({
+		name: 'langbase-docs-server',
+		version: '0.1.0'
+	});
+
+	server.tool(
+		'docs-route-finder',
+		"Searches through all available documentation routes and returns relevant paths based on the user's query. This tool helps navigate the documentation by finding the most appropriate sections that match the search criteria.",
+		{
+			query: z.string()
+				.describe(`A refined search term extracted from the user's question.
+            For example, if user asks 'How do I create a pipe?', the query would be 'SDK Pipe'.
+            This should be the specific concept or topic to search for in the documentation.
+            Treat keyword add as create if user ask for Eg. 'How do I add memory to pipe?' the query should be 'create memory'`)
+		},
+		async ({ query }) => {
+			const docs = await fetchDocsList();
+			// search through the docs and return the most relevent path based on the query
+			const docLines = docs.split('\n').filter(line => line.trim());
+
+			// Score and sort the documentation entries
+			const scoredDocs = docLines
+				.map(line => ({
+					line,
+					score: getRelevanceScore(line, query)
+				}))
+				.sort((a, b) => b.score - a.score)
+				.filter(doc => doc.score > 0)
+				.slice(0, 3); // Get top 3 most relevant results
+
+			const hasRelevantDocs = scoredDocs.length === 0;
+
+			if (hasRelevantDocs) {
+				return {
+					content: [
+						{
+							type: 'text',
+							text:
+								'No relevant documentation found for the query: ' +
+								query
+						}
+					]
+				};
+			}
+
+			const results = scoredDocs.map(doc => doc.line).join('\n');
+
+			return {
+				content: [
+					{
+						type: 'text',
+						text: results
+					}
+				]
+			};
+		}
+	);
+
+	server.tool(
+		'sdk-documentation-fetcher',
+		'Fetches detailed SDK documentation, specializing in implementation guides for core features like pipes, memory, and tools. This is the primary source for the latest SDK documentation and should be consulted first for questions about creating or implementing SDK components. Use this tool for detailed step-by-step instructions on building pipes, configuring memory systems, and developing custom tools.',
+		{
+			url: z
+				.string()
+				.describe(
+					'URL of a specific SDK page to fetch. Format: /sdk/...'
+				)
+		},
+		async ({ url }) => {
+			const content = await fetchDocsPost(
+				`https://langbase.com/docs${url}`
+			);
+			return {
+				content: [
+					{
+						type: 'text',
+						text: content
+					}
+				]
+			};
+		}
+	);
+
+	server.tool(
+		'examples-tool',
+		'Fetches code examples and sample implementations from the documentation. Use this tool when users specifically request examples, sample code, or implementation demonstrations. This tool provides practical code snippets and complete working examples that demonstrate how to implement various features.',
+		{
+			url: z
+				.string()
+				.describe(
+					'URL of a specific examples page to fetch. Format: /examples/...'
+				)
+		},
+		async ({ url }) => {
+			const content = await fetchDocsPost(
+				`https://langbase.com/docs${url}`
+			);
+			return {
+				content: [
+					{
+						type: 'text',
+						text: content
+					}
+				]
+			};
+		}
+	);
+
+	server.tool(
+		'guide-tool',
+		'Fetches detailed guides and tutorials from the documentation. Use this tool when users explicitly request guides, tutorials, or how-to content. This tool provides step-by-step instructions and practical examples for implementing various features.',
+		{
+			url: z
+				.string()
+				.describe(
+					'URL of a specific guide page to fetch. Format: /guides/...'
+				)
+		},
+		async ({ url }) => {
+			const content = await fetchDocsPost(
+				`https://langbase.com/docs${url}`
+			);
+			return {
+				content: [
+					{
+						type: 'text',
+						text: content
+					}
+				]
+			};
+		}
+	);
+
+	server.tool(
+		'api-reference-tool',
+		'Fetches API reference documentation. Use this tool ONLY when the user explicitly asks about API endpoints, REST API calls, or programmatically creating/updating/deleting resources (like pipes, memory, etc.) through the API interface. For general SDK implementation questions, use the sdk-documentation-fetcher instead.',
+		{
+			url: z
+				.string()
+				.describe(
+					'URL of a specific api-reference page to fetch. Format: /api-reference/...'
+				)
+		},
+		async ({ url }) => {
+			const content = await fetchDocsPost(
+				`https://langbase.com/docs${url}`
+			);
+			return {
+				content: [
+					{
+						type: 'text',
+						text: content
+					}
+				]
+			};
+		}
+	);
+
+	async function main() {
+		const transport = new StdioServerTransport();
+
+		try {
+			await server.connect(transport);
+		} catch (error) {
+			console.error('Error connecting to transport:', error);
+			process.exit(1);
+		}
+	}
+
+	main().catch(error => {
+		console.error('Something went wrong:', error);
+		process.exit(1);
+	});
+}

packages/cli/src/index.ts

@@ -2,6 +2,7 @@
 import { auth } from './auth';
 import { build } from './build';
 import { deploy } from './deploy';
+import { docsMcpServer } from './docs-mcp-server';
 import cli from './utils/cli';
 import debugMode from './utils/debug-mode';
 import cliInit from './utils/init';
@@ -16,7 +17,10 @@ const command = (cmd: string): boolean => input.includes(cmd);
 const flag = (flg: string): boolean => Boolean(flags[flg]);
 
 (async () => {
-	await cliInit({ clear });
+	// Skip welcome message for docs-mcp-server command
+	if (!command('docs-mcp-server')) {
+		await cliInit({ clear });
+	}
 	if (debug) debugMode(cli);
 
 	if (command('help')) {
@@ -42,4 +46,8 @@ const flag = (flg: string): boolean => Boolean(flags[flg]);
 
 		await deploy({ isDev, agent, filePath, apiKey });
 	}
+
+	if (command('docs-mcp-server')) {
+		await docsMcpServer();
+	}
 })();

packages/cli/src/utils/cli.ts

@@ -1,4 +1,5 @@
 import meowHelp from 'cli-meow-help';
+// @ts-ignore
 import meow from 'meow';
 
 const flags = {
@@ -37,7 +38,7 @@ const commands = {
 };
 
 const helpText = meowHelp({
-	name: `baseai`,
+	name: `@langbase/cli`,
 	flags,
 	commands,
 	desc: false,