opik 0.0.1

package/src/opik/rest_api/core/schemas/builders/record/record.ts

@@ -0,0 +1,130 @@
+import { MaybeValid, Schema, SchemaType, ValidationError } from "../../Schema";
+import { entries } from "../../utils/entries";
+import { getErrorMessageForIncorrectType } from "../../utils/getErrorMessageForIncorrectType";
+import { isPlainObject } from "../../utils/isPlainObject";
+import { maybeSkipValidation } from "../../utils/maybeSkipValidation";
+import { getSchemaUtils } from "../schema-utils";
+import { BaseRecordSchema, RecordSchema } from "./types";
+
+export function record<RawKey extends string | number, RawValue, ParsedValue, ParsedKey extends string | number>(
+    keySchema: Schema<RawKey, ParsedKey>,
+    valueSchema: Schema<RawValue, ParsedValue>
+): RecordSchema<RawKey, RawValue, ParsedKey, ParsedValue> {
+    const baseSchema: BaseRecordSchema<RawKey, RawValue, ParsedKey, ParsedValue> = {
+        parse: (raw, opts) => {
+            return validateAndTransformRecord({
+                value: raw,
+                isKeyNumeric: keySchema.getType() === SchemaType.NUMBER,
+                transformKey: (key) =>
+                    keySchema.parse(key, {
+                        ...opts,
+                        breadcrumbsPrefix: [...(opts?.breadcrumbsPrefix ?? []), `${key} (key)`],
+                    }),
+                transformValue: (value, key) =>
+                    valueSchema.parse(value, {
+                        ...opts,
+                        breadcrumbsPrefix: [...(opts?.breadcrumbsPrefix ?? []), `${key}`],
+                    }),
+                breadcrumbsPrefix: opts?.breadcrumbsPrefix,
+            });
+        },
+        json: (parsed, opts) => {
+            return validateAndTransformRecord({
+                value: parsed,
+                isKeyNumeric: keySchema.getType() === SchemaType.NUMBER,
+                transformKey: (key) =>
+                    keySchema.json(key, {
+                        ...opts,
+                        breadcrumbsPrefix: [...(opts?.breadcrumbsPrefix ?? []), `${key} (key)`],
+                    }),
+                transformValue: (value, key) =>
+                    valueSchema.json(value, {
+                        ...opts,
+                        breadcrumbsPrefix: [...(opts?.breadcrumbsPrefix ?? []), `${key}`],
+                    }),
+                breadcrumbsPrefix: opts?.breadcrumbsPrefix,
+            });
+        },
+        getType: () => SchemaType.RECORD,
+    };
+
+    return {
+        ...maybeSkipValidation(baseSchema),
+        ...getSchemaUtils(baseSchema),
+    };
+}
+
+function validateAndTransformRecord<TransformedKey extends string | number, TransformedValue>({
+    value,
+    isKeyNumeric,
+    transformKey,
+    transformValue,
+    breadcrumbsPrefix = [],
+}: {
+    value: unknown;
+    isKeyNumeric: boolean;
+    transformKey: (key: string | number) => MaybeValid<TransformedKey>;
+    transformValue: (value: unknown, key: string | number) => MaybeValid<TransformedValue>;
+    breadcrumbsPrefix: string[] | undefined;
+}): MaybeValid<Record<TransformedKey, TransformedValue>> {
+    if (!isPlainObject(value)) {
+        return {
+            ok: false,
+            errors: [
+                {
+                    path: breadcrumbsPrefix,
+                    message: getErrorMessageForIncorrectType(value, "object"),
+                },
+            ],
+        };
+    }
+
+    return entries(value).reduce<MaybeValid<Record<TransformedKey, TransformedValue>>>(
+        (accPromise, [stringKey, value]) => {
+            // skip nullish keys
+            if (value == null) {
+                return accPromise;
+            }
+
+            const acc = accPromise;
+
+            let key: string | number = stringKey;
+            if (isKeyNumeric) {
+                const numberKey = stringKey.length > 0 ? Number(stringKey) : NaN;
+                if (!isNaN(numberKey)) {
+                    key = numberKey;
+                }
+            }
+            const transformedKey = transformKey(key);
+
+            const transformedValue = transformValue(value, key);
+
+            if (acc.ok && transformedKey.ok && transformedValue.ok) {
+                return {
+                    ok: true,
+                    value: {
+                        ...acc.value,
+                        [transformedKey.value]: transformedValue.value,
+                    },
+                };
+            }
+
+            const errors: ValidationError[] = [];
+            if (!acc.ok) {
+                errors.push(...acc.errors);
+            }
+            if (!transformedKey.ok) {
+                errors.push(...transformedKey.errors);
+            }
+            if (!transformedValue.ok) {
+                errors.push(...transformedValue.errors);
+            }
+
+            return {
+                ok: false,
+                errors,
+            };
+        },
+        { ok: true, value: {} as Record<TransformedKey, TransformedValue> }
+    );
+}

