@getzep/zep-cloud 2.12.2 → 2.13.0

package/dist/wrapper/ontology.d.ts

@@ -297,12 +297,135 @@ export declare const EntityTypeSchema: z.ZodObject<{
     }>;
     description?: string | undefined;
 }>;
+export declare const EdgeTypeSchema: z.ZodObject<z.objectUtil.extendShape<{
+    description: z.ZodDefault<z.ZodString>;
+    fields: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<z.objectUtil.extendShape<{
+        type: z.ZodNativeEnum<typeof EntityPropertyType>;
+        description: z.ZodString;
+    }, {
+        type: z.ZodLiteral<EntityPropertyType.Text>;
+        value: z.ZodOptional<z.ZodString>;
+    }>, "strip", z.ZodTypeAny, {
+        type: EntityPropertyType.Text;
+        description: string;
+        value?: string | undefined;
+    }, {
+        type: EntityPropertyType.Text;
+        description: string;
+        value?: string | undefined;
+    }>, z.ZodObject<z.objectUtil.extendShape<{
+        type: z.ZodNativeEnum<typeof EntityPropertyType>;
+        description: z.ZodString;
+    }, {
+        type: z.ZodLiteral<EntityPropertyType.Int>;
+        value: z.ZodOptional<z.ZodNumber>;
+    }>, "strip", z.ZodTypeAny, {
+        type: EntityPropertyType.Int;
+        description: string;
+        value?: number | undefined;
+    }, {
+        type: EntityPropertyType.Int;
+        description: string;
+        value?: number | undefined;
+    }>, z.ZodObject<z.objectUtil.extendShape<{
+        type: z.ZodNativeEnum<typeof EntityPropertyType>;
+        description: z.ZodString;
+    }, {
+        type: z.ZodLiteral<EntityPropertyType.Float>;
+        value: z.ZodOptional<z.ZodNumber>;
+    }>, "strip", z.ZodTypeAny, {
+        type: EntityPropertyType.Float;
+        description: string;
+        value?: number | undefined;
+    }, {
+        type: EntityPropertyType.Float;
+        description: string;
+        value?: number | undefined;
+    }>, z.ZodObject<z.objectUtil.extendShape<{
+        type: z.ZodNativeEnum<typeof EntityPropertyType>;
+        description: z.ZodString;
+    }, {
+        type: z.ZodLiteral<EntityPropertyType.Boolean>;
+        value: z.ZodOptional<z.ZodBoolean>;
+    }>, "strip", z.ZodTypeAny, {
+        type: EntityPropertyType.Boolean;
+        description: string;
+        value?: boolean | undefined;
+    }, {
+        type: EntityPropertyType.Boolean;
+        description: string;
+        value?: boolean | undefined;
+    }>]>>;
+}, {
+    sourceTargets: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        source: z.ZodOptional<z.ZodString>;
+        target: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        source?: string | undefined;
+        target?: string | undefined;
+    }, {
+        source?: string | undefined;
+        target?: string | undefined;
+    }>, "many">>;
+}>, "strip", z.ZodTypeAny, {
+    description: string;
+    fields: Record<string, {
+        type: EntityPropertyType.Text;
+        description: string;
+        value?: string | undefined;
+    } | {
+        type: EntityPropertyType.Int;
+        description: string;
+        value?: number | undefined;
+    } | {
+        type: EntityPropertyType.Float;
+        description: string;
+        value?: number | undefined;
+    } | {
+        type: EntityPropertyType.Boolean;
+        description: string;
+        value?: boolean | undefined;
+    }>;
+    sourceTargets?: {
+        source?: string | undefined;
+        target?: string | undefined;
+    }[] | undefined;
+}, {
+    fields: Record<string, {
+        type: EntityPropertyType.Text;
+        description: string;
+        value?: string | undefined;
+    } | {
+        type: EntityPropertyType.Int;
+        description: string;
+        value?: number | undefined;
+    } | {
+        type: EntityPropertyType.Float;
+        description: string;
+        value?: number | undefined;
+    } | {
+        type: EntityPropertyType.Boolean;
+        description: string;
+        value?: boolean | undefined;
+    }>;
+    description?: string | undefined;
+    sourceTargets?: {
+        source?: string | undefined;
+        target?: string | undefined;
+    }[] | undefined;
+}>;
 export type EntityType = z.infer<typeof EntityTypeSchema>;
