Add support for fixed pipeline ID via environment variable (#5)

* Add support for fixed pipeline ID via environment variable

Co-Authored-By: [email protected] <[email protected]>

* Update README to document VECTORIZE_PIPELINE_ID environment variable

Co-Authored-By: [email protected] <[email protected]>

* Update README.md

---------

Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Co-authored-by: [email protected] <[email protected]>
Co-authored-by: Nicolò Boschi <[email protected]>

Author: 3 people

README.md

@@ -15,10 +15,12 @@ A Model Context Protocol (MCP) server implementation that integrates with [Vecto
 ```bash
 export VECTORIZE_ORG_ID=YOUR_ORG_ID
 export VECTORIZE_TOKEN=YOUR_TOKEN
+# Optional: Set a fixed pipeline ID to use for all requests
+export VECTORIZE_PIPELINE_ID=YOUR_PIPELINE_ID
 npx -y @vectorize-io/vectorize-mcp-server
 ```
 
-## Configuration on Claude/Windsurf
+## Configuration on Claude/Windsurf/Cursor/Cline
 
 ```json
 {
@@ -28,7 +30,8 @@ npx -y @vectorize-io/vectorize-mcp-server
       "args": ["-y", "@vectorize-io/vectorize-mcp-server"],
       "env": {
         "VECTORIZE_ORG_ID": "your-org-id",
-        "VECTORIZE_TOKEN": "your-token"
+        "VECTORIZE_TOKEN": "your-token",
+        "VECTORIZE_PIPELINE_ID": "your-pipeline-id"
       }
     }
   }
@@ -94,4 +97,4 @@ npm run build
 
 1. Fork the repository
 2. Create your feature branch
-3. Submit a pull request
+3. Submit a pull request

src/index.ts

@@ -32,7 +32,7 @@ const RETRIEVAL_TOOL: Tool = {
         description: 'The number of documents to retrieve.',
       },
     },
-    required: ['pipelineId', 'question', 'k'],
+    required: process.env.VECTORIZE_PIPELINE_ID ? ['question', 'k'] : ['pipelineId', 'question', 'k'],
   },
 };
 
@@ -56,7 +56,7 @@ const DEEP_RESEARCH_TOOL: Tool = {
         description: 'Whether to perform a web search.',
       },
     },
-    required: ['pipelineId', 'query', 'webSearch'],
+    required: process.env.VECTORIZE_PIPELINE_ID ? ['query', 'webSearch'] : ['pipelineId', 'query', 'webSearch'],
   },
 };
 
@@ -97,6 +97,7 @@ const server = new Server(
 // Get optional API URL
 const VECTORIZE_ORG_ID = process.env.VECTORIZE_ORG_ID;
 const VECTORIZE_TOKEN = process.env.VECTORIZE_TOKEN;
+const VECTORIZE_PIPELINE_ID = process.env.VECTORIZE_PIPELINE_ID;
 // Check if API key is required (only for cloud service)
 if (!VECTORIZE_ORG_ID || !VECTORIZE_TOKEN) {
   console.error(
@@ -250,7 +251,7 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       case 'retrieve': {
         return await performRetrieval(
           VECTORIZE_ORG_ID,
-          args.pipelineId + '',
+          args.pipelineId ? (args.pipelineId + '') : (VECTORIZE_PIPELINE_ID || ''),
           args.question + '',
           Number(args.k)
         );
@@ -265,7 +266,7 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       case 'deep-research': {
         return await performDeepResearch(
           VECTORIZE_ORG_ID,
-          args.pipelineId + '',
+          args.pipelineId ? (args.pipelineId + '') : (VECTORIZE_PIPELINE_ID || ''),
           args.query + '',
           Boolean(args.webSearch)
         );
@@ -305,6 +306,13 @@ async function runServer() {
     data: `Configuration: Organization ID: ${VECTORIZE_ORG_ID || 'default'}`,
   });
 
+  if (VECTORIZE_PIPELINE_ID) {
+    server.sendLoggingMessage({
+      level: 'info',
+      data: `Configuration: Using fixed Pipeline ID: ${VECTORIZE_PIPELINE_ID}`,
+    });
+  }
+
   console.error('Vectorize MCP Server running');
 }