@proofhound/web-ui 0.1.6

package/dist/hooks/connector.js

@@ -0,0 +1,157 @@
+import { connectorClient, } from '@proofhound/api-client';
+import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+const listKey = (projectId, query) => [
+    'connectors',
+    projectId,
+    query?.direction ?? 'all',
+    query?.type ?? 'all',
+    query?.healthStatus ?? 'all',
+    query?.search ?? '',
+];
+const detailKey = (projectId, connectorId) => ['connectors', projectId, connectorId];
+const referencesKey = (projectId, connectorId) => ['connector-refs', projectId, connectorId];
+const webhookTokensKey = (projectId, connectorId) => ['connector-webhook-tokens', projectId, connectorId];
+export function useConnectors(projectId, query) {
+    return useQuery({
+        queryKey: listKey(projectId, query),
+        queryFn: () => connectorClient.list(projectId, query),
+        enabled: projectId.length > 0,
+        placeholderData: (previousData) => previousData,
+    });
+}
+export function useConnector(projectId, connectorId) {
+    return useQuery({
+        queryKey: detailKey(projectId, connectorId),
+        queryFn: () => connectorClient.get(projectId, connectorId),
+        enabled: projectId.length > 0 && connectorId.length > 0,
+    });
+}
+export function useConnectorReferences(projectId, connectorId, enabled = true) {
+    return useQuery({
+        queryKey: referencesKey(projectId, connectorId),
+        queryFn: () => connectorClient.getReferences(projectId, connectorId),
+        enabled: enabled && projectId.length > 0 && connectorId.length > 0,
+    });
+}
+export function useCreateConnector(projectId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: (body) => connectorClient.create(projectId, body),
+        onSuccess: (created) => {
+            void qc.invalidateQueries({ queryKey: ['connectors', projectId], exact: false });
+            qc.setQueryData(detailKey(projectId, created.id), created);
+        },
+    });
+}
+export function useUpdateConnector(projectId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: ({ connectorId, body }) => connectorClient.update(projectId, connectorId, body),
+        onSuccess: (updated, { connectorId }) => {
+            void qc.invalidateQueries({ queryKey: ['connectors', projectId], exact: false });
+            qc.setQueryData(detailKey(projectId, connectorId), updated);
+        },
+    });
+}
+export function useDeleteConnector(projectId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: ({ connectorId, options }) => connectorClient.delete(projectId, connectorId, options),
+        onSuccess: (_data, { connectorId }) => {
+            void qc.invalidateQueries({ queryKey: ['connectors', projectId], exact: false });
+            void qc.removeQueries({ queryKey: detailKey(projectId, connectorId) });
+        },
+    });
+}
+export function useBulkDeleteConnectors(projectId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: (body) => connectorClient.bulkDelete(projectId, body),
+        onSuccess: () => void qc.invalidateQueries({ queryKey: ['connectors', projectId], exact: false }),
+    });
+}
+export function useProbeConnector(projectId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: (connectorId) => connectorClient.probe(projectId, connectorId),
+        onSuccess: (_data, connectorId) => {
+            void qc.invalidateQueries({ queryKey: ['connectors', projectId], exact: false });
+            void qc.invalidateQueries({ queryKey: detailKey(projectId, connectorId) });
+        },
+    });
+}
+export function usePeekConnector(projectId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: ({ connectorId, body }) => connectorClient.peek(projectId, connectorId, body ?? { limit: 5 }),
+        onSuccess: (data, { connectorId }) => {
+            qc.setQueryData(detailKey(projectId, connectorId), (current) => current ? applyPeekResultToConnector(current, data) : current);
+            qc.setQueriesData({ queryKey: ['connectors', projectId], exact: false }, (current) => current && Array.isArray(current.data)
+                ? {
+                    ...current,
+                    data: current.data.map((item) => item.id === connectorId ? applyPeekResultToConnector(item, data) : item),
+                }
+                : current);
+            void qc.invalidateQueries({ queryKey: ['connectors', projectId], exact: false });
+            void qc.invalidateQueries({ queryKey: detailKey(projectId, connectorId) });
+        },
+    });
+}
+// per-connector webhook tokens hooks
+// See packages/api-client/src/connector.ts + docs/specs/26-connectors.md
+export function useConnectorWebhookTokens(projectId, connectorId, enabled = true) {
+    return useQuery({
+        queryKey: webhookTokensKey(projectId, connectorId),
+        queryFn: () => connectorClient.listWebhookTokens(projectId, connectorId),
+        enabled: enabled && projectId.length > 0 && connectorId.length > 0,
+    });
+}
+export function useCreateConnectorWebhookToken(projectId, connectorId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: (body) => connectorClient.createWebhookToken(projectId, connectorId, body),
+        onSuccess: () => {
+            void qc.invalidateQueries({ queryKey: webhookTokensKey(projectId, connectorId) });
+            void qc.invalidateQueries({ queryKey: detailKey(projectId, connectorId) });
+        },
+    });
+}
+export function useRevokeConnectorWebhookToken(projectId, connectorId) {
+    const qc = useQueryClient();
+    return useMutation({
+        mutationFn: (tokenId) => connectorClient.revokeWebhookToken(projectId, connectorId, tokenId),
+        onSuccess: () => {
+            void qc.invalidateQueries({ queryKey: webhookTokensKey(projectId, connectorId) });
+            void qc.invalidateQueries({ queryKey: detailKey(projectId, connectorId) });
+        },
+    });
+}
+export function useRevealConnectorWebhookToken(projectId, connectorId) {
+    return useMutation({
+        mutationFn: (tokenId) => connectorClient.revealWebhookToken(projectId, connectorId, tokenId),
+    });
+}
+function applyPeekResultToConnector(connector, result) {
+    const healthStatus = result.error ? 'unhealthy' : 'healthy';
+    const base = {
+        ...connector,
+        healthStatus,
+        lastProbedAt: result.fetchedAt,
+        lastProbeError: result.error,
+    };
+    if (!('config' in connector)) {
+        return base;
+    }
+    const config = connector.config && typeof connector.config === 'object' ? connector.config : {};
+    return {
+        ...base,
+        config: {
+            ...config,
+            lastPeekPayloadSchema: result.payloadSchema,
+            lastPeekMessage: result.messages[0] ?? null,
+            lastPeekedAt: result.fetchedAt,
+            lastPeekMessageCount: result.messages.length,
+        },
+    };
+}
+//# sourceMappingURL=connector.js.map

