@arizeai/phoenix-client
Version:
A client for the Phoenix API
48 lines • 1.86 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 tiny_invariant_1 = __importDefault(require("tiny-invariant"));
const client_1 = require("../client");
const getDatasetInfoByName_1 = require("./getDatasetInfoByName");
/**
* Get examples from a dataset
* @param dataset - Dataset selector (ID, name, or version ID)
* @param versionId - Optional specific version ID (ignored if dataset selector is datasetVersionId)
*/
async function getDatasetExamples({ client: _client, dataset, versionId, }) {
var _a;
const client = _client || (0, client_1.createClient)();
let datasetId;
if ("datasetName" in dataset) {
const datasetInfo = await (0, getDatasetInfoByName_1.getDatasetInfoByName)({
client,
datasetName: dataset.datasetName,
});
datasetId = datasetInfo.id;
}
else {
datasetId = dataset.datasetId;
}
const response = await client.GET("/v1/datasets/{id}/examples", {
params: {
path: {
id: datasetId,
},
query: versionId
? {
version_id: versionId,
}
: undefined,
},
});
(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