import {z} from 'zod'

export type GetApiV1LogsLogIdArgs = {log_id: string};

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

export const functionParameters = z.object({type: z.string().optional(), properties: z.record(z.string(), z.object({})), required: z.array(z.string()).nullable()});

export const toolFunctionDetail = z.object({name: z.string(), description: z.string().nullable().optional(), parameters: functionParameters});

export const toolFunctionOutput = z.object({type: z.literal("function").optional(), function: toolFunctionDetail});

export const toolChoiceFunction = z.object({name: z.string()});

export const toolChoice = z.object({type: z.enum(["function", "json_object"]).nullable().optional(), function: toolChoiceFunction.nullable().optional()});

export const requestResponseFormat = z.object({type: z.enum(["text", "json_object"])});

export const roleContentChatChoiceContentImageUrl = z.object({url: z.string()});

export const roleContentChatChoiceContent = z.object({type: z.enum(["text", "image_url"]), text: z.string().nullable().optional(), image_url: roleContentChatChoiceContentImageUrl.nullable().optional()});

export const functionCallArguments = z.object({arguments: z.string(), name: z.string()});

export const toolCall = z.object({function: functionCallArguments, id: z.string(), type: z.literal("function").optional()});

export const roleContentChatChoiceOutput = z.object({role: z.enum(["user", "assistant", "system"]), content: z.union([z.string(), z.array(roleContentChatChoiceContent)]).nullable(), tool_calls: z.array(toolCall).nullable().optional()});

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 pulzeFeatureFlags = z.object({citations: z.boolean().optional(), auto_tools: z.boolean().optional(), smart_learn: z.boolean().optional()});

export const pulzeCustomHeadersData = z.object({labels: z.record(z.string(), z.string()).nullable().optional(), headers: z.record(z.string(), z.string()).nullable().optional(), weights: lLMModelWeights.nullable().optional(), policies: lLMModelPolicies.nullable().optional(), feature_flags: pulzeFeatureFlags.optional()});

export const pulzeCompletionRequest = z.object({model: z.string().nullable().optional(), max_tokens: z.number().int().nullable().optional(), temperature: z.number().gte(0).lte(1).nullable().optional(), top_p: z.number().nullable().optional(), tools: z.array(toolFunctionOutput).nullable().optional(), tool_choice: z.union([z.enum(["none", "auto"]), toolChoice]).optional(), n: z.number().int().gte(1).nullable().optional(), stream: z.boolean().nullable().optional(), logprobs: z.number().int().gte(0).lte(5).nullable().optional(), stop: z.union([z.string(), z.array(z.string())]).nullable().optional(), presence_penalty: z.number().gte(-2).lte(2).nullable().optional(), frequency_penalty: z.number().gte(-2).lte(2).nullable().optional(), best_of: z.number().int().gte(1).nullable().optional(), logit_bias: z.object({}).nullable().optional(), response_format: requestResponseFormat.nullable().optional(), messages: z.array(roleContentChatChoiceOutput).nullable().optional(), assistant_id: z.string().nullable().optional(), assistant_version_id: z.string().nullable().optional(), conversation_id: z.string().nullable().optional(), parent_request_id: z.string().nullable().optional(), plugins: z.array(z.string()).optional(), images: z.array(z.string()).nullable().optional(), files: z.array(z.string()).nullable().optional(), custom_headers: pulzeCustomHeadersData.nullable().optional()});

export const responseCompletionChatChoice = z.object({index: z.number().int(), finish_reason: z.string().nullable().optional(), message: roleContentChatChoiceOutput});

export const responseCompletionChoice = z.object({index: z.number().int(), finish_reason: z.string().nullable().optional(), text: z.string(), logprobs: z.number().int().nullable().optional()});

export const modelParts = z.object({model: z.string(), provider: z.string().nullable().optional(), owner: z.string().nullable().optional(), namespace: z.string(), at: z.string().nullable().optional()});

export const pulzeEngineTokens = z.object({total_tokens: z.number().optional(), prompt_tokens: z.number().optional(), completion_tokens: z.number().optional()});

