@arizeai/phoenix-client
Version:
A client for the Phoenix API
45 lines • 1.87 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.getDatasetExamples = getDatasetExamples;
const client_1 = require("../client");
const getDatasetInfoByName_1 = require("./getDatasetInfoByName");
const tiny_invariant_1 = __importDefault(require("tiny-invariant"));
/**
* Get examples from a dataset
* @param dataset - Dataset selector (ID, name, version ID, or splits)
* @returns Dataset examples
*/
async function getDatasetExamples({ client: _client, dataset: datasetSelector, }) {
var _a;
const client = _client || (0, client_1.createClient)();
let datasetId;
if ("datasetName" in datasetSelector) {
const datasetInfo = await (0, getDatasetInfoByName_1.getDatasetInfoByName)({
client,
datasetName: datasetSelector.datasetName,
});
datasetId = datasetInfo.id;
}
else {
datasetId = datasetSelector.datasetId;
}
const { versionId, splits } = datasetSelector;
const response = await client.GET("/v1/datasets/{id}/examples", {
params: {
path: {
id: datasetId,
},
query: Object.assign(Object.assign({}, (versionId ? { version_id: versionId } : {})), (splits ? { split: splits } : {})),
},
});
(0, tiny_invariant_1.default)((_a = response.data) === null || _a === void 0 ? void 0 : _a.data, "Failed to get dataset examples");
const examplesData = response.data.data;
return {
versionId: examplesData.version_id,
examples: examplesData.examples.map((example) => (Object.assign(Object.assign({}, example), { updatedAt: new Date(example.updated_at) }))),
};
}
//# sourceMappingURL=getDatasetExamples.js.map