@aeriondyseti/vector-memory-mcp 0.9.0-dev.5 → 0.9.0-dev.8
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/package.json +72 -0
- package/dist/scripts/test-runner.d.ts +9 -0
- package/dist/scripts/test-runner.d.ts.map +1 -0
- package/dist/scripts/test-runner.js +61 -0
- package/dist/scripts/test-runner.js.map +1 -0
- package/dist/scripts/warmup.d.ts +8 -0
- package/dist/scripts/warmup.d.ts.map +1 -0
- package/dist/scripts/warmup.js +61 -0
- package/dist/scripts/warmup.js.map +1 -0
- package/dist/src/config/index.d.ts +24 -0
- package/dist/src/config/index.d.ts.map +1 -0
- package/dist/src/config/index.js +48 -0
- package/dist/src/config/index.js.map +1 -0
- package/dist/src/db/connection.d.ts +3 -0
- package/dist/src/db/connection.d.ts.map +1 -0
- package/dist/src/db/connection.js +10 -0
- package/dist/src/db/connection.js.map +1 -0
- package/dist/src/db/memory.repository.d.ts +38 -0
- package/dist/src/db/memory.repository.d.ts.map +1 -0
- package/dist/src/db/memory.repository.js +176 -0
- package/dist/src/db/memory.repository.js.map +1 -0
- package/dist/src/db/schema.d.ts +4 -0
- package/dist/src/db/schema.d.ts.map +1 -0
- package/dist/src/db/schema.js +15 -0
- package/dist/src/db/schema.js.map +1 -0
- package/dist/src/http/mcp-transport.d.ts +19 -0
- package/dist/src/http/mcp-transport.d.ts.map +1 -0
- package/dist/src/http/mcp-transport.js +191 -0
- package/dist/src/http/mcp-transport.js.map +1 -0
- package/dist/src/http/server.d.ts +13 -0
- package/dist/src/http/server.d.ts.map +1 -0
- package/dist/src/http/server.js +210 -0
- package/dist/src/http/server.js.map +1 -0
- package/dist/src/index.d.ts +3 -0
- package/dist/src/index.d.ts.map +1 -0
- package/dist/src/index.js +59 -0
- package/dist/src/index.js.map +1 -0
- package/dist/src/mcp/handlers.d.ts +12 -0
- package/dist/src/mcp/handlers.d.ts.map +1 -0
- package/dist/src/mcp/handlers.js +198 -0
- package/dist/src/mcp/handlers.js.map +1 -0
- package/dist/src/mcp/server.d.ts +5 -0
- package/dist/src/mcp/server.d.ts.map +1 -0
- package/dist/src/mcp/server.js +22 -0
- package/dist/src/mcp/server.js.map +1 -0
- package/dist/src/mcp/tools.d.ts +10 -0
- package/dist/src/mcp/tools.d.ts.map +1 -0
- package/dist/src/mcp/tools.js +272 -0
- package/dist/src/mcp/tools.js.map +1 -0
- package/dist/src/services/embeddings.service.d.ts +12 -0
- package/dist/src/services/embeddings.service.d.ts.map +1 -0
- package/dist/src/services/embeddings.service.js +37 -0
- package/dist/src/services/embeddings.service.js.map +1 -0
- package/dist/src/services/memory.service.d.ts +33 -0
- package/dist/src/services/memory.service.d.ts.map +1 -0
- package/dist/src/services/memory.service.js +200 -0
- package/dist/src/services/memory.service.js.map +1 -0
- package/dist/src/types/memory.d.ts +28 -0
- package/dist/src/types/memory.d.ts.map +1 -0
- package/dist/src/types/memory.js +18 -0
- package/dist/src/types/memory.js.map +1 -0
- package/package.json +11 -8
- package/src/config/index.ts +5 -2
- package/src/http/server.ts +85 -12
- package/scripts/publish.ts +0 -61
|
@@ -0,0 +1,191 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* MCP HTTP Transport Handler
|
|
3
|
+
*
|
|
4
|
+
* Provides StreamableHTTP transport for MCP over HTTP.
|
|
5
|
+
* and other HTTP-based MCP clients to connect to the memory server.
|
|
6
|
+
*
|
|
7
|
+
* This implementation handles the MCP protocol directly using Hono's streaming
|
|
8
|
+
* capabilities, since StreamableHTTPServerTransport expects Node.js req/res objects.
|
|
9
|
+
*/
|
|
10
|
+
import { Hono } from "hono";
|
|
11
|
+
import { streamSSE } from "hono/streaming";
|
|
12
|
+
import { randomUUID } from "node:crypto";
|
|
13
|
+
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
|
|
14
|
+
import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
|
|
15
|
+
import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
|
|
16
|
+
import { tools } from "../mcp/tools.js";
|
|
17
|
+
import { handleToolCall } from "../mcp/handlers.js";
|
|
18
|
+
/**
|
|
19
|
+
* Creates MCP routes for a Hono app.
|
|
20
|
+
*
|
|
21
|
+
* Uses InMemoryTransport internally and bridges to HTTP/SSE manually,
|
|
22
|
+
* since StreamableHTTPServerTransport requires Node.js req/res objects.
|
|
23
|
+
*/
|
|
24
|
+
export function createMcpRoutes(memoryService) {
|
|
25
|
+
const app = new Hono();
|
|
26
|
+
// Store active sessions by session ID
|
|
27
|
+
const sessions = new Map();
|
|
28
|
+
/**
|
|
29
|
+
* Creates a new MCP server instance configured with memory tools.
|
|
30
|
+
*/
|
|
31
|
+
async function createSession() {
|
|
32
|
+
const server = new Server({ name: "vector-memory-mcp", version: "0.6.0" }, { capabilities: { tools: {} } });
|
|
33
|
+
server.setRequestHandler(ListToolsRequestSchema, async () => {
|
|
34
|
+
return { tools };
|
|
35
|
+
});
|
|
36
|
+
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
37
|
+
const { name, arguments: args } = request.params;
|
|
38
|
+
return handleToolCall(name, args, memoryService);
|
|
39
|
+
});
|
|
40
|
+
// Create linked in-memory transports
|
|
41
|
+
const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
|
|
42
|
+
// Connect server to its transport
|
|
43
|
+
await server.connect(serverTransport);
|
|
44
|
+
const session = {
|
|
45
|
+
server,
|
|
46
|
+
serverTransport,
|
|
47
|
+
clientTransport,
|
|
48
|
+
pendingResponses: new Map(),
|
|
49
|
+
sseClients: new Set(),
|
|
50
|
+
};
|
|
51
|
+
// Handle messages from server (responses and notifications)
|
|
52
|
+
clientTransport.onmessage = (message) => {
|
|
53
|
+
// Check if this is a response to a pending request
|
|
54
|
+
if ("id" in message && message.id !== undefined) {
|
|
55
|
+
const resolver = session.pendingResponses.get(message.id);
|
|
56
|
+
if (resolver) {
|
|
57
|
+
resolver(message);
|
|
58
|
+
session.pendingResponses.delete(message.id);
|
|
59
|
+
return;
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
// Otherwise, broadcast to SSE clients (notifications)
|
|
63
|
+
for (const sendToClient of session.sseClients) {
|
|
64
|
+
sendToClient(message);
|
|
65
|
+
}
|
|
66
|
+
};
|
|
67
|
+
return session;
|
|
68
|
+
}
|
|
69
|
+
/**
|
|
70
|
+
* Handle POST requests - session initialization and message handling
|
|
71
|
+
*/
|
|
72
|
+
app.post("/mcp", async (c) => {
|
|
73
|
+
const sessionId = c.req.header("mcp-session-id");
|
|
74
|
+
const body = await c.req.json();
|
|
75
|
+
let session;
|
|
76
|
+
let newSessionId;
|
|
77
|
+
if (sessionId && sessions.has(sessionId)) {
|
|
78
|
+
// Reuse existing session
|
|
79
|
+
session = sessions.get(sessionId);
|
|
80
|
+
}
|
|
81
|
+
else if (isInitializeRequest(body)) {
|
|
82
|
+
// New session initialization
|
|
83
|
+
newSessionId = randomUUID();
|
|
84
|
+
session = await createSession();
|
|
85
|
+
sessions.set(newSessionId, session);
|
|
86
|
+
console.error(`[vector-memory-mcp] MCP session initialized: ${newSessionId}`);
|
|
87
|
+
}
|
|
88
|
+
else {
|
|
89
|
+
// Invalid request - no session ID and not an initialize request
|
|
90
|
+
return c.json({
|
|
91
|
+
jsonrpc: "2.0",
|
|
92
|
+
error: {
|
|
93
|
+
code: -32000,
|
|
94
|
+
message: "Invalid session. Send initialize request without session ID to start.",
|
|
95
|
+
},
|
|
96
|
+
id: body.id ?? null,
|
|
97
|
+
}, 400);
|
|
98
|
+
}
|
|
99
|
+
// Send message to server and wait for response
|
|
100
|
+
const response = await sendAndWaitForResponse(session, body);
|
|
101
|
+
// Include session ID header for new sessions
|
|
102
|
+
if (newSessionId) {
|
|
103
|
+
c.header("mcp-session-id", newSessionId);
|
|
104
|
+
}
|
|
105
|
+
return c.json(response);
|
|
106
|
+
});
|
|
107
|
+
/**
|
|
108
|
+
* Handle GET requests - SSE stream for server-to-client notifications
|
|
109
|
+
*/
|
|
110
|
+
app.get("/mcp", async (c) => {
|
|
111
|
+
const sessionId = c.req.header("mcp-session-id");
|
|
112
|
+
if (!sessionId || !sessions.has(sessionId)) {
|
|
113
|
+
return c.json({
|
|
114
|
+
jsonrpc: "2.0",
|
|
115
|
+
error: { code: -32000, message: "Invalid or missing session ID" },
|
|
116
|
+
id: null,
|
|
117
|
+
}, 400);
|
|
118
|
+
}
|
|
119
|
+
const session = sessions.get(sessionId);
|
|
120
|
+
return streamSSE(c, async (stream) => {
|
|
121
|
+
// Register this SSE client
|
|
122
|
+
const sendMessage = (message) => {
|
|
123
|
+
stream.writeSSE({
|
|
124
|
+
data: JSON.stringify(message),
|
|
125
|
+
event: "message",
|
|
126
|
+
});
|
|
127
|
+
};
|
|
128
|
+
session.sseClients.add(sendMessage);
|
|
129
|
+
// Keep connection open
|
|
130
|
+
try {
|
|
131
|
+
// Send a ping every 30 seconds to keep connection alive
|
|
132
|
+
while (true) {
|
|
133
|
+
await stream.sleep(30000);
|
|
134
|
+
await stream.writeSSE({ event: "ping", data: "" });
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
finally {
|
|
138
|
+
session.sseClients.delete(sendMessage);
|
|
139
|
+
}
|
|
140
|
+
});
|
|
141
|
+
});
|
|
142
|
+
/**
|
|
143
|
+
* Handle DELETE requests - session termination
|
|
144
|
+
*/
|
|
145
|
+
app.delete("/mcp", async (c) => {
|
|
146
|
+
const sessionId = c.req.header("mcp-session-id");
|
|
147
|
+
if (!sessionId || !sessions.has(sessionId)) {
|
|
148
|
+
return c.json({
|
|
149
|
+
jsonrpc: "2.0",
|
|
150
|
+
error: { code: -32000, message: "Invalid or missing session ID" },
|
|
151
|
+
id: null,
|
|
152
|
+
}, 400);
|
|
153
|
+
}
|
|
154
|
+
const session = sessions.get(sessionId);
|
|
155
|
+
// Close transports
|
|
156
|
+
await session.clientTransport.close();
|
|
157
|
+
await session.serverTransport.close();
|
|
158
|
+
await session.server.close();
|
|
159
|
+
sessions.delete(sessionId);
|
|
160
|
+
console.error(`[vector-memory-mcp] MCP session closed: ${sessionId}`);
|
|
161
|
+
return c.json({ success: true });
|
|
162
|
+
});
|
|
163
|
+
return app;
|
|
164
|
+
}
|
|
165
|
+
/**
|
|
166
|
+
* Send a message to the server and wait for its response.
|
|
167
|
+
*/
|
|
168
|
+
async function sendAndWaitForResponse(session, message) {
|
|
169
|
+
return new Promise((resolve) => {
|
|
170
|
+
// Register response handler for requests (messages with id)
|
|
171
|
+
if ("id" in message && message.id !== undefined) {
|
|
172
|
+
session.pendingResponses.set(message.id, resolve);
|
|
173
|
+
}
|
|
174
|
+
// Send message to server
|
|
175
|
+
session.clientTransport.send(message);
|
|
176
|
+
// For notifications (no id), resolve immediately with empty response
|
|
177
|
+
if (!("id" in message) || message.id === undefined) {
|
|
178
|
+
resolve({ jsonrpc: "2.0" });
|
|
179
|
+
}
|
|
180
|
+
});
|
|
181
|
+
}
|
|
182
|
+
/**
|
|
183
|
+
* Check if a message is an initialize request.
|
|
184
|
+
*/
|
|
185
|
+
function isInitializeRequest(body) {
|
|
186
|
+
return (typeof body === "object" &&
|
|
187
|
+
body !== null &&
|
|
188
|
+
"method" in body &&
|
|
189
|
+
body.method === "initialize");
|
|
190
|
+
}
|
|
191
|
+
//# sourceMappingURL=mcp-transport.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-transport.js","sourceRoot":"","sources":["../../../src/http/mcp-transport.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GAIvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAE1E,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAWpD;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,aAA4B;IAC1D,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IAEvB,sCAAsC;IACtC,MAAM,QAAQ,GAAyB,IAAI,GAAG,EAAE,CAAC;IAEjD;;OAEG;IACH,KAAK,UAAU,aAAa;QAC1B,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,OAAO,EAAE,EAC/C,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAChC,CAAC;QAEF,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YAC1D,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;YACjD,OAAO,cAAc,CAAC,IAAI,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,CAAC,eAAe,EAAE,eAAe,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAEhF,kCAAkC;QAClC,MAAM,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEtC,MAAM,OAAO,GAAY;YACvB,MAAM;YACN,eAAe;YACf,eAAe;YACf,gBAAgB,EAAE,IAAI,GAAG,EAAE;YAC3B,UAAU,EAAE,IAAI,GAAG,EAAE;SACtB,CAAC;QAEF,4DAA4D;QAC5D,eAAe,CAAC,SAAS,GAAG,CAAC,OAAuB,EAAE,EAAE;YACtD,mDAAmD;YACnD,IAAI,IAAI,IAAI,OAAO,IAAI,OAAO,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;gBAChD,MAAM,QAAQ,GAAG,OAAO,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC1D,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,OAAO,CAAC,CAAC;oBAClB,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC5C,OAAO;gBACT,CAAC;YACH,CAAC;YAED,sDAAsD;YACtD,KAAK,MAAM,YAAY,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBAC9C,YAAY,CAAC,OAAO,CAAC,CAAC;YACxB,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC3B,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACjD,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QAEhC,IAAI,OAA4B,CAAC;QACjC,IAAI,YAAgC,CAAC;QAErC,IAAI,SAAS,IAAI,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACzC,yBAAyB;YACzB,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;QACrC,CAAC;aAAM,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,6BAA6B;YAC7B,YAAY,GAAG,UAAU,EAAE,CAAC;YAC5B,OAAO,GAAG,MAAM,aAAa,EAAE,CAAC;YAChC,QAAQ,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACpC,OAAO,CAAC,KAAK,CAAC,gDAAgD,YAAY,EAAE,CAAC,CAAC;QAChF,CAAC;aAAM,CAAC;YACN,gEAAgE;YAChE,OAAO,CAAC,CAAC,IAAI,CACX;gBACE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE;oBACL,IAAI,EAAE,CAAC,KAAK;oBACZ,OAAO,EAAE,uEAAuE;iBACjF;gBACD,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,IAAI;aACpB,EACD,GAAG,CACJ,CAAC;QACJ,CAAC;QAED,+CAA+C;QAC/C,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAE7D,6CAA6C;QAC7C,IAAI,YAAY,EAAE,CAAC;YACjB,CAAC,CAAC,MAAM,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC1B,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAEjD,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3C,OAAO,CAAC,CAAC,IAAI,CACX;gBACE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,+BAA+B,EAAE;gBACjE,EAAE,EAAE,IAAI;aACT,EACD,GAAG,CACJ,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;QAEzC,OAAO,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACnC,2BAA2B;YAC3B,MAAM,WAAW,GAAG,CAAC,OAAuB,EAAE,EAAE;gBAC9C,MAAM,CAAC,QAAQ,CAAC;oBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;oBAC7B,KAAK,EAAE,SAAS;iBACjB,CAAC,CAAC;YACL,CAAC,CAAC;YAEF,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAEpC,uBAAuB;YACvB,IAAI,CAAC;gBACH,wDAAwD;gBACxD,OAAO,IAAI,EAAE,CAAC;oBACZ,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC1B,MAAM,MAAM,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YACzC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH;;OAEG;IACH,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC7B,MAAM,SAAS,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAEjD,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3C,OAAO,CAAC,CAAC,IAAI,CACX;gBACE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,+BAA+B,EAAE;gBACjE,EAAE,EAAE,IAAI;aACT,EACD,GAAG,CACJ,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;QAEzC,mBAAmB;QACnB,MAAM,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QACtC,MAAM,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QACtC,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAE7B,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3B,OAAO,CAAC,KAAK,CAAC,2CAA2C,SAAS,EAAE,CAAC,CAAC;QAEtE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,sBAAsB,CACnC,OAAgB,EAChB,OAA6C;IAE7C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,4DAA4D;QAC5D,IAAI,IAAI,IAAI,OAAO,IAAI,OAAO,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YAChD,OAAO,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;QAED,yBAAyB;QACzB,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEtC,qEAAqE;QACrE,IAAI,CAAC,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,OAAO,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YACnD,OAAO,CAAC,EAAE,OAAO,EAAE,KAAK,EAAoB,CAAC,CAAC;QAChD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,IAAa;IACxC,OAAO,CACL,OAAO,IAAI,KAAK,QAAQ;QACxB,IAAI,KAAK,IAAI;QACb,QAAQ,IAAI,IAAI;QACf,IAA2B,CAAC,MAAM,KAAK,YAAY,CACrD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import { Hono } from "hono";
|
|
2
|
+
import type { MemoryService } from "../services/memory.service.js";
|
|
3
|
+
import type { Config } from "../config/index.js";
|
|
4
|
+
export interface HttpServerOptions {
|
|
5
|
+
memoryService: MemoryService;
|
|
6
|
+
config: Config;
|
|
7
|
+
}
|
|
8
|
+
export declare function createHttpApp(memoryService: MemoryService, config: Config): Hono;
|
|
9
|
+
export declare function startHttpServer(memoryService: MemoryService, config: Config): Promise<{
|
|
10
|
+
stop: () => void;
|
|
11
|
+
port: number;
|
|
12
|
+
}>;
|
|
13
|
+
//# sourceMappingURL=server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../../src/http/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAI5B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAqDjD,MAAM,WAAW,iBAAiB;IAChC,aAAa,EAAE,aAAa,CAAC;IAC7B,MAAM,EAAE,MAAM,CAAC;CAChB;AAKD,wBAAgB,aAAa,CAAC,aAAa,EAAE,aAAa,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAyJhF;AAED,wBAAsB,eAAe,CACnC,aAAa,EAAE,aAAa,EAC5B,MAAM,EAAE,MAAM,GACb,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,IAAI,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC,CAuC7C"}
|
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
import { Hono } from "hono";
|
|
2
|
+
import { cors } from "hono/cors";
|
|
3
|
+
import { serve as nodeServe } from "@hono/node-server";
|
|
4
|
+
import { createServer } from "net";
|
|
5
|
+
import { isDeleted } from "../types/memory.js";
|
|
6
|
+
import { createMcpRoutes } from "./mcp-transport.js";
|
|
7
|
+
// Detect runtime
|
|
8
|
+
const isBun = typeof globalThis.Bun !== "undefined";
|
|
9
|
+
/**
|
|
10
|
+
* Check if a port is available by attempting to bind to it
|
|
11
|
+
*/
|
|
12
|
+
async function isPortAvailable(port, host) {
|
|
13
|
+
return new Promise((resolve) => {
|
|
14
|
+
const server = createServer();
|
|
15
|
+
server.once("error", () => {
|
|
16
|
+
resolve(false);
|
|
17
|
+
});
|
|
18
|
+
server.once("listening", () => {
|
|
19
|
+
server.close(() => resolve(true));
|
|
20
|
+
});
|
|
21
|
+
server.listen(port, host);
|
|
22
|
+
});
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* Find an available port, starting with the preferred port.
|
|
26
|
+
* If preferred port is unavailable, picks a random available port.
|
|
27
|
+
*/
|
|
28
|
+
async function findAvailablePort(preferredPort, host) {
|
|
29
|
+
if (await isPortAvailable(preferredPort, host)) {
|
|
30
|
+
return preferredPort;
|
|
31
|
+
}
|
|
32
|
+
console.error(`[vector-memory-mcp] Port ${preferredPort} is in use, finding an available port...`);
|
|
33
|
+
// Let the OS pick a random available port
|
|
34
|
+
return new Promise((resolve, reject) => {
|
|
35
|
+
const server = createServer();
|
|
36
|
+
server.once("error", reject);
|
|
37
|
+
server.once("listening", () => {
|
|
38
|
+
const address = server.address();
|
|
39
|
+
const port = typeof address === "object" && address ? address.port : 0;
|
|
40
|
+
server.close(() => resolve(port));
|
|
41
|
+
});
|
|
42
|
+
server.listen(0, host);
|
|
43
|
+
});
|
|
44
|
+
}
|
|
45
|
+
// Track server start time for uptime calculation
|
|
46
|
+
const startedAt = Date.now();
|
|
47
|
+
export function createHttpApp(memoryService, config) {
|
|
48
|
+
const app = new Hono();
|
|
49
|
+
// Enable CORS for local development
|
|
50
|
+
app.use("/*", cors());
|
|
51
|
+
// Mount MCP routes for StreamableHTTP transport
|
|
52
|
+
const mcpApp = createMcpRoutes(memoryService);
|
|
53
|
+
app.route("/", mcpApp);
|
|
54
|
+
// Health check endpoint with config info
|
|
55
|
+
app.get("/health", (c) => {
|
|
56
|
+
return c.json({
|
|
57
|
+
status: "ok",
|
|
58
|
+
timestamp: new Date().toISOString(),
|
|
59
|
+
pid: process.pid,
|
|
60
|
+
uptime: Math.floor((Date.now() - startedAt) / 1000),
|
|
61
|
+
config: {
|
|
62
|
+
dbPath: config.dbPath,
|
|
63
|
+
embeddingModel: config.embeddingModel,
|
|
64
|
+
embeddingDimension: config.embeddingDimension,
|
|
65
|
+
},
|
|
66
|
+
});
|
|
67
|
+
});
|
|
68
|
+
// Search endpoint
|
|
69
|
+
app.post("/search", async (c) => {
|
|
70
|
+
try {
|
|
71
|
+
const body = await c.req.json();
|
|
72
|
+
const query = body.query;
|
|
73
|
+
const intent = body.intent ?? "fact_check";
|
|
74
|
+
const limit = body.limit ?? 10;
|
|
75
|
+
if (!query || typeof query !== "string") {
|
|
76
|
+
return c.json({ error: "Missing or invalid 'query' field" }, 400);
|
|
77
|
+
}
|
|
78
|
+
const memories = await memoryService.search(query, intent, limit);
|
|
79
|
+
return c.json({
|
|
80
|
+
memories: memories.map((m) => ({
|
|
81
|
+
id: m.id,
|
|
82
|
+
content: m.content,
|
|
83
|
+
metadata: m.metadata,
|
|
84
|
+
createdAt: m.createdAt.toISOString(),
|
|
85
|
+
})),
|
|
86
|
+
count: memories.length,
|
|
87
|
+
});
|
|
88
|
+
}
|
|
89
|
+
catch (error) {
|
|
90
|
+
const message = error instanceof Error ? error.message : "Unknown error";
|
|
91
|
+
return c.json({ error: message }, 500);
|
|
92
|
+
}
|
|
93
|
+
});
|
|
94
|
+
// Store endpoint
|
|
95
|
+
app.post("/store", async (c) => {
|
|
96
|
+
try {
|
|
97
|
+
const body = await c.req.json();
|
|
98
|
+
const { content, metadata, embeddingText } = body;
|
|
99
|
+
if (!content || typeof content !== "string") {
|
|
100
|
+
return c.json({ error: "Missing or invalid 'content' field" }, 400);
|
|
101
|
+
}
|
|
102
|
+
const memory = await memoryService.store(content, metadata ?? {}, embeddingText);
|
|
103
|
+
return c.json({
|
|
104
|
+
id: memory.id,
|
|
105
|
+
createdAt: memory.createdAt.toISOString(),
|
|
106
|
+
});
|
|
107
|
+
}
|
|
108
|
+
catch (error) {
|
|
109
|
+
const message = error instanceof Error ? error.message : "Unknown error";
|
|
110
|
+
return c.json({ error: message }, 500);
|
|
111
|
+
}
|
|
112
|
+
});
|
|
113
|
+
// Delete endpoint
|
|
114
|
+
app.delete("/memories/:id", async (c) => {
|
|
115
|
+
try {
|
|
116
|
+
const id = c.req.param("id");
|
|
117
|
+
const deleted = await memoryService.delete(id);
|
|
118
|
+
if (!deleted) {
|
|
119
|
+
return c.json({ error: "Memory not found" }, 404);
|
|
120
|
+
}
|
|
121
|
+
return c.json({ deleted: true });
|
|
122
|
+
}
|
|
123
|
+
catch (error) {
|
|
124
|
+
const message = error instanceof Error ? error.message : "Unknown error";
|
|
125
|
+
return c.json({ error: message }, 500);
|
|
126
|
+
}
|
|
127
|
+
});
|
|
128
|
+
// Get latest handoff
|
|
129
|
+
app.get("/handoff", async (c) => {
|
|
130
|
+
try {
|
|
131
|
+
const handoff = await memoryService.getLatestHandoff();
|
|
132
|
+
if (!handoff) {
|
|
133
|
+
return c.json({ error: "No handoff found" }, 404);
|
|
134
|
+
}
|
|
135
|
+
// Fetch referenced memories if any
|
|
136
|
+
const memoryIds = handoff.metadata.memory_ids ?? [];
|
|
137
|
+
const referencedMemories = [];
|
|
138
|
+
for (const id of memoryIds) {
|
|
139
|
+
const memory = await memoryService.get(id);
|
|
140
|
+
if (memory && !isDeleted(memory)) {
|
|
141
|
+
referencedMemories.push({ id: memory.id, content: memory.content });
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
return c.json({
|
|
145
|
+
content: handoff.content,
|
|
146
|
+
metadata: handoff.metadata,
|
|
147
|
+
referencedMemories,
|
|
148
|
+
updatedAt: handoff.updatedAt.toISOString(),
|
|
149
|
+
});
|
|
150
|
+
}
|
|
151
|
+
catch (error) {
|
|
152
|
+
const message = error instanceof Error ? error.message : "Unknown error";
|
|
153
|
+
return c.json({ error: message }, 500);
|
|
154
|
+
}
|
|
155
|
+
});
|
|
156
|
+
// Get single memory
|
|
157
|
+
app.get("/memories/:id", async (c) => {
|
|
158
|
+
try {
|
|
159
|
+
const id = c.req.param("id");
|
|
160
|
+
const memory = await memoryService.get(id);
|
|
161
|
+
if (!memory || isDeleted(memory)) {
|
|
162
|
+
return c.json({ error: "Memory not found" }, 404);
|
|
163
|
+
}
|
|
164
|
+
return c.json({
|
|
165
|
+
id: memory.id,
|
|
166
|
+
content: memory.content,
|
|
167
|
+
metadata: memory.metadata,
|
|
168
|
+
createdAt: memory.createdAt.toISOString(),
|
|
169
|
+
updatedAt: memory.updatedAt.toISOString(),
|
|
170
|
+
});
|
|
171
|
+
}
|
|
172
|
+
catch (error) {
|
|
173
|
+
const message = error instanceof Error ? error.message : "Unknown error";
|
|
174
|
+
return c.json({ error: message }, 500);
|
|
175
|
+
}
|
|
176
|
+
});
|
|
177
|
+
return app;
|
|
178
|
+
}
|
|
179
|
+
export async function startHttpServer(memoryService, config) {
|
|
180
|
+
const app = createHttpApp(memoryService, config);
|
|
181
|
+
// Find an available port (uses configured port if available, otherwise picks a random one)
|
|
182
|
+
const actualPort = await findAvailablePort(config.httpPort, config.httpHost);
|
|
183
|
+
if (isBun) {
|
|
184
|
+
// Use Bun's native server
|
|
185
|
+
const server = Bun.serve({
|
|
186
|
+
port: actualPort,
|
|
187
|
+
hostname: config.httpHost,
|
|
188
|
+
fetch: app.fetch,
|
|
189
|
+
});
|
|
190
|
+
console.error(`[vector-memory-mcp] HTTP server listening on http://${config.httpHost}:${actualPort}`);
|
|
191
|
+
return {
|
|
192
|
+
stop: () => server.stop(),
|
|
193
|
+
port: actualPort,
|
|
194
|
+
};
|
|
195
|
+
}
|
|
196
|
+
else {
|
|
197
|
+
// Use Node.js server via @hono/node-server
|
|
198
|
+
const server = nodeServe({
|
|
199
|
+
fetch: app.fetch,
|
|
200
|
+
port: actualPort,
|
|
201
|
+
hostname: config.httpHost,
|
|
202
|
+
});
|
|
203
|
+
console.error(`[vector-memory-mcp] HTTP server listening on http://${config.httpHost}:${actualPort}`);
|
|
204
|
+
return {
|
|
205
|
+
stop: () => server.close(),
|
|
206
|
+
port: actualPort,
|
|
207
|
+
};
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../src/http/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,KAAK,IAAI,SAAS,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,KAAK,CAAC;AAGnC,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD,iBAAiB;AACjB,MAAM,KAAK,GAAG,OAAO,UAAU,CAAC,GAAG,KAAK,WAAW,CAAC;AAEpD;;GAEG;AACH,KAAK,UAAU,eAAe,CAAC,IAAY,EAAE,IAAY;IACvD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE;YACxB,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE;YAC5B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,iBAAiB,CAC9B,aAAqB,EACrB,IAAY;IAEZ,IAAI,MAAM,eAAe,CAAC,aAAa,EAAE,IAAI,CAAC,EAAE,CAAC;QAC/C,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,OAAO,CAAC,KAAK,CACX,4BAA4B,aAAa,0CAA0C,CACpF,CAAC;IAEF,0CAA0C;IAC1C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7B,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE;YAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACzB,CAAC,CAAC,CAAC;AACL,CAAC;AAOD,iDAAiD;AACjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;AAE7B,MAAM,UAAU,aAAa,CAAC,aAA4B,EAAE,MAAc;IACxE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IAEvB,oCAAoC;IACpC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAEtB,gDAAgD;IAChD,MAAM,MAAM,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;IAC9C,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAEvB,yCAAyC;IACzC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE;QACvB,OAAO,CAAC,CAAC,IAAI,CAAC;YACZ,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;YACnD,MAAM,EAAE;gBACN,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,cAAc,EAAE,MAAM,CAAC,cAAc;gBACrC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;aAC9C;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,kBAAkB;IAClB,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC9B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YAChC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;YACzB,MAAM,MAAM,GAAI,IAAI,CAAC,MAAuB,IAAI,YAAY,CAAC;YAC7D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YAE/B,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACxC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,kCAAkC,EAAE,EAAE,GAAG,CAAC,CAAC;YACpE,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;YAElE,OAAO,CAAC,CAAC,IAAI,CAAC;gBACZ,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC7B,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,OAAO,EAAE,CAAC,CAAC,OAAO;oBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;oBACpB,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;iBACrC,CAAC,CAAC;gBACH,KAAK,EAAE,QAAQ,CAAC,MAAM;aACvB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,iBAAiB;IACjB,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC7B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YAChC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;YAElD,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC5C,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,oCAAoC,EAAE,EAAE,GAAG,CAAC,CAAC;YACtE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,KAAK,CACtC,OAAO,EACP,QAAQ,IAAI,EAAE,EACd,aAAa,CACd,CAAC;YAEF,OAAO,CAAC,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE;aAC1C,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,kBAAkB;IAClB,GAAG,CAAC,MAAM,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QACtC,IAAI,CAAC;YACH,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAE/C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,EAAE,GAAG,CAAC,CAAC;YACpD,CAAC;YAED,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,qBAAqB;IACrB,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QAC9B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,gBAAgB,EAAE,CAAC;YAEvD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,EAAE,GAAG,CAAC,CAAC;YACpD,CAAC;YAED,mCAAmC;YACnC,MAAM,SAAS,GAAI,OAAO,CAAC,QAAQ,CAAC,UAAmC,IAAI,EAAE,CAAC;YAC9E,MAAM,kBAAkB,GAA2C,EAAE,CAAC;YAEtE,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;gBAC3B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC3C,IAAI,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;oBACjC,kBAAkB,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAED,OAAO,CAAC,CAAC,IAAI,CAAC;gBACZ,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,kBAAkB;gBAClB,SAAS,EAAE,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE;aAC3C,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QACnC,IAAI,CAAC;YACH,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAE3C,IAAI,CAAC,MAAM,IAAI,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;gBACjC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,EAAE,GAAG,CAAC,CAAC;YACpD,CAAC;YAED,OAAO,CAAC,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE;gBACzC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE;aAC1C,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;QACzC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,aAA4B,EAC5B,MAAc;IAEd,MAAM,GAAG,GAAG,aAAa,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAEjD,2FAA2F;IAC3F,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;IAE7E,IAAI,KAAK,EAAE,CAAC;QACV,0BAA0B;QAC1B,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC;YACvB,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,KAAK,EAAE,GAAG,CAAC,KAAK;SACjB,CAAC,CAAC;QAEH,OAAO,CAAC,KAAK,CACX,uDAAuD,MAAM,CAAC,QAAQ,IAAI,UAAU,EAAE,CACvF,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE;YACzB,IAAI,EAAE,UAAU;SACjB,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,2CAA2C;QAC3C,MAAM,MAAM,GAAG,SAAS,CAAC;YACvB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC,CAAC;QAEH,OAAO,CAAC,KAAK,CACX,uDAAuD,MAAM,CAAC,QAAQ,IAAI,UAAU,EAAE,CACvF,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE;YAC1B,IAAI,EAAE,UAAU;SACjB,CAAC;IACJ,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { loadConfig, parseCliArgs } from "./config/index.js";
|
|
3
|
+
import { connectToDatabase } from "./db/connection.js";
|
|
4
|
+
import { MemoryRepository } from "./db/memory.repository.js";
|
|
5
|
+
import { EmbeddingsService } from "./services/embeddings.service.js";
|
|
6
|
+
import { MemoryService } from "./services/memory.service.js";
|
|
7
|
+
import { startServer } from "./mcp/server.js";
|
|
8
|
+
import { startHttpServer } from "./http/server.js";
|
|
9
|
+
async function main() {
|
|
10
|
+
const args = process.argv.slice(2);
|
|
11
|
+
// Check for warmup command
|
|
12
|
+
if (args[0] === "warmup") {
|
|
13
|
+
const { warmup } = await import("../scripts/warmup.js");
|
|
14
|
+
await warmup();
|
|
15
|
+
return;
|
|
16
|
+
}
|
|
17
|
+
// Parse CLI args and load config
|
|
18
|
+
const overrides = parseCliArgs(args);
|
|
19
|
+
const config = loadConfig(overrides);
|
|
20
|
+
// Initialize database
|
|
21
|
+
const db = await connectToDatabase(config.dbPath);
|
|
22
|
+
// Initialize layers
|
|
23
|
+
const repository = new MemoryRepository(db);
|
|
24
|
+
const embeddings = new EmbeddingsService(config.embeddingModel, config.embeddingDimension);
|
|
25
|
+
const memoryService = new MemoryService(repository, embeddings);
|
|
26
|
+
// Track cleanup functions
|
|
27
|
+
let httpStop = null;
|
|
28
|
+
// Graceful shutdown handler
|
|
29
|
+
const shutdown = () => {
|
|
30
|
+
console.error("[vector-memory-mcp] Shutting down...");
|
|
31
|
+
if (httpStop)
|
|
32
|
+
httpStop();
|
|
33
|
+
db.close();
|
|
34
|
+
process.exit(0);
|
|
35
|
+
};
|
|
36
|
+
// Handle signals and stdin close (parent process exit)
|
|
37
|
+
process.on("SIGTERM", shutdown);
|
|
38
|
+
process.on("SIGINT", shutdown);
|
|
39
|
+
process.stdin.on("close", shutdown);
|
|
40
|
+
process.stdin.on("end", shutdown);
|
|
41
|
+
// Start HTTP server if transport mode includes it
|
|
42
|
+
if (config.enableHttp) {
|
|
43
|
+
const http = await startHttpServer(memoryService, config);
|
|
44
|
+
httpStop = http.stop;
|
|
45
|
+
console.error(`[vector-memory-mcp] MCP available at http://${config.httpHost}:${config.httpPort}/mcp`);
|
|
46
|
+
}
|
|
47
|
+
// Start stdio transport unless in HTTP-only mode
|
|
48
|
+
if (config.transportMode !== "http") {
|
|
49
|
+
await startServer(memoryService);
|
|
50
|
+
}
|
|
51
|
+
else {
|
|
52
|
+
// In HTTP-only mode, keep the process running
|
|
53
|
+
console.error("[vector-memory-mcp] Running in HTTP-only mode (no stdio)");
|
|
54
|
+
// Keep process alive - the HTTP server runs indefinitely
|
|
55
|
+
await new Promise(() => { });
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
main().catch(console.error);
|
|
59
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAEnD,KAAK,UAAU,IAAI;IACjB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAEnC,2BAA2B;IAC3B,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;QACzB,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QACxD,MAAM,MAAM,EAAE,CAAC;QACf,OAAO;IACT,CAAC;IAED,iCAAiC;IACjC,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IAErC,sBAAsB;IACtB,MAAM,EAAE,GAAG,MAAM,iBAAiB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAElD,oBAAoB;IACpB,MAAM,UAAU,GAAG,IAAI,gBAAgB,CAAC,EAAE,CAAC,CAAC;IAC5C,MAAM,UAAU,GAAG,IAAI,iBAAiB,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,kBAAkB,CAAC,CAAC;IAC3F,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IAEhE,0BAA0B;IAC1B,IAAI,QAAQ,GAAwB,IAAI,CAAC;IAEzC,4BAA4B;IAC5B,MAAM,QAAQ,GAAG,GAAG,EAAE;QACpB,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACtD,IAAI,QAAQ;YAAE,QAAQ,EAAE,CAAC;QACzB,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC;IAEF,uDAAuD;IACvD,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/B,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACpC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAElC,kDAAkD;IAClD,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QACtB,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAC1D,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;QACrB,OAAO,CAAC,KAAK,CACX,+CAA+C,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,MAAM,CACxF,CAAC;IACJ,CAAC;IAED,iDAAiD;IACjD,IAAI,MAAM,CAAC,aAAa,KAAK,MAAM,EAAE,CAAC;QACpC,MAAM,WAAW,CAAC,aAAa,CAAC,CAAC;IACnC,CAAC;SAAM,CAAC;QACN,8CAA8C;QAC9C,OAAO,CAAC,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC1E,yDAAyD;QACzD,MAAM,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;AACH,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
|
|
2
|
+
import type { MemoryService } from "../services/memory.service.js";
|
|
3
|
+
export declare function handleStoreMemories(args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
4
|
+
export declare function handleDeleteMemories(args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
5
|
+
export declare function handleUpdateMemories(args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
6
|
+
export declare function handleSearchMemories(args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
7
|
+
export declare function handleGetMemories(args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
8
|
+
export declare function handleReportMemoryUsefulness(args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
9
|
+
export declare function handleStoreHandoff(args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
10
|
+
export declare function handleGetHandoff(_args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
11
|
+
export declare function handleToolCall(name: string, args: Record<string, unknown> | undefined, service: MemoryService): Promise<CallToolResult>;
|
|
12
|
+
//# sourceMappingURL=handlers.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"handlers.d.ts","sourceRoot":"","sources":["../../../src/mcp/handlers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAGnE,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CA4BzB;AAED,wBAAsB,oBAAoB,CACxC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CAmBzB;AAGD,wBAAsB,oBAAoB,CACxC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CAgCzB;AAED,wBAAsB,oBAAoB,CACxC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CA4BzB;AAED,wBAAsB,iBAAiB,CACrC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CAgCzB;AAED,wBAAsB,4BAA4B,CAChD,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CAqBzB;AAED,wBAAsB,kBAAkB,CACtC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CAgBzB;AAED,wBAAsB,gBAAgB,CACpC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC1C,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CA6BzB;AAED,wBAAsB,cAAc,CAClC,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,cAAc,CAAC,CAwBzB"}
|