/*
 * 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 { ClosedEnum } from "../../types/enums.js";
import { Result as SafeParseResult } from "../../types/fp.js";
import { SDKValidationError } from "../errors/sdkvalidationerror.js";
import {
  CodeInterpreterTool,
  CodeInterpreterTool$inboundSchema,
} from "./codeinterpretertool.js";
import {
  CompletionArgs,
  CompletionArgs$inboundSchema,
} from "./completionargs.js";
import {
  DocumentLibraryTool,
  DocumentLibraryTool$inboundSchema,
} from "./documentlibrarytool.js";
import { FunctionTool, FunctionTool$inboundSchema } from "./functiontool.js";
import {
  ImageGenerationTool,
  ImageGenerationTool$inboundSchema,
} from "./imagegenerationtool.js";
import {
  WebSearchPremiumTool,
  WebSearchPremiumTool$inboundSchema,
} from "./websearchpremiumtool.js";
import { WebSearchTool, WebSearchTool$inboundSchema } from "./websearchtool.js";

export type ModelConversationTools =
  | (CodeInterpreterTool & { type: "code_interpreter" })
  | (DocumentLibraryTool & { type: "document_library" })
  | (FunctionTool & { type: "function" })
  | (ImageGenerationTool & { type: "image_generation" })
  | (WebSearchTool & { type: "web_search" })
  | (WebSearchPremiumTool & { type: "web_search_premium" });

export const ModelConversationObject = {
  Conversation: "conversation",
} as const;
export type ModelConversationObject = ClosedEnum<
  typeof ModelConversationObject
>;

export type ModelConversation = {
  /**
   * Instruction prompt the model will follow during the conversation.
   */
  instructions?: string | null | undefined;
  /**
   * List of tools which are available to the model during the conversation.
   */
  tools?:
    | Array<
      | (CodeInterpreterTool & { type: "code_interpreter" })
      | (DocumentLibraryTool & { type: "document_library" })
      | (FunctionTool & { type: "function" })
      | (ImageGenerationTool & { type: "image_generation" })
      | (WebSearchTool & { type: "web_search" })
      | (WebSearchPremiumTool & { type: "web_search_premium" })
    >
    | undefined;
  /**
   * White-listed arguments from the completion API
   */
  completionArgs?: CompletionArgs | undefined;
  /**
   * Name given to the conversation.
   */
  name?: string | null | undefined;
  /**
   * Description of the what the conversation is about.
   */
  description?: string | null | undefined;
  /**
   * Custom metadata for the conversation.
   */
  metadata?: { [k: string]: any } | null | undefined;
  object: ModelConversationObject | undefined;
  id: string;
  createdAt: Date;
  updatedAt: Date;
  model: string;
};

/** @internal */
export const ModelConversationTools$inboundSchema: z.ZodType<
  ModelConversationTools,
  z.ZodTypeDef,
  unknown
> = z.union([
  CodeInterpreterTool$inboundSchema.and(
    z.object({ type: z.literal("code_interpreter") }),
  ),
  DocumentLibraryTool$inboundSchema.and(
    z.object({ type: z.literal("document_library") }),
  ),
  FunctionTool$inboundSchema.and(z.object({ type: z.literal("function") })),
  ImageGenerationTool$inboundSchema.and(
    z.object({ type: z.literal("image_generation") }),
  ),
  WebSearchTool$inboundSchema.and(z.object({ type: z.literal("web_search") })),
  WebSearchPremiumTool$inboundSchema.and(
    z.object({ type: z.literal("web_search_premium") }),
  ),
]);

export function modelConversationToolsFromJSON(
  jsonString: string,
): SafeParseResult<ModelConversationTools, SDKValidationError> {
  return safeParse(
    jsonString,
    (x) => ModelConversationTools$inboundSchema.parse(JSON.parse(x)),
    `Failed to parse 'ModelConversationTools' from JSON`,
  );
}

/** @internal */
export const ModelConversationObject$inboundSchema: z.ZodNativeEnum<
  typeof ModelConversationObject
> = z.nativeEnum(ModelConversationObject);

/** @internal */
export const ModelConversation$inboundSchema: z.ZodType<
  ModelConversation,
  z.ZodTypeDef,
  unknown
> = z.object({
  instructions: z.nullable(z.string()).optional(),
  tools: z.array(
    z.union([
      CodeInterpreterTool$inboundSchema.and(
        z.object({ type: z.literal("code_interpreter") }),
      ),
      DocumentLibraryTool$inboundSchema.and(
        z.object({ type: z.literal("document_library") }),
      ),
      FunctionTool$inboundSchema.and(z.object({ type: z.literal("function") })),
      ImageGenerationTool$inboundSchema.and(
        z.object({ type: z.literal("image_generation") }),
      ),
      WebSearchTool$inboundSchema.and(
        z.object({ type: z.literal("web_search") }),
      ),
      WebSearchPremiumTool$inboundSchema.and(
        z.object({ type: z.literal("web_search_premium") }),
      ),
    ]),
  ).optional(),
  completion_args: CompletionArgs$inboundSchema.optional(),
  name: z.nullable(z.string()).optional(),
  description: z.nullable(z.string()).optional(),
  metadata: z.nullable(z.record(z.any())).optional(),
  object: ModelConversationObject$inboundSchema.default("conversation"),
  id: z.string(),
  created_at: z.string().datetime({ offset: true }).transform(v => new Date(v)),
  updated_at: z.string().datetime({ offset: true }).transform(v => new Date(v)),
  model: z.string(),
}).transform((v) => {
  return remap$(v, {
    "completion_args": "completionArgs",
    "created_at": "createdAt",
    "updated_at": "updatedAt",
  });
});

export function modelConversationFromJSON(
  jsonString: string,
): SafeParseResult<ModelConversation, SDKValidationError> {
  return safeParse(
    jsonString,
    (x) => ModelConversation$inboundSchema.parse(JSON.parse(x)),
    `Failed to parse 'ModelConversation' from JSON`,
  );
}
