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

package/src/controlplane/funcs/remove-memory-type.ts

@@ -0,0 +1,249 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { AgentMemoryServiceCore } from "../core.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";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.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 operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Unregister a custom memory type without deleting existing memories or Search
+ * index fields. First disable its session extraction strategy and remove every
+ * document Extractor bound to it; otherwise this returns 409. The removed name
+ * is permanently reserved within this store to protect stored attribute schemas.
+ * Requests or extraction jobs already admitted with an older store configuration
+ * may finish. This operation is not a write barrier or a data purge.
+ * An optional expectedRevision protects against stale deletion (412 on mismatch).
+ * Even when omitted, a concurrent registry change may return 412; read and retry.
+ */
+export function removeMemoryType(
+  client: AgentMemoryServiceCore,
+  storeId: string,
+  memoryTypeName: string,
+  projectId?: string | undefined,
+  expectedRevision?: string | undefined,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    void,
+    | errors.BadRequestErrorResponseContent
+    | errors.AuthenticationErrorResponseContent
+    | errors.ForbiddenErrorResponseContent
+    | errors.NotFoundErrorResponseContent
+    | errors.ConflictErrorResponseContent
+    | errors.PreconditionFailedErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
+    | errors.FailedDependencyErrorResponseContent
+    | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
+    | errors.ServiceUnavailableErrorResponseContent
+    | AgentMemoryServiceError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    storeId,
+    memoryTypeName,
+    projectId,
+    expectedRevision,
+    options,
+  ));
+}
+
+async function $do(
+  client: AgentMemoryServiceCore,
+  storeId: string,
+  memoryTypeName: string,
+  projectId?: string | undefined,
+  expectedRevision?: string | undefined,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      void,
+      | errors.BadRequestErrorResponseContent
+      | errors.AuthenticationErrorResponseContent
+      | errors.ForbiddenErrorResponseContent
+      | errors.NotFoundErrorResponseContent
+      | errors.ConflictErrorResponseContent
+      | errors.PreconditionFailedErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
+      | errors.FailedDependencyErrorResponseContent
+      | errors.UnexpectedErrorResponseContent
+      | errors.NotImplementedErrorResponseContent
+      | errors.ServiceUnavailableErrorResponseContent
+      | AgentMemoryServiceError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const input: operations.RemoveMemoryTypeRequest = {
+    storeId: storeId,
+    memoryTypeName: memoryTypeName,
+    projectId: projectId,
+    expectedRevision: expectedRevision,
+  };
+
+  const parsed = safeParse(
+    input,
+    (value) => operations.RemoveMemoryTypeRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = null;
+
+  const pathParams = {
+    memoryTypeName: encodeSimple("memoryTypeName", payload.memoryTypeName, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+    storeId: encodeSimple("storeId", payload.storeId, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc(
+    "/iris/v1/agent-memory/stores/{storeId}/memory-types/{memoryTypeName}",
+  )(pathParams);
+
+  const query = encodeFormQuery({
+    "expectedRevision": payload.expectedRevision,
+    "projectId": payload.projectId,
+  });
+
+  const headers = new Headers(compactMap({
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.token);
+  const securityInput = secConfig == null ? {} : { token: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "RemoveMemoryType",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.token,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "DELETE",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    query: query,
+    body: body,
+    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<
+    void,
+    | errors.BadRequestErrorResponseContent
+    | errors.AuthenticationErrorResponseContent
+    | errors.ForbiddenErrorResponseContent
+    | errors.NotFoundErrorResponseContent
+    | errors.ConflictErrorResponseContent
+    | errors.PreconditionFailedErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
+    | errors.FailedDependencyErrorResponseContent
+    | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
+    | errors.ServiceUnavailableErrorResponseContent
+    | AgentMemoryServiceError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.nil(204, z.void()),
+    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(409, errors.ConflictErrorResponseContent$inboundSchema),
+    M.jsonErr(412, errors.PreconditionFailedErrorResponseContent$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"),
+  )(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/update-extractor.ts

@@ -3,7 +3,7 @@
  */
 
 import { AgentMemoryServiceCore } from "../core.js";
-import { encodeJSON, encodeSimple } from "../lib/encodings.js";
+import { encodeFormQuery, encodeJSON, encodeSimple } from "../lib/encodings.js";
 import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
@@ -29,24 +29,28 @@ import { Result } from "../types/fp.js";
 
 /**
  * Update an extractor. `expectedRevision` must match the extractor's current
- * revision; otherwise the request fails with 409.
+ * revision; otherwise the request fails with 412 `/errors/revision-conflict`.
  */
 export function updateExtractor(
   client: AgentMemoryServiceCore,
   storeId: string,
   extractorId: string,
   body: models.UpdateExtractorRequestContent,
+  projectId?: string | undefined,
   options?: RequestOptions,
 ): APIPromise<
   Result<
     models.UpdateExtractorResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
     | errors.ConflictErrorResponseContent
+    | errors.PreconditionFailedErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.FailedDependencyErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -63,6 +67,7 @@ export function updateExtractor(
     storeId,
     extractorId,
     body,
+    projectId,
     options,
   ));
 }
@@ -72,18 +77,22 @@ async function $do(
   storeId: string,
   extractorId: string,
   body: models.UpdateExtractorRequestContent,
+  projectId?: string | undefined,
   options?: RequestOptions,
 ): Promise<
   [
     Result<
       models.UpdateExtractorResponseContent,
-      | errors.MemoryBadRequestErrorResponseContent
+      | errors.BadRequestErrorResponseContent
       | errors.AuthenticationErrorResponseContent
       | errors.ForbiddenErrorResponseContent
       | errors.NotFoundErrorResponseContent
       | errors.ConflictErrorResponseContent
+      | errors.PreconditionFailedErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
       | errors.FailedDependencyErrorResponseContent
       | errors.UnexpectedErrorResponseContent
+      | errors.NotImplementedErrorResponseContent
       | errors.ServiceUnavailableErrorResponseContent
       | AgentMemoryServiceError
       | ResponseValidationError
@@ -101,6 +110,7 @@ async function $do(
     storeId: storeId,
     extractorId: extractorId,
     body: body,
+    projectId: projectId,
   };
 
   const parsed = safeParse(
@@ -124,9 +134,13 @@ async function $do(
       charEncoding: "percent",
     }),
   };
-  const path = pathToFunc("/v2/stores/{storeId}/extractors/{extractorId}")(
-    pathParams,
-  );
+  const path = pathToFunc(
+    "/iris/v1/agent-memory/stores/{storeId}/extractors/{extractorId}",
+  )(pathParams);
+
+  const query = encodeFormQuery({
+    "projectId": payload.projectId,
+  });
 
   const headers = new Headers(compactMap({
     "Content-Type": "application/json",
@@ -158,6 +172,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,
@@ -185,13 +200,16 @@ async function $do(
 
   const [result] = await M.match<
     models.UpdateExtractorResponseContent,
-    | errors.MemoryBadRequestErrorResponseContent
+    | errors.BadRequestErrorResponseContent
     | errors.AuthenticationErrorResponseContent
     | errors.ForbiddenErrorResponseContent
     | errors.NotFoundErrorResponseContent
     | errors.ConflictErrorResponseContent
+    | errors.PreconditionFailedErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
     | errors.FailedDependencyErrorResponseContent
     | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
     | errors.ServiceUnavailableErrorResponseContent
     | AgentMemoryServiceError
     | ResponseValidationError
@@ -203,13 +221,16 @@ async function $do(
     | SDKValidationError
   >(
     M.json(200, models.UpdateExtractorResponseContent$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(409, errors.ConflictErrorResponseContent$inboundSchema),
+    M.jsonErr(412, errors.PreconditionFailedErrorResponseContent$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/update-memory-type.ts

@@ -0,0 +1,241 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { AgentMemoryServiceCore } from "../core.js";
+import { encodeFormQuery, encodeJSON, encodeSimple } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.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 * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Change a custom memory type's extraction strategy: its prompt, `enabled`, or both.
+ * Omitted members are unchanged. The type must already have a strategy.
+ * An optional expectedRevision protects against stale edits (412 on mismatch).
+ * Even when omitted, a concurrent registry change may return 412; read and retry.
+ */
+export function updateMemoryType(
+  client: AgentMemoryServiceCore,
+  storeId: string,
+  memoryTypeName: string,
+  body: models.UpdateMemoryTypeRequestContent,
+  projectId?: string | undefined,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    models.UpdateMemoryTypeResponseContent,
+    | errors.BadRequestErrorResponseContent
+    | errors.AuthenticationErrorResponseContent
+    | errors.ForbiddenErrorResponseContent
+    | errors.NotFoundErrorResponseContent
+    | errors.PreconditionFailedErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
+    | errors.FailedDependencyErrorResponseContent
+    | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
+    | errors.ServiceUnavailableErrorResponseContent
+    | AgentMemoryServiceError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    storeId,
+    memoryTypeName,
+    body,
+    projectId,
+    options,
+  ));
+}
+
+async function $do(
+  client: AgentMemoryServiceCore,
+  storeId: string,
+  memoryTypeName: string,
+  body: models.UpdateMemoryTypeRequestContent,
+  projectId?: string | undefined,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      models.UpdateMemoryTypeResponseContent,
+      | errors.BadRequestErrorResponseContent
+      | errors.AuthenticationErrorResponseContent
+      | errors.ForbiddenErrorResponseContent
+      | errors.NotFoundErrorResponseContent
+      | errors.PreconditionFailedErrorResponseContent
+      | errors.PayloadTooLargeErrorResponseContent
+      | errors.FailedDependencyErrorResponseContent
+      | errors.UnexpectedErrorResponseContent
+      | errors.NotImplementedErrorResponseContent
+      | errors.ServiceUnavailableErrorResponseContent
+      | AgentMemoryServiceError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const input: operations.UpdateMemoryTypeRequest = {
+    storeId: storeId,
+    memoryTypeName: memoryTypeName,
+    body: body,
+    projectId: projectId,
+  };
+
+  const parsed = safeParse(
+    input,
+    (value) => operations.UpdateMemoryTypeRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body$ = encodeJSON("body", payload.body, { explode: true });
+
+  const pathParams = {
+    memoryTypeName: encodeSimple("memoryTypeName", payload.memoryTypeName, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+    storeId: encodeSimple("storeId", payload.storeId, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc(
+    "/iris/v1/agent-memory/stores/{storeId}/memory-types/{memoryTypeName}",
+  )(pathParams);
+
+  const query = encodeFormQuery({
+    "projectId": payload.projectId,
+  });
+
+  const headers = new Headers(compactMap({
+    "Content-Type": "application/json",
+    Accept: "application/json",
+  }));
+
+  const secConfig = await extractSecurity(client._options.token);
+  const securityInput = secConfig == null ? {} : { token: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "UpdateMemoryType",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.token,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "PATCH",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    query: query,
+    body: body$,
+    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.UpdateMemoryTypeResponseContent,
+    | errors.BadRequestErrorResponseContent
+    | errors.AuthenticationErrorResponseContent
+    | errors.ForbiddenErrorResponseContent
+    | errors.NotFoundErrorResponseContent
+    | errors.PreconditionFailedErrorResponseContent
+    | errors.PayloadTooLargeErrorResponseContent
+    | errors.FailedDependencyErrorResponseContent
+    | errors.UnexpectedErrorResponseContent
+    | errors.NotImplementedErrorResponseContent
+    | errors.ServiceUnavailableErrorResponseContent
+    | AgentMemoryServiceError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, models.UpdateMemoryTypeResponseContent$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(412, errors.PreconditionFailedErrorResponseContent$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"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}