package/dist/hooks/connector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connector.js","sourceRoot":"","sources":["../../src/hooks/connector.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,GAEhB,MAAM,wBAAwB,CAAC;AAYhC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE9E,MAAM,OAAO,GAAG,CAAC,SAAiB,EAAE,KAA6B,EAAE,EAAE,CACnE;IACE,YAAY;IACZ,SAAS;IACT,KAAK,EAAE,SAAS,IAAI,KAAK;IACzB,KAAK,EAAE,IAAI,IAAI,KAAK;IACpB,KAAK,EAAE,YAAY,IAAI,KAAK;IAC5B,KAAK,EAAE,MAAM,IAAI,EAAE;CACX,CAAC;AAEb,MAAM,SAAS,GAAG,CAAC,SAAiB,EAAE,WAAmB,EAAE,EAAE,CAC3D,CAAC,YAAY,EAAE,SAAS,EAAE,WAAW,CAAU,CAAC;AAElD,MAAM,aAAa,GAAG,CAAC,SAAiB,EAAE,WAAmB,EAAE,EAAE,CAC/D,CAAC,gBAAgB,EAAE,SAAS,EAAE,WAAW,CAAU,CAAC;AAEtD,MAAM,gBAAgB,GAAG,CAAC,SAAiB,EAAE,WAAmB,EAAE,EAAE,CAClE,CAAC,0BAA0B,EAAE,SAAS,EAAE,WAAW,CAAU,CAAC;AAEhE,MAAM,UAAU,aAAa,CAAC,SAAiB,EAAE,KAA6B;IAC5E,OAAO,QAAQ,CAAC;QACd,QAAQ,EAAE,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC;QACnC,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC;QACrD,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC;QAC7B,eAAe,EAAE,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY;KAChD,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,SAAiB,EAAE,WAAmB;IACjE,OAAO,QAAQ,CAAC;QACd,QAAQ,EAAE,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC;QAC3C,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,WAAW,CAAC;QAC1D,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC;KACxD,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,SAAiB,EAAE,WAAmB,EAAE,OAAO,GAAG,IAAI;IAC3F,OAAO,QAAQ,CAAC;QACd,QAAQ,EAAE,aAAa,CAAC,SAAS,EAAE,WAAW,CAAC;QAC/C,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,aAAa,CAAC,SAAS,EAAE,WAAW,CAAC;QACpE,OAAO,EAAE,OAAO,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC;KACnE,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAiB;IAClD,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,IAAwB,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC;QACjF,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;YACrB,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACjF,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;QAC7D,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAiB;IAClD,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,EAAqD,EAAE,EAAE,CACvF,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,EAAE,IAAI,CAAC;QACtD,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;YACtC,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACjF,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,CAAC;QAC9D,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAiB;IAClD,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,OAAO,EAA6D,EAAE,EAAE,CAClG,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,EAAE,OAAO,CAAC;QACzD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;YACpC,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACjF,KAAK,EAAE,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QACzE,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,SAAiB;IACvD,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,IAAoC,EAAE,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC,SAAS,EAAE,IAAI,CAAC;QACjG,SAAS,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;KAClG,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,SAAiB;IACjD,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,WAAmB,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,SAAS,EAAE,WAAW,CAAC;QAClF,SAAS,EAAE,CAAC,KAAK,EAAE,WAAW,EAAE,EAAE;YAChC,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACjF,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QAC7E,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,EAA2D,EAAE,EAAE,CAC7F,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,EAAE,IAAI,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACpE,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;YACnC,EAAE,CAAC,YAAY,CAAiC,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,CAC7F,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9D,CAAC;YACF,EAAE,CAAC,cAAc,CACf,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EACrD,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;gBACpC,CAAC,CAAC;oBACE,GAAG,OAAO;oBACV,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAC9B,IAAI,CAAC,EAAE,KAAK,WAAW,CAAC,CAAC,CAAC,0BAA0B,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CACxE;iBACF;gBACH,CAAC,CAAC,OAAO,CACd,CAAC;YACF,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACjF,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QAC7E,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,qCAAqC;AACrC,yEAAyE;AACzE,MAAM,UAAU,yBAAyB,CAAC,SAAiB,EAAE,WAAmB,EAAE,OAAO,GAAG,IAAI;IAC9F,OAAO,QAAQ,CAAC;QACd,QAAQ,EAAE,gBAAgB,CAAC,SAAS,EAAE,WAAW,CAAC;QAClD,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC,SAAS,EAAE,WAAW,CAAC;QACxE,OAAO,EAAE,OAAO,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC;KACnE,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAC,SAAiB,EAAE,WAAmB;IACnF,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,IAA2B,EAAE,EAAE,CAC1C,eAAe,CAAC,kBAAkB,CAAC,SAAS,EAAE,WAAW,EAAE,IAAI,CAAC;QAClE,SAAS,EAAE,GAAG,EAAE;YACd,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,gBAAgB,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;YAClF,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QAC7E,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAC,SAAiB,EAAE,WAAmB;IACnF,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;IAC5B,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,OAAe,EAAE,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,SAAS,EAAE,WAAW,EAAE,OAAO,CAAC;QACpG,SAAS,EAAE,GAAG,EAAE;YACd,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,gBAAgB,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;YAClF,KAAK,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;QAC7E,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAC,SAAiB,EAAE,WAAmB;IACnF,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,OAAe,EAAE,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,SAAS,EAAE,WAAW,EAAE,OAAO,CAAC;KACrG,CAAC,CAAC;AACL,CAAC;AAED,SAAS,0BAA0B,CACjC,SAAY,EACZ,MAAgC;IAEhC,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5D,MAAM,IAAI,GAAG;QACX,GAAG,SAAS;QACZ,YAAY;QACZ,YAAY,EAAE,MAAM,CAAC,SAAS;QAC9B,cAAc,EAAE,MAAM,CAAC,KAAK;KAC7B,CAAC;IAEF,IAAI,CAAC,CAAC,QAAQ,IAAI,SAAS,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,IAAI,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IAChG,OAAO;QACL,GAAG,IAAI;QACP,MAAM,EAAE;YACN,GAAG,MAAM;YACT,qBAAqB,EAAE,MAAM,CAAC,aAAa;YAC3C,eAAe,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI;YAC3C,YAAY,EAAE,MAAM,CAAC,SAAS;YAC9B,oBAAoB,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM;SAC7C;KACG,CAAC;AACT,CAAC"}