package/src/opik/rest_api/core/schemas/builders/record/types.ts

@@ -0,0 +1,17 @@
+import { BaseSchema } from "../../Schema";
+import { SchemaUtils } from "../schema-utils";
+
+export type RecordSchema<
+    RawKey extends string | number,
+    RawValue,
+    ParsedKey extends string | number,
+    ParsedValue
+> = BaseRecordSchema<RawKey, RawValue, ParsedKey, ParsedValue> &
+    SchemaUtils<Record<RawKey, RawValue>, Record<ParsedKey, ParsedValue>>;
+
+export type BaseRecordSchema<
+    RawKey extends string | number,
+    RawValue,
+    ParsedKey extends string | number,
+    ParsedValue
+> = BaseSchema<Record<RawKey, RawValue>, Record<ParsedKey, ParsedValue>>;

package/src/opik/rest_api/core/schemas/builders/schema-utils/JsonError.ts

@@ -0,0 +1,9 @@
+import { ValidationError } from "../../Schema";
+import { stringifyValidationError } from "./stringifyValidationErrors";
+
+export class JsonError extends Error {
+    constructor(public readonly errors: ValidationError[]) {
+        super(errors.map(stringifyValidationError).join("; "));
+        Object.setPrototypeOf(this, JsonError.prototype);
+    }
+}

package/src/opik/rest_api/core/schemas/builders/schema-utils/ParseError.ts

@@ -0,0 +1,9 @@
+import { ValidationError } from "../../Schema";
+import { stringifyValidationError } from "./stringifyValidationErrors";
+
+export class ParseError extends Error {
+    constructor(public readonly errors: ValidationError[]) {
+        super(errors.map(stringifyValidationError).join("; "));
+        Object.setPrototypeOf(this, ParseError.prototype);
+    }
+}

package/src/opik/rest_api/core/schemas/builders/schema-utils/getSchemaUtils.ts

