@mastra/react 1.12.0 → 1.13.0-alpha.0

package/dist/hooks/traces/index.js

@@ -1,8 +1,8 @@
 import { n as useMastraClient } from "../../mastra-client-context-Bx_g84YB.js";
 import { t as useInView } from "../../use-in-view-C9sYPGRm.js";
+import { t as getOrCreate } from "../../map-CDv4fbu4.js";
 import { n as is403ForbiddenError } from "../../query-utils-D1NgFnWU.js";
 import { t as getFeedbackRefetchInterval } from "../../feedback-refetch-interval-DLHzQoVV.js";
-import { t as getOrCreate } from "../../map-CDv4fbu4.js";
 import { MastraClient, MastraClientError } from "@mastra/client-js";
 import { useEffect, useMemo, useRef, useState } from "react";
 import { keepPreviousData, queryOptions, skipToken, useInfiniteQuery, useMutation, useQueries, useQuery, useQueryClient } from "@tanstack/react-query";
@@ -627,6 +627,110 @@ function useTraceQueryLatest({ query, limit = 25, enabled = true, refetchInterva
 	});
 }
 //#endregion
+//#region src/hooks/traces/use-query-spans.ts
+const SPAN_QUERY_PER_PAGE = 25;
+const selectSpans = (data) => data.pages.flatMap((page) => page.spans);
+/** Queries completed spans with cursor pagination and viewport-driven loading. */
+function useQuerySpans({ query, limit = 25, enabled = true, queryOptions }) {
+	const client = useMastraClient();
+	const { inView, setRef: setEndOfListElement } = useInView();
+	const { data, hasNextPage, isFetchingNextPage, fetchNextPage, isLoading, isFetching, isError, error, refetch, isFetchNextPageError } = useInfiniteQuery({
+		queryKey: [
+			"span-query",
+			query,
+			limit
+		],
+		queryFn: query ? ({ pageParam }) => client.querySpans({
+			...query,
+			page: pageParam ? {
+				limit,
+				after: pageParam
+			} : { limit }
+		}) : skipToken,
+		initialPageParam: void 0,
+		getNextPageParam: (lastPage) => lastPage.page.next ?? void 0,
+		select: selectSpans,
+		retry: false,
+		placeholderData: keepPreviousData,
+		enabled,
+		...queryOptions
+	});
+	useEffect(() => {
+		if (enabled && inView && hasNextPage && !isFetching && !isFetchNextPageError) fetchNextPage();
+	}, [
+		enabled,
+		inView,
+		hasNextPage,
+		isFetching,
+		isFetchNextPageError,
+		fetchNextPage
+	]);
+	return {
+		spans: data,
+		hasNextPage,
+		isFetchingNextPage,
+		fetchNextPage: () => void fetchNextPage(),
+		setEndOfListElement,
+		isLoading,
+		isFetching,
+		isError,
+		error,
+		refetch: () => void refetch()
+	};
+}
+//#endregion
+//#region src/hooks/traces/use-query-threads.ts
+const THREAD_QUERY_PER_PAGE = 25;
+const selectThreads = (data) => data.pages.flatMap((page) => page.threads);
+/** Queries thread rows with cursor pagination and viewport-driven loading. */
+function useQueryThreads({ query, limit = 25, enabled = true, queryOptions }) {
+	const client = useMastraClient();
+	const { inView, setRef: setEndOfListElement } = useInView();
+	const { data, hasNextPage, isFetchingNextPage, fetchNextPage, isLoading, isFetching, isError, error, refetch, isFetchNextPageError } = useInfiniteQuery({
+		queryKey: [
+			"thread-query",
+			query,
+			limit
+		],
+		queryFn: query ? ({ pageParam }) => client.queryTraceThreads({
+			...query,
+			page: pageParam ? {
+				limit,
+				after: pageParam
+			} : { limit }
+		}) : skipToken,
+		initialPageParam: void 0,
+		getNextPageParam: (lastPage) => lastPage.page.next ?? void 0,
+		select: selectThreads,
+		retry: false,
+		placeholderData: keepPreviousData,
+		enabled,
+		...queryOptions
+	});
+	useEffect(() => {
+		if (enabled && inView && hasNextPage && !isFetching && !isFetchNextPageError) fetchNextPage();
+	}, [
+		enabled,
+		inView,
+		hasNextPage,
+		isFetching,
+		isFetchNextPageError,
+		fetchNextPage
+	]);
+	return {
+		threads: data,
+		hasNextPage,
+		isFetchingNextPage,
+		fetchNextPage: () => void fetchNextPage(),
+		setEndOfListElement,
+		isLoading,
+		isFetching,
+		isError,
+		error,
+		refetch: () => void refetch()
+	};
+}
+//#endregion
 //#region src/hooks/traces/use-trace-usage.ts
 const INPUT_TOKEN_METRIC = "mastra_model_total_input_tokens";
 const OUTPUT_TOKEN_METRIC = "mastra_model_total_output_tokens";
