@xyd-js/mcp-server 0.0.0-build-be800ef-20251021211339 → 0.0.0-build-a1d398e-20251028174055

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xyd-js/mcp-server",
-  "version": "0.0.0-build-be800ef-20251021211339",
+  "version": "0.0.0-build-a1d398e-20251028174055",
   "type": "module",
   "description": "MCP server for xyd",
   "main": "dist/index.js",
@@ -21,7 +21,7 @@
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.18.1",
     "express": "^5.1.0",
-    "@xyd-js/mcp": "0.0.0-build-be800ef-20251021211339"
+    "@xyd-js/mcp": "0.0.0-build-a1d398e-20251028174055"
   },
   "devDependencies": {
     "@types/express": "^4.17.21",

package/src/index.ts

@@ -2,7 +2,15 @@ import express from "express";
 
 import { MCPServer } from "./mcp";
 
-const mcp = new MCPServer();
+// Support both uniform sources and llms.txt sources
+const uniformSources = process.argv[2];
+const llmsSources = process.argv[3];
+
+const mcp = new MCPServer({
+    uniformSources,
+    llmsSources,
+    openAIApiKey: process.env.OPENAI_API_KEY || "", // TODO: configurable
+});
 const app = express();
 
 app.use(express.json());

package/src/mcp.ts

@@ -6,7 +6,7 @@ import { isInitializeRequest } from "@modelcontextprotocol/sdk/types.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 
-import { mcpUniformResources, mcpUniformTools } from "@xyd-js/mcp";
+import { mcpUniformResources, mcpUniformTools, mcpLLMsResources, mcpLLMsTools } from "@xyd-js/mcp";
 
 // Extend Express Request type to include pendingToken
 declare global {
@@ -17,6 +17,12 @@ declare global {
   }
 }
 
+interface MCPServerOptions {
+  uniformSources: string | string[]
+  llmsSources: string | string[]
+  openAIApiKey: string // TODO: support other LLMS embeddings
+}
+
 export class MCPServer {
   private transports: { [sessionId: string]: StreamableHTTPServerTransport } =
     {};
@@ -25,8 +31,11 @@ export class MCPServer {
   private sessionTokens: { [sessionId: string]: string } = {};
 
   private uniformSources: string[] = []
+  private llmsSources: string[] = []
+
+  constructor(protected options: MCPServerOptions) {
+    const { uniformSources, llmsSources } = options;
 
-  constructor(uniformSources: string | string[] = process.argv[2]) {
     this.connect = this.connect.bind(this);
     this.handleConnectionRequest = this.handleConnectionRequest.bind(this);
     this.handleSessionRequest = this.handleSessionRequest.bind(this);
@@ -36,6 +45,13 @@ export class MCPServer {
     } else if (uniformSources && Array.isArray(uniformSources)) {
       this.uniformSources.push(...uniformSources);
     }
+
+    // Handle llms.txt sources
+    if (llmsSources && typeof llmsSources === "string") {
+      this.llmsSources.push(llmsSources);
+    } else if (llmsSources && Array.isArray(llmsSources)) {
+      this.llmsSources.push(...llmsSources);
+    }
   }
 
   public async handleConnectionRequest(
@@ -134,6 +150,12 @@ export class MCPServer {
     
     await mcpUniformTools(server, source || "", token || "");
 
+    // Add llms.txt resources and tools
+    if (this.llmsSources.length > 0) {
+      // await mcpLLMsResources(server, this.llmsSources);
+      await mcpLLMsTools(server, this.llmsSources, this.options.openAIApiKey);
+    }
+
     // Add simple token tool
     this.addSimpleTokenTool(server);
 
@@ -220,4 +242,5 @@ export class MCPServer {
     this.transports = {};
     this.sessionTokens = {};
   }
+
 }