package/dist/hooks/dataset.d.ts

@@ -0,0 +1,138 @@
+import { type DatasetTransferOptions } from '@proofhound/api-client';
+import type { CreateDatasetDto, DatasetExportFormatDto, DatasetListItemDto, UpdateDatasetMetadataDto } from '@proofhound/shared';
+interface CreateDatasetVariables extends DatasetTransferOptions {
+    body: CreateDatasetDto;
+}
+interface DownloadDatasetVariables extends DatasetTransferOptions {
+    datasetId: string;
+    format: DatasetExportFormatDto;
+}
+interface UpdateDatasetVariables {
+    datasetId: string;
+    body: UpdateDatasetMetadataDto;
+}
+export declare function useDatasets(projectId: string): import("@tanstack/react-query").UseQueryResult<{
+    data: DatasetListItemDto[];
+    total: number;
+}, Error>;
+export declare function useDataset(projectId: string, datasetId: string): import("@tanstack/react-query").UseQueryResult<{
+    id: string;
+    projectId: string;
+    name: string;
+    description: string | null;
+    sampleCount: number;
+    fieldSchema: {
+        name: string;
+        role: "text" | "metadata" | "image" | "image_url" | "image_base64" | "expected_output";
+        type: "string" | "number" | "boolean" | "object" | "unknown" | "null" | "array";
+    }[];
+    categoryDistribution: {
+        field: string | null;
+        total: number;
+        categories: {
+            label: string;
+            count: number;
+        }[];
+    };
+    references: {
+        experiments: number;
+        optimizations: number;
+    };
+    hasImages: boolean;
+    storagePrefix: string | null;
+    createdBy: string;
+    createdByDisplayName: string | null;
+    createdAt: string;
+    updatedAt: string;
+    deletedAt: string | null;
+}, Error>;
+export interface DatasetSamplesQuery {
+    page: number;
+    pageSize: number;
+    search: string;
+}
+export declare function useDatasetSamples(projectId: string, datasetId: string, query: DatasetSamplesQuery): import("@tanstack/react-query").UseQueryResult<{
+    data: {
+        id: string;
+        datasetId: string;
+        data: Record<string, unknown>;
+        externalId: string | null;
+        createdAt: string;
+        updatedAt: string;
+    }[];
+    total: number;
+}, Error>;
+export declare function useCreateDataset(projectId: string): import("@tanstack/react-query").UseMutationResult<{
+    dataset: {
+        id: string;
+        projectId: string;
+        name: string;
+        description: string | null;
+        sampleCount: number;
+        fieldSchema: {
+            name: string;
+            role: "text" | "metadata" | "image" | "image_url" | "image_base64" | "expected_output";
+            type: "string" | "number" | "boolean" | "object" | "unknown" | "null" | "array";
+        }[];
+        categoryDistribution: {
+            field: string | null;
+            total: number;
+            categories: {
+                label: string;
+                count: number;
+            }[];
+        };
+        references: {
+            experiments: number;
+            optimizations: number;
+        };
+        hasImages: boolean;
+        storagePrefix: string | null;
+        createdBy: string;
+        createdByDisplayName: string | null;
+        createdAt: string;
+        updatedAt: string;
+        deletedAt: string | null;
+    };
+    sampleCount: number;
+}, Error, CreateDatasetVariables, unknown>;
+export declare function useUpdateDataset(projectId: string): import("@tanstack/react-query").UseMutationResult<{
+    id: string;
+    projectId: string;
+    name: string;
+    description: string | null;
+    sampleCount: number;
+    fieldSchema: {
+        name: string;
+        role: "text" | "metadata" | "image" | "image_url" | "image_base64" | "expected_output";
+        type: "string" | "number" | "boolean" | "object" | "unknown" | "null" | "array";
+    }[];
+    categoryDistribution: {
+        field: string | null;
+        total: number;
+        categories: {
+            label: string;
+            count: number;
+        }[];
+    };
+    references: {
+        experiments: number;
+        optimizations: number;
+    };
+    hasImages: boolean;
+    storagePrefix: string | null;
+    createdBy: string;
+    createdByDisplayName: string | null;
+    createdAt: string;
+    updatedAt: string;
+    deletedAt: string | null;
+}, Error, UpdateDatasetVariables, unknown>;
+export declare function useDownloadDataset(projectId: string): import("@tanstack/react-query").UseMutationResult<import("@proofhound/api-client").DatasetDownloadResult, Error, DownloadDatasetVariables, unknown>;
+export declare function useDeleteDataset(projectId: string): import("@tanstack/react-query").UseMutationResult<undefined, Error, string, unknown>;
+export declare function useDeleteDatasetSamples(projectId: string, datasetId: string): import("@tanstack/react-query").UseMutationResult<{
+    deleted: number;
+}, Error, {
+    sampleIds: string[];
+}, unknown>;
+export {};
+//# sourceMappingURL=dataset.d.ts.map

