import {z} from 'zod'

export type LLMModelWeights = {cost?: number | undefined, latency?: number | undefined, quality?: number | undefined};

export type LLMModelPolicies = {privacy_level?: (1 | 2 | 3) | undefined, max_cost?: number | undefined, max_switch_model_retries?: number | undefined, max_same_model_retries?: number | undefined, optimize_internal_requests?: (0 | 1) | undefined, prompt_id?: (string | null) | undefined};

export type LLMModelSettings = {assistant_id?: (string | null) | undefined, assistant_version_id?: (string | null) | undefined, instructions?: (string | null) | undefined, max_tokens?: number | undefined, temperature?: number | undefined};

export type AppUpdate = {id: string, sandbox_of?: (string | null) | undefined, use_parent_custom_data?: (boolean | null) | undefined, benchmark_model_id?: (string | null) | undefined, name?: (string | null) | undefined, description?: (string | null) | undefined, app_logo?: (string | null) | undefined, weights?: (LLMModelWeights | null) | undefined, policies?: (LLMModelPolicies | null) | undefined, has_failover_chain?: (boolean | null) | undefined, settings?: (LLMModelSettings | null) | undefined};

export type UpdateApiV1AppsUpdateArgs = {body: AppUpdate};

export type UpdateApiV1AppsUpdateOptions = {baseUrl: string, HTTPBearer?: string | undefined, fetch?: typeof fetch | undefined};

export const lLMModelWeights = z.object({cost: z.number().optional(), latency: z.number().optional(), quality: z.number().optional()});

export const lLMModelPolicies = z.object({privacy_level: z.union([z.literal(1), z.literal(2), z.literal(3)]).optional(), max_cost: z.number().gte(0.0001).optional(), max_switch_model_retries: z.number().int().gte(0).lte(5).optional(), max_same_model_retries: z.number().int().gte(0).lte(3).optional(), optimize_internal_requests: z.union([z.literal(0), z.literal(1)]).optional(), prompt_id: z.string().nullable().optional()});

export const lLMModelSettings = z.object({assistant_id: z.string().nullable().optional(), assistant_version_id: z.string().nullable().optional(), instructions: z.string().nullable().optional(), max_tokens: z.number().int().optional(), temperature: z.number().gte(0).lte(1).optional()});

export const modelData: z.ZodType<ModelData> = z.object({supports_functions: z.boolean(), supports_json: z.boolean(), supports_n: z.boolean(), supports_penalties: z.boolean(), supports_stream: z.boolean(), supports_vision: z.boolean(), completion_token_cost: z.number(), prompt_token_cost: z.number(), base_cost: z.number(), price_unit: z.enum(["tokens", "characters"]), model: z.string(), provider: z.string().nullable().optional(), owner: z.string().nullable().optional(), namespace: z.string(), at: z.string().nullable().optional(), context_window: z.number().int(), until: z.string().nullable(), description: z.string(), url: z.string(), api_target: z.string().optional(), deprecated_on: z.string().nullable().optional(), parent_id: z.string().nullable().optional(), parent: z.lazy(() => modelData.nullable().optional()), prompt_id: z.string().nullable().optional(), is_rag: z.boolean(), is_ft: z.boolean(), is_open_source: z.boolean(), is_gdpr: z.boolean(), is_chat: z.boolean(), id: z.string().nullable(), app_id: z.string().nullable(), org_id: z.string().nullable(), added_by: z.string().nullable(), added_on: z.string().nullable(), modified_on: z.string().nullable(), is_public: z.boolean(), is_test_model: z.boolean(), is_pulze_owner: z.boolean(), default_active: z.boolean()});

export const prompt = z.object({prompt: z.string(), title: z.string().min(1).max(60), description: z.string().min(1).max(200), added_on: z.string(), modified_on: z.string(), org_id: z.string(), tokens: z.number().int(), published_on: z.string().nullable(), reviewed_on: z.string().nullable(), approved_on: z.string().nullable(), decline_reason: z.string().nullable().optional(), id: z.string()});

export const app = z.object({auth0_id: z.string(), modified_by: z.string(), org_id: z.string().optional(), prompt_id: z.string().nullable().optional(), hashed_key: z.string(), key_end: z.string(), name: z.string(), description: z.string().nullable().optional(), app_logo: z.string().nullable().optional(), is_active: z.boolean(), rate_limit: z.number().int().optional(), has_failover_chain: z.boolean(), weights: lLMModelWeights.optional(), policies: lLMModelPolicies.optional(), settings: lLMModelSettings.optional(), sandbox_of: z.string().nullable().optional(), use_parent_custom_data: z.boolean().nullable().optional(), added_on: z.string().optional(), modified_on: z.string().optional(), benchmark_model_id: z.string(), id: z.string(), benchmark_model: modelData, prompt: prompt.nullable()});

export const pulzeModelActiveForApp = z.object({model: z.string(), provider: z.string().nullable().optional(), owner: z.string().nullable().optional(), namespace: z.string(), at: z.string().nullable().optional(), context_window: z.number().int(), until: z.string().nullable(), description: z.string(), url: z.string(), api_target: z.string().optional(), deprecated_on: z.string().nullable().optional(), parent_id: z.string().nullable().optional(), parent: modelData.nullable().optional(), prompt_id: z.string().nullable().optional(), id: z.string().nullable(), active: z.boolean().optional(), globally_disabled: z.boolean().nullable().optional()});

export const appSettingsData = z.object({app: app, active_models: z.number().int(), base_model_settings: z.array(pulzeModelActiveForApp), custom_model_settings: z.array(pulzeModelActiveForApp), failover_models: z.array(modelData), current_user_permissions: z.array(z.string()).nullable().optional()});

export const appSettings = z.object({main: appSettingsData, sandbox: appSettingsData.nullable(), sandbox_status: z.literal("outdated").nullable()});

export const validateResponse = async <Output>(
  schema: {
    readonly '~standard': {
      readonly validate: (value: unknown) => ({ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }) | Promise<{ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }>
    }
  },
  value: unknown
): Promise<Output> => {
  const result = await schema['~standard'].validate(value)

  if (result.issues !== undefined) {
    const detail = result.issues
      .map(({ message, path }) =>
        path?.length
          ? `${path.map((segment) => String(typeof segment === 'object' ? segment.key : segment)).join('.')}: ${message}`
          : message
      )
      .join('; ')

    throw new Error(`Response does not match the schema: ${detail}`, { cause: { issues: result.issues } })
  }

  return result.value
};

/**
 * Update Space And Models
 * 
 * PUT /v1/apps/update
 */
export const updateApiV1AppsUpdate = async (args: UpdateApiV1AppsUpdateArgs, options: UpdateApiV1AppsUpdateOptions) => {
const {baseUrl, fetch: fetchFn = fetch, HTTPBearer} = options ?? {}

const query = new URLSearchParams()
const headers = new Headers()
headers.set('content-type', "application/json")
headers.set('accept', "application/json")

if (HTTPBearer !== undefined) headers.set('authorization', `Bearer ${HTTPBearer}`)

const url = `${baseUrl.replace(/\/+$/, '')}/v1/apps/update${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "PUT", headers, body: JSON.stringify(args.body)})

if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}})

return validateResponse(appSettings, await res.json())
};