Minor protocol fixes

Author: pwilkin

grizabella/mcp/server.py

@@ -393,7 +393,7 @@ async def mcp_create_embedding_definition(embedding_def: EmbeddingDefinition) ->
 async def mcp_begin_bulk_addition():
     try:
         get_grizabella_client().begin_bulk_addition()
-        return "Bulk addition mode started."
+        return  # Return None for successful bulk addition start
     except Exception as e:
         logger.error(f"MCP: Error starting bulk addition: {e}")
         raise
@@ -406,7 +406,7 @@ async def mcp_begin_bulk_addition():
 async def mcp_finish_bulk_addition():
     try:
         get_grizabella_client().finish_bulk_addition()
-        return "Bulk addition mode finished and embeddings generated."
+        return  # Return None for successful bulk addition completion
     except Exception as e:
         logger.error(f"MCP: Error finishing bulk addition: {e}")
         raise
@@ -966,7 +966,12 @@ def shutdown_handler(signum, frame):
         print(f"Received signal {signum}, shutting down...", file=sys.stderr)
     except Exception:
         # sys.stderr might not be available during shutdown
-        print(f"Received signal {signum}, shutting down...")
+        # Using stderr even for the fallback to avoid stdout contamination
+        try:
+            print(f"Received signal {signum}, shutting down...", file=sys.stderr)
+        except Exception:
+            # If even stderr fails, just use logger
+            pass
 
     logger.info(f"Received signal {signum}, shutting down...")
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "grizabella"
-version = "0.6.0"
+version = "0.6.1"
 description = "A tri-layer memory framework for LLM solutions."
 authors = ["Grizabella Project Contributors <contributors@example.com>"]
 readme = "README.md"

typescript/package.json

@@ -1,6 +1,6 @@
 {
   "name": "grizabella-typescript-api",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "type": "module",
   "description": "TypeScript API client for Grizabella multi-database knowledge graph operations through MCP communication",
   "main": "dist/index.js",

typescript/src/client/GrizabellaClient.ts

@@ -142,7 +142,7 @@ export class GrizabellaClient {
    * ```
    */
   constructor(config: GrizabellaClientConfig) {
-    process.stdout.write('=== GrizabellaClient constructor called with config: ' + JSON.stringify(config, null, 2) + '\n');
+    process.stderr.write('=== GrizabellaClient constructor called with config: ' + JSON.stringify(config, null, 2) + '\n');
 
     // Store and resolve database configuration
     this._dbNameOrPath = GrizabellaClient.resolveDatabasePath(config.dbNameOrPath ?? 'default');

typescript/src/client/MCPClient.ts

@@ -153,9 +153,9 @@ export class MCPClient {
     this.maxReconnectAttempts = this.config.maxReconnectAttempts;
     this.reconnectDelay = this.config.reconnectDelay;
 
-    process.stdout.write('MCPClient final config debug=' + this.config.debug + '\n');
+    process.stderr.write('MCPClient final config debug=' + this.config.debug + '\n');
     if (this.config.debug) {
-      process.stdout.write('MCPClient created with debug enabled, config:' + JSON.stringify(this.config, null, 2) + '\n');
+      process.stderr.write('MCPClient created with debug enabled, config:' + JSON.stringify(this.config, null, 2) + '\n');
     }
   }
 
@@ -923,7 +923,7 @@ export class MCPClient {
    */
    async executeComplexQuery(params: ExecuteComplexQueryParams): Promise<QueryResult> {
       if (this.config.debug) {
-        process.stdout.write('MCPClient.executeComplexQuery called with:' + JSON.stringify(params, null, 2) + '\n');
+        process.stderr.write('MCPClient.executeComplexQuery called with:' + JSON.stringify(params, null, 2) + '\n');
       }
 
       const rawResult = await this.callTool<unknown>('execute_complex_query', {

typescript/src/client/errors.ts

@@ -817,7 +817,7 @@ export interface ErrorLogger {
  */
 export class ConsoleErrorLogger implements ErrorLogger {
   log(error: GrizabellaError): void {
-    console.log(error.getDetailedMessage(), error.toJSON());
+    console.error(error.getDetailedMessage(), error.toJSON());
   }
 
   warn(error: GrizabellaError): void {

typescript/src/index.ts

@@ -176,11 +176,11 @@ export { GrizabellaClient, GrizabellaClientConfig, ClientConnectionState } from
 // Main initialization function (to be implemented)
 export function initialize(): void {
   // TODO: Initialize the Grizabella client
-  console.log('Grizabella TypeScript API initialized');
+  console.error('Grizabella TypeScript API initialized');
 }
 
 // Main cleanup function (to be implemented)
 export function cleanup(): void {
   // TODO: Clean up resources
-  console.log('Grizabella TypeScript API cleanup');
+  console.error('Grizabella TypeScript API cleanup');
 }