package/dist/hooks/dataset.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dataset.d.ts","sourceRoot":"","sources":["../../src/hooks/dataset.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AACpF,OAAO,KAAK,EACV,gBAAgB,EAChB,sBAAsB,EACtB,kBAAkB,EAElB,wBAAwB,EACzB,MAAM,oBAAoB,CAAC;AAG5B,UAAU,sBAAuB,SAAQ,sBAAsB;IAC7D,IAAI,EAAE,gBAAgB,CAAC;CACxB;AAED,UAAU,wBAAyB,SAAQ,sBAAsB;IAC/D,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,sBAAsB,CAAC;CAChC;AAED,UAAU,sBAAsB;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,wBAAwB,CAAC;CAChC;AAID,wBAAgB,WAAW,CAAC,SAAS,EAAE,MAAM;;;UAO5C;AAED,wBAAgB,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAM9D;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,mBAAmB;;;;;;;;;;UAajG;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2CAUjD;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2CAoBjD;AAED,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,MAAM,uJAKnD;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,wFAWjD;AAED,wBAAgB,uBAAuB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;;;;YAW3E"}

package/dist/hooks/dataset.js

@@ -0,0 +1,84 @@
+import { datasetClient } from '@proofhound/api-client';
+import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+export function useDatasets(projectId) {
+    return useQuery({
+        queryKey: ['datasets', projectId],
+        queryFn: () => datasetClient.listDatasets(projectId),
+        enabled: projectId.length > 0,
+        placeholderData: (previousData) => previousData,
+    });
+}
+export function useDataset(projectId, datasetId) {
+    return useQuery({
+        queryKey: ['datasets', projectId, datasetId],
+        queryFn: () => datasetClient.getDataset(projectId, datasetId),
+        enabled: projectId.length > 0 && datasetId.length > 0,
+    });
+}
+export function useDatasetSamples(projectId, datasetId, query) {
+    return useQuery({
+        queryKey: ['dataset-samples', projectId, datasetId, query.page, query.pageSize, query.search],
+        queryFn: () => datasetClient.listDatasetSamples(projectId, datasetId, {
+            page: query.page,
+            pageSize: query.pageSize,
+            search: query.search || undefined,
+        }),
+        enabled: projectId.length > 0 && datasetId.length > 0,
+        // Keep the previous page visible while the next page / search result loads.
+        placeholderData: (previousData) => previousData,
+    });
+}
+export function useCreateDataset(projectId) {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: ({ body, onProgress }) => datasetClient.createDataset(projectId, body, { onProgress }),
+        onSuccess: () => {
+            void queryClient.invalidateQueries({ queryKey: ['datasets', projectId] });
+        },
+    });
+}
+export function useUpdateDataset(projectId) {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: ({ datasetId, body }) => datasetClient.updateDataset(projectId, datasetId, body),
+        onSuccess: (dataset) => {
+            queryClient.setQueryData(['datasets', projectId], (current) => current
+                ? {
+                    ...current,
+                    data: current.data.map((item) => (item.id === dataset.id ? dataset : item)),
+                }
+                : current);
+            queryClient.setQueryData(['datasets', projectId, dataset.id], dataset);
+            void queryClient.invalidateQueries({ queryKey: ['datasets', projectId] });
+            void queryClient.invalidateQueries({ queryKey: ['datasets', projectId, dataset.id] });
+        },
+    });
+}
+export function useDownloadDataset(projectId) {
+    return useMutation({
+        mutationFn: ({ datasetId, format, onProgress }) => datasetClient.downloadDataset(projectId, datasetId, format, { onProgress }),
+    });
+}
+export function useDeleteDataset(projectId) {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: (datasetId) => datasetClient.deleteDataset(projectId, datasetId),
+        onSuccess: (_data, datasetId) => {
+            void queryClient.invalidateQueries({ queryKey: ['datasets', projectId] });
+            void queryClient.removeQueries({ queryKey: ['datasets', projectId, datasetId] });
+            void queryClient.removeQueries({ queryKey: ['dataset-samples', projectId, datasetId] });
+        },
+    });
+}
+export function useDeleteDatasetSamples(projectId, datasetId) {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: (body) => datasetClient.deleteDatasetSamples(projectId, datasetId, body),
+        onSuccess: () => {
+            void queryClient.invalidateQueries({ queryKey: ['datasets', projectId] });
+            void queryClient.invalidateQueries({ queryKey: ['datasets', projectId, datasetId] });
+            void queryClient.invalidateQueries({ queryKey: ['dataset-samples', projectId, datasetId] });
+        },
+    });
+}
+//# sourceMappingURL=dataset.js.map

