@redis-iris/agent-memory 0.4.0-beta.3 → 1.0.0-beta.2

package/src/controlplane/funcs/get-service-readiness.ts

@@ -0,0 +1,148 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { AgentMemoryServiceCore } from "../core.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { pathToFunc } from "../lib/url.js";
+import { AgentMemoryServiceError } from "../models/errors/agent-memory-service-error.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/http-client-errors.js";
+import * as errors from "../models/errors/index.js";
+import { ResponseValidationError } from "../models/errors/response-validation-error.js";
+import { SDKValidationError } from "../models/errors/sdk-validation-error.js";
+import * as models from "../models/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Evaluates the same serving dependencies as the operational readiness probe. Returns 503 without dependency details when the service cannot serve requests.
+ */
+export function getServiceReadiness(
+  client: AgentMemoryServiceCore,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    models.GetServiceReadinessResponseContent,
+    | errors.PayloadTooLargeErrorResponseContent
+    | errors.ServiceUnavailableErrorResponseContent
+    | AgentMemoryServiceError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    options,
+  ));
+}
+
+async function $do(
+  client: AgentMemoryServiceCore,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      models.GetServiceReadinessResponseContent,
+      | errors.PayloadTooLargeErrorResponseContent
+      | errors.ServiceUnavailableErrorResponseContent
+      | AgentMemoryServiceError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const path = pathToFunc("/iris/v1/agent-memory/health/ready")();
+
+  const headers = new Headers(compactMap({
+    Accept: "application/json",
+  }));
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "GetServiceReadiness",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: null,
+
+    securitySource: null,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    method: "GET",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    models.GetServiceReadinessResponseContent,
+    | errors.PayloadTooLargeErrorResponseContent
+    | errors.ServiceUnavailableErrorResponseContent
+    | AgentMemoryServiceError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, models.GetServiceReadinessResponseContent$inboundSchema),
+    M.jsonErr(413, errors.PayloadTooLargeErrorResponseContent$inboundSchema),
+    M.jsonErr(503, errors.ServiceUnavailableErrorResponseContent$inboundSchema),
+    M.fail("4XX"),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/controlplane/funcs/get-store-metrics.ts

@@ -40,16 +40,19 @@ export function getStoreMetrics(
   from: Date,
   to: Date,
   period: models.MetricsPeriod,
+  projectId?: string | undefined,
   options?: RequestOptions,
 ): APIPromise<
   Result<
     models.GetStoreMetricsResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.FailedDependencyErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -67,6 +70,7 @@ export function getStoreMetrics(
     from,
     to,
     period,
+    projectId,
     options,
   ));
 }
@@ -77,17 +81,20 @@ async function $do(
   from: Date,
   to: Date,
   period: models.MetricsPeriod,
+  projectId?: string | undefined,
   options?: RequestOptions,
 ): Promise<
   [
     Result<
       models.GetStoreMetricsResponseContent,
-      | errors.MemoryBadRequestErrorResponseContent
+      | errors.BadRequestErrorResponseContent
       | errors.AuthenticationErrorResponseContent
       | errors.ForbiddenErrorResponseContent
       | errors.NotFoundErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
       | errors.FailedDependencyErrorResponseContent
       | errors.UnexpectedErrorResponseContent
+      | errors.NotImplementedErrorResponseContent
       | errors.ServiceUnavailableErrorResponseContent
       | AgentMemoryServiceError
       | ResponseValidationError
@@ -106,6 +113,7 @@ async function $do(
     from: from,
     to: to,
     period: period,
+    projectId: projectId,
   };
 
   const parsed = safeParse(
@@ -125,11 +133,14 @@ async function $do(
       charEncoding: "percent",
     }),
   };
-  const path = pathToFunc("/v2/stores/{storeId}/metrics")(pathParams);
+  const path = pathToFunc("/iris/v1/agent-memory/stores/{storeId}/metrics")(
+    pathParams,
+  );
 
   const query = encodeFormQuery({
     "from": payload.from,
     "period": payload.period,
+    "projectId": payload.projectId,
     "to": payload.to,
   });
 
