@posthog/wizard/dist/src/utils/query.js

The PostHog wizard helps you to configure your project

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.query = void 0;
const axios_1 = __importDefault(require("axios"));
const zod_to_json_schema_1 = require("zod-to-json-schema");
const urls_1 = require("./urls");
const analytics_1 = require("./analytics");
const axios_2 = require("axios");
const debug_1 = require("./debug");
const query = async ({ message, model = 'o4-mini', region, schema, wizardHash, }) => {
    const fullSchema = (0, zod_to_json_schema_1.zodToJsonSchema)(schema, 'schema');
    const jsonSchema = fullSchema.definitions;
    (0, debug_1.debug)('Full schema:', JSON.stringify(fullSchema, null, 2));
    (0, debug_1.debug)('Query request:', {
        url: `${(0, urls_1.getCloudUrlFromRegion)(region)}/api/wizard/query`,
        wizardHash,
        message: message.substring(0, 100) + '...',
        json_schema: { ...jsonSchema, name: 'schema', strict: true },
    });
    const response = await axios_1.default
        .post(`${(0, urls_1.getCloudUrlFromRegion)(region)}/api/wizard/query`, {
        message,
        model,
        json_schema: { ...jsonSchema, name: 'schema', strict: true },
    }, {
        headers: {
            'X-PostHog-Wizard-Hash': wizardHash,
            ...(process.env.RECORD_FIXTURES === 'true'
                ? { 'X-PostHog-Wizard-Fixture-Generation': true }
                : {}),
        },
    })
        .catch((error) => {
        (0, debug_1.debug)('Query error:', error);
        if (error instanceof axios_2.AxiosError) {
            analytics_1.analytics.captureException(error, {
                response_status_code: error.response?.status,
                message,
                model,
                json_schema: jsonSchema,
                type: 'wizard_query_error',
            });
        }
        throw error;
    });
    (0, debug_1.debug)('Query response:', {
        status: response.status,
        data: response.data,
    });
    const validation = schema.safeParse(response.data.data);
    if (!validation.success) {
        (0, debug_1.debug)('Validation error:', validation.error);
        throw new Error(`Invalid response from wizard: ${validation.error.message}`);
    }
    if (process.env.NODE_ENV === 'test') {
        const { fixtureTracker } = await import('../../e2e-tests/mocks/fixture-tracker.js');
        const requestBody = JSON.stringify({
            message,
            model,
            json_schema: { ...jsonSchema, name: 'schema', strict: true },
        });
        fixtureTracker.saveQueryFixture(requestBody, validation.data);
    }
    return validation.data;
};
exports.query = query;
//# sourceMappingURL=query.js.map