@probelabs/probe 0.6.0-rc125 → 0.6.0-rc127

package/build/agent/storage/InMemoryStorageAdapter.js

@@ -0,0 +1,49 @@
+import { StorageAdapter } from './StorageAdapter.js';
+
+/**
+ * Default in-memory storage adapter
+ * This is the default behavior - stores history in a Map in memory
+ */
+export class InMemoryStorageAdapter extends StorageAdapter {
+  constructor() {
+    super();
+    this.sessions = new Map(); // sessionId -> {messages: [], metadata: {}}
+  }
+
+  async loadHistory(sessionId) {
+    const session = this.sessions.get(sessionId);
+    return session ? session.messages : [];
+  }
+
+  async saveMessage(sessionId, message) {
+    if (!this.sessions.has(sessionId)) {
+      this.sessions.set(sessionId, {
+        messages: [],
+        metadata: {
+          createdAt: new Date().toISOString(),
+          lastActivity: new Date().toISOString()
+        }
+      });
+    }
+
+    const session = this.sessions.get(sessionId);
+    session.messages.push(message);
+    session.metadata.lastActivity = new Date().toISOString();
+  }
+
+  async clearHistory(sessionId) {
+    this.sessions.delete(sessionId);
+  }
+
+  async getSessionMetadata(sessionId) {
+    const session = this.sessions.get(sessionId);
+    return session ? session.metadata : null;
+  }
+
+  async updateSessionActivity(sessionId) {
+    const session = this.sessions.get(sessionId);
+    if (session) {
+      session.metadata.lastActivity = new Date().toISOString();
+    }
+  }
+}

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,