agent-memory-client 0.2.0

package/dist/index.js

@@ -0,0 +1,646 @@
+'use strict';
+
+// src/errors.ts
+var MemoryClientError = class _MemoryClientError extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "MemoryClientError";
+    if (Error.captureStackTrace) {
+      Error.captureStackTrace(this, _MemoryClientError);
+    }
+  }
+};
+var MemoryValidationError = class extends MemoryClientError {
+  constructor(message) {
+    super(message);
+    this.name = "MemoryValidationError";
+  }
+};
+var MemoryNotFoundError = class extends MemoryClientError {
+  constructor(message) {
+    super(message);
+    this.name = "MemoryNotFoundError";
+  }
+};
+var MemoryServerError = class extends MemoryClientError {
+  statusCode;
+  constructor(message, statusCode) {
+    super(message);
+    this.name = "MemoryServerError";
+    this.statusCode = statusCode;
+  }
+};
+
+// src/filters.ts
+var SessionId = class {
+  eq;
+  in_;
+  not_eq;
+  not_in;
+  ne;
+  constructor(options = {}) {
+    this.eq = options.eq;
+    this.in_ = options.in_;
+    this.not_eq = options.not_eq;
+    this.not_in = options.not_in;
+    this.ne = options.ne;
+  }
+  toJSON() {
+    const result = {};
+    if (this.eq !== void 0) result.eq = this.eq;
+    if (this.in_ !== void 0) result.in_ = this.in_;
+    if (this.not_eq !== void 0) result.not_eq = this.not_eq;
+    if (this.not_in !== void 0) result.not_in = this.not_in;
+    if (this.ne !== void 0) result.ne = this.ne;
+    return result;
+  }
+};
+var Namespace = class {
+  eq;
+  in_;
+  not_eq;
+  not_in;
+  constructor(options = {}) {
+    this.eq = options.eq;
+    this.in_ = options.in_;
+    this.not_eq = options.not_eq;
+    this.not_in = options.not_in;
+  }
+  toJSON() {
+    const result = {};
+    if (this.eq !== void 0) result.eq = this.eq;
+    if (this.in_ !== void 0) result.in_ = this.in_;
+    if (this.not_eq !== void 0) result.not_eq = this.not_eq;
+    if (this.not_in !== void 0) result.not_in = this.not_in;
+    return result;
+  }
+};
+var UserId = class {
+  eq;
+  in_;
+  not_eq;
+  not_in;
+  constructor(options = {}) {
+    this.eq = options.eq;
+    this.in_ = options.in_;
+    this.not_eq = options.not_eq;
+    this.not_in = options.not_in;
+  }
+  toJSON() {
+    const result = {};
+    if (this.eq !== void 0) result.eq = this.eq;
+    if (this.in_ !== void 0) result.in_ = this.in_;
+    if (this.not_eq !== void 0) result.not_eq = this.not_eq;
+    if (this.not_in !== void 0) result.not_in = this.not_in;
+    return result;
+  }
+};
+var Topics = class {
+  any;
+  all;
+  none;
+  constructor(options = {}) {
+    this.any = options.any;
+    this.all = options.all;
+    this.none = options.none;
+  }
+  toJSON() {
+    const result = {};
+    if (this.any !== void 0) result.any = this.any;
+    if (this.all !== void 0) result.all = this.all;
+    if (this.none !== void 0) result.none = this.none;
+    return result;
+  }
+};
+var Entities = class {
+  any;
+  all;
+  none;
+  constructor(options = {}) {
+    this.any = options.any;
+    this.all = options.all;
+    this.none = options.none;
+  }
+  toJSON() {
+    const result = {};
+    if (this.any !== void 0) result.any = this.any;
+    if (this.all !== void 0) result.all = this.all;
+    if (this.none !== void 0) result.none = this.none;
+    return result;
+  }
+};
+var CreatedAt = class {
+  gte;
+  lte;
+  eq;
+  constructor(options = {}) {
+    this.gte = options.gte;
+    this.lte = options.lte;
+    this.eq = options.eq;
+  }
+  toJSON() {
+    const result = {};
+    if (this.gte !== void 0)
+      result.gte = this.gte instanceof Date ? this.gte.toISOString() : this.gte;
+    if (this.lte !== void 0)
+      result.lte = this.lte instanceof Date ? this.lte.toISOString() : this.lte;
+    if (this.eq !== void 0)
+      result.eq = this.eq instanceof Date ? this.eq.toISOString() : this.eq;
+    return result;
+  }
+};
+var LastAccessed = class {
+  gte;
+  lte;
+  eq;
+  constructor(options = {}) {
+    this.gte = options.gte;
+    this.lte = options.lte;
+    this.eq = options.eq;
+  }
+  toJSON() {
+    const result = {};
+    if (this.gte !== void 0)
+      result.gte = this.gte instanceof Date ? this.gte.toISOString() : this.gte;
+    if (this.lte !== void 0)
+      result.lte = this.lte instanceof Date ? this.lte.toISOString() : this.lte;
+    if (this.eq !== void 0)
+      result.eq = this.eq instanceof Date ? this.eq.toISOString() : this.eq;
+    return result;
+  }
+};
+var EventDate = class {
+  gte;
+  lte;
+  eq;
+  constructor(options = {}) {
+    this.gte = options.gte;
+    this.lte = options.lte;
+    this.eq = options.eq;
+  }
+  toJSON() {
+    const result = {};
+    if (this.gte !== void 0)
+      result.gte = this.gte instanceof Date ? this.gte.toISOString() : this.gte;
+    if (this.lte !== void 0)
+      result.lte = this.lte instanceof Date ? this.lte.toISOString() : this.lte;
+    if (this.eq !== void 0)
+      result.eq = this.eq instanceof Date ? this.eq.toISOString() : this.eq;
+    return result;
+  }
+};
+var MemoryType = class {
+  eq;
+  in_;
+  not_eq;
+  not_in;
+  constructor(options = {}) {
+    this.eq = options.eq;
+    this.in_ = options.in_;
+    this.not_eq = options.not_eq;
+    this.not_in = options.not_in;
+  }
+  toJSON() {
+    const result = {};
+    if (this.eq !== void 0) result.eq = this.eq;
+    if (this.in_ !== void 0) result.in_ = this.in_;
+    if (this.not_eq !== void 0) result.not_eq = this.not_eq;
+    if (this.not_in !== void 0) result.not_in = this.not_in;
+    return result;
+  }
+};
+
+// src/client.ts
+var VERSION = "0.1.0";
+var MemoryAPIClient = class {
+  config;
+  fetchFn;
+  constructor(config) {
+    this.config = {
+      timeout: 3e4,
+      ...config,
+      baseUrl: config.baseUrl.replace(/\/$/, "")
+      // Remove trailing slash
+    };
+    this.fetchFn = config.fetch ?? fetch;
+  }
+  /**
+   * Get default headers for requests
+   */
+  getHeaders() {
+    const headers = {
+      "Content-Type": "application/json",
+      "User-Agent": `agent-memory-client-js/${VERSION}`,
+      "X-Client-Version": VERSION
+    };
+    if (this.config.apiKey) {
+      headers["X-API-Key"] = this.config.apiKey;
+    }
+    if (this.config.bearerToken) {
+      headers["Authorization"] = `Bearer ${this.config.bearerToken}`;
+    }
+    return headers;
+  }
+  /**
+   * Make an HTTP request with error handling
+   */
+  async request(method, path, options = {}) {
+    const url = new URL(path, this.config.baseUrl);
+    if (options.params) {
+      for (const [key, value] of Object.entries(options.params)) {
+        if (value !== void 0) {
+          if (Array.isArray(value)) {
+            for (const item of value) {
+              url.searchParams.append(key, String(item));
+            }
+          } else {
+            url.searchParams.set(key, String(value));
+          }
+        }
+      }
+    }
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
+    try {
+      const response = await this.fetchFn(url.toString(), {
+        method,
+        headers: this.getHeaders(),
+        body: options.body ? JSON.stringify(options.body) : void 0,
+        signal: controller.signal
+      });
+      clearTimeout(timeoutId);
+      if (!response.ok) {
+        return this.handleHttpError(response);
+      }
+      return await response.json();
+    } catch (error) {
+      clearTimeout(timeoutId);
+      if (error instanceof MemoryClientError) {
+        throw error;
+      }
+      if (error instanceof Error && error.name === "AbortError") {
+        throw new MemoryClientError(`Request timeout after ${this.config.timeout}ms`);
+      }
+      throw new MemoryClientError(`Request failed: ${String(error)}`);
+    }
+  }
+  /**
+   * Handle HTTP error responses
+   */
+  async handleHttpError(response) {
+    let message;
+    try {
+      const body = await response.json();
+      message = body.detail || body.message || JSON.stringify(body);
+    } catch {
+      message = await response.text();
+    }
+    if (response.status === 404) {
+      throw new MemoryNotFoundError(message);
+    }
+    throw new MemoryServerError(message, response.status);
+  }
+  // ==================== Health ====================
+  /**
+   * Check server health
+   */
+  async healthCheck() {
+    return this.request("GET", "/v1/health");
+  }
+  // ==================== Working Memory ====================
+  /**
+   * List all session IDs
+   */
+  async listSessions(options = {}) {
+    return this.request("GET", "/v1/working-memory/", {
+      params: {
+        namespace: options.namespace ?? this.config.defaultNamespace,
+        limit: options.limit,
+        offset: options.offset
+      }
+    });
+  }
+  /**
+   * Get working memory for a session
+   */
+  async getWorkingMemory(sessionId, options = {}) {
+    try {
+      return await this.request(
+        "GET",
+        `/v1/working-memory/${encodeURIComponent(sessionId)}`,
+        {
+          params: {
+            namespace: options.namespace ?? this.config.defaultNamespace,
+            model_name: options.modelName ?? this.config.defaultModelName,
+            context_window_max: options.contextWindowMax ?? this.config.defaultContextWindowMax
+          }
+        }
+      );
+    } catch (error) {
+      if (error instanceof MemoryNotFoundError) {
+        return null;
+      }
+      throw error;
+    }
+  }
+  /**
+   * Get or create working memory for a session
+   */
+  async getOrCreateWorkingMemory(sessionId, options = {}) {
+    const existing = await this.getWorkingMemory(sessionId, options);
+    if (existing) {
+      return existing;
+    }
+    const workingMemory = {
+      session_id: sessionId,
+      namespace: options.namespace ?? this.config.defaultNamespace,
+      user_id: options.userId,
+      messages: [],
+      memories: [],
+      ttl_seconds: options.ttlSeconds,
+      long_term_memory_strategy: options.longTermMemoryStrategy
+    };
+    return this.putWorkingMemory(sessionId, workingMemory, options);
+  }
+  /**
+   * Create or update working memory for a session
+   */
+  async putWorkingMemory(sessionId, workingMemory, options = {}) {
+    const body = {
+      session_id: sessionId,
+      ...workingMemory,
+      namespace: workingMemory.namespace ?? options.namespace ?? this.config.defaultNamespace
+    };
+    return this.request(
+      "PUT",
+      `/v1/working-memory/${encodeURIComponent(sessionId)}`,
+      {
+        body,
+        params: {
+          model_name: options.modelName ?? this.config.defaultModelName,
+          context_window_max: options.contextWindowMax ?? this.config.defaultContextWindowMax,
+          background: options.background
+        }
+      }
+    );
+  }
+  /**
+   * Delete working memory for a session
+   */
+  async deleteWorkingMemory(sessionId, options = {}) {
+    return this.request(
+      "DELETE",
+      `/v1/working-memory/${encodeURIComponent(sessionId)}`,
+      {
+        params: {
+          namespace: options.namespace ?? this.config.defaultNamespace
+        }
+      }
+    );
+  }
+  // ==================== Long-term Memory ====================
+  /**
+   * Create long-term memory records
+   */
+  async createLongTermMemory(memories, options = {}) {
+    return this.request("POST", "/v1/long-term-memory/", {
+      body: { memories },
+      params: {
+        namespace: options.namespace ?? this.config.defaultNamespace
+      }
+    });
+  }
+  /**
+   * Search long-term memory
+   */
+  async searchLongTermMemory(options) {
+    const body = {
+      text: options.text,
+      limit: options.limit,
+      offset: options.offset,
+      distance_threshold: options.distanceThreshold
+    };
+    if (options.sessionId) {
+      body.session_id = options.sessionId instanceof SessionId ? options.sessionId.toJSON() : options.sessionId;
+    }
+    if (options.namespace) {
+      body.namespace = options.namespace instanceof Namespace ? options.namespace.toJSON() : options.namespace;
+    }
+    if (options.topics) {
+      body.topics = options.topics instanceof Topics ? options.topics.toJSON() : options.topics;
+    }
+    if (options.entities) {
+      body.entities = options.entities instanceof Entities ? options.entities.toJSON() : options.entities;
+    }
+    if (options.createdAt) {
+      body.created_at = options.createdAt instanceof CreatedAt ? options.createdAt.toJSON() : options.createdAt;
+    }
+    if (options.lastAccessed) {
+      body.last_accessed = options.lastAccessed instanceof LastAccessed ? options.lastAccessed.toJSON() : options.lastAccessed;
+    }
+    if (options.userId) {
+      body.user_id = options.userId instanceof UserId ? options.userId.toJSON() : options.userId;
+    }
+    if (options.memoryType) {
+      body.memory_type = options.memoryType instanceof MemoryType ? options.memoryType.toJSON() : options.memoryType;
+    }
+    if (options.eventDate) {
+      body.event_date = options.eventDate instanceof EventDate ? options.eventDate.toJSON() : options.eventDate;
+    }
+    if (options.recency) {
+      body.recency_boost = options.recency.recency_boost;
+      body.recency_semantic_weight = options.recency.semantic_weight;
+      body.recency_recency_weight = options.recency.recency_weight;
+      body.recency_freshness_weight = options.recency.freshness_weight;
+      body.recency_novelty_weight = options.recency.novelty_weight;
+      body.recency_half_life_last_access_days = options.recency.half_life_last_access_days;
+      body.recency_half_life_created_days = options.recency.half_life_created_days;
+      body.server_side_recency = options.recency.server_side_recency;
+    }
+    return this.request("POST", "/v1/long-term-memory/search", {
+      body
+    });
+  }
+  /**
+   * Get a long-term memory by ID
+   */
+  async getLongTermMemory(memoryId, options = {}) {
+    try {
+      return await this.request(
+        "GET",
+        `/v1/long-term-memory/${encodeURIComponent(memoryId)}`,
+        {
+          params: {
+            namespace: options.namespace ?? this.config.defaultNamespace
+          }
+        }
+      );
+    } catch (error) {
+      if (error instanceof MemoryNotFoundError) {
+        return null;
+      }
+      throw error;
+    }
+  }
+  /**
+   * Delete long-term memories by IDs
+   */
+  async deleteLongTermMemories(memoryIds, options = {}) {
+    return this.request("DELETE", "/v1/long-term-memory", {
+      params: {
+        memory_ids: memoryIds,
+        namespace: options.namespace ?? this.config.defaultNamespace
+      }
+    });
+  }
+  // ==================== Memory Prompt ====================
+  /**
+   * Get memory-enhanced prompt
+   */
+  async memoryPrompt(request) {
+    return this.request("POST", "/v1/memory/prompt", {
+      body: request
+    });
+  }
+  // ==================== Edit Long-term Memory ====================
+  /**
+   * Edit a long-term memory by ID
+   */
+  async editLongTermMemory(memoryId, updates) {
+    return this.request(
+      "PATCH",
+      `/v1/long-term-memory/${encodeURIComponent(memoryId)}`,
+      { body: updates }
+    );
+  }
+  // ==================== Forget ====================
+  /**
+   * Run a forgetting pass with the provided policy
+   */
+  async forgetLongTermMemories(options) {
+    return this.request("POST", "/v1/long-term-memory/forget", {
+      params: {
+        namespace: options.namespace,
+        user_id: options.userId,
+        session_id: options.sessionId,
+        limit: options.limit,
+        dry_run: options.dryRun
+      },
+      body: {
+        policy: options.policy,
+        pinned_ids: options.pinnedIds
+      }
+    });
+  }
+  // ==================== Summary Views ====================
+  /**
+   * List all summary views
+   */
+  async listSummaryViews() {
+    return this.request("GET", "/v1/summary-views");
+  }
+  /**
+   * Create a new summary view
+   */
+  async createSummaryView(request) {
+    return this.request("POST", "/v1/summary-views", {
+      body: request
+    });
+  }
+  /**
+   * Get a summary view by ID
+   */
+  async getSummaryView(viewId) {
+    try {
+      return await this.request(
+        "GET",
+        `/v1/summary-views/${encodeURIComponent(viewId)}`
+      );
+    } catch (error) {
+      if (error instanceof MemoryNotFoundError) {
+        return null;
+      }
+      throw error;
+    }
+  }
+  /**
+   * Delete a summary view
+   */
+  async deleteSummaryView(viewId) {
+    return this.request(
+      "DELETE",
+      `/v1/summary-views/${encodeURIComponent(viewId)}`
+    );
+  }
+  /**
+   * Run a summary view partition
+   */
+  async runSummaryViewPartition(viewId, group) {
+    return this.request(
+      "POST",
+      `/v1/summary-views/${encodeURIComponent(viewId)}/partitions/run`,
+      { body: { group } }
+    );
+  }
+  /**
+   * List summary view partitions
+   */
+  async listSummaryViewPartitions(viewId, options = {}) {
+    return this.request(
+      "GET",
+      `/v1/summary-views/${encodeURIComponent(viewId)}/partitions`,
+      {
+        params: {
+          namespace: options.namespace,
+          user_id: options.userId,
+          session_id: options.sessionId,
+          memory_type: options.memoryType
+        }
+      }
+    );
+  }
+  /**
+   * Run a full summary view (async task)
+   */
+  async runSummaryView(viewId, options = {}) {
+    return this.request(
+      "POST",
+      `/v1/summary-views/${encodeURIComponent(viewId)}/run`,
+      { body: options }
+    );
+  }
+  // ==================== Tasks ====================
+  /**
+   * Get a task by ID
+   */
+  async getTask(taskId) {
+    try {
+      return await this.request(
+        "GET",
+        `/v1/tasks/${encodeURIComponent(taskId)}`
+      );
+    } catch (error) {
+      if (error instanceof MemoryNotFoundError) {
+        return null;
+      }
+      throw error;
+    }
+  }
+};
+
+exports.CreatedAt = CreatedAt;
+exports.Entities = Entities;
+exports.EventDate = EventDate;
+exports.LastAccessed = LastAccessed;
+exports.MemoryAPIClient = MemoryAPIClient;
+exports.MemoryClientError = MemoryClientError;
+exports.MemoryNotFoundError = MemoryNotFoundError;
+exports.MemoryServerError = MemoryServerError;
+exports.MemoryType = MemoryType;
+exports.MemoryValidationError = MemoryValidationError;
+exports.Namespace = Namespace;
+exports.SessionId = SessionId;
+exports.Topics = Topics;
+exports.UserId = UserId;
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map