@getzep/zep-cloud 3.0.5 → 3.1.0

package/dist/cjs/Client.js

@@ -47,8 +47,8 @@ class ZepClient {
         this._options = Object.assign(Object.assign({}, _options), { headers: (0, headers_js_1.mergeHeaders)({
                 "X-Fern-Language": "JavaScript",
                 "X-Fern-SDK-Name": "zep-cloud",
-                "X-Fern-SDK-Version": "3.0.5",
-                "User-Agent": "zep-cloud/3.0.5",
+                "X-Fern-SDK-Version": "3.1.0",
+                "User-Agent": "zep-cloud/3.1.0",
                 "X-Fern-Runtime": core.RUNTIME.type,
                 "X-Fern-Runtime-Version": core.RUNTIME.version,
             }, _options === null || _options === void 0 ? void 0 : _options.headers) });

package/dist/cjs/api/errors/index.d.ts

@@ -1,3 +1,3 @@
+export * from "./BadRequestError.js";
 export * from "./NotFoundError.js";
 export * from "./InternalServerError.js";
-export * from "./BadRequestError.js";

package/dist/cjs/api/errors/index.js

@@ -14,6 +14,6 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./BadRequestError.js"), exports);
 __exportStar(require("./NotFoundError.js"), exports);
 __exportStar(require("./InternalServerError.js"), exports);
-__exportStar(require("./BadRequestError.js"), exports);

package/dist/cjs/api/resources/graph/client/Client.d.ts

@@ -38,20 +38,22 @@ export declare class Graph {
     get episode(): Episode;
     get node(): Node;
     /**
-     * Returns all entity types for a project.
+     * Returns all entity types for a project, user, or graph.
      *
+     * @param {Zep.GraphListEntityTypesRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
+     * @throws {@link Zep.BadRequestError}
      * @throws {@link Zep.NotFoundError}
      * @throws {@link Zep.InternalServerError}
      *
      * @example
      *     await client.graph.listEntityTypes()
      */
-    listEntityTypes(requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.EntityTypeResponse>;
+    listEntityTypes(request?: Zep.GraphListEntityTypesRequest, requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.EntityTypeResponse>;
     private __listEntityTypes;
     /**
-     * Sets the entity types for a project, replacing any existing ones.
+     * Sets the entity types for multiple users and graphs, replacing any existing ones.
      *
      * @param {Zep.EntityTypeRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.

package/dist/cjs/api/resources/graph/client/Client.js

@@ -72,26 +72,37 @@ class Graph {
         return ((_a = this._node) !== null && _a !== void 0 ? _a : (this._node = new Client_js_3.Node(this._options)));
     }
     /**
-     * Returns all entity types for a project.
+     * Returns all entity types for a project, user, or graph.
      *
+     * @param {Zep.GraphListEntityTypesRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
+     * @throws {@link Zep.BadRequestError}
      * @throws {@link Zep.NotFoundError}
      * @throws {@link Zep.InternalServerError}
      *
      * @example
      *     await client.graph.listEntityTypes()
      */
-    listEntityTypes(requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__listEntityTypes(requestOptions));
+    listEntityTypes(request = {}, requestOptions) {
+        return core.HttpResponsePromise.fromPromise(this.__listEntityTypes(request, requestOptions));
     }
-    __listEntityTypes(requestOptions) {
-        return __awaiter(this, void 0, void 0, function* () {
+    __listEntityTypes() {
+        return __awaiter(this, arguments, void 0, function* (request = {}, requestOptions) {
             var _a, _b, _c, _d;
+            const { userId, graphId } = request;
+            const _queryParams = {};
+            if (userId != null) {
+                _queryParams["user_id"] = userId;
+            }
+            if (graphId != null) {
+                _queryParams["graph_id"] = graphId;
+            }
             const _response = yield ((_a = this._options.fetcher) !== null && _a !== void 0 ? _a : core.fetcher)({
                 url: core.url.join((_c = (_b = (yield core.Supplier.get(this._options.baseUrl))) !== null && _b !== void 0 ? _b : (yield core.Supplier.get(this._options.environment))) !== null && _c !== void 0 ? _c : environments.ZepEnvironment.Default, "entity-types"),
                 method: "GET",
                 headers: (0, headers_js_1.mergeHeaders)((_d = this._options) === null || _d === void 0 ? void 0 : _d.headers, (0, headers_js_1.mergeOnlyDefinedHeaders)(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers),
+                queryParameters: _queryParams,
                 timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
                 maxRetries: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
@@ -110,6 +121,14 @@ class Graph {
             }
             if (_response.error.reason === "status-code") {
                 switch (_response.error.statusCode) {
+                    case 400:
+                        throw new Zep.BadRequestError(serializers.ApiError.parseOrThrow(_response.error.body, {
+                            unrecognizedObjectKeys: "passthrough",
+                            allowUnrecognizedUnionMembers: true,
+                            allowUnrecognizedEnumValues: true,
+                            skipValidation: true,
+                            breadcrumbsPrefix: ["response"],
+                        }), _response.rawResponse);
                     case 404:
                         throw new Zep.NotFoundError(serializers.ApiError.parseOrThrow(_response.error.body, {
                             unrecognizedObjectKeys: "passthrough",
@@ -152,7 +171,7 @@ class Graph {
         });
     }
     /**
-     * Sets the entity types for a project, replacing any existing ones.
+     * Sets the entity types for multiple users and graphs, replacing any existing ones.
      *
      * @param {Zep.EntityTypeRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.

package/dist/cjs/api/resources/graph/client/requests/EntityTypeRequest.d.ts

@@ -9,4 +9,6 @@ import * as Zep from "../../../../index.js";
 export interface EntityTypeRequest {
     edgeTypes?: Zep.EdgeType[];
     entityTypes?: Zep.EntityType[];
+    graphIds?: string[];
+    userIds?: string[];
 }

package/dist/cjs/api/resources/graph/client/requests/GraphListEntityTypesRequest.d.ts

@@ -0,0 +1,17 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+/**
+ * @example
+ *     {}
+ */
+export interface GraphListEntityTypesRequest {
+    /**
+     * User ID to get user-specific entity types
+     */
+    userId?: string;
+    /**
+     * Graph ID to get graph-specific entity types
+     */
+    graphId?: string;
+}

package/dist/cjs/api/resources/graph/client/requests/GraphListEntityTypesRequest.js

@@ -0,0 +1,5 @@
+"use strict";
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/cjs/api/resources/graph/client/requests/index.d.ts

@@ -1,3 +1,4 @@
+export { type GraphListEntityTypesRequest } from "./GraphListEntityTypesRequest.js";
 export { type EntityTypeRequest } from "./EntityTypeRequest.js";
 export { type AddDataRequest } from "./AddDataRequest.js";
 export { type AddDataBatchRequest } from "./AddDataBatchRequest.js";

package/dist/cjs/serialization/resources/graph/client/requests/EntityTypeRequest.d.ts

@@ -11,5 +11,7 @@ export declare namespace EntityTypeRequest {
     interface Raw {
         edge_types?: EdgeType.Raw[] | null;
         entity_types?: EntityType.Raw[] | null;
+        graph_ids?: string[] | null;
+        user_ids?: string[] | null;
     }
 }

package/dist/cjs/serialization/resources/graph/client/requests/EntityTypeRequest.js

@@ -43,4 +43,6 @@ const EntityType_js_1 = require("../../../../types/EntityType.js");
 exports.EntityTypeRequest = core.serialization.object({
     edgeTypes: core.serialization.property("edge_types", core.serialization.list(EdgeType_js_1.EdgeType).optional()),
     entityTypes: core.serialization.property("entity_types", core.serialization.list(EntityType_js_1.EntityType).optional()),
+    graphIds: core.serialization.property("graph_ids", core.serialization.list(core.serialization.string()).optional()),
+    userIds: core.serialization.property("user_ids", core.serialization.list(core.serialization.string()).optional()),
 });

package/dist/cjs/version.d.ts

@@ -1 +1 @@
-export declare const SDK_VERSION = "3.0.5";
+export declare const SDK_VERSION = "3.1.0";

package/dist/cjs/version.js

@@ -1,4 +1,4 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SDK_VERSION = void 0;
-exports.SDK_VERSION = "3.0.5";
+exports.SDK_VERSION = "3.1.0";

package/dist/cjs/wrapper/graph.d.ts

@@ -1,12 +1,17 @@
 import { Graph as BaseGraph } from "../api/resources/graph/client/Client.js";
 import { Zep } from "../index.js";
 import { EdgeType, EntityType } from "./ontology.js";
+interface OntologyTargets {
+    userIds?: string[];
+    graphIds?: string[];
+}
 export declare class Graph extends BaseGraph {
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -38,14 +43,15 @@ export declare class Graph extends BaseGraph {
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setEntityTypes(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
+    setEntityTypes(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, ontologyTargets?: OntologyTargets, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -73,11 +79,12 @@ export declare class Graph extends BaseGraph {
      *         ]
      *     }
      *
-     *     await client.graph.setOntology({
+     *     await client.graph.setEntityTypes({
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setOntology(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
+    setOntology(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, ontologyTargets?: OntologyTargets, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
 }
+export {};

package/dist/cjs/wrapper/graph.js

@@ -14,10 +14,11 @@ const Client_js_1 = require("../api/resources/graph/client/Client.js");
 const ontology_js_1 = require("./ontology.js");
 class Graph extends Client_js_1.Graph {
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -49,9 +50,9 @@ class Graph extends Client_js_1.Graph {
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setEntityTypes(entityTypes, edgeTypes, requestOptions) {
+    setEntityTypes(entityTypes, edgeTypes, ontologyTargets, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
             const validatedEntityTypes = Object.keys(entityTypes).map((key) => {
                 const schema = entityTypes[key];
@@ -64,14 +65,17 @@ class Graph extends Client_js_1.Graph {
             return this.setEntityTypesInternal({
                 entityTypes: validatedEntityTypes,
                 edgeTypes: validatedEdgeTypes,
+                userIds: ontologyTargets === null || ontologyTargets === void 0 ? void 0 : ontologyTargets.userIds,
+                graphIds: ontologyTargets === null || ontologyTargets === void 0 ? void 0 : ontologyTargets.graphIds,
             }, requestOptions);
         });
     }
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -99,15 +103,15 @@ class Graph extends Client_js_1.Graph {
      *         ]
      *     }
      *
-     *     await client.graph.setOntology({
+     *     await client.graph.setEntityTypes({
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setOntology(entityTypes, edgeTypes, requestOptions) {
+    setOntology(entityTypes, edgeTypes, ontologyTargets, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            return this.setEntityTypes(entityTypes, edgeTypes, requestOptions);
+            return this.setEntityTypes(entityTypes, edgeTypes, ontologyTargets, requestOptions);
         });
     }
 }

package/dist/esm/Client.mjs

@@ -11,8 +11,8 @@ export class ZepClient {
         this._options = Object.assign(Object.assign({}, _options), { headers: mergeHeaders({
                 "X-Fern-Language": "JavaScript",
                 "X-Fern-SDK-Name": "zep-cloud",
-                "X-Fern-SDK-Version": "3.0.5",
-                "User-Agent": "zep-cloud/3.0.5",
+                "X-Fern-SDK-Version": "3.1.0",
+                "User-Agent": "zep-cloud/3.1.0",
                 "X-Fern-Runtime": core.RUNTIME.type,
                 "X-Fern-Runtime-Version": core.RUNTIME.version,
             }, _options === null || _options === void 0 ? void 0 : _options.headers) });

package/dist/esm/api/errors/index.d.mts

@@ -1,3 +1,3 @@
+export * from "./BadRequestError.mjs";
 export * from "./NotFoundError.mjs";
 export * from "./InternalServerError.mjs";
-export * from "./BadRequestError.mjs";

package/dist/esm/api/errors/index.mjs

@@ -1,3 +1,3 @@
+export * from "./BadRequestError.mjs";
 export * from "./NotFoundError.mjs";
 export * from "./InternalServerError.mjs";
-export * from "./BadRequestError.mjs";

package/dist/esm/api/resources/graph/client/Client.d.mts

@@ -38,20 +38,22 @@ export declare class Graph {
     get episode(): Episode;
     get node(): Node;
     /**
-     * Returns all entity types for a project.
+     * Returns all entity types for a project, user, or graph.
      *
+     * @param {Zep.GraphListEntityTypesRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
+     * @throws {@link Zep.BadRequestError}
      * @throws {@link Zep.NotFoundError}
      * @throws {@link Zep.InternalServerError}
      *
      * @example
      *     await client.graph.listEntityTypes()
      */
-    listEntityTypes(requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.EntityTypeResponse>;
+    listEntityTypes(request?: Zep.GraphListEntityTypesRequest, requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.EntityTypeResponse>;
     private __listEntityTypes;
     /**
-     * Sets the entity types for a project, replacing any existing ones.
+     * Sets the entity types for multiple users and graphs, replacing any existing ones.
      *
      * @param {Zep.EntityTypeRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.

package/dist/esm/api/resources/graph/client/Client.mjs

@@ -36,26 +36,37 @@ export class Graph {
         return ((_a = this._node) !== null && _a !== void 0 ? _a : (this._node = new Node(this._options)));
     }
     /**
-     * Returns all entity types for a project.
+     * Returns all entity types for a project, user, or graph.
      *
+     * @param {Zep.GraphListEntityTypesRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
+     * @throws {@link Zep.BadRequestError}
      * @throws {@link Zep.NotFoundError}
      * @throws {@link Zep.InternalServerError}
      *
      * @example
      *     await client.graph.listEntityTypes()
      */
-    listEntityTypes(requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__listEntityTypes(requestOptions));
+    listEntityTypes(request = {}, requestOptions) {
+        return core.HttpResponsePromise.fromPromise(this.__listEntityTypes(request, requestOptions));
     }
-    __listEntityTypes(requestOptions) {
-        return __awaiter(this, void 0, void 0, function* () {
+    __listEntityTypes() {
+        return __awaiter(this, arguments, void 0, function* (request = {}, requestOptions) {
             var _a, _b, _c, _d;
+            const { userId, graphId } = request;
+            const _queryParams = {};
+            if (userId != null) {
+                _queryParams["user_id"] = userId;
+            }
+            if (graphId != null) {
+                _queryParams["graph_id"] = graphId;
+            }
             const _response = yield ((_a = this._options.fetcher) !== null && _a !== void 0 ? _a : core.fetcher)({
                 url: core.url.join((_c = (_b = (yield core.Supplier.get(this._options.baseUrl))) !== null && _b !== void 0 ? _b : (yield core.Supplier.get(this._options.environment))) !== null && _c !== void 0 ? _c : environments.ZepEnvironment.Default, "entity-types"),
                 method: "GET",
                 headers: mergeHeaders((_d = this._options) === null || _d === void 0 ? void 0 : _d.headers, mergeOnlyDefinedHeaders(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers),
+                queryParameters: _queryParams,
                 timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
                 maxRetries: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
@@ -74,6 +85,14 @@ export class Graph {
             }
             if (_response.error.reason === "status-code") {
                 switch (_response.error.statusCode) {
+                    case 400:
+                        throw new Zep.BadRequestError(serializers.ApiError.parseOrThrow(_response.error.body, {
+                            unrecognizedObjectKeys: "passthrough",
+                            allowUnrecognizedUnionMembers: true,
+                            allowUnrecognizedEnumValues: true,
+                            skipValidation: true,
+                            breadcrumbsPrefix: ["response"],
+                        }), _response.rawResponse);
                     case 404:
                         throw new Zep.NotFoundError(serializers.ApiError.parseOrThrow(_response.error.body, {
                             unrecognizedObjectKeys: "passthrough",
@@ -116,7 +135,7 @@ export class Graph {
         });
     }
     /**
-     * Sets the entity types for a project, replacing any existing ones.
+     * Sets the entity types for multiple users and graphs, replacing any existing ones.
      *
      * @param {Zep.EntityTypeRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.

package/dist/esm/api/resources/graph/client/requests/EntityTypeRequest.d.mts

@@ -9,4 +9,6 @@ import * as Zep from "../../../../index.mjs";
 export interface EntityTypeRequest {
     edgeTypes?: Zep.EdgeType[];
     entityTypes?: Zep.EntityType[];
+    graphIds?: string[];
+    userIds?: string[];
 }

package/dist/esm/api/resources/graph/client/requests/GraphListEntityTypesRequest.d.mts

@@ -0,0 +1,17 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+/**
+ * @example
+ *     {}
+ */
+export interface GraphListEntityTypesRequest {
+    /**
+     * User ID to get user-specific entity types
+     */
+    userId?: string;
+    /**
+     * Graph ID to get graph-specific entity types
+     */
+    graphId?: string;
+}

package/dist/esm/api/resources/graph/client/requests/GraphListEntityTypesRequest.mjs

@@ -0,0 +1,4 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+export {};

package/dist/esm/api/resources/graph/client/requests/index.d.mts

@@ -1,3 +1,4 @@
+export { type GraphListEntityTypesRequest } from "./GraphListEntityTypesRequest.mjs";
 export { type EntityTypeRequest } from "./EntityTypeRequest.mjs";
 export { type AddDataRequest } from "./AddDataRequest.mjs";
 export { type AddDataBatchRequest } from "./AddDataBatchRequest.mjs";

package/dist/esm/serialization/resources/graph/client/requests/EntityTypeRequest.d.mts

@@ -11,5 +11,7 @@ export declare namespace EntityTypeRequest {
     interface Raw {
         edge_types?: EdgeType.Raw[] | null;
         entity_types?: EntityType.Raw[] | null;
+        graph_ids?: string[] | null;
+        user_ids?: string[] | null;
     }
 }

package/dist/esm/serialization/resources/graph/client/requests/EntityTypeRequest.mjs

@@ -7,4 +7,6 @@ import { EntityType } from "../../../../types/EntityType.mjs";
 export const EntityTypeRequest = core.serialization.object({
     edgeTypes: core.serialization.property("edge_types", core.serialization.list(EdgeType).optional()),
     entityTypes: core.serialization.property("entity_types", core.serialization.list(EntityType).optional()),
+    graphIds: core.serialization.property("graph_ids", core.serialization.list(core.serialization.string()).optional()),
+    userIds: core.serialization.property("user_ids", core.serialization.list(core.serialization.string()).optional()),
 });

package/dist/esm/version.d.mts

@@ -1 +1 @@
-export declare const SDK_VERSION = "3.0.5";
+export declare const SDK_VERSION = "3.1.0";

package/dist/esm/version.mjs

@@ -1 +1 @@
-export const SDK_VERSION = "3.0.5";
+export const SDK_VERSION = "3.1.0";

package/dist/esm/wrapper/graph.d.mts

@@ -1,12 +1,17 @@
 import { Graph as BaseGraph } from "../api/resources/graph/client/Client.mjs";
 import { Zep } from "../index.mjs";
 import { EdgeType, EntityType } from "./ontology.mjs";
+interface OntologyTargets {
+    userIds?: string[];
+    graphIds?: string[];
+}
 export declare class Graph extends BaseGraph {
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -38,14 +43,15 @@ export declare class Graph extends BaseGraph {
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setEntityTypes(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
+    setEntityTypes(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, ontologyTargets?: OntologyTargets, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -73,11 +79,12 @@ export declare class Graph extends BaseGraph {
      *         ]
      *     }
      *
-     *     await client.graph.setOntology({
+     *     await client.graph.setEntityTypes({
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setOntology(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
+    setOntology(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, ontologyTargets?: OntologyTargets, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
 }
+export {};

package/dist/esm/wrapper/graph.mjs

@@ -11,10 +11,11 @@ import { Graph as BaseGraph } from "../api/resources/graph/client/Client.mjs";
 import { entityModelToAPISchema, edgeModelToAPISchema } from "./ontology.mjs";
 export class Graph extends BaseGraph {
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -46,9 +47,9 @@ export class Graph extends BaseGraph {
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setEntityTypes(entityTypes, edgeTypes, requestOptions) {
+    setEntityTypes(entityTypes, edgeTypes, ontologyTargets, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
             const validatedEntityTypes = Object.keys(entityTypes).map((key) => {
                 const schema = entityTypes[key];
@@ -61,14 +62,17 @@ export class Graph extends BaseGraph {
             return this.setEntityTypesInternal({
                 entityTypes: validatedEntityTypes,
                 edgeTypes: validatedEdgeTypes,
+                userIds: ontologyTargets === null || ontologyTargets === void 0 ? void 0 : ontologyTargets.userIds,
+                graphIds: ontologyTargets === null || ontologyTargets === void 0 ? void 0 : ontologyTargets.graphIds,
             }, requestOptions);
         });
     }
     /**
-     * Sets the entity and edge types for a project, replacing any existing ones.
+     * Sets the entity and edge types for the specified targets, replacing any existing ones in those targets. If no targets are specified, it sets the ontology for the entire project.
      *
      * @param {Record<string, EntityType>} entityTypes
      * @param {Record<string, EdgeType>} edgeTypes
+     * @param {OntologyTargets} [ontologyTargets] - The targets for which to set the ontology. Can include userIds or graphIds. If none specified, sets for the entire project.
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
@@ -96,15 +100,15 @@ export class Graph extends BaseGraph {
      *         ]
      *     }
      *
-     *     await client.graph.setOntology({
+     *     await client.graph.setEntityTypes({
      *         TravelDestination: travelDestinationSchema,
      *     }, {
      *         IS_TRAVELING_TO: isTravelingTo,
-     *     });
+     *     }, {userIds: ['user_1234']});
      */
-    setOntology(entityTypes, edgeTypes, requestOptions) {
+    setOntology(entityTypes, edgeTypes, ontologyTargets, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            return this.setEntityTypes(entityTypes, edgeTypes, requestOptions);
+            return this.setEntityTypes(entityTypes, edgeTypes, ontologyTargets, requestOptions);
         });
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@getzep/zep-cloud",
-    "version": "3.0.5",
+    "version": "3.1.0",
     "private": false,
     "repository": "https://github.com/getzep/zep-js",
     "description": "Zep: Fast, scalable building blocks for production LLM apps",

package/reference.md

@@ -2,7 +2,7 @@
 
 ## Graph
 
-<details><summary><code>client.graph.<a href="/src/api/resources/graph/client/Client.ts">listEntityTypes</a>() -> Zep.EntityTypeResponse</code></summary>
+<details><summary><code>client.graph.<a href="/src/api/resources/graph/client/Client.ts">listEntityTypes</a>({ ...params }) -> Zep.EntityTypeResponse</code></summary>
 <dl>
 <dd>
 
@@ -14,7 +14,7 @@
 <dl>
 <dd>
 
-Returns all entity types for a project.
+Returns all entity types for a project, user, or graph.
 
 </dd>
 </dl>
@@ -46,6 +46,14 @@ await client.graph.listEntityTypes();
 <dl>
 <dd>
 
+**request:** `Zep.GraphListEntityTypesRequest`
+
+</dd>
+</dl>
+
+<dl>
+<dd>
+
 **requestOptions:** `Graph.RequestOptions`
 
 </dd>
@@ -69,7 +77,7 @@ await client.graph.listEntityTypes();
 <dl>
 <dd>
 
-Sets the entity types for a project, replacing any existing ones.
+Sets the entity types for multiple users and graphs, replacing any existing ones.
 
 </dd>
 </dl>