Initial commit

src/mcp/brainyMCPAdapter.ts

@@ -0,0 +1,203 @@
+/**
+ * BrainyMCPAdapter
+ * 
+ * This class provides an adapter for accessing Brainy data through the Model Control Protocol (MCP).
+ * It wraps a BrainyData instance and exposes methods for getting vectors, searching similar items,
+ * and getting relationships.
+ */
+
+import { v4 as uuidv4 } from 'uuid'
+import { BrainyDataInterface } from '../types/brainyDataInterface.js'
+import { 
+  MCPRequest, 
+  MCPResponse, 
+  MCPDataAccessRequest,
+  MCPRequestType,
+  MCP_VERSION
+} from '../types/mcpTypes.js'
+
+export class BrainyMCPAdapter {
+  private brainyData: BrainyDataInterface
+
+  /**
+   * Creates a new BrainyMCPAdapter
+   * @param brainyData The BrainyData instance to wrap
+   */
+  constructor(brainyData: BrainyDataInterface) {
+    this.brainyData = brainyData
+  }
+
+  /**
+   * Handles an MCP data access request
+   * @param request The MCP request
+   * @returns An MCP response
+   */
+  async handleRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
+    try {
+      switch (request.operation) {
+        case 'get':
+          return await this.handleGetRequest(request)
+        case 'search':
+          return await this.handleSearchRequest(request)
+        case 'add':
+          return await this.handleAddRequest(request)
+        case 'getRelationships':
+          return await this.handleGetRelationshipsRequest(request)
+        default:
+          return this.createErrorResponse(
+            request.requestId,
+            'UNSUPPORTED_OPERATION',
+            `Operation ${request.operation} is not supported`
+          )
+      }
+    } catch (error) {
+      return this.createErrorResponse(
+        request.requestId,
+        'INTERNAL_ERROR',
+        error instanceof Error ? error.message : String(error)
+      )
+    }
+  }
+
+  /**
+   * Handles a get request
+   * @param request The MCP request
+   * @returns An MCP response
+   */
+  private async handleGetRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
+    const { id } = request.parameters
+
+    if (!id) {
+      return this.createErrorResponse(
+        request.requestId,
+        'MISSING_PARAMETER',
+        'Parameter "id" is required'
+      )
+    }
+
+    const noun = await this.brainyData.get(id)
+    
+    if (!noun) {
+      return this.createErrorResponse(
+        request.requestId,
+        'NOT_FOUND',
+        `No noun found with id ${id}`
+      )
+    }
+
+    return this.createSuccessResponse(request.requestId, noun)
+  }
+
+  /**
+   * Handles a search request
+   * @param request The MCP request
+   * @returns An MCP response
+   */
+  private async handleSearchRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
+    const { query, k = 10 } = request.parameters
+
+    if (!query) {
+      return this.createErrorResponse(
+        request.requestId,
+        'MISSING_PARAMETER',
+        'Parameter "query" is required'
+      )
+    }
+
+    const results = await this.brainyData.searchText(query, k)
+    return this.createSuccessResponse(request.requestId, results)
+  }
+
+  /**
+   * Handles an add request
+   * @param request The MCP request
+   * @returns An MCP response
+   */
+  private async handleAddRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
+    const { text, metadata } = request.parameters
+
+    if (!text) {
+      return this.createErrorResponse(
+        request.requestId,
+        'MISSING_PARAMETER',
+        'Parameter "text" is required'
+      )
+    }
+
+    const id = await this.brainyData.add(text, metadata)
+    return this.createSuccessResponse(request.requestId, { id })
+  }
+
+  /**
+   * Handles a getRelationships request
+   * @param request The MCP request
+   * @returns An MCP response
+   */
+  private async handleGetRelationshipsRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
+    const { id } = request.parameters
+
+    if (!id) {
+      return this.createErrorResponse(
+        request.requestId,
+        'MISSING_PARAMETER',
+        'Parameter "id" is required'
+      )
+    }
+
+    // This is a simplified implementation - in a real implementation, we would
+    // need to check if these methods exist on the BrainyDataInterface
+    const outgoing = await (this.brainyData as any).getVerbsBySource?.(id) || []
+    const incoming = await (this.brainyData as any).getVerbsByTarget?.(id) || []
+
+    return this.createSuccessResponse(request.requestId, { outgoing, incoming })
+  }
+
+  /**
+   * Creates a success response
+   * @param requestId The request ID
+   * @param data The response data
+   * @returns An MCP response
+   */
+  private createSuccessResponse(requestId: string, data: any): MCPResponse {
+    return {
+      success: true,
+      requestId,
+      version: MCP_VERSION,
+      data
+    }
+  }
+
+  /**
+   * Creates an error response
+   * @param requestId The request ID
+   * @param code The error code
+   * @param message The error message
+   * @param details Optional error details
+   * @returns An MCP response
+   */
+  private createErrorResponse(
+    requestId: string,
+    code: string,
+    message: string,
+    details?: any
+  ): MCPResponse {
+    return {
+      success: false,
+      requestId,
+      version: MCP_VERSION,
+      error: {
+        code,
+        message,
+        details
+      }
+    }
+  }
+
+  /**
+   * Creates a new request ID
+   * @returns A new UUID
+   */
+  generateRequestId(): string {
+    return uuidv4()
+  }
+}