diff --git a/.env.example b/.env.example index dd664fe..f311255 100644 --- a/.env.example +++ b/.env.example @@ -1,4 +1,4 @@ -VERSION=0.1.3 +VERSION=0.1.6 # POSTGRES POSTGRES_USER=docker diff --git a/mcp/package.json b/mcp/package.json index 7d33954..065ff12 100644 --- a/mcp/package.json +++ b/mcp/package.json @@ -1,6 +1,6 @@ { "name": "@redplanethq/core-mcp", - "version": "0.1.0", + "version": "0.1.3", "type": "module", "main": "dist/index.js", "bin": { diff --git a/mcp/src/index.ts b/mcp/src/index.ts index e0b57ac..8a057d8 100644 --- a/mcp/src/index.ts +++ b/mcp/src/index.ts @@ -31,8 +31,8 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { inputSchema: zodToJsonSchema(SearchKGSchema), }, { - name: "ingest_memory", - description: "Ingest data into the memory graph pipeline", + name: "add_memory", + description: "Add data into the memory graph pipeline", inputSchema: zodToJsonSchema(IngestKGSchema), }, ], @@ -51,7 +51,7 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => { const result = await searchKnowledgeGraph(args); return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] }; } - case "ingest_memory": { + case "add_memory": { const args = IngestKGSchema.parse(request.params.arguments); const result = await ingestKnowledgeGraph(args); return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] }; diff --git a/mcp/src/operations/graph.ts b/mcp/src/operations/graph.ts index c98568d..ef658e2 100644 --- a/mcp/src/operations/graph.ts +++ b/mcp/src/operations/graph.ts @@ -12,10 +12,10 @@ export async function searchKnowledgeGraph(args: SearchKG) { export async function ingestKnowledgeGraph(args: IngestKG) { const response = await axiosInstance.post(`/ingest`, { - data: args.data, + episodeBody: args.message, source: process.env.SOURCE, referenceTime: args.referenceTime, - sessionId: args.sessionId ? args.sessionId : undefined, + sessionId: undefined, }); return response.data; } diff --git a/mcp/src/types/graph.ts b/mcp/src/types/graph.ts index 4bc3f7c..9bb06d5 100644 --- a/mcp/src/types/graph.ts +++ b/mcp/src/types/graph.ts @@ -8,9 +8,8 @@ export const SearchKGSchema = z.object({ }); export const IngestKGSchema = z.object({ - data: z.string().describe("The data to ingest in text format"), + message: z.string().describe("The data to ingest in text format"), referenceTime: z.string().describe("The reference time in ISO format"), - sessionId: z.string().optional().describe("The session id of the conversation"), }); export type SearchKG = z.infer; diff --git a/package.json b/package.json index ef217fb..8432232 100644 --- a/package.json +++ b/package.json @@ -1,7 +1,7 @@ { "name": "core", "private": true, - "version": "0.1.5", + "version": "0.1.6", "workspaces": [ "apps/*", "packages/*"