@@ -0,0 +1,105 @@
+import { BaseSchema, Schema, SchemaOptions, SchemaType } from "../../Schema";
+import { JsonError } from "./JsonError";
+import { ParseError } from "./ParseError";
+
+export interface SchemaUtils<Raw, Parsed> {
+    optional: () => Schema<Raw | null | undefined, Parsed | undefined>;
+    transform: <Transformed>(transformer: SchemaTransformer<Parsed, Transformed>) => Schema<Raw, Transformed>;
+    parseOrThrow: (raw: unknown, opts?: SchemaOptions) => Parsed;
+    jsonOrThrow: (raw: unknown, opts?: SchemaOptions) => Raw;
+}
+
+export interface SchemaTransformer<Parsed, Transformed> {
+    transform: (parsed: Parsed) => Transformed;
+    untransform: (transformed: any) => Parsed;
+}
+
+export function getSchemaUtils<Raw, Parsed>(schema: BaseSchema<Raw, Parsed>): SchemaUtils<Raw, Parsed> {
+    return {
+        optional: () => optional(schema),
+        transform: (transformer) => transform(schema, transformer),
+        parseOrThrow: (raw, opts) => {
+            const parsed = schema.parse(raw, opts);
+            if (parsed.ok) {
+                return parsed.value;
+            }
+            throw new ParseError(parsed.errors);
+        },
+        jsonOrThrow: (parsed, opts) => {
+            const raw = schema.json(parsed, opts);
+            if (raw.ok) {
+                return raw.value;
+            }
+            throw new JsonError(raw.errors);
+        },
+    };
+}
+
+/**
+ * schema utils are defined in one file to resolve issues with circular imports
+ */
+
+export function optional<Raw, Parsed>(
+    schema: BaseSchema<Raw, Parsed>
+): Schema<Raw | null | undefined, Parsed | undefined> {
+    const baseSchema: BaseSchema<Raw | null | undefined, Parsed | undefined> = {
+        parse: (raw, opts) => {
+            if (raw == null) {
+                return {
+                    ok: true,
+                    value: undefined,
+                };
+            }
+            return schema.parse(raw, opts);
+        },
+        json: (parsed, opts) => {
+            if (opts?.omitUndefined && parsed === undefined) {
+                return {
+                    ok: true,
+                    value: undefined,
+                };
+            }
+            if (parsed == null) {
+                return {
+                    ok: true,
+                    value: null,
+                };
+            }
+            return schema.json(parsed, opts);
+        },
+        getType: () => SchemaType.OPTIONAL,
+    };
+
+    return {
+        ...baseSchema,
+        ...getSchemaUtils(baseSchema),
+    };
+}
+
+export function transform<Raw, Parsed, Transformed>(
+    schema: BaseSchema<Raw, Parsed>,
+    transformer: SchemaTransformer<Parsed, Transformed>
+): Schema<Raw, Transformed> {
+    const baseSchema: BaseSchema<Raw, Transformed> = {
+        parse: (raw, opts) => {
+            const parsed = schema.parse(raw, opts);
+            if (!parsed.ok) {
+                return parsed;
+            }
+            return {
+                ok: true,
+                value: transformer.transform(parsed.value),
+            };
+        },
+        json: (transformed, opts) => {
+            const parsed = transformer.untransform(transformed);
+            return schema.json(parsed, opts);
+        },
+        getType: () => schema.getType(),
+    };
+
+    return {
+        ...baseSchema,
+        ...getSchemaUtils(baseSchema),
+    };
+}

package/src/opik/rest_api/core/schemas/builders/schema-utils/index.ts

@@ -0,0 +1,4 @@
+export { getSchemaUtils, optional, transform } from "./getSchemaUtils";
+export type { SchemaUtils } from "./getSchemaUtils";
+export { JsonError } from "./JsonError";
+export { ParseError } from "./ParseError";

package/src/opik/rest_api/core/schemas/builders/schema-utils/stringifyValidationErrors.ts

@@ -0,0 +1,8 @@
+import { ValidationError } from "../../Schema";
+
+export function stringifyValidationError(error: ValidationError): string {
+    if (error.path.length === 0) {
+        return error.message;
+    }
+    return `${error.path.join(" -> ")}: ${error.message}`;
+}

package/src/opik/rest_api/core/schemas/builders/set/index.ts

@@ -0,0 +1 @@
+export { set } from "./set";

package/src/opik/rest_api/core/schemas/builders/set/set.ts

