@probelabs/probe 0.6.0-rc124 → 0.6.0-rc126

package/build/agent/storage/StorageAdapter.js

@@ -0,0 +1,51 @@
+/**
+ * Base class for storage adapters
+ * Implement this interface to provide custom storage backends for ProbeAgent history
+ */
+export class StorageAdapter {
+  /**
+   * Load conversation history for a session
+   * @param {string} sessionId - Session identifier
+   * @returns {Promise<Array<Object>>} Array of message objects with {role, content, ...}
+   */
+  async loadHistory(sessionId) {
+    throw new Error('StorageAdapter.loadHistory() must be implemented by subclass');
+  }
+
+  /**
+   * Save a message to storage
+   * @param {string} sessionId - Session identifier
+   * @param {Object} message - Message object { role, content, ... }
+   * @returns {Promise<void>}
+   */
+  async saveMessage(sessionId, message) {
+    throw new Error('StorageAdapter.saveMessage() must be implemented by subclass');
+  }
+
+  /**
+   * Clear history for a session
+   * @param {string} sessionId - Session identifier
+   * @returns {Promise<void>}
+   */
+  async clearHistory(sessionId) {
+    throw new Error('StorageAdapter.clearHistory() must be implemented by subclass');
+  }
+
+  /**
+   * Get session metadata (optional)
+   * @param {string} sessionId - Session identifier
+   * @returns {Promise<Object|null>} Session metadata or null
+   */
+  async getSessionMetadata(sessionId) {
+    return null;
+  }
+
+  /**
+   * Update session activity timestamp (optional)
+   * @param {string} sessionId - Session identifier
+   * @returns {Promise<void>}
+   */
+  async updateSessionActivity(sessionId) {
+    // Optional - implement if you want to track session activity
+  }
+}

package/build/agent/storage/index.js

@@ -0,0 +1,2 @@
+export { StorageAdapter } from './StorageAdapter.js';
+export { InMemoryStorageAdapter } from './InMemoryStorageAdapter.js';

package/build/index.js

@@ -36,6 +36,8 @@ import { bashTool } from './tools/bash.js';
 import { ProbeAgent } from './agent/ProbeAgent.js';
 import { SimpleTelemetry, SimpleAppTracer, initializeSimpleTelemetryFromOptions } from './agent/simpleTelemetry.js';
 import { listFilesToolInstance, searchFilesToolInstance } from './agent/probeTool.js';
+import { StorageAdapter, InMemoryStorageAdapter } from './agent/storage/index.js';
+import { HookManager, HOOK_TYPES } from './agent/hooks/index.js';
 
 export {
 	search,
@@ -50,6 +52,12 @@ export {
 	DEFAULT_SYSTEM_MESSAGE,
 	// Export AI Agent (NEW!)
 	ProbeAgent,
+	// Export storage adapters
+	StorageAdapter,
+	InMemoryStorageAdapter,
+	// Export hooks
+	HookManager,
+	HOOK_TYPES,
 	// Export telemetry classes
 	SimpleTelemetry,
 	SimpleAppTracer,