@mnexium/sdk 0.1.0 → 0.2.1

package/dist/index.mjs

@@ -61,7 +61,9 @@ var Chat = class {
       temperature: options.temperature ?? this.options.temperature,
       stream: options.stream,
       metadata: options.metadata ?? this.options.metadata,
-      regenerateKey: options.regenerateKey
+      regenerateKey: options.regenerateKey,
+      memoryPolicy: options.memoryPolicy ?? this.options.memoryPolicy,
+      records: options.records ?? this.options.records
     });
   }
 };
@@ -81,7 +83,8 @@ var EventStream = class {
   }
   async *[Symbol.asyncIterator]() {
     const response = await this.client._requestRaw("GET", "/events/memories", {
-      query: { subject_id: this.subjectId }
+      query: { subject_id: this.subjectId },
+      signal: this.abortController.signal
     });
     if (!response.body) {
       throw new Error("Response body is null \u2014 SSE not supported");
@@ -170,7 +173,8 @@ var SubjectMemoriesResource = class {
         text,
         source: options?.source,
         visibility: options?.visibility,
-        metadata: options?.metadata
+        metadata: options?.metadata,
+        no_supersede: options?.no_supersede
       }
     });
   }
@@ -566,6 +570,11 @@ var ChatCompletionsResource = class {
     } else if (options.googleKey) {
       headers["x-google-key"] = options.googleKey;
     }
+    if (options.memoryPolicy === false) {
+      headers["x-mnx-memory-policy"] = "false";
+    } else if (typeof options.memoryPolicy === "string" && options.memoryPolicy.trim()) {
+      headers["x-mnx-memory-policy"] = options.memoryPolicy;
+    }
     const body = {
       model: options.model,
       messages: options.messages,
@@ -584,9 +593,24 @@ var ChatCompletionsResource = class {
         state: options.state,
         system_prompt: options.systemPrompt,
         metadata: options.metadata,
-        regenerate_key: options.regenerateKey
+        regenerate_key: options.regenerateKey,
+        memory_policy: options.memoryPolicy,
+        records: options.records
       }
     };
+    if (options.stream) {
+      const response = await this.client._requestRaw("POST", "/chat/completions", {
+        body,
+        headers
+      });
+      return new StreamResponse(response, {
+        chatId: response.headers.get("X-Mnx-Chat-Id") || options.chatId || "",
+        subjectId: response.headers.get("X-Mnx-Subject-Id") || options.subjectId || "",
+        model: options.model,
+        provisionedKey: response.headers.get("X-Mnx-Key-Provisioned") || void 0,
+        claimUrl: response.headers.get("X-Mnx-Claim-Url") || void 0
+      });
+    }
     return this.client._request("POST", "/chat/completions", {
       body,
       headers
@@ -621,20 +645,18 @@ var MemoriesResource = class {
         offset: options?.offset
       }
     });
