@getzep/zep-cloud 2.12.3 → 2.14.0

package/examples/graph/entity_type_example.ts

@@ -1,5 +1,5 @@
 import { ZepClient } from "../../src";
-import { EntityData, entityFields, EntityType } from "../../src/wrapper/ontology";
+import { EntityData, entityFields, EntityType, EdgeType } from "../../src/wrapper/ontology";
 
 const API_KEY = process.env.ZEP_API_KEY;
 
@@ -8,33 +8,37 @@ async function main() {
         apiKey: API_KEY,
     });
 
-    const purchaseSchema: EntityType = {
-        description: "A purchase",
+    const travelDestinationSchema: EntityType = {
+        description: "A travel destination entity",
         fields: {
-            product: entityFields.text("The product purchased"),
-            quantity: entityFields.integer("The quantity purchased"),
-            price: entityFields.float("The price of the product"),
+            destination_name: entityFields.text("The name of travel destination"),
         },
     };
 
-    type Purchase = EntityData<typeof purchaseSchema>;
+    type TravelDestination = EntityData<typeof travelDestinationSchema>;
 
-    await client.graph.setEntityTypes({
-        Purchase: purchaseSchema,
-    });
-
-    const { nodes } = await client.graph.search({
-        userId: "<user_id>",
-        scope: "nodes",
-        query: "ticket purchases",
-        searchFilters: {
-            nodeLabels: ["Purchase"],
+    const isTravelingTo: EdgeType = {
+        description: "An edge representing a traveler going to a destination.",
+        fields: {
+            travel_date: entityFields.text("The date of the travel"),
+            purpose: entityFields.text("The purpose of the travel"),
         },
-    });
+        sourceTargets: [
+            {
+                source: "User",
+                target: "TravelDestination",
+            }
+        ]
+    }
 
-    const purchases: Purchase[] = nodes?.map((node) => node.attributes as Purchase) ?? [];
+    await client.graph.setEntityTypes({
+        TravelDestination: travelDestinationSchema,
+    }, {
+        IS_TRAVELING_TO: isTravelingTo,
+    });
 
-    console.log(purchases);
+    const customTypes = await client.graph.listEntityTypes();
+    console.log(JSON.stringify(customTypes, null, 2));
 }
 
 main().catch(console.error);

package/package.json

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

package/reference.md

@@ -4262,7 +4262,7 @@ await client.graph.episode.delete("uuid");
 </dl>
 </details>
 
-<details><summary><code>client.graph.episode.<a href="/src/api/resources/graph/resources/episode/client/Client.ts">getNodesAndEdges</a>(uuid) -> Zep.GraphSearchResults</code></summary>
+<details><summary><code>client.graph.episode.<a href="/src/api/resources/graph/resources/episode/client/Client.ts">getNodesAndEdges</a>(uuid) -> Zep.EpisodeMentions</code></summary>
 <dl>
 <dd>
 

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

@@ -4,10 +4,12 @@
 import * as serializers from "../../../../index";
 import * as Zep from "../../../../../api/index";
 import * as core from "../../../../../core";
+import { EdgeType } from "../../../../types/EdgeType";
 import { EntityType } from "../../../../types/EntityType";
 export declare const EntityTypeRequest: core.serialization.Schema<serializers.EntityTypeRequest.Raw, Zep.EntityTypeRequest>;
 export declare namespace EntityTypeRequest {
     interface Raw {
+        edge_types?: EdgeType.Raw[] | null;
         entity_types?: EntityType.Raw[] | null;
     }
 }

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

@@ -38,7 +38,9 @@ var __importStar = (this && this.__importStar) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.EntityTypeRequest = void 0;
 const core = __importStar(require("../../../../../core"));
+const EdgeType_1 = require("../../../../types/EdgeType");
 const EntityType_1 = require("../../../../types/EntityType");
 exports.EntityTypeRequest = core.serialization.object({
+    edgeTypes: core.serialization.property("edge_types", core.serialization.list(EdgeType_1.EdgeType).optional()),
     entityTypes: core.serialization.property("entity_types", core.serialization.list(EntityType_1.EntityType).optional()),
 });

package/serialization/types/EdgeType.d.ts

@@ -0,0 +1,17 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+import * as serializers from "../index";
+import * as Zep from "../../api/index";
+import * as core from "../../core";
+import { EntityProperty } from "./EntityProperty";
+import { EntityEdgeSourceTarget } from "./EntityEdgeSourceTarget";
+export declare const EdgeType: core.serialization.ObjectSchema<serializers.EdgeType.Raw, Zep.EdgeType>;
+export declare namespace EdgeType {
+    interface Raw {
+        description: string;
+        name: string;
+        properties?: EntityProperty.Raw[] | null;
+        source_targets?: EntityEdgeSourceTarget.Raw[] | null;
+    }
+}

package/serialization/types/EdgeType.js

@@ -0,0 +1,48 @@
+"use strict";
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EdgeType = void 0;
+const core = __importStar(require("../../core"));
+const EntityProperty_1 = require("./EntityProperty");
+const EntityEdgeSourceTarget_1 = require("./EntityEdgeSourceTarget");
+exports.EdgeType = core.serialization.object({
+    description: core.serialization.string(),
+    name: core.serialization.string(),
+    properties: core.serialization.list(EntityProperty_1.EntityProperty).optional(),
+    sourceTargets: core.serialization.property("source_targets", core.serialization.list(EntityEdgeSourceTarget_1.EntityEdgeSourceTarget).optional()),
+});

package/serialization/types/EntityEdge.d.ts

@@ -7,6 +7,7 @@ import * as core from "../../core";
 export declare const EntityEdge: core.serialization.ObjectSchema<serializers.EntityEdge.Raw, Zep.EntityEdge>;
 export declare namespace EntityEdge {
     interface Raw {
+        attributes?: Record<string, unknown> | null;
         created_at: string;
         episodes?: string[] | null;
         expired_at?: string | null;

package/serialization/types/EntityEdge.js

@@ -39,6 +39,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.EntityEdge = void 0;
 const core = __importStar(require("../../core"));
 exports.EntityEdge = core.serialization.object({
+    attributes: core.serialization.record(core.serialization.string(), core.serialization.unknown()).optional(),
     createdAt: core.serialization.property("created_at", core.serialization.string()),
     episodes: core.serialization.list(core.serialization.string()).optional(),
     expiredAt: core.serialization.property("expired_at", core.serialization.string().optional()),

package/serialization/types/EntityEdgeSourceTarget.d.ts

@@ -0,0 +1,13 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+import * as serializers from "../index";
+import * as Zep from "../../api/index";
+import * as core from "../../core";
+export declare const EntityEdgeSourceTarget: core.serialization.ObjectSchema<serializers.EntityEdgeSourceTarget.Raw, Zep.EntityEdgeSourceTarget>;
+export declare namespace EntityEdgeSourceTarget {
+    interface Raw {
+        source?: string | null;
+        target?: string | null;
+    }
+}

package/serialization/types/EntityEdgeSourceTarget.js

@@ -0,0 +1,44 @@
+"use strict";
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EntityEdgeSourceTarget = void 0;
+const core = __importStar(require("../../core"));
+exports.EntityEdgeSourceTarget = core.serialization.object({
+    source: core.serialization.string().optional(),
+    target: core.serialization.string().optional(),
+});

package/serialization/types/EntityTypeResponse.d.ts

@@ -4,10 +4,12 @@
 import * as serializers from "../index";
 import * as Zep from "../../api/index";
 import * as core from "../../core";
+import { EdgeType } from "./EdgeType";
 import { EntityType } from "./EntityType";
 export declare const EntityTypeResponse: core.serialization.ObjectSchema<serializers.EntityTypeResponse.Raw, Zep.EntityTypeResponse>;
 export declare namespace EntityTypeResponse {
     interface Raw {
+        edge_types?: EdgeType.Raw[] | null;
         entity_types?: EntityType.Raw[] | null;
     }
 }

package/serialization/types/EntityTypeResponse.js

@@ -38,7 +38,9 @@ var __importStar = (this && this.__importStar) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.EntityTypeResponse = void 0;
 const core = __importStar(require("../../core"));
+const EdgeType_1 = require("./EdgeType");
 const EntityType_1 = require("./EntityType");
 exports.EntityTypeResponse = core.serialization.object({
+    edgeTypes: core.serialization.property("edge_types", core.serialization.list(EdgeType_1.EdgeType).optional()),
     entityTypes: core.serialization.property("entity_types", core.serialization.list(EntityType_1.EntityType).optional()),
 });

package/serialization/types/Episode.d.ts

@@ -4,6 +4,7 @@
 import * as serializers from "../index";
 import * as Zep from "../../api/index";
 import * as core from "../../core";
+import { RoleType } from "./RoleType";
 import { GraphDataType } from "./GraphDataType";
 export declare const Episode: core.serialization.ObjectSchema<serializers.Episode.Raw, Zep.Episode>;
 export declare namespace Episode {
@@ -11,6 +12,8 @@ export declare namespace Episode {
         content: string;
         created_at: string;
         processed?: boolean | null;
+        role?: string | null;
+        role_type?: RoleType.Raw | null;
         session_id?: string | null;
         source?: GraphDataType.Raw | null;
         source_description?: string | null;

package/serialization/types/Episode.js

@@ -38,11 +38,14 @@ var __importStar = (this && this.__importStar) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Episode = void 0;
 const core = __importStar(require("../../core"));
+const RoleType_1 = require("./RoleType");
 const GraphDataType_1 = require("./GraphDataType");
 exports.Episode = core.serialization.object({
     content: core.serialization.string(),
     createdAt: core.serialization.property("created_at", core.serialization.string()),
     processed: core.serialization.boolean().optional(),
+    role: core.serialization.string().optional(),
+    roleType: core.serialization.property("role_type", RoleType_1.RoleType.optional()),
     sessionId: core.serialization.property("session_id", core.serialization.string().optional()),
     source: GraphDataType_1.GraphDataType.optional(),
     sourceDescription: core.serialization.property("source_description", core.serialization.string().optional()),

package/serialization/types/EpisodeMentions.d.ts

@@ -0,0 +1,15 @@
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+import * as serializers from "../index";
+import * as Zep from "../../api/index";
+import * as core from "../../core";
+import { EntityEdge } from "./EntityEdge";
+import { EntityNode } from "./EntityNode";
+export declare const EpisodeMentions: core.serialization.ObjectSchema<serializers.EpisodeMentions.Raw, Zep.EpisodeMentions>;
+export declare namespace EpisodeMentions {
+    interface Raw {
+        edges?: EntityEdge.Raw[] | null;
+        nodes?: EntityNode.Raw[] | null;
+    }
+}

package/serialization/types/EpisodeMentions.js

@@ -0,0 +1,46 @@
+"use strict";
+/**
+ * This file was auto-generated by Fern from our API Definition.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EpisodeMentions = void 0;
+const core = __importStar(require("../../core"));
+const EntityEdge_1 = require("./EntityEdge");
+const EntityNode_1 = require("./EntityNode");
+exports.EpisodeMentions = core.serialization.object({
+    edges: core.serialization.list(EntityEdge_1.EntityEdge).optional(),
+    nodes: core.serialization.list(EntityNode_1.EntityNode).optional(),
+});

package/serialization/types/GraphSearchResults.d.ts

@@ -5,11 +5,13 @@ import * as serializers from "../index";
 import * as Zep from "../../api/index";
 import * as core from "../../core";
 import { EntityEdge } from "./EntityEdge";
+import { Episode } from "./Episode";
 import { EntityNode } from "./EntityNode";
 export declare const GraphSearchResults: core.serialization.ObjectSchema<serializers.GraphSearchResults.Raw, Zep.GraphSearchResults>;
 export declare namespace GraphSearchResults {
     interface Raw {
         edges?: EntityEdge.Raw[] | null;
+        episodes?: Episode.Raw[] | null;
         nodes?: EntityNode.Raw[] | null;
     }
 }

package/serialization/types/GraphSearchResults.js

@@ -39,8 +39,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.GraphSearchResults = void 0;
 const core = __importStar(require("../../core"));
 const EntityEdge_1 = require("./EntityEdge");
+const Episode_1 = require("./Episode");
 const EntityNode_1 = require("./EntityNode");
 exports.GraphSearchResults = core.serialization.object({
     edges: core.serialization.list(EntityEdge_1.EntityEdge).optional(),
+    episodes: core.serialization.list(Episode_1.Episode).optional(),
     nodes: core.serialization.list(EntityNode_1.EntityNode).optional(),
 });

package/serialization/types/GraphSearchScope.d.ts

@@ -6,5 +6,5 @@ import * as Zep from "../../api/index";
 import * as core from "../../core";
 export declare const GraphSearchScope: core.serialization.Schema<serializers.GraphSearchScope.Raw, Zep.GraphSearchScope>;
 export declare namespace GraphSearchScope {
-    type Raw = "edges" | "nodes";
+    type Raw = "edges" | "nodes" | "episodes";
 }

package/serialization/types/GraphSearchScope.js

@@ -38,4 +38,4 @@ var __importStar = (this && this.__importStar) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GraphSearchScope = void 0;
 const core = __importStar(require("../../core"));
-exports.GraphSearchScope = core.serialization.enum_(["edges", "nodes"]);
+exports.GraphSearchScope = core.serialization.enum_(["edges", "nodes", "episodes"]);

package/serialization/types/SearchFilters.d.ts

@@ -7,6 +7,7 @@ import * as core from "../../core";
 export declare const SearchFilters: core.serialization.ObjectSchema<serializers.SearchFilters.Raw, Zep.SearchFilters>;
 export declare namespace SearchFilters {
     interface Raw {
+        edge_types?: string[] | null;
         node_labels?: string[] | null;
     }
 }

package/serialization/types/SearchFilters.js

@@ -39,5 +39,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.SearchFilters = void 0;
 const core = __importStar(require("../../core"));
 exports.SearchFilters = core.serialization.object({
+    edgeTypes: core.serialization.property("edge_types", core.serialization.list(core.serialization.string()).optional()),
     nodeLabels: core.serialization.property("node_labels", core.serialization.list(core.serialization.string()).optional()),
 });

package/serialization/types/index.d.ts

@@ -4,12 +4,15 @@ export * from "./ApidataDocument";
 export * from "./ApidataDocumentCollection";
 export * from "./ApidataDocumentSearchResponse";
 export * from "./ApidataDocumentWithScore";
+export * from "./EdgeType";
 export * from "./EndSessionResponse";
 export * from "./EndSessionsResponse";
+export * from "./EntityEdgeSourceTarget";
 export * from "./EntityProperty";
 export * from "./EntityType";
 export * from "./EntityTypeResponse";
 export * from "./EpisodeData";
+export * from "./EpisodeMentions";
 export * from "./Fact";
 export * from "./FactRatingInstruction";
 export * from "./FactResponse";
@@ -18,6 +21,7 @@ export * from "./GraphEdgesRequest";
 export * from "./Episode";
 export * from "./EpisodeResponse";
 export * from "./GraphNodesRequest";
+export * from "./GraphSearchResults";
 export * from "./Group";
 export * from "./GroupListResponse";
 export * from "./Memory";
@@ -41,7 +45,6 @@ export * from "./UserNodeResponse";
 export * from "./AddTripleResponse";
 export * from "./EntityEdge";
 export * from "./EntityNode";
-export * from "./GraphSearchResults";
 export * from "./GraphSearchScope";
 export * from "./Reranker";
 export * from "./SearchFilters";

package/serialization/types/index.js

@@ -20,12 +20,15 @@ __exportStar(require("./ApidataDocument"), exports);
 __exportStar(require("./ApidataDocumentCollection"), exports);
 __exportStar(require("./ApidataDocumentSearchResponse"), exports);
 __exportStar(require("./ApidataDocumentWithScore"), exports);
+__exportStar(require("./EdgeType"), exports);
 __exportStar(require("./EndSessionResponse"), exports);
 __exportStar(require("./EndSessionsResponse"), exports);
+__exportStar(require("./EntityEdgeSourceTarget"), exports);
 __exportStar(require("./EntityProperty"), exports);
 __exportStar(require("./EntityType"), exports);
 __exportStar(require("./EntityTypeResponse"), exports);
 __exportStar(require("./EpisodeData"), exports);
+__exportStar(require("./EpisodeMentions"), exports);
 __exportStar(require("./Fact"), exports);
 __exportStar(require("./FactRatingInstruction"), exports);
 __exportStar(require("./FactResponse"), exports);
@@ -34,6 +37,7 @@ __exportStar(require("./GraphEdgesRequest"), exports);
 __exportStar(require("./Episode"), exports);
 __exportStar(require("./EpisodeResponse"), exports);
 __exportStar(require("./GraphNodesRequest"), exports);
+__exportStar(require("./GraphSearchResults"), exports);
 __exportStar(require("./Group"), exports);
 __exportStar(require("./GroupListResponse"), exports);
 __exportStar(require("./Memory"), exports);
@@ -57,7 +61,6 @@ __exportStar(require("./UserNodeResponse"), exports);
 __exportStar(require("./AddTripleResponse"), exports);
 __exportStar(require("./EntityEdge"), exports);
 __exportStar(require("./EntityNode"), exports);
-__exportStar(require("./GraphSearchResults"), exports);
 __exportStar(require("./GraphSearchScope"), exports);
 __exportStar(require("./Reranker"), exports);
 __exportStar(require("./SearchFilters"), exports);

package/version.d.ts

@@ -1 +1 @@
-export declare const SDK_VERSION = "2.12.3";
+export declare const SDK_VERSION = "2.14.0";

package/version.js

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

package/wrapper/graph.d.ts

@@ -1,6 +1,44 @@
 import { Graph as BaseGraph } from "../api/resources/graph/client/Client";
 import { Zep } from "../index";
-import { EntityType } from "./ontology";
+import { EdgeType, EntityType } from "./ontology";
 export declare class Graph extends BaseGraph {
-    setEntityTypes(entityTypes: Record<string, EntityType>, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
+    /**
+     * Sets the entity and edge types for a project, replacing any existing ones.
+     *
+     * @param {Record<string, EntityType>} entityTypes
+     * @param {Record<string, EdgeType>} edgeTypes
+     * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link Zep.BadRequestError}
+     * @throws {@link Zep.InternalServerError}
+     *
+     * @example
+     *     const travelDestinationSchema: EntityType = {
+     *         description: "A travel destination entity",
+     *         fields: {
+     *             destination_name: entityFields.text("The name of travel destination"),
+     *         },
+     *     };
+     *
+     *     const isTravelingTo: EdgeType = {
+     *         description: "An edge representing a traveler going to a destination.",
+     *         fields: {
+     *             travel_date: entityFields.text("The date of the travel"),
+     *             purpose: entityFields.text("The purpose of the travel"),
+     *         },
+     *         sourceTargets: [
+     *             {
+     *                 source: "User",
+     *                 target: "TravelDestination",
+     *             }
+     *         ]
+     *     }
+     *
+     *     await client.graph.setEntityTypes({
+     *         TravelDestination: travelDestinationSchema,
+     *     }, {
+     *         IS_TRAVELING_TO: isTravelingTo,
+     *     });
+     */
+    setEntityTypes(entityTypes: Record<string, EntityType>, edgeTypes: Record<string, EdgeType>, requestOptions?: BaseGraph.RequestOptions): Promise<Zep.SuccessResponse>;
 }

package/wrapper/graph.js

@@ -13,15 +13,57 @@ exports.Graph = void 0;
 const Client_1 = require("../api/resources/graph/client/Client");
 const ontology_1 = require("./ontology");
 class Graph extends Client_1.Graph {
-    setEntityTypes(entityTypes, requestOptions) {
+    /**
+     * Sets the entity and edge types for a project, replacing any existing ones.
+     *
+     * @param {Record<string, EntityType>} entityTypes
+     * @param {Record<string, EdgeType>} edgeTypes
+     * @param {Graph.RequestOptions} requestOptions - Request-specific configuration.
+     *
+     * @throws {@link Zep.BadRequestError}
+     * @throws {@link Zep.InternalServerError}
+     *
+     * @example
+     *     const travelDestinationSchema: EntityType = {
+     *         description: "A travel destination entity",
+     *         fields: {
+     *             destination_name: entityFields.text("The name of travel destination"),
+     *         },
+     *     };
+     *
+     *     const isTravelingTo: EdgeType = {
+     *         description: "An edge representing a traveler going to a destination.",
+     *         fields: {
+     *             travel_date: entityFields.text("The date of the travel"),
+     *             purpose: entityFields.text("The purpose of the travel"),
+     *         },
+     *         sourceTargets: [
+     *             {
+     *                 source: "User",
+     *                 target: "TravelDestination",
+     *             }
+     *         ]
+     *     }
+     *
+     *     await client.graph.setEntityTypes({
+     *         TravelDestination: travelDestinationSchema,
+     *     }, {
+     *         IS_TRAVELING_TO: isTravelingTo,
+     *     });
+     */
+    setEntityTypes(entityTypes, edgeTypes, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
             const validatedEntityTypes = Object.keys(entityTypes).map((key) => {
                 const schema = entityTypes[key];
-                const entityType = (0, ontology_1.entityModelToAPISchema)(schema, key);
-                return entityType;
+                return (0, ontology_1.entityModelToAPISchema)(schema, key);
+            });
+            const validatedEdgeTypes = Object.keys(edgeTypes).map((key) => {
+                const schema = edgeTypes[key];
+                return (0, ontology_1.edgeModelToAPISchema)(schema, key);
             });
             return this.setEntityTypesInternal({
                 entityTypes: validatedEntityTypes,
+                edgeTypes: validatedEdgeTypes,
             }, requestOptions);
         });
     }