import {z} from 'zod' export type AnalysisGroupStatus = 'active' | 'deleted'; export type AnalysisGroupReadinessStatus = 'pending' | 'failed' | 'ready'; export type Interval = 'day' | 'week'; export type Industry = 'accommodation' | 'cpg' | 'tourism' | 'marketing' | 'parking' | 'restaurants' | 'retail' | 'transportation' | 'other'; export type AnalysisGroupSortField = 'name' | 'created' | 'updated' | '-name' | '-created' | '-updated'; export type GetApiV1BeamAnalysisGroupsArgs = {'updated.gt'?: string | undefined, 'updated.gte'?: string | undefined, 'updated.lt'?: string | undefined, 'updated.lte'?: string | undefined, q?: string | undefined, status?: Array | undefined, readiness_status?: Array | undefined, 'demand_type.interval'?: Array | undefined, 'demand_type.industry'?: Array | undefined, sort?: Array | undefined, offset?: number | undefined, limit?: number | undefined}; export type GetApiV1BeamAnalysisGroupsOptions = {baseUrl?: string | undefined, BearerAuth?: string | undefined, fetch?: typeof fetch | undefined}; export const analysisExcludedReason = z.enum(["analysis_not_found", "analysis_deleted", "analysis_failed", "analysis_not_ready", "analysis_demand_type_inconsistent", "analysis_feature_importance_incomplete", "analysis_value_quant_incomplete"]); export const excludedFromReason = z.enum(["feature_importance", "value_quant"]); export const excludedAnalysis = z.object({analysis_id: z.string(), reason: analysisExcludedReason, excluded_from: z.array(excludedFromReason).optional()}); export const analysisGroupProcessingCompleted = z.object({feature_importance: z.boolean().optional(), value_quant: z.boolean().optional(), excluded_analyses: z.array(excludedAnalysis).optional()}); export const analysisGroupReadinessStatus = z.enum(["pending", "failed", "ready"]); export const industry = z.enum(["accommodation", "cpg", "tourism", "marketing", "parking", "restaurants", "retail", "transportation", "other"]); export const interval = z.enum(["day", "week"]); export const weekday = z.enum(["monday", "tuesday", "wednesday", "thursday", "friday", "saturday", "sunday"]); export const currencyCode = z.enum(["USD", "EUR", "GBP", "AUD", "CAD"]); export const demandTypeResponse = z.object({industry: industry.optional(), interval: interval.optional(), week_start_day: weekday.optional(), unit_descriptor: z.string().optional(), currency_code: currencyCode.optional(), unit_currency_multiplier: z.number().optional()}); export const analysisGroupStatus = z.enum(["active", "deleted"]); export const analysisGroupResponse = z.object({group_id: z.string().optional(), name: z.string().min(1), analysis_ids: z.array(z.string()).optional(), user_id: z.string().optional(), processing_completed: analysisGroupProcessingCompleted.optional(), readiness_status: analysisGroupReadinessStatus.optional(), demand_type: demandTypeResponse.optional(), status: analysisGroupStatus.optional(), create_dt: z.string().optional(), update_dt: z.string().optional(), processed_dt: z.string().optional()}); export const multiAnalysisGroupResponse = z.object({count: z.number().int(), groups: z.array(analysisGroupResponse)}); export const validateResponse = async ( schema: { readonly '~standard': { readonly validate: (value: unknown) => ({ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray | undefined }> }) | Promise<{ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray | undefined }> }> } }, value: unknown ): Promise => { 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 }; /** * Search Analysis Groups * * GET /v1/beam/analysis-groups */ export const getApiV1BeamAnalysisGroups = async (args: GetApiV1BeamAnalysisGroupsArgs, options?: GetApiV1BeamAnalysisGroupsOptions) => { const {baseUrl = "https://api.predicthq.com", fetch: fetchFn = fetch, BearerAuth} = options ?? {} const query = new URLSearchParams() const headers = new Headers() headers.set('accept', "application/json") if (args['updated.gt'] != null) query.append("updated.gt", String(args['updated.gt'])) if (args['updated.gte'] != null) query.append("updated.gte", String(args['updated.gte'])) if (args['updated.lt'] != null) query.append("updated.lt", String(args['updated.lt'])) if (args['updated.lte'] != null) query.append("updated.lte", String(args['updated.lte'])) if (args.q != null) query.append("q", String(args.q)) if (args.status != null) query.append("status", args.status.join(',')) if (args.readiness_status != null) query.append("readiness_status", args.readiness_status.join(',')) if (args['demand_type.interval'] != null) query.append("demand_type.interval", args['demand_type.interval'].join(',')) if (args['demand_type.industry'] != null) query.append("demand_type.industry", args['demand_type.industry'].join(',')) if (args.sort != null) query.append("sort", args.sort.join(',')) if (args.offset != null) query.append("offset", String(args.offset)) if (args.limit != null) query.append("limit", String(args.limit)) if (BearerAuth !== undefined) headers.set('authorization', `Bearer ${BearerAuth}`) const url = `${baseUrl.replace(/\/+$/, '')}/v1/beam/analysis-groups${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(multiAnalysisGroupResponse, await res.json()) };