@albatrossai/albatross-sdk 0.2.0-c → 0.2.2

package/dist/index.cjs

@@ -809,6 +809,15 @@ var FilterConditionSchema = zod.z.object({
         lt: zod.z.number().optional()
       })
     }),
+    zod.z.object({
+      type: zod.z.literal("DatetimeRange"),
+      value: zod.z.object({
+        gte: zod.z.string().optional(),
+        lte: zod.z.string().optional(),
+        gt: zod.z.string().optional(),
+        lt: zod.z.string().optional()
+      })
+    }),
     zod.z.object({
       type: zod.z.literal("GeoRadius"),
       value: zod.z.object({
@@ -1029,7 +1038,7 @@ var Client = class {
     if (request.options?.image) {
       this.validateImageSize(request.options.image);
     }
-    const url = `${this.baseUrl}/use-case/content-selection`;
+    const url = `${this.baseUrl}/prediction/content-selection`;
     return this.makeRequest({
       method: "POST",
       url,
@@ -1474,7 +1483,7 @@ var Client = class {
    *
    * @example
    * ```typescript
-   * const prediction = await client.prediction({
+   * const ranking = await client.ranking({
    *   useCase: { uuid: "ranking-use-case" },
    *   context: { user_id: "user-123" },
    *   actions: [
@@ -1484,9 +1493,9 @@ var Client = class {
    * });
    * ```
    */
-  prediction = async (payload) => {
+  ranking = async (payload) => {
     PredictionPayloadSchema.parse(payload);
-    const url = `${this.baseUrl}/use-case/prediction`;
+    const url = `${this.baseUrl}/prediction/ranker`;
     const data = this.predictionPreProcessing(payload);
     return this.makeRequest({
       method: "POST",
@@ -1501,7 +1510,7 @@ var Client = class {
    * submits outcome feedback for a previous prediction.
    *
    * @param payload - feedback payload
-   * @param payload.predictionUuid - id of the prediction to provide feedback for (from prediction() response)
+   * @param payload.predictionUuid - id of the prediction to provide feedback for (from ranking() response)
    * @param payload.value - feedback data (e.g., { reward: 1 }, { clicked: true }, { rating: 4.5 })
    * @param payload.modelUuid - optional specific model uuid to send feedback to
    *
@@ -1513,7 +1522,7 @@ var Client = class {
    *
    * @example
    * ```typescript
-   * const pred = await client.prediction({...});
+   * const pred = await client.ranking({...});
    * // later, after user interaction
    * await client.feedback({
    *   predictionUuid: pred.id,
@@ -1611,6 +1620,44 @@ var FilterBuilder = class {
     else this.must_not.push(condition);
     return this;
   }
+  /**
+   * add datetime range filter condition
+   *
+   * filters items where datetime field falls within specified range.
+   * values should be RFC 3339 formatted strings (e.g., "2025-12-15T00:00:00Z").
+   *
+   * @param field - datetime field name to filter on
+   * @param value - range specification with operators (RFC 3339 strings)
+   * @param value.gte - greater than or equal to
+   * @param value.lte - less than or equal to
+   * @param value.gt - greater than
+   * @param value.lt - less than
+   * @param type - filter type: "must", "should", or "must_not". default: "must"
+   *
+   * @returns this filter builder for method chaining
+   *
+   * @example
+   * ```typescript
+   * // events between two dates
+   * builder.datetimeRange("start_date", {
+   *   gte: "2025-12-15T00:00:00Z",
+   *   lte: "2025-12-31T23:59:59Z"
+   * });
+   *
+   * // events after a specific date
+   * builder.datetimeRange("created_at", { gt: "2025-01-01T00:00:00Z" });
+   * ```
+   */
+  datetimeRange(field, value, type = "must") {
+    const condition = {
+      field,
+      condition: { type: "DatetimeRange", value }
+    };
+    if (type === "must") this.must.push(condition);
+    else if (type === "should") this.should.push(condition);
+    else this.must_not.push(condition);
+    return this;
+  }
   /**
    * add geo-radius filter condition
    *