import {z} from 'zod'

export type UnversionedEntityRequestDtoExperimentTemplateIndexEntityExperimentTemplateAnnotationsExperimentTemplatePropertiesExtensibleObject = {unversionedEntityIds?: (Array<string> | null) | undefined};

export type CreateApiFlowV10FlowexperimenttemplatesGetIndexEntitiesArgs = {subscriptionId: string, resourceGroupName: string, workspaceName: string, body: UnversionedEntityRequestDtoExperimentTemplateIndexEntityExperimentTemplateAnnotationsExperimentTemplatePropertiesExtensibleObject};

export type CreateApiFlowV10FlowexperimenttemplatesGetIndexEntitiesOptions = {baseUrl: string, azureAuth?: string | undefined, fetch?: typeof fetch | undefined};

export const entityKind = z.enum(["Invalid", "LineageRoot", "Versioned", "Unversioned"]);

export const experimentNodeType = z.enum(["Flow", "ChatGroup", "Command"]);

export const sessionRuntimeResources = z.object({vmSize: z.string().nullable().optional(), maxIdleTimeSeconds: z.number().int().nullable().optional(), identity: z.string().nullable().optional(), computeName: z.string().nullable().optional(), enableMultiContainer: z.boolean().optional()});

export const chatGroupRole = z.object({name: z.string().nullable().optional(), role: z.string().nullable().optional(), stop_signal: z.string().nullable().optional(), path: z.string().nullable().optional(), variant: z.string().nullable().optional(), connections: z.record(z.string(), z.record(z.string(), z.string())).nullable().optional(), environment_variables: z.record(z.string(), z.string()).nullable().optional(), display_name: z.string().nullable().optional(), description: z.string().nullable().optional(), tags: z.record(z.string(), z.string()).nullable().optional(), properties: z.record(z.string(), z.string()).nullable().optional(), resources: sessionRuntimeResources.optional(), inputs: z.record(z.string(), z.unknown()).nullable().optional()});

export const experimentNodeOutput = z.object({type: z.string().nullable().optional(), mode: z.string().nullable().optional(), uri: z.string().nullable().optional(), name: z.string().nullable().optional(), version: z.string().nullable().optional()});

export const experimentNode = z.object({name: z.string().nullable().optional(), type: experimentNodeType.optional(), max_turns: z.number().int().nullable().optional(), roles: z.array(chatGroupRole).nullable().optional(), command: z.string().nullable().optional(), code: z.string().nullable().optional(), environment: z.string().nullable().optional(), outputs: z.record(z.string(), experimentNodeOutput).nullable().optional(), path: z.string().nullable().optional(), variant: z.string().nullable().optional(), connections: z.record(z.string(), z.record(z.string(), z.string())).nullable().optional(), environment_variables: z.record(z.string(), z.string()).nullable().optional(), display_name: z.string().nullable().optional(), description: z.string().nullable().optional(), tags: z.record(z.string(), z.string()).nullable().optional(), properties: z.record(z.string(), z.string()).nullable().optional(), resources: sessionRuntimeResources.optional(), inputs: z.record(z.string(), z.unknown()).nullable().optional()});

export const experimentData = z.object({name: z.string().nullable().optional(), path: z.string().nullable().optional()});

export const schemaContractsCreatedBy = z.object({userObjectId: z.string().nullable().optional(), userTenantId: z.string().nullable().optional(), userName: z.string().nullable().optional(), userPrincipalName: z.string().nullable().optional()});

export const experimentTemplateAnnotations = z.object({nodes: z.array(experimentNode).nullable().optional(), data: z.array(experimentData).nullable().optional(), createdDate: z.string().nullable().optional(), lastModifiedDate: z.string().nullable().optional(), owner: schemaContractsCreatedBy.optional(), name: z.string().nullable().optional(), description: z.string().nullable().optional(), archived: z.boolean().optional(), tags: z.record(z.string(), z.string().nullable()).nullable().optional()});

export const creationContext = z.object({createdTime: z.string().optional(), createdBy: schemaContractsCreatedBy.optional(), creationSource: z.string().nullable().optional()});

export const experimentTemplateProperties = z.object({id: z.string().nullable().optional(), creationContext: creationContext.optional()});

export const extensibleObject = z.object({});

export const relationship = z.object({relationType: z.string().nullable().optional(), targetEntityId: z.string().nullable().optional(), assetId: z.string().nullable().optional(), entityType: z.string().nullable().optional(), direction: z.string().nullable().optional(), entityContainerId: z.string().nullable().optional()});

export const entityUsage = z.object({totalCount: z.number().int().nullable().optional()});

export const experimentTemplateIndexEntity = z.object({schemaId: z.string().nullable().optional(), entityId: z.string().nullable().optional(), kind: entityKind.optional(), annotations: experimentTemplateAnnotations.optional(), properties: experimentTemplateProperties.optional(), internal: extensibleObject.optional(), updateSequence: z.number().int().optional(), type: z.string().nullable().optional(), version: z.string().nullable().optional(), entityContainerId: z.string().nullable().optional(), entityObjectId: z.string().nullable().optional(), resourceType: z.string().nullable().optional(), relationships: z.array(relationship).nullable().optional(), assetId: z.string().nullable().optional(), usage: entityUsage.optional(), isAFragment: z.boolean().optional(), fragmentId: z.string().nullable().optional()});

export const unversionedEntityResponseDtoExperimentTemplateIndexEntityExperimentTemplateAnnotationsExperimentTemplatePropertiesExtensibleObject = z.object({unversionedEntities: z.array(experimentTemplateIndexEntity).nullable().optional(), unversionedEntityJsonSchema: z.unknown(), normalizedRequestCharge: z.number().optional(), normalizedRequestChargePeriod: z.string().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
};

/**
 * POST /flow/v1.0/flowexperimenttemplates/getIndexEntities
 */
export const createApiFlowV10FlowexperimenttemplatesGetIndexEntities = async (args: CreateApiFlowV10FlowexperimenttemplatesGetIndexEntitiesArgs, options: CreateApiFlowV10FlowexperimenttemplatesGetIndexEntitiesOptions) => {
const {baseUrl, fetch: fetchFn = fetch, azureAuth} = options ?? {}

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

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

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

const res = await fetchFn(url, {method: "POST", 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(unversionedEntityResponseDtoExperimentTemplateIndexEntityExperimentTemplateAnnotationsExperimentTemplatePropertiesExtensibleObject, await res.json())
};