export const pulzeGeneratedArtifacts = z.object({items: z.array(z.object({})).optional()});

export const pulzeSearchResults = z.object({items: z.array(z.object({})).optional()});

export const rankedScoringModels = z.object({best_models: z.array(z.record(z.string(), z.number())).optional(), candidates: z.array(z.record(z.string(), z.number())).nullable().optional()});

export const pulzeEngineResponseMetadata = z.object({app_id: z.string().optional(), model: modelParts.nullable().optional(), costs: pulzeEngineTokens.optional(), cost_savings: pulzeEngineTokens.optional(), generated_artifacts: pulzeGeneratedArtifacts.optional(), search_results: pulzeSearchResults.optional(), latency: z.number().nullable().optional(), labels: z.record(z.string(), z.string()).nullable().optional(), error: z.string().nullable().optional(), scores: rankedScoringModels.nullable().optional(), score: z.number().nullable().optional(), temperature: z.union([z.number(), z.number().int()]).nullable().optional(), max_tokens: z.number().int().gte(-1).nullable().optional(), status_code: z.number().int().nullable().optional(), retries: z.number().int().optional(), extra: z.object({}).nullable().optional(), warning: z.string().nullable().optional()});

export const pulzeEngineResponse = z.object({choices: z.union([z.array(responseCompletionChatChoice), z.array(responseCompletionChoice)]).optional(), created: z.number().int().optional(), metadata: pulzeEngineResponseMetadata.optional(), id: z.string().optional(), usage: pulzeEngineTokens.optional(), model: z.string(), object: z.enum(["text_completion", "chat.completion"])}).and(z.record(z.string(), z.unknown()));

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 appCollection = z.object({name: z.string(), description: z.string().nullable().optional(), id: z.string(), request_count: z.number().int().nullable().optional()});

export const appCollectionRequest = z.object({name: z.string(), collection: appCollection});

export const logRequest: z.ZodType<LogRequest> = z.object({conversation_id: z.string().nullable(), parent_request_id: z.string().nullable(), good_answer: z.boolean().nullable().optional(), feedback: z.string().nullable().optional(), prompt_tokens: z.number().int().optional(), completion_tokens: z.number().int().optional(), total_tokens: z.number().int().optional(), prompt_tokens_cost: z.number().optional(), completion_tokens_cost: z.number().optional(), total_tokens_cost: z.number().optional(), prompt_tokens_cost_savings: z.number().optional(), completion_tokens_cost_savings: z.number().optional(), total_tokens_cost_savings: z.number().optional(), app_id: z.string().nullable(), auth0_id: z.string().nullable(), costs_incurred: z.boolean().optional(), namespace: z.string().nullable().optional(), payload: pulzeCompletionRequest.optional(), privacy_level: z.union([z.literal(1), z.literal(2), z.literal(3)]).optional(), prompt: z.string().nullable().optional(), request_type: z.enum(["completions", "chat_completions"]).nullable().optional(), response_text: z.string().optional(), response: pulzeEngineResponse, status_code: z.number().int().optional(), is_test: z.boolean().optional(), added_on: z.string().nullable().optional(), latency: z.number().optional(), timestamp: z.number().int(), parent_id: z.string().nullable().optional(), id: z.string(), is_saved_for_later: z.boolean().optional(), comment_count: z.number().int().optional(), parent: z.lazy(() => logRequest.nullable().optional()), children: z.array(z.lazy(() => logRequest)).nullable().optional(), model: modelData.nullable().optional(), collections: z.array(appCollectionRequest).nullable().optional()});

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
};

/**
 * Get Log By Id
 * 
 * GET /v1/logs/{log_id}
 */
export const getApiV1LogsLogId = async (args: GetApiV1LogsLogIdArgs, options: GetApiV1LogsLogIdOptions) => {
const {baseUrl, fetch: fetchFn = fetch, HTTPBearer} = options ?? {}

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

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

const url = `${baseUrl.replace(/\/+$/, '')}/v1/logs/${encodeURIComponent(String(args.log_id))}${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "GET", headers})

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

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