@getzep/zep-cloud 2.12.3 → 2.13.0

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