+export type EdgeType = z.infer<typeof EdgeTypeSchema>;
 export type EntityData<T extends EntityType> = {
     [P in keyof T["fields"]]: T["fields"][P] extends {
         value?: infer V;
     } ? V : never;
 };
+export type EdgeData<T extends EdgeType> = {
+    [P in keyof T["fields"]]: T["fields"][P] extends {
+        value?: infer V;
+    } ? V : never;
+};
 export declare const entityFields: {
     text: (description: string) => EntityTextField;
     integer: (description: string) => EntityIntField;
@@ -310,3 +433,4 @@ export declare const entityFields: {
     boolean: (description: string) => EntityBooleanField;
 };
 export declare function entityModelToAPISchema(entityType: EntityType, name: string): Zep.EntityType;
+export declare function edgeModelToAPISchema(entityType: EdgeType, name: string): Zep.EdgeType;

package/dist/wrapper/ontology.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.entityFields = exports.EntityTypeSchema = exports.EntityFields = exports.EntityTypeFieldSchema = exports.EntityTypeBooleanFieldSchema = exports.EntityTypeFloatFieldSchema = exports.EntityTypeIntFieldSchema = exports.EntityTypeTextFieldSchema = exports.EntityTypeBaseField = exports.EntityPropertyType = void 0;
+exports.entityFields = exports.EdgeTypeSchema = exports.EntityTypeSchema = exports.EntityFields = exports.EntityTypeFieldSchema = exports.EntityTypeBooleanFieldSchema = exports.EntityTypeFloatFieldSchema = exports.EntityTypeIntFieldSchema = exports.EntityTypeTextFieldSchema = exports.EntityTypeBaseField = exports.EntityPropertyType = void 0;
 exports.entityModelToAPISchema = entityModelToAPISchema;
+exports.edgeModelToAPISchema = edgeModelToAPISchema;
 const zod_1 = require("zod");
 var EntityPropertyType;
 (function (EntityPropertyType) {
@@ -41,6 +42,13 @@ exports.EntityTypeSchema = zod_1.z.object({
     description: zod_1.z.string().default(""),
     fields: zod_1.z.record(exports.EntityTypeFieldSchema),
 });
+const EdgeSourceTarget = zod_1.z.object({
+    source: zod_1.z.string().optional(),
+    target: zod_1.z.string().optional(),
+});
+exports.EdgeTypeSchema = exports.EntityTypeSchema.extend({
+    sourceTargets: zod_1.z.array(EdgeSourceTarget).optional(),
+});
 exports.entityFields = {
     text: (description) => {
         return exports.EntityTypeTextFieldSchema.parse({
@@ -78,3 +86,15 @@ function entityModelToAPISchema(entityType, name) {
         })),
     };
 }
+function edgeModelToAPISchema(entityType, name) {
+    return {
+        name,
+        description: entityType.description,
+        sourceTargets: entityType.sourceTargets,
+        properties: Object.entries(entityType.fields).map(([fieldName, fieldDef]) => ({
+            name: fieldName,
+            type: fieldDef.type,
+            description: fieldDef.description,
+        })),
+    };
+}

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.2",
+    "version": "2.13.0",
     "private": false,
     "repository": "https://github.com/getzep/zep-js",
     "description": "Zep: Fast, scalable building blocks for production LLM apps",

package/reference.md

@@ -1041,7 +1041,7 @@ await client.graph.setEntityTypesInternal();
 <dl>
 <dd>
 
-Add data to the graph. Note: each subscription tier has different limits on the amount of data that can be added to the graph please refer to the pricing page for more information.
+Add data to the graph.
 
 </dd>
 </dl>
@@ -1107,7 +1107,7 @@ await client.graph.add({
 <dl>
 <dd>
 
-Add data to the graph in batch mode (each episode processed concurrently). Note: each subscription tier has different limits on the amount of data that can be added to the graph please refer to the pricing page for more information.
+Add data to the graph in batch mode, processing episodes concurrently. Use only for data that is insensitive to processing order.
 
 </dd>
 </dl>

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/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,8 +4,10 @@ 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";

package/serialization/types/index.js

@@ -20,8 +20,10 @@ __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);

package/version.d.ts

@@ -1 +1 @@
-export declare const SDK_VERSION = "2.12.2";
+export declare const SDK_VERSION = "2.13.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.2";
+exports.SDK_VERSION = "2.13.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);
         });
     }