iotexproject
diff --git a/‎.dockerignore‎
Lines changed: 1 addition & 0 deletions b/‎.dockerignore‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎.gitignore‎
Lines changed: 1 addition & 0 deletions b/‎.gitignore‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎README.md‎
Lines changed: 82 additions & 20 deletions b/‎README.md‎
Lines changed: 82 additions & 20 deletions
diff --git a/‎mcpServer.ts‎
Lines changed: 141 additions & 0 deletions b/‎mcpServer.ts‎
Lines changed: 141 additions & 0 deletions
diff --git a/‎package.json‎
Lines changed: 3 additions & 1 deletion b/‎package.json‎
Lines changed: 3 additions & 1 deletion
@@ -5,5 +5,6 @@ npm-debug.log
 .env.*
 
 configs/.env.*
+configs/*.json
 
 .cursor
@@ -25,6 +25,7 @@ yarn-error.log*
 configs/*.env
 !configs/template.env
 configs/.env.*
+configs/*.json
 
 # Temporary files
 *~
 
@@ -152,13 +152,33 @@ graph TD
    bun run start
    ```
 
-6. Test API query:
+6. Run in MCP (Model Context Protocol) mode:
+
+   ```bash
+   bun run start:mcp
+   ```
+   
+   This starts Quicksilver in MCP compatibility mode on port 3000 by default. To connect an MCP-compatible client, add the following to your client configuration:
+
+   ```json
+   {
+     "mcpServers": {
+       "askSentai": {
+         "url": "http://yourServerUrl/sse"
+       }
+     }
+   }
+   ```
+
+   Note that the standard API endpoints (`/ask`, `/stream`, etc.) are not available in MCP mode.
+
+7. Test API query:
 
    ```bash
     curl http://localhost:8000/ask -X POST -H "Content-Type: application/json" -d '{"q": "What is the weather in San Francisco?"}'
    ```
 
-7. Access raw tool data:
+8. Access raw tool data:
 
    ```bash
    # Get raw weather data for San Francisco
@@ -171,7 +191,6 @@ graph TD
    curl "http://localhost:8000/raw?tool=depin-metrics&isLatest=true"
    ```
 
-
 ---
 
 ## Configuration
@@ -217,7 +236,7 @@ Quicksilver supports running multiple instances with different tool configuratio
 #### Configuration Structure
 
 ```bash
-configs/ # Your instance-specific configuration (gitignored) 
+configs/ # Your instance-specific configuration (gitignored)
     ├── .env.weather # Instance with only weather-related tools
     ├── .env.news    # Instance with news and analytics tools
     └── .env.full    # Instance with all tools enabled
@@ -318,22 +337,22 @@ PORT=8003
 
 The following tools can be enabled in your configuration:
 
-| Tool Name          | Description           | Required Environment Variables         |
-| ------------------ | --------------------- | -------------------------------------- |
-| `news`             | News API integration  | `NEWSAPI_API_KEY`                      |
-| `weather-current`  | Current weather data  | `NUBILA_API_KEY`                       |
-| `weather-forecast` | Weather forecasts     | `NUBILA_API_KEY`                       |
-| `depin-metrics`    | DePIN network metrics | `DEPIN_API_KEY`                        |
-| `depin-projects`   | DePIN project data    | `DEPIN_API_KEY`                        |
-| `l1data`           | L1 blockchain data    | `API_V2_KEY`                          |
-| `dimo`             | Vehicle IoT data      | `CLIENT_ID`, `REDIRECT_URI`, `API_KEY` |
-| `nuclear`          | Nuclear outage data   | `EIA_API_KEY`                          |
-| `mapbox`           | Mapbox API integration | `MAPBOX_ACCESS_TOKEN`                  |
-| `messari`          | Messari API integration | `MESSARI_API_KEY`                     |
-| `thirdweb`         | Thirdweb API integration | `THIRDWEB_SECRET_KEY`, `THIRDWEB_SESSION_ID` |
-| `cmc`              | CoinMarketCap API integration | `CMC_API_KEY`                          |
-| `airquality`       | Air quality data      | `AIRVISUAL_API_KEY`                   |
-| `depinninja`       | Depin Ninja API integration | `DEPINNINJA_API_KEY`                  |
+| Tool Name          | Description                   | Required Environment Variables               |
+| ------------------ | ----------------------------- | -------------------------------------------- |
+| `news`             | News API integration          | `NEWSAPI_API_KEY`                            |
+| `weather-current`  | Current weather data          | `NUBILA_API_KEY`                             |
+| `weather-forecast` | Weather forecasts             | `NUBILA_API_KEY`                             |
+| `depin-metrics`    | DePIN network metrics         | `DEPIN_API_KEY`                              |
+| `depin-projects`   | DePIN project data            | `DEPIN_API_KEY`                              |
+| `l1data`           | L1 blockchain data            | `API_V2_KEY`                                 |
+| `dimo`             | Vehicle IoT data              | `CLIENT_ID`, `REDIRECT_URI`, `API_KEY`       |
+| `nuclear`          | Nuclear outage data           | `EIA_API_KEY`                                |
+| `mapbox`           | Mapbox API integration        | `MAPBOX_ACCESS_TOKEN`                        |
+| `messari`          | Messari API integration       | `MESSARI_API_KEY`                            |
+| `thirdweb`         | Thirdweb API integration      | `THIRDWEB_SECRET_KEY`, `THIRDWEB_SESSION_ID` |
+| `cmc`              | CoinMarketCap API integration | `CMC_API_KEY`                                |
+| `airquality`       | Air quality data              | `AIRVISUAL_API_KEY`                          |
+| `depinninja`       | Depin Ninja API integration   | `DEPINNINJA_API_KEY`                         |
 
 ### Best Practices
 
@@ -387,6 +406,7 @@ curl -H "API-KEY: your_api_key" http://localhost:8000/endpoint
 Simple health check endpoint.
 
 **Response**
+
 ```
 hello world, Sentient AI!
 ```
@@ -396,9 +416,11 @@ hello world, Sentient AI!
 Send a query to the Sentient AI system.
 
 **Parameters**
+
 - `q` or `content` (string, required) - The query text
 
 **Request Examples**
+
 ```bash
 # URL parameter
 curl "http://localhost:8000/ask?q=What%20is%20the%20weather%20in%20San%20Francisco?"
@@ -411,6 +433,7 @@ curl http://localhost:8000/ask \
 ```
 
 **Response**
+
 ```json
 {
   "data": "The current weather in San Francisco is 62°F with partly cloudy conditions..."
@@ -422,10 +445,12 @@ curl http://localhost:8000/ask \
 Stream a response from the Sentient AI system.
 
 **Parameters**
+
 - `text` (form data, required) - The query text
 - `recentMessages` (form data, optional) - Previous conversation context
 
 **Request Example**
+
 ```bash
 curl http://localhost:8000/stream \
   -X POST \
@@ -442,10 +467,12 @@ Stream of text data.
 Get raw data from a specific tool without LLM processing.
 
 **Parameters**
+
 - `tool` (string, required) - The tool name to query
 - Additional parameters specific to each tool
 
 **Request Examples**
+
 ```bash
 # Get current weather
 curl "http://localhost:8000/raw?tool=weather-current&lat=37.7749&lon=-122.4194"
@@ -458,12 +485,47 @@ curl "http://localhost:8000/raw?tool=depin-metrics&isLatest=true"
 ```
 
 **Response**
+
 ```json
 {
   "data": "Tool-specific response data"
 }
 ```
 
+#### GET `/sse` (MCP Mode)
+
+Establishes a Server-Sent Events (SSE) connection for Model Context Protocol interactions. Available when running in MCP mode.
+
+**Request Example**
+
+```bash
+# Connect to the MCP server via SSE
+curl -N http://localhost:3000/sse
+```
+
+**Response**
+Stream of SSE events with tool capabilities and message exchange.
+
+#### POST `/messages` (MCP Mode)
+
+Endpoint for sending messages to the MCP server after establishing an SSE connection.
+
+**Parameters**
+- `sessionId` (query parameter, required) - The session ID received from the SSE connection
+
+**Request Example**
+
+```bash
+# Send a message to the MCP server (session ID will be provided by the SSE connection)
+curl http://localhost:3000/messages?sessionId=YOUR_SESSION_ID \
+  -X POST \
+  -H "Content-Type: application/json" \
+  -d '{"jsonrpc":"2.0","method":"invoke","params":{"name":"tool_name","arguments":{}},"id":"request-id"}'
+```
+
+**Response**
+Confirmation of message receipt or error information.
+
 ---
 
 ## Integrations
 
@@ -0,0 +1,141 @@
+import { Hono } from 'hono';
+import { streamSSE } from 'hono/streaming';
+
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import { JSONRPCMessage, JSONRPCMessageSchema } from '@modelcontextprotocol/sdk/types.js';
+
+import { ToolRegistry } from './src/registry/registry.js';
+
+const app = new Hono();
+
+class HonoSSETransport implements Transport {
+  sessionId: string;
+  private sseStream: any; // Hono SSE stream type
+  private endpoint: string;
+
+  onclose?: () => void;
+  onerror?: (error: Error) => void;
+  onmessage?: (message: JSONRPCMessage) => void;
+
+  constructor(endpoint: string, stream: any) {
+    this.endpoint = endpoint;
+    this.sessionId = crypto.randomUUID();
+    this.sseStream = stream;
+  }
+
+  async start(): Promise<void> {
+    // Send initial ping
+    await this.sseStream.writeSSE({ data: 'ping', event: 'heartbeat' });
+
+    // Send endpoint information
+    await this.sseStream.writeSSE({
+      data: `${encodeURI(this.endpoint)}?sessionId=${this.sessionId}`,
+      event: 'endpoint',
+    });
+  }
+
+  async handlePostMessage(request: Request): Promise<Response> {
+    if (!this.sseStream) {
+      return new Response('SSE connection not established', { status: 500 });
+    }
+
+    try {
+      const contentType = request.headers.get('content-type');
+      if (contentType !== 'application/json') {
+        throw new Error(`Unsupported content-type: ${contentType}`);
+      }
+
+      const body = await request.json();
+      let parsedMessage: JSONRPCMessage;
+
+      try {
+        parsedMessage = JSONRPCMessageSchema.parse(body);
+      } catch (error) {
+        this.onerror?.(error as Error);
+        throw error;
+      }
+
+      this.onmessage?.(parsedMessage);
+      return new Response('Accepted', { status: 202 });
+    } catch (error) {
+      this.onerror?.(error as Error);
+      return new Response(String(error), { status: 400 });
+    }
+  }
+
+  async close(): Promise<void> {
+    this.onclose?.();
+  }
+
+  async send(message: JSONRPCMessage): Promise<void> {
+    if (!this.sseStream) {
+      throw new Error('Not connected');
+    }
+
+    await this.sseStream.writeSSE({
+      data: JSON.stringify(message),
+      event: 'message',
+    });
+  }
+
+  get id(): string {
+    return this.sessionId;
+  }
+}
+
+// Create server instance
+const server = new McpServer({
+  name: 'askSentientAI',
+  version: '1.0.0',
+  capabilities: {
+    tools: {},
+  },
+});
+
+ToolRegistry.registerMcpTools(server);
+
+// Store active transports by session ID
+const transports: Record<string, HonoSSETransport> = {};
+
+app.get('/sse', c => {
+  return streamSSE(c, async stream => {
+    const transport = new HonoSSETransport('/messages', stream);
+
+    // Store the transport with its session ID
+    transports[transport.id] = transport;
+
+    // Clean up when connection closes
+    stream.onAbort(() => {
+      delete transports[transport.id];
+      transport.close();
+    });
+
+    // Connect to MCP server
+    await server.connect(transport);
+    await transport.start();
+
+    // Keep the connection alive
+    while (true) {
+      await stream.sleep(30000);
+      await stream.writeSSE({ data: 'ping', event: 'heartbeat' });
+    }
+  });
+});
+
+app.post('/messages', async c => {
+  const sessionId = c.req.query('sessionId');
+  const transport = sessionId ? transports[sessionId] : null;
+
+  if (!transport) {
+    return c.text('No active SSE connection for the provided session ID', 400);
+  }
+
+  return transport.handlePostMessage(c.req.raw);
+});
+
+export default {
+  port: process.env.PORT ? parseInt(process.env.PORT) : 3000,
+  fetch: app.fetch,
+  idleTimeout: 120,
+};
@@ -6,6 +6,7 @@
   "type": "module",
   "scripts": {
     "start": "bun run server.ts --server-options=\"{\\\"idleTimeout\\\": 120000}\"",
+    "start:mcp": "bun run mcpServer.ts --server-options=\"{\\\"idleTimeout\\\": 120000}\"",
     "dev": "bun run --watch server.ts --server-options=\"{\\\"idleTimeout\\\": 120000}\"",
     "test": "vitest run",
     "test:watch": "vitest",
@@ -27,9 +28,10 @@
     "@langchain/core": "^0.3.40",
     "@langchain/openai": "^0.4.4",
     "@langchain/qdrant": "^0.1.1",
+    "@modelcontextprotocol/sdk": "^1.10.2",
     "@qdrant/js-client-rest": "^1.13.0",
     "@upstash/redis": "^1.34.4",
-    "ai": "^4.1.45",
+    "ai": "^4.3",
     "axios": "^1.7.9",
     "chalk": "^4.1.2",
     "dotenv": "^16.4.7",