@getzep/zep-cloud 3.0.4 → 3.1.0

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.4",
-                "User-Agent": "zep-cloud/3.0.4",
+                "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.
@@ -132,22 +134,6 @@ export declare class Graph {
      */
     clone(request?: Zep.CloneGraphRequest, requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.CloneGraphResponse>;
     private __clone;
-    /**
-     * Perform a graph search query.
-     *
-     * @param {Zep.GraphSearchQuery} request
-     * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
-     *
-     * @throws {@link Zep.BadRequestError}
-     * @throws {@link Zep.InternalServerError}
-     *
-     * @example
-     *     await client.graph.search({
-     *         query: "query"
-     *     })
-     */
-    search(request: Zep.GraphSearchQuery, requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.GraphSearchResults>;
-    private __search;
     /**
      * Creates a new graph.
      *
@@ -178,6 +164,22 @@ export declare class Graph {
      */
     listAll(request?: Zep.GraphListAllRequest, requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.GraphListResponse>;
     private __listAll;
+    /**
+     * Perform a graph search query.
+     *
+     * @param {Zep.GraphSearchQuery} request
+     * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link Zep.BadRequestError}
+     * @throws {@link Zep.InternalServerError}
+     *
+     * @example
+     *     await client.graph.search({
+     *         query: "query"
+     *     })
+     */
+    search(request: Zep.GraphSearchQuery, requestOptions?: Graph.RequestOptions): core.HttpResponsePromise<Zep.GraphSearchResults>;
+    private __search;
     /**
      * Returns a graph.
      *

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.
@@ -563,32 +582,32 @@ export class Graph {
         });
     }
     /**
-     * Perform a graph search query.
+     * Creates a new graph.
      *
-     * @param {Zep.GraphSearchQuery} request
+     * @param {Zep.CreateGraphRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
      * @throws {@link Zep.InternalServerError}
      *
      * @example
-     *     await client.graph.search({
-     *         query: "query"
+     *     await client.graph.create({
+     *         graphId: "graph_id"
      *     })
      */
-    search(request, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__search(request, requestOptions));
+    create(request, requestOptions) {
+        return core.HttpResponsePromise.fromPromise(this.__create(request, requestOptions));
     }
-    __search(request, requestOptions) {
+    __create(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
             var _a, _b, _c, _d;
             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, "graph/search"),
+                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, "graph/create"),
                 method: "POST",
                 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),
                 contentType: "application/json",
                 requestType: "json",
-                body: serializers.GraphSearchQuery.jsonOrThrow(request, {
+                body: serializers.CreateGraphRequest.jsonOrThrow(request, {
                     unrecognizedObjectKeys: "strip",
                     omitUndefined: true,
                 }),
@@ -598,7 +617,7 @@ export class Graph {
             });
             if (_response.ok) {
                 return {
-                    data: serializers.GraphSearchResults.parseOrThrow(_response.body, {
+                    data: serializers.Graph.parseOrThrow(_response.body, {
                         unrecognizedObjectKeys: "passthrough",
                         allowUnrecognizedUnionMembers: true,
                         allowUnrecognizedEnumValues: true,
@@ -642,7 +661,7 @@ export class Graph {
                         rawResponse: _response.rawResponse,
                     });
                 case "timeout":
-                    throw new errors.ZepTimeoutError("Timeout exceeded when calling POST /graph/search.");
+                    throw new errors.ZepTimeoutError("Timeout exceeded when calling POST /graph/create.");
                 case "unknown":
                     throw new errors.ZepError({
                         message: _response.error.errorMessage,
@@ -652,42 +671,43 @@ export class Graph {
         });
     }
     /**
-     * Creates a new graph.
+     * Returns all graphs.
      *
-     * @param {Zep.CreateGraphRequest} request
+     * @param {Zep.GraphListAllRequest} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
      * @throws {@link Zep.InternalServerError}
      *
      * @example
-     *     await client.graph.create({
-     *         graphId: "graph_id"
-     *     })
+     *     await client.graph.listAll()
      */
-    create(request, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__create(request, requestOptions));
+    listAll(request = {}, requestOptions) {
+        return core.HttpResponsePromise.fromPromise(this.__listAll(request, requestOptions));
     }
-    __create(request, requestOptions) {
-        return __awaiter(this, void 0, void 0, function* () {
+    __listAll() {
+        return __awaiter(this, arguments, void 0, function* (request = {}, requestOptions) {
             var _a, _b, _c, _d;
+            const { pageNumber, pageSize } = request;
+            const _queryParams = {};
+            if (pageNumber != null) {
+                _queryParams["pageNumber"] = pageNumber.toString();
+            }
+            if (pageSize != null) {
+                _queryParams["pageSize"] = pageSize.toString();
+            }
             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, "graphs"),
-                method: "POST",
+                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, "graph/list-all"),
+                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),
-                contentType: "application/json",
-                requestType: "json",
-                body: serializers.CreateGraphRequest.jsonOrThrow(request, {
-                    unrecognizedObjectKeys: "strip",
-                    omitUndefined: true,
-                }),
+                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,
             });
             if (_response.ok) {
                 return {
-                    data: serializers.Graph.parseOrThrow(_response.body, {
+                    data: serializers.GraphListResponse.parseOrThrow(_response.body, {
                         unrecognizedObjectKeys: "passthrough",
                         allowUnrecognizedUnionMembers: true,
                         allowUnrecognizedEnumValues: true,
@@ -731,7 +751,7 @@ export class Graph {
                         rawResponse: _response.rawResponse,
                     });
                 case "timeout":
-                    throw new errors.ZepTimeoutError("Timeout exceeded when calling POST /graphs.");
+                    throw new errors.ZepTimeoutError("Timeout exceeded when calling GET /graph/list-all.");
                 case "unknown":
                     throw new errors.ZepError({
                         message: _response.error.errorMessage,
@@ -741,43 +761,42 @@ export class Graph {
         });
     }
     /**
-     * Returns all graphs.
+     * Perform a graph search query.
      *
-     * @param {Zep.GraphListAllRequest} request
+     * @param {Zep.GraphSearchQuery} request
      * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link Zep.BadRequestError}
      * @throws {@link Zep.InternalServerError}
      *
      * @example
-     *     await client.graph.listAll()
+     *     await client.graph.search({
+     *         query: "query"
+     *     })
      */
-    listAll(request = {}, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__listAll(request, requestOptions));
+    search(request, requestOptions) {
+        return core.HttpResponsePromise.fromPromise(this.__search(request, requestOptions));
     }
-    __listAll() {
-        return __awaiter(this, arguments, void 0, function* (request = {}, requestOptions) {
+    __search(request, requestOptions) {
+        return __awaiter(this, void 0, void 0, function* () {
             var _a, _b, _c, _d;
-            const { pageNumber, pageSize } = request;
-            const _queryParams = {};
-            if (pageNumber != null) {
-                _queryParams["pageNumber"] = pageNumber.toString();
-            }
-            if (pageSize != null) {
-                _queryParams["pageSize"] = pageSize.toString();
-            }
             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, "graphs/list-all"),
-                method: "GET",
+                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, "graph/search"),
+                method: "POST",
                 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,
+                contentType: "application/json",
+                requestType: "json",
+                body: serializers.GraphSearchQuery.jsonOrThrow(request, {
+                    unrecognizedObjectKeys: "strip",
+                    omitUndefined: true,
+                }),
                 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,
             });
             if (_response.ok) {
                 return {
-                    data: serializers.GraphListResponse.parseOrThrow(_response.body, {
+                    data: serializers.GraphSearchResults.parseOrThrow(_response.body, {
                         unrecognizedObjectKeys: "passthrough",
                         allowUnrecognizedUnionMembers: true,
                         allowUnrecognizedEnumValues: true,
@@ -821,7 +840,7 @@ export class Graph {
                         rawResponse: _response.rawResponse,
                     });
                 case "timeout":
-                    throw new errors.ZepTimeoutError("Timeout exceeded when calling GET /graphs/list-all.");
+                    throw new errors.ZepTimeoutError("Timeout exceeded when calling POST /graph/search.");
                 case "unknown":
                     throw new errors.ZepError({
                         message: _response.error.errorMessage,
@@ -849,7 +868,7 @@ export class Graph {
         return __awaiter(this, void 0, void 0, function* () {
             var _a, _b, _c, _d;
             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, `graphs/${encodeURIComponent(graphId)}`),
+                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, `graph/${encodeURIComponent(graphId)}`),
                 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),
                 timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
@@ -902,7 +921,7 @@ export class Graph {
                         rawResponse: _response.rawResponse,
                     });
                 case "timeout":
-                    throw new errors.ZepTimeoutError("Timeout exceeded when calling GET /graphs/{graphId}.");
+                    throw new errors.ZepTimeoutError("Timeout exceeded when calling GET /graph/{graphId}.");
                 case "unknown":
                     throw new errors.ZepError({
                         message: _response.error.errorMessage,
@@ -931,7 +950,7 @@ export class Graph {
         return __awaiter(this, void 0, void 0, function* () {
             var _a, _b, _c, _d;
             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, `graphs/${encodeURIComponent(graphId)}`),
+                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, `graph/${encodeURIComponent(graphId)}`),
                 method: "DELETE",
                 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),
                 timeoutMs: (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) != null ? requestOptions.timeoutInSeconds * 1000 : 60000,
@@ -992,7 +1011,7 @@ export class Graph {
                         rawResponse: _response.rawResponse,
                     });
                 case "timeout":
-                    throw new errors.ZepTimeoutError("Timeout exceeded when calling DELETE /graphs/{graphId}.");
+                    throw new errors.ZepTimeoutError("Timeout exceeded when calling DELETE /graph/{graphId}.");
                 case "unknown":
                     throw new errors.ZepError({
                         message: _response.error.errorMessage,
@@ -1022,7 +1041,7 @@ export class Graph {
         return __awaiter(this, arguments, void 0, function* (graphId, request = {}, requestOptions) {
             var _a, _b, _c, _d;
             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, `graphs/${encodeURIComponent(graphId)}`),
+                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, `graph/${encodeURIComponent(graphId)}`),
                 method: "PATCH",
                 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),
                 contentType: "application/json",
@@ -1089,7 +1108,7 @@ export class Graph {
                         rawResponse: _response.rawResponse,
                     });
                 case "timeout":
-                    throw new errors.ZepTimeoutError("Timeout exceeded when calling PATCH /graphs/{graphId}.");
+                    throw new errors.ZepTimeoutError("Timeout exceeded when calling PATCH /graph/{graphId}.");
                 case "unknown":
                     throw new errors.ZepError({
                         message: _response.error.errorMessage,

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,9 +1,10 @@
+export { type GraphListEntityTypesRequest } from "./GraphListEntityTypesRequest.mjs";
 export { type EntityTypeRequest } from "./EntityTypeRequest.mjs";
 export { type AddDataRequest } from "./AddDataRequest.mjs";
 export { type AddDataBatchRequest } from "./AddDataBatchRequest.mjs";
 export { type AddTripleRequest } from "./AddTripleRequest.mjs";
 export { type CloneGraphRequest } from "./CloneGraphRequest.mjs";
-export { type GraphSearchQuery } from "./GraphSearchQuery.mjs";
 export { type CreateGraphRequest } from "./CreateGraphRequest.mjs";
 export { type GraphListAllRequest } from "./GraphListAllRequest.mjs";
+export { type GraphSearchQuery } from "./GraphSearchQuery.mjs";
 export { type UpdateGraphRequest } from "./UpdateGraphRequest.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/serialization/resources/graph/client/requests/index.d.mts

@@ -3,6 +3,6 @@ export { AddDataRequest } from "./AddDataRequest.mjs";
 export { AddDataBatchRequest } from "./AddDataBatchRequest.mjs";
 export { AddTripleRequest } from "./AddTripleRequest.mjs";
 export { CloneGraphRequest } from "./CloneGraphRequest.mjs";
-export { GraphSearchQuery } from "./GraphSearchQuery.mjs";
 export { CreateGraphRequest } from "./CreateGraphRequest.mjs";
+export { GraphSearchQuery } from "./GraphSearchQuery.mjs";
 export { UpdateGraphRequest } from "./UpdateGraphRequest.mjs";

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

@@ -3,6 +3,6 @@ export { AddDataRequest } from "./AddDataRequest.mjs";
 export { AddDataBatchRequest } from "./AddDataBatchRequest.mjs";
 export { AddTripleRequest } from "./AddTripleRequest.mjs";
 export { CloneGraphRequest } from "./CloneGraphRequest.mjs";
-export { GraphSearchQuery } from "./GraphSearchQuery.mjs";
 export { CreateGraphRequest } from "./CreateGraphRequest.mjs";
+export { GraphSearchQuery } from "./GraphSearchQuery.mjs";
 export { UpdateGraphRequest } from "./UpdateGraphRequest.mjs";

package/dist/esm/version.d.mts

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

package/dist/esm/version.mjs

@@ -1 +1 @@
-export const SDK_VERSION = "3.0.4";
+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 {};