package/dist/hooks/dataset.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dataset.js","sourceRoot":"","sources":["../../src/hooks/dataset.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAA+B,MAAM,wBAAwB,CAAC;AAQpF,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAkB9E,MAAM,UAAU,WAAW,CAAC,SAAiB;IAC3C,OAAO,QAAQ,CAAC;QACd,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC;QACjC,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,YAAY,CAAC,SAAS,CAAC;QACpD,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC;QAC7B,eAAe,EAAE,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY;KAChD,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,SAAiB,EAAE,SAAiB;IAC7D,OAAO,QAAQ,CAAC;QACd,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,SAAS,CAAC;QAC5C,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,UAAU,CAAC,SAAS,EAAE,SAAS,CAAC;QAC7D,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;KACtD,CAAC,CAAC;AACL,CAAC;AAQD,MAAM,UAAU,iBAAiB,CAAC,SAAiB,EAAE,SAAiB,EAAE,KAA0B;IAChG,OAAO,QAAQ,CAAC;QACd,QAAQ,EAAE,CAAC,iBAAiB,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC;QAC7F,OAAO,EAAE,GAAG,EAAE,CACZ,aAAa,CAAC,kBAAkB,CAAC,SAAS,EAAE,SAAS,EAAE;YACrD,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,SAAS;SAClC,CAAC;QACJ,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;QACrD,4EAA4E;QAC5E,eAAe,EAAE,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY;KAChD,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAA0B,EAAE,EAAE,CAC3D,aAAa,CAAC,aAAa,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE,UAAU,EAAE,CAAC;QAC9D,SAAS,EAAE,GAAG,EAAE;YACd,KAAK,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QAC5E,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAA0B,EAAE,EAAE,CAC1D,aAAa,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC;QACzD,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;YACrB,WAAW,CAAC,YAAY,CAAsB,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,CACjF,OAAO;gBACL,CAAC,CAAC;oBACE,GAAG,OAAO;oBACV,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;iBAC5E;gBACH,CAAC,CAAC,OAAO,CACZ,CAAC;YACF,WAAW,CAAC,YAAY,CAAC,CAAC,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;YACvE,KAAK,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YAC1E,KAAK,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACxF,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAiB;IAClD,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAA4B,EAAE,EAAE,CAC1E,aAAa,CAAC,eAAe,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,UAAU,EAAE,CAAC;KAC9E,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,SAAiB,EAAE,EAAE,CAAC,aAAa,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC;QACpF,SAAS,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;YAC9B,KAAK,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YAC1E,KAAK,WAAW,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YACjF,KAAK,WAAW,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,CAAC,iBAAiB,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,SAAiB,EAAE,SAAiB;IAC1E,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,OAAO,WAAW,CAAC;QACjB,UAAU,EAAE,CAAC,IAA6B,EAAE,EAAE,CAAC,aAAa,CAAC,oBAAoB,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC;QAC7G,SAAS,EAAE,GAAG,EAAE;YACd,KAAK,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YAC1E,KAAK,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;YACrF,KAAK,WAAW,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,iBAAiB,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QAC9F,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}