-    return response.memories;
+    return response.data || [];
   }
   async search(options) {
-    const response = await this.client._request("POST", "/memories/search", {
-      body: {
+    const response = await this.client._request("GET", "/memories/search", {
+      query: {
         subject_id: options.subjectId,
-        query: options.query,
+        q: options.query,
         limit: options.limit,
-        min_score: options.minScore,
-        include_deleted: options.includeDeleted,
-        include_superseded: options.includeSuperseded
+        min_score: options.minScore
       }
     });
-    return response.results;
+    return response.data || [];
   }
   async delete(id) {
     await this.client._request("DELETE", `/memories/${id}`);
@@ -699,7 +721,7 @@ var StateResource = class {
   async get(key, subjectId) {
     try {
       return await this.client._request("GET", `/state/${key}`, {
-        query: subjectId ? { subject_id: subjectId } : void 0
+        headers: subjectId ? { "x-subject-id": subjectId } : void 0
       });
     } catch (error) {
       if (error instanceof NotFoundError) {
@@ -709,17 +731,21 @@ var StateResource = class {
     }
   }
   async set(options) {
-    return this.client._request("POST", `/state/${options.key}`, {
+    const headers = {};
+    if (options.subjectId) {
+      headers["x-subject-id"] = options.subjectId;
+    }
+    return this.client._request("PUT", `/state/${options.key}`, {
+      headers,
       body: {
         value: options.value,
-        subject_id: options.subjectId,
         ttl_seconds: options.ttlSeconds
       }
     });
   }
   async delete(key, subjectId) {
     await this.client._request("DELETE", `/state/${key}`, {
-      query: subjectId ? { subject_id: subjectId } : void 0
+      headers: subjectId ? { "x-subject-id": subjectId } : void 0
     });
   }
 };
@@ -769,6 +795,136 @@ var PromptsResource = class {
   }
 };
 
+// src/resources/records.ts
+var RecordsResource = class {
+  constructor(client) {
+    this.client = client;
+  }
+  /**
+   * Define or update a record schema
+   *
+   * @example
+   * await mnx.records.defineSchema('deal', {
+   *   title: { type: 'string', required: true },
+   *   value: { type: 'number' },
+   *   stage: { type: 'string' },
+   *   account_id: { type: 'ref:account' },
+   * }, { displayName: 'Deal', description: 'Sales deals' });
+   */
+  async defineSchema(typeName, fields, opts) {
+    return this.client._request("POST", "/records/schemas", {
+      body: {
+        type_name: typeName,
+        fields,
+        display_name: opts?.displayName,
+        description: opts?.description
+      }
+    });
+  }
+  /**
+   * Get a schema by type name
+   */
+  async getSchema(typeName) {
+    try {
+      return await this.client._request("GET", `/records/schemas/${typeName}`);
+    } catch (error) {
+      if (error instanceof NotFoundError) return null;
+      throw error;
+    }
+  }
+  /**
+   * List all schemas for the project
+   */
+  async listSchemas() {
+    const response = await this.client._request("GET", "/records/schemas");
+    return response.schemas;
+  }
+  /**
+   * Insert a new record
+   *
+   * @example
+   * const deal = await mnx.records.insert('deal', {
+   *   title: 'Acme Renewal',
+   *   value: 500000,
+   *   stage: 'negotiation',
+   * });
+   */
+  async insert(typeName, data, opts) {
+    return this.client._request("POST", `/records/${typeName}`, {
+      body: {
+        data,
+        owner_id: opts?.ownerId,
+        visibility: opts?.visibility,
+        collaborators: opts?.collaborators
+      }
+    });
+  }
+  /**
+   * Get a record by ID
+   */
+  async get(typeName, recordId) {
+    try {
+      return await this.client._request("GET", `/records/${typeName}/${recordId}`);
+    } catch (error) {
+      if (error instanceof NotFoundError) return null;
+      throw error;
+    }
+  }
+  /**
+   * Update a record (partial merge)
+   *
+   * @example
+   * await mnx.records.update('deal', 'rec_abc', { stage: 'closed_won', value: 550000 });
+   */
+  async update(typeName, recordId, data) {
+    return this.client._request("PUT", `/records/${typeName}/${recordId}`, {
+      body: { data }
+    });
+  }
+  /**
+   * Soft-delete a record
+   */
+  async delete(typeName, recordId) {
+    await this.client._request("DELETE", `/records/${typeName}/${recordId}`);
+  }
+  /**
+   * Query records with JSONB filters
+   *
+   * @example
+   * const deals = await mnx.records.query('deal', {
+   *   where: { stage: 'closed_won' },
+   *   orderBy: '-value',
+   *   limit: 10,
+   * });
+   */
+  async query(typeName, opts) {
+    const response = await this.client._request("POST", `/records/${typeName}/query`, {
+      body: {
+        where: opts?.where,
+        order_by: opts?.orderBy,
+        limit: opts?.limit,
+        offset: opts?.offset
+      }
+    });
+    return response.records;
+  }
+  /**
+   * Semantic search across records
+   *
+   * @example
+   * const results = await mnx.records.search('deal', 'large enterprise renewal');
+   */
+  async search(typeName, query, opts) {
+    const response = await this.client._request("POST", `/records/${typeName}/search`, {
+      body: {
+        query,
+        limit: opts?.limit
+      }
+    });
+    return response.records;
+  }
+};
+
 // src/client.ts
 var DEFAULT_BASE_URL = "https://mnexium.com/api/v1";
 var DEFAULT_TIMEOUT = 3e4;
@@ -798,6 +954,7 @@ var Mnexium = class {
     this.profiles = new ProfilesResource(this);
     this.state = new StateResource(this);
     this.prompts = new PromptsResource(this);
+    this.records = new RecordsResource(this);
   }
   async process(input) {
     const options = typeof input === "string" ? { content: input } : input;
@@ -815,6 +972,8 @@ var Mnexium = class {
     const maxTokens = options.maxTokens || this.defaults.maxTokens;
     const temperature = options.temperature ?? this.defaults.temperature;
     const regenerateKey = options.regenerateKey ?? this.defaults.regenerateKey ?? false;
+    const memoryPolicy = options.memoryPolicy ?? this.defaults.memoryPolicy;
+    const records = options.records ?? this.defaults.records;
     const headers = {};
     const provider = detectProvider(model);
     if (provider === "anthropic" && this.anthropicConfig?.apiKey) {
@@ -828,6 +987,11 @@ var Mnexium = class {
     } else if (this.googleConfig?.apiKey) {
       headers["x-google-key"] = this.googleConfig.apiKey;
     }
+    if (memoryPolicy === false) {
+      headers["x-mnx-memory-policy"] = "false";
+    } else if (typeof memoryPolicy === "string" && memoryPolicy.trim()) {
+      headers["x-mnx-memory-policy"] = memoryPolicy;
+    }
     const body = {
       model,
       messages: [{ role: "user", content: options.content }],
@@ -845,7 +1009,9 @@ var Mnexium = class {
         summarize,
         system_prompt: systemPrompt,
         metadata,
-        regenerate_key: regenerateKey
+        regenerate_key: regenerateKey,
+        memory_policy: memoryPolicy,
+        records
       }
     };
     if (options.stream) {
@@ -878,6 +1044,7 @@ var Mnexium = class {
       } : void 0,
       provisionedKey: raw.mnx.provisioned_key,
       claimUrl: raw.mnx.claim_url,
+      records: raw.mnx.records,
       raw
     };
   }
@@ -968,13 +1135,20 @@ var Mnexium = class {
     }
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), this.timeout);
-    const response = await fetch(url.toString(), {
-      method,
-      headers,
-      body: options.body ? JSON.stringify(options.body) : void 0,
-      signal: controller.signal
-    });
-    clearTimeout(timeoutId);
+    if (options.signal) {
+      options.signal.addEventListener("abort", () => controller.abort(), { once: true });
+    }
+    let response;
+    try {
+      response = await fetch(url.toString(), {
+        method,
+        headers,
+        body: options.body ? JSON.stringify(options.body) : void 0,
+        signal: controller.signal
+      });
+    } finally {
+      clearTimeout(timeoutId);
+    }
     const provisionedKey = response.headers.get("x-mnx-key-provisioned");
     if (provisionedKey) {
       this._setProvisionedKey(provisionedKey);
@@ -1007,16 +1181,15 @@ var Mnexium = class {
     }
     let lastError;
     for (let attempt = 0; attempt <= this.maxRetries; attempt++) {
+      const controller = new AbortController();
+      const timeoutId = setTimeout(() => controller.abort(), this.timeout);
       try {
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), this.timeout);
         const response = await fetch(url.toString(), {
           method,
           headers,
           body: options.body ? JSON.stringify(options.body) : void 0,
           signal: controller.signal
         });
-        clearTimeout(timeoutId);
         const provisionedKey = response.headers.get("x-mnx-key-provisioned");
         if (provisionedKey) {
           this._setProvisionedKey(provisionedKey);
@@ -1025,7 +1198,14 @@ var Mnexium = class {
           const errorBody = await response.json().catch(() => ({}));
           throw this._handleErrorResponse(response.status, errorBody);
         }
-        return await response.json();
+        if (response.status === 204) {
+          return void 0;
+        }
+        const text = await response.text();
+        if (!text) {
+          return void 0;
+        }
+        return JSON.parse(text);
       } catch (error) {
         lastError = error;
         if (error instanceof APIError && error.status < 500 && !(error instanceof RateLimitError)) {
@@ -1035,6 +1215,8 @@ var Mnexium = class {
           throw error;
         }
         await this._sleep(Math.pow(2, attempt) * 1e3);
+      } finally {
+        clearTimeout(timeoutId);
       }
     }
     throw lastError || new MnexiumError("Request failed");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mnexium/sdk",
-  "version": "0.1.0",
+  "version": "0.2.1",
   "description": "Official Mnexium SDK for JavaScript/TypeScript - Add memory to your AI applications",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -19,7 +19,7 @@
     "build": "tsup src/index.ts --format cjs,esm --dts --clean",
     "dev": "tsup src/index.ts --format cjs,esm --dts --watch",
     "test": "vitest",
-    "lint": "eslint src/",
+    "lint": "eslint --ext .ts src/",
     "prepublishOnly": "npm run build",
     "examples:install": "npm run build && cd examples && npm pkg set dependencies.@mnexium/sdk=file:.. && npm install",
     "example:hello": "cd examples && npm run hello",
@@ -28,6 +28,7 @@
     "example:claims": "cd examples && npm run claims",
     "example:state": "cd examples && npm run state",
     "example:profile": "cd examples && npm run profile",
+    "example:records": "cd examples && npm run records",
     "example:prompts": "cd examples && npm run prompts",
     "example:streaming": "cd examples && npm run streaming",
     "example:events": "cd examples && npm run events",
@@ -56,6 +57,7 @@
   },
   "devDependencies": {
     "@types/node": "^20.0.0",
+    "@typescript-eslint/parser": "^6.0.0",
     "eslint": "^8.0.0",
     "tsup": "^8.0.0",
     "typescript": "^5.0.0",