/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 */

import * as z from "zod/v3";
import { remap as remap$ } from "../../lib/primitives.js";
import { safeParse } from "../../lib/schemas.js";
import { Result as SafeParseResult } from "../../types/fp.js";
import { SDKValidationError } from "../errors/sdkvalidationerror.js";
import {
  Link,
  Link$inboundSchema,
  Link$Outbound,
  Link$outboundSchema,
} from "./link.js";

export type ScoredChunk = {
  text: string;
  score: number;
  id: string;
  index: number;
  metadata?: { [k: string]: any } | undefined;
  documentId: string;
  documentName: string;
  documentMetadata: { [k: string]: any };
  links: { [k: string]: Link };
};

/** @internal */
export const ScoredChunk$inboundSchema: z.ZodType<
  ScoredChunk,
  z.ZodTypeDef,
  unknown
> = z.object({
  text: z.string(),
  score: z.number(),
  id: z.string(),
  index: z.number().int(),
  metadata: z.record(z.any()).optional(),
  document_id: z.string(),
  document_name: z.string(),
  document_metadata: z.record(z.any()),
  links: z.record(Link$inboundSchema),
}).transform((v) => {
  return remap$(v, {
    "document_id": "documentId",
    "document_name": "documentName",
    "document_metadata": "documentMetadata",
  });
});
/** @internal */
export type ScoredChunk$Outbound = {
  text: string;
  score: number;
  id: string;
  index: number;
  metadata?: { [k: string]: any } | undefined;
  document_id: string;
  document_name: string;
  document_metadata: { [k: string]: any };
  links: { [k: string]: Link$Outbound };
};

/** @internal */
export const ScoredChunk$outboundSchema: z.ZodType<
  ScoredChunk$Outbound,
  z.ZodTypeDef,
  ScoredChunk
> = z.object({
  text: z.string(),
  score: z.number(),
  id: z.string(),
  index: z.number().int(),
  metadata: z.record(z.any()).optional(),
  documentId: z.string(),
  documentName: z.string(),
  documentMetadata: z.record(z.any()),
  links: z.record(Link$outboundSchema),
}).transform((v) => {
  return remap$(v, {
    documentId: "document_id",
    documentName: "document_name",
    documentMetadata: "document_metadata",
  });
});

export function scoredChunkToJSON(scoredChunk: ScoredChunk): string {
  return JSON.stringify(ScoredChunk$outboundSchema.parse(scoredChunk));
}
export function scoredChunkFromJSON(
  jsonString: string,
): SafeParseResult<ScoredChunk, SDKValidationError> {
  return safeParse(
    jsonString,
    (x) => ScoredChunk$inboundSchema.parse(JSON.parse(x)),
    `Failed to parse 'ScoredChunk' from JSON`,
  );
}