64 lines
No EOL
1.8 KiB
TypeScript
64 lines
No EOL
1.8 KiB
TypeScript
import {
|
|
type UseQueryOptions,
|
|
useQuery,
|
|
useQueryClient,
|
|
} from "@tanstack/react-query";
|
|
|
|
export interface DoclingHealthResponse {
|
|
status: "healthy" | "unhealthy" | "backend-unavailable";
|
|
message?: string;
|
|
}
|
|
|
|
export const useDoclingHealthQuery = (
|
|
options?: Omit<UseQueryOptions<DoclingHealthResponse>, "queryKey" | "queryFn">,
|
|
) => {
|
|
const queryClient = useQueryClient();
|
|
|
|
async function checkDoclingHealth(): Promise<DoclingHealthResponse> {
|
|
try {
|
|
// Call backend proxy endpoint instead of direct localhost
|
|
const response = await fetch("/api/docling/health", {
|
|
method: "GET",
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
},
|
|
});
|
|
|
|
if (response.ok) {
|
|
return { status: "healthy" };
|
|
} else if (response.status === 503) {
|
|
// Backend is up but docling is down (backend returns 503 for docling issues)
|
|
return {
|
|
status: "unhealthy",
|
|
message: `Health check failed with status: ${response.status}`,
|
|
};
|
|
} else {
|
|
// Other backend errors - treat as docling unhealthy
|
|
return {
|
|
status: "unhealthy",
|
|
message: `Health check failed with status: ${response.status}`,
|
|
};
|
|
}
|
|
} catch (error) {
|
|
// Network error - backend is likely down, don't show docling banner
|
|
return {
|
|
status: "backend-unavailable",
|
|
message: error instanceof Error ? error.message : "Connection failed",
|
|
};
|
|
}
|
|
}
|
|
|
|
const queryResult = useQuery(
|
|
{
|
|
queryKey: ["docling-health"],
|
|
queryFn: checkDoclingHealth,
|
|
retry: 1,
|
|
refetchInterval: 30000, // Check every 30 seconds
|
|
staleTime: 25000, // Consider data stale after 25 seconds
|
|
...options,
|
|
},
|
|
queryClient,
|
|
);
|
|
|
|
return queryResult;
|
|
}; |