@@ -190,12 +201,14 @@ async function $do(
 
   const [result] = await M.match<
     models.GetStoreMetricsResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.FailedDependencyErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -207,12 +220,14 @@ async function $do(
     | SDKValidationError
   >(
     M.json(200, models.GetStoreMetricsResponseContent$inboundSchema),
-    M.jsonErr(400, errors.MemoryBadRequestErrorResponseContent$inboundSchema),
+    M.jsonErr(400, errors.BadRequestErrorResponseContent$inboundSchema),
     M.jsonErr(401, errors.AuthenticationErrorResponseContent$inboundSchema),
     M.jsonErr(403, errors.ForbiddenErrorResponseContent$inboundSchema),
     M.jsonErr(404, errors.NotFoundErrorResponseContent$inboundSchema),
+    M.jsonErr(413, errors.PayloadTooLargeErrorResponseContent$inboundSchema),
     M.jsonErr(424, errors.FailedDependencyErrorResponseContent$inboundSchema),
     M.jsonErr(500, errors.UnexpectedErrorResponseContent$inboundSchema),
+    M.jsonErr(501, errors.NotImplementedErrorResponseContent$inboundSchema),
     M.jsonErr(503, errors.ServiceUnavailableErrorResponseContent$inboundSchema),
     M.fail("4XX"),
     M.fail("5XX"),

package/src/controlplane/funcs/get-store.ts

@@ -3,7 +3,7 @@
  */
 
 import { AgentMemoryServiceCore } from "../core.js";
-import { encodeSimple } from "../lib/encodings.js";
+import { encodeFormQuery, encodeSimple } from "../lib/encodings.js";
 import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
@@ -33,15 +33,18 @@ import { Result } from "../types/fp.js";
 export function getStore(
   client: AgentMemoryServiceCore,
   storeId: string,
+  projectId?: string | undefined,
   options?: RequestOptions,
 ): APIPromise<
   Result<
     models.GetStoreResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -56,6 +59,7 @@ export function getStore(
   return new APIPromise($do(
     client,
     storeId,
+    projectId,
     options,
   ));
 }
@@ -63,16 +67,19 @@ export function getStore(
 async function $do(
   client: AgentMemoryServiceCore,
   storeId: string,
+  projectId?: string | undefined,
   options?: RequestOptions,
 ): Promise<
   [
     Result<
       models.GetStoreResponseContent,
-      | errors.MemoryBadRequestErrorResponseContent
+      | errors.BadRequestErrorResponseContent
       | errors.AuthenticationErrorResponseContent
       | errors.ForbiddenErrorResponseContent
       | errors.NotFoundErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
       | errors.UnexpectedErrorResponseContent
+      | errors.NotImplementedErrorResponseContent
       | errors.ServiceUnavailableErrorResponseContent
       | AgentMemoryServiceError
       | ResponseValidationError
@@ -88,6 +95,7 @@ async function $do(
 > {
   const input: operations.GetStoreRequest = {
     storeId: storeId,
+    projectId: projectId,
   };
 
   const parsed = safeParse(
@@ -107,7 +115,11 @@ async function $do(
       charEncoding: "percent",
     }),
   };
-  const path = pathToFunc("/v2/stores/{storeId}")(pathParams);
+  const path = pathToFunc("/iris/v1/agent-memory/stores/{storeId}")(pathParams);
+
+  const query = encodeFormQuery({
+    "projectId": payload.projectId,
+  });
 
   const headers = new Headers(compactMap({
     Accept: "application/json",
@@ -138,6 +150,7 @@ async function $do(
     baseURL: options?.serverURL,
     path: path,
     headers: headers,
+    query: query,
     body: body,
     userAgent: client._options.userAgent,
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
@@ -165,11 +178,13 @@ async function $do(
 
   const [result] = await M.match<
     models.GetStoreResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -181,11 +196,13 @@ async function $do(
     | SDKValidationError
   >(
     M.json(200, models.GetStoreResponseContent$inboundSchema),
-    M.jsonErr(400, errors.MemoryBadRequestErrorResponseContent$inboundSchema),
+    M.jsonErr(400, errors.BadRequestErrorResponseContent$inboundSchema),
     M.jsonErr(401, errors.AuthenticationErrorResponseContent$inboundSchema),
     M.jsonErr(403, errors.ForbiddenErrorResponseContent$inboundSchema),
     M.jsonErr(404, errors.NotFoundErrorResponseContent$inboundSchema),
+    M.jsonErr(413, errors.PayloadTooLargeErrorResponseContent$inboundSchema),
     M.jsonErr(500, errors.UnexpectedErrorResponseContent$inboundSchema),
+    M.jsonErr(501, errors.NotImplementedErrorResponseContent$inboundSchema),
     M.jsonErr(503, errors.ServiceUnavailableErrorResponseContent$inboundSchema),
     M.fail("4XX"),
     M.fail("5XX"),

package/src/controlplane/funcs/list-built-in-detectors.ts

@@ -37,6 +37,7 @@ export function listBuiltInDetectors(
     models.ListBuiltInDetectorsResponseContent,
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
@@ -64,6 +65,7 @@ async function $do(
       models.ListBuiltInDetectorsResponseContent,
       | errors.AuthenticationErrorResponseContent
       | errors.ForbiddenErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
       | errors.UnexpectedErrorResponseContent
       | errors.ServiceUnavailableErrorResponseContent
       | AgentMemoryServiceError
@@ -78,7 +80,7 @@ async function $do(
     APICall,
   ]
 > {
-  const path = pathToFunc("/v2/detectors")();
+  const path = pathToFunc("/iris/v1/agent-memory/detectors")();
 
   const headers = new Headers(compactMap({
     Accept: "application/json",
@@ -137,6 +139,7 @@ async function $do(
     models.ListBuiltInDetectorsResponseContent,
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
@@ -151,6 +154,7 @@ async function $do(
     M.json(200, models.ListBuiltInDetectorsResponseContent$inboundSchema),
     M.jsonErr(401, errors.AuthenticationErrorResponseContent$inboundSchema),
     M.jsonErr(403, errors.ForbiddenErrorResponseContent$inboundSchema),
+    M.jsonErr(413, errors.PayloadTooLargeErrorResponseContent$inboundSchema),
     M.jsonErr(500, errors.UnexpectedErrorResponseContent$inboundSchema),
     M.jsonErr(503, errors.ServiceUnavailableErrorResponseContent$inboundSchema),
     M.fail("4XX"),

package/src/controlplane/funcs/list-databases.ts

@@ -38,9 +38,10 @@ export function listDatabases(
 ): APIPromise<
   Result<
     models.ListDatabasesResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -65,9 +66,10 @@ async function $do(
   [
     Result<
       models.ListDatabasesResponseContent,
-      | errors.MemoryBadRequestErrorResponseContent
+      | errors.BadRequestErrorResponseContent
       | errors.AuthenticationErrorResponseContent
       | errors.ForbiddenErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
       | errors.UnexpectedErrorResponseContent
       | AgentMemoryServiceError
       | ResponseValidationError
@@ -81,7 +83,7 @@ async function $do(
     APICall,
   ]
 > {
-  const path = pathToFunc("/v2/account/databases")();
+  const path = pathToFunc("/iris/v1/agent-memory/databases")();
 
   const headers = new Headers(compactMap({
     Accept: "application/json",
@@ -138,9 +140,10 @@ async function $do(
 
   const [result] = await M.match<
     models.ListDatabasesResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -152,9 +155,10 @@ async function $do(
     | SDKValidationError
   >(
     M.json(200, models.ListDatabasesResponseContent$inboundSchema),
-    M.jsonErr(400, errors.MemoryBadRequestErrorResponseContent$inboundSchema),
+    M.jsonErr(400, errors.BadRequestErrorResponseContent$inboundSchema),
     M.jsonErr(401, errors.AuthenticationErrorResponseContent$inboundSchema),
     M.jsonErr(403, errors.ForbiddenErrorResponseContent$inboundSchema),
+    M.jsonErr(413, errors.PayloadTooLargeErrorResponseContent$inboundSchema),
     M.jsonErr(500, errors.UnexpectedErrorResponseContent$inboundSchema),
     M.fail("4XX"),
     M.fail("5XX"),

package/src/controlplane/funcs/list-extractors.ts

@@ -3,7 +3,7 @@
  */
 
 import { AgentMemoryServiceCore } from "../core.js";
-import { encodeSimple } from "../lib/encodings.js";
+import { encodeFormQuery, encodeSimple } from "../lib/encodings.js";
 import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
@@ -33,15 +33,20 @@ import { Result } from "../types/fp.js";
 export function listExtractors(
   client: AgentMemoryServiceCore,
   storeId: string,
+  projectId?: string | undefined,
+  limit?: number | undefined,
+  cursor?: string | undefined,
   options?: RequestOptions,
 ): APIPromise<
   Result<
     models.ListExtractorsResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -56,6 +61,9 @@ export function listExtractors(
   return new APIPromise($do(
     client,
     storeId,
+    projectId,
+    limit,
+    cursor,
     options,
   ));
 }
@@ -63,16 +71,21 @@ export function listExtractors(
 async function $do(
   client: AgentMemoryServiceCore,
   storeId: string,
+  projectId?: string | undefined,
+  limit?: number | undefined,
+  cursor?: string | undefined,
   options?: RequestOptions,
 ): Promise<
   [
     Result<
       models.ListExtractorsResponseContent,
-      | errors.MemoryBadRequestErrorResponseContent
+      | errors.BadRequestErrorResponseContent
       | errors.AuthenticationErrorResponseContent
       | errors.ForbiddenErrorResponseContent
       | errors.NotFoundErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
       | errors.UnexpectedErrorResponseContent
+      | errors.NotImplementedErrorResponseContent
       | errors.ServiceUnavailableErrorResponseContent
       | AgentMemoryServiceError
       | ResponseValidationError
@@ -88,6 +101,9 @@ async function $do(
 > {
   const input: operations.ListExtractorsRequest = {
     storeId: storeId,
+    projectId: projectId,
+    limit: limit,
+    cursor: cursor,
   };
 
   const parsed = safeParse(
@@ -107,7 +123,15 @@ async function $do(
       charEncoding: "percent",
     }),
   };
-  const path = pathToFunc("/v2/stores/{storeId}/extractors")(pathParams);
+  const path = pathToFunc("/iris/v1/agent-memory/stores/{storeId}/extractors")(
+    pathParams,
+  );
+
+  const query = encodeFormQuery({
+    "cursor": payload.cursor,
+    "limit": payload.limit,
+    "projectId": payload.projectId,
+  });
 
   const headers = new Headers(compactMap({
     Accept: "application/json",
@@ -138,6 +162,7 @@ async function $do(
     baseURL: options?.serverURL,
     path: path,
     headers: headers,
+    query: query,
     body: body,
     userAgent: client._options.userAgent,
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
@@ -165,11 +190,13 @@ async function $do(
 
   const [result] = await M.match<
     models.ListExtractorsResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -181,11 +208,13 @@ async function $do(
     | SDKValidationError
   >(
     M.json(200, models.ListExtractorsResponseContent$inboundSchema),
-    M.jsonErr(400, errors.MemoryBadRequestErrorResponseContent$inboundSchema),
+    M.jsonErr(400, errors.BadRequestErrorResponseContent$inboundSchema),
     M.jsonErr(401, errors.AuthenticationErrorResponseContent$inboundSchema),
     M.jsonErr(403, errors.ForbiddenErrorResponseContent$inboundSchema),
     M.jsonErr(404, errors.NotFoundErrorResponseContent$inboundSchema),
+    M.jsonErr(413, errors.PayloadTooLargeErrorResponseContent$inboundSchema),
     M.jsonErr(500, errors.UnexpectedErrorResponseContent$inboundSchema),
+    M.jsonErr(501, errors.NotImplementedErrorResponseContent$inboundSchema),
     M.jsonErr(503, errors.ServiceUnavailableErrorResponseContent$inboundSchema),
     M.fail("4XX"),
     M.fail("5XX"),