@getzep/zep-cloud 2.12.3 → 2.14.0

package/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/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/dist/version.d.ts

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

package/dist/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/dist/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/dist/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);
         });
     }

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,
+        })),
+    };
+}