@@ -0,0 +1,43 @@
+import { BaseSchema, Schema, SchemaType } from "../../Schema";
+import { getErrorMessageForIncorrectType } from "../../utils/getErrorMessageForIncorrectType";
+import { maybeSkipValidation } from "../../utils/maybeSkipValidation";
+import { list } from "../list";
+import { getSchemaUtils } from "../schema-utils";
+
+export function set<Raw, Parsed>(schema: Schema<Raw, Parsed>): Schema<Raw[], Set<Parsed>> {
+    const listSchema = list(schema);
+    const baseSchema: BaseSchema<Raw[], Set<Parsed>> = {
+        parse: (raw, opts) => {
+            const parsedList = listSchema.parse(raw, opts);
+            if (parsedList.ok) {
+                return {
+                    ok: true,
+                    value: new Set(parsedList.value),
+                };
+            } else {
+                return parsedList;
+            }
+        },
+        json: (parsed, opts) => {
+            if (!(parsed instanceof Set)) {
+                return {
+                    ok: false,
+                    errors: [
+                        {
+                            path: opts?.breadcrumbsPrefix ?? [],
+                            message: getErrorMessageForIncorrectType(parsed, "Set"),
+                        },
+                    ],
+                };
+            }
+            const jsonList = listSchema.json([...parsed], opts);
+            return jsonList;
+        },
+        getType: () => SchemaType.SET,
+    };
+
+    return {
+        ...maybeSkipValidation(baseSchema),
+        ...getSchemaUtils(baseSchema),
+    };
+}

package/src/opik/rest_api/core/schemas/builders/undiscriminated-union/index.ts

@@ -0,0 +1,6 @@
+export type {
+    inferParsedUnidiscriminatedUnionSchema,
+    inferRawUnidiscriminatedUnionSchema,
+    UndiscriminatedUnionSchema,
+} from "./types";
+export { undiscriminatedUnion } from "./undiscriminatedUnion";

package/src/opik/rest_api/core/schemas/builders/undiscriminated-union/types.ts

@@ -0,0 +1,10 @@
+import { inferParsed, inferRaw, Schema } from "../../Schema";
+
+export type UndiscriminatedUnionSchema<Schemas extends [...Schema[]]> = Schema<
+    inferRawUnidiscriminatedUnionSchema<Schemas>,
+    inferParsedUnidiscriminatedUnionSchema<Schemas>
+>;
+
+export type inferRawUnidiscriminatedUnionSchema<Schemas extends [...Schema[]]> = inferRaw<Schemas[number]>;
+
+export type inferParsedUnidiscriminatedUnionSchema<Schemas extends [...Schema[]]> = inferParsed<Schemas[number]>;

package/src/opik/rest_api/core/schemas/builders/undiscriminated-union/undiscriminatedUnion.ts

@@ -0,0 +1,60 @@
+import { BaseSchema, MaybeValid, Schema, SchemaOptions, SchemaType, ValidationError } from "../../Schema";
+import { maybeSkipValidation } from "../../utils/maybeSkipValidation";
+import { getSchemaUtils } from "../schema-utils";
+import { inferParsedUnidiscriminatedUnionSchema, inferRawUnidiscriminatedUnionSchema } from "./types";
+
+export function undiscriminatedUnion<Schemas extends [Schema<any, any>, ...Schema<any, any>[]]>(
+    schemas: Schemas
+): Schema<inferRawUnidiscriminatedUnionSchema<Schemas>, inferParsedUnidiscriminatedUnionSchema<Schemas>> {
+    const baseSchema: BaseSchema<
+        inferRawUnidiscriminatedUnionSchema<Schemas>,
+        inferParsedUnidiscriminatedUnionSchema<Schemas>
+    > = {
+        parse: (raw, opts) => {
+            return validateAndTransformUndiscriminatedUnion<inferParsedUnidiscriminatedUnionSchema<Schemas>>(
+                (schema, opts) => schema.parse(raw, opts),
+                schemas,
+                opts
+            );
+        },
+        json: (parsed, opts) => {
+            return validateAndTransformUndiscriminatedUnion<inferRawUnidiscriminatedUnionSchema<Schemas>>(
+                (schema, opts) => schema.json(parsed, opts),
+                schemas,
+                opts
+            );
+        },
+        getType: () => SchemaType.UNDISCRIMINATED_UNION,
+    };
+
+    return {
+        ...maybeSkipValidation(baseSchema),
+        ...getSchemaUtils(baseSchema),
+    };
+}
+
+function validateAndTransformUndiscriminatedUnion<Transformed>(
+    transform: (schema: Schema<any, any>, opts: SchemaOptions) => MaybeValid<Transformed>,
+    schemas: Schema<any, any>[],
+    opts: SchemaOptions | undefined
+): MaybeValid<Transformed> {
+    const errors: ValidationError[] = [];
+    for (const [index, schema] of schemas.entries()) {
+        const transformed = transform(schema, { ...opts, skipValidation: false });
+        if (transformed.ok) {
+            return transformed;
+        } else {
+            for (const error of transformed.errors) {
+                errors.push({
+                    path: error.path,
+                    message: `[Variant ${index}] ${error.message}`,
+                });
+            }
+        }
+    }
+
+    return {
+        ok: false,
+        errors,
+    };
+}

