mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-06-10 17:35:57 +00:00
105db00987
* feat: show token usage per assistant response * fix: align client models response with token usage * fix: address token usage review feedback * docs: clarify token usage config example --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
19 lines
458 B
TypeScript
19 lines
458 B
TypeScript
import { useQuery } from "@tanstack/react-query";
|
|
|
|
import { loadModels } from "./api";
|
|
|
|
export function useModels({ enabled = true }: { enabled?: boolean } = {}) {
|
|
const { data, isLoading, error } = useQuery({
|
|
queryKey: ["models"],
|
|
queryFn: () => loadModels(),
|
|
enabled,
|
|
refetchOnWindowFocus: false,
|
|
});
|
|
return {
|
|
models: data?.models ?? [],
|
|
tokenUsageEnabled: data?.token_usage.enabled ?? false,
|
|
isLoading,
|
|
error,
|
|
};
|
|
}
|