@@ -1300,6 +1404,6 @@ function useTraceColumnPreferencesStorageKey() {
 	return `mastra:traces:columns:${client.options.baseUrl || (typeof window === "undefined" ? "local" : window.location.origin || "local")}:${client.options.apiPrefix ?? "/api"}`;
 }
 //#endregion
-export { DISCOVERY_STALE_TIME, ROOT_ENTITY_TYPES, TRACES_PER_PAGE, TRACE_QUERY_DELTA_LIMIT, TRACE_QUERY_PER_PAGE, createTraceQueryValuesResolver, getLegacyTraceQueryNextPageParam, getTraceQueryNextPageParam, getTracesNextPageParam, lightSpanToTraceQueryTrace, mergeDeltaIntoPage0, mergeTraceQueryDelta, refreshPage0Rows, selectLegacyTraceQueryTraces, selectTraceQueryTraces, selectUniqueTraces, shouldPinLightList, shouldResetAfterIdle, traceQueryFieldsQueryKey, traceQueryKey, traceSpansQueryOptions, useBranch, useCreateFeedback, useDeleteFeedback, useEntityNames, useEnvironments, useFetchTraceSpans, useServiceNames, useSpanDetail, useSpanFeedback, useTags, useThreadHasOtherTraces, useTraceColumnPreferencesStorageKey, useTraceFeedback, useTraceLightSpans, useTraceMetadataFilterFields, useTraceOrBranchSpans, useTraceQuery, useTraceQueryLatest, useTraceSpans, useTraceSpansQueries, useTraceUsage, useTraces };
+export { DISCOVERY_STALE_TIME, ROOT_ENTITY_TYPES, SPAN_QUERY_PER_PAGE, THREAD_QUERY_PER_PAGE, TRACES_PER_PAGE, TRACE_QUERY_DELTA_LIMIT, TRACE_QUERY_PER_PAGE, createTraceQueryValuesResolver, getLegacyTraceQueryNextPageParam, getTraceQueryNextPageParam, getTracesNextPageParam, lightSpanToTraceQueryTrace, mergeDeltaIntoPage0, mergeTraceQueryDelta, refreshPage0Rows, selectLegacyTraceQueryTraces, selectTraceQueryTraces, selectUniqueTraces, shouldPinLightList, shouldResetAfterIdle, traceQueryFieldsQueryKey, traceQueryKey, traceSpansQueryOptions, useBranch, useCreateFeedback, useDeleteFeedback, useEntityNames, useEnvironments, useFetchTraceSpans, useQuerySpans, useQueryThreads, useServiceNames, useSpanDetail, useSpanFeedback, useTags, useThreadHasOtherTraces, useTraceColumnPreferencesStorageKey, useTraceFeedback, useTraceLightSpans, useTraceMetadataFilterFields, useTraceOrBranchSpans, useTraceQuery, useTraceQueryLatest, useTraceSpans, useTraceSpansQueries, useTraceUsage, useTraces };
 
 //# sourceMappingURL=index.js.map