package/src/opik/rest_api/core/schemas/builders/union/discriminant.ts

@@ -0,0 +1,14 @@
+export function discriminant<RawDiscriminant extends string, ParsedDiscriminant extends string>(
+    parsedDiscriminant: ParsedDiscriminant,
+    rawDiscriminant: RawDiscriminant
+): Discriminant<RawDiscriminant, ParsedDiscriminant> {
+    return {
+        parsedDiscriminant,
+        rawDiscriminant,
+    };
+}
+
+export interface Discriminant<RawDiscriminant extends string, ParsedDiscriminant extends string> {
+    parsedDiscriminant: ParsedDiscriminant;
+    rawDiscriminant: RawDiscriminant;
+}

package/src/opik/rest_api/core/schemas/builders/union/index.ts

@@ -0,0 +1,10 @@
+export { discriminant } from "./discriminant";
+export type { Discriminant } from "./discriminant";
+export type {
+    inferParsedDiscriminant,
+    inferParsedUnion,
+    inferRawDiscriminant,
+    inferRawUnion,
+    UnionSubtypes,
+} from "./types";
+export { union } from "./union";

package/src/opik/rest_api/core/schemas/builders/union/types.ts

@@ -0,0 +1,26 @@
+import { inferParsedObject, inferRawObject, ObjectSchema } from "../object";
+import { Discriminant } from "./discriminant";
+
+export type UnionSubtypes<DiscriminantValues extends string | number | symbol> = {
+    [K in DiscriminantValues]: ObjectSchema<any, any>;
+};
+
+export type inferRawUnion<D extends string | Discriminant<any, any>, U extends UnionSubtypes<keyof U>> = {
+    [K in keyof U]: Record<inferRawDiscriminant<D>, K> & inferRawObject<U[K]>;
+}[keyof U];
+
+export type inferParsedUnion<D extends string | Discriminant<any, any>, U extends UnionSubtypes<keyof U>> = {
+    [K in keyof U]: Record<inferParsedDiscriminant<D>, K> & inferParsedObject<U[K]>;
+}[keyof U];
+
+export type inferRawDiscriminant<D extends string | Discriminant<any, any>> = D extends string
+    ? D
+    : D extends Discriminant<infer Raw, any>
+    ? Raw
+    : never;
+
+export type inferParsedDiscriminant<D extends string | Discriminant<any, any>> = D extends string
+    ? D
+    : D extends Discriminant<any, infer Parsed>
+    ? Parsed
+    : never;

package/src/opik/rest_api/core/schemas/builders/union/union.ts

@@ -0,0 +1,170 @@
+import { BaseSchema, MaybeValid, SchemaType } from "../../Schema";
+import { getErrorMessageForIncorrectType } from "../../utils/getErrorMessageForIncorrectType";
+import { isPlainObject } from "../../utils/isPlainObject";
+import { keys } from "../../utils/keys";
+import { maybeSkipValidation } from "../../utils/maybeSkipValidation";
+import { enum_ } from "../enum";
+import { ObjectSchema } from "../object";
+import { getObjectLikeUtils, ObjectLikeSchema } from "../object-like";
+import { getSchemaUtils } from "../schema-utils";
+import { Discriminant } from "./discriminant";
+import { inferParsedDiscriminant, inferParsedUnion, inferRawDiscriminant, inferRawUnion, UnionSubtypes } from "./types";
+
+export function union<D extends string | Discriminant<any, any>, U extends UnionSubtypes<any>>(
+    discriminant: D,
+    union: U
+): ObjectLikeSchema<inferRawUnion<D, U>, inferParsedUnion<D, U>> {
+    const rawDiscriminant =
+        typeof discriminant === "string" ? discriminant : (discriminant.rawDiscriminant as inferRawDiscriminant<D>);
+    const parsedDiscriminant =
+        typeof discriminant === "string"
+            ? discriminant
+            : (discriminant.parsedDiscriminant as inferParsedDiscriminant<D>);
+
+    const discriminantValueSchema = enum_(keys(union) as string[]);
+
+    const baseSchema: BaseSchema<inferRawUnion<D, U>, inferParsedUnion<D, U>> = {
+        parse: (raw, opts) => {
+            return transformAndValidateUnion({
+                value: raw,
+                discriminant: rawDiscriminant,
+                transformedDiscriminant: parsedDiscriminant,
+                transformDiscriminantValue: (discriminantValue) =>
+                    discriminantValueSchema.parse(discriminantValue, {
+                        allowUnrecognizedEnumValues: opts?.allowUnrecognizedUnionMembers,
+                        breadcrumbsPrefix: [...(opts?.breadcrumbsPrefix ?? []), rawDiscriminant],
+                    }),
+                getAdditionalPropertiesSchema: (discriminantValue) => union[discriminantValue],
+                allowUnrecognizedUnionMembers: opts?.allowUnrecognizedUnionMembers,
+                transformAdditionalProperties: (additionalProperties, additionalPropertiesSchema) =>
+                    additionalPropertiesSchema.parse(additionalProperties, opts),
+                breadcrumbsPrefix: opts?.breadcrumbsPrefix,
+            });
+        },
+        json: (parsed, opts) => {
+            return transformAndValidateUnion({
+                value: parsed,
+                discriminant: parsedDiscriminant,
+                transformedDiscriminant: rawDiscriminant,
+                transformDiscriminantValue: (discriminantValue) =>
+                    discriminantValueSchema.json(discriminantValue, {
+                        allowUnrecognizedEnumValues: opts?.allowUnrecognizedUnionMembers,
+                        breadcrumbsPrefix: [...(opts?.breadcrumbsPrefix ?? []), parsedDiscriminant],
+                    }),
+                getAdditionalPropertiesSchema: (discriminantValue) => union[discriminantValue],
+                allowUnrecognizedUnionMembers: opts?.allowUnrecognizedUnionMembers,
+                transformAdditionalProperties: (additionalProperties, additionalPropertiesSchema) =>
+                    additionalPropertiesSchema.json(additionalProperties, opts),
+                breadcrumbsPrefix: opts?.breadcrumbsPrefix,
+            });
+        },
+        getType: () => SchemaType.UNION,
+    };
+
+    return {
+        ...maybeSkipValidation(baseSchema),
+        ...getSchemaUtils(baseSchema),
+        ...getObjectLikeUtils(baseSchema),
+    };
+}
+
+function transformAndValidateUnion<
+    TransformedDiscriminant extends string,
+    TransformedDiscriminantValue extends string,
+    TransformedAdditionalProperties
+>({
+    value,
+    discriminant,
+    transformedDiscriminant,
+    transformDiscriminantValue,
+    getAdditionalPropertiesSchema,
+    allowUnrecognizedUnionMembers = false,
+    transformAdditionalProperties,
+    breadcrumbsPrefix = [],
+}: {
+    value: unknown;
+    discriminant: string;
+    transformedDiscriminant: TransformedDiscriminant;
+    transformDiscriminantValue: (discriminantValue: unknown) => MaybeValid<TransformedDiscriminantValue>;
+    getAdditionalPropertiesSchema: (discriminantValue: string) => ObjectSchema<any, any> | undefined;
+    allowUnrecognizedUnionMembers: boolean | undefined;
+    transformAdditionalProperties: (
+        additionalProperties: unknown,
+        additionalPropertiesSchema: ObjectSchema<any, any>
+    ) => MaybeValid<TransformedAdditionalProperties>;
+    breadcrumbsPrefix: string[] | undefined;
+}): MaybeValid<Record<TransformedDiscriminant, TransformedDiscriminantValue> & TransformedAdditionalProperties> {
+    if (!isPlainObject(value)) {
+        return {
+            ok: false,
+            errors: [
+                {
+                    path: breadcrumbsPrefix,
+                    message: getErrorMessageForIncorrectType(value, "object"),
+                },
+            ],
+        };
+    }
+
+    const { [discriminant]: discriminantValue, ...additionalProperties } = value;
+
+    if (discriminantValue == null) {
+        return {
+            ok: false,
+            errors: [
+                {
+                    path: breadcrumbsPrefix,
+                    message: `Missing discriminant ("${discriminant}")`,
+                },
+            ],
+        };
+    }
+
+    const transformedDiscriminantValue = transformDiscriminantValue(discriminantValue);
+    if (!transformedDiscriminantValue.ok) {
+        return {
+            ok: false,
+            errors: transformedDiscriminantValue.errors,
+        };
+    }
+
+    const additionalPropertiesSchema = getAdditionalPropertiesSchema(transformedDiscriminantValue.value);
+
+    if (additionalPropertiesSchema == null) {
+        if (allowUnrecognizedUnionMembers) {
+            return {
+                ok: true,
+                value: {
+                    [transformedDiscriminant]: transformedDiscriminantValue.value,
+                    ...additionalProperties,
+                } as Record<TransformedDiscriminant, TransformedDiscriminantValue> & TransformedAdditionalProperties,
+            };
+        } else {
+            return {
+                ok: false,
+                errors: [
+                    {
+                        path: [...breadcrumbsPrefix, discriminant],
+                        message: "Unexpected discriminant value",
+                    },
+                ],
+            };
+        }
+    }
+
+    const transformedAdditionalProperties = transformAdditionalProperties(
+        additionalProperties,
+        additionalPropertiesSchema
+    );
+    if (!transformedAdditionalProperties.ok) {
+        return transformedAdditionalProperties;
+    }
+
+    return {
+        ok: true,
+        value: {
+            [transformedDiscriminant]: discriminantValue,
+            ...transformedAdditionalProperties.value,
+        } as Record<TransformedDiscriminant, TransformedDiscriminantValue> & TransformedAdditionalProperties,
+    };
+}

package/src/opik/rest_api/core/schemas/index.ts

@@ -0,0 +1,2 @@
+export * from "./builders";
+export type { inferParsed, inferRaw, Schema, SchemaOptions } from "./Schema";

package/src/opik/rest_api/core/schemas/utils/MaybePromise.ts

@@ -0,0 +1 @@
+export type MaybePromise<T> = T | Promise<T>;

package/src/opik/rest_api/core/schemas/utils/addQuestionMarksToNullableProperties.ts

@@ -0,0 +1,15 @@
+export type addQuestionMarksToNullableProperties<T> = {
+    [K in OptionalKeys<T>]?: T[K];
+} & Pick<T, RequiredKeys<T>>;
+
+export type OptionalKeys<T> = {
+    [K in keyof T]-?: undefined extends T[K]
+        ? K
+        : null extends T[K]
+        ? K
+        : 1 extends (any extends T[K] ? 0 : 1)
+        ? never
+        : K;
+}[keyof T];
+
+export type RequiredKeys<T> = Exclude<keyof T, OptionalKeys<T>>;