dmitry.galkin/chatbot-template
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: remove tokenlens and token usage tracking (#1357)

Browse source
This commit is contained in:
josh 2025-12-15 16:56:10 +00:00 • committed by GitHub
parent 5f9e231788
commit 7942e97095
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
17 changed files with 60 additions and 773 deletions
Download patch file Download diff file Expand all files Collapse all files

11
components/data-stream-handler.tsx
View file

@ -1,12 +1,16 @@
"use client";
import { useEffect } from "react";
import { useSWRConfig } from "swr";
import { unstable_serialize } from "swr/infinite";
import { initialArtifactData, useArtifact } from "@/hooks/use-artifact";
import { artifactDefinitions } from "./artifact";
import { useDataStream } from "./data-stream-provider";
import { getChatHistoryPaginationKey } from "./sidebar-history";
export function DataStreamHandler() {
const { dataStream, setDataStream } = useDataStream();
const { mutate } = useSWRConfig();
const { artifact, setArtifact, setMetadata } = useArtifact();
@ -19,6 +23,11 @@ export function DataStreamHandler() {
setDataStream([]);
for (const delta of newDeltas) {
// Handle chat title updates
if (delta.type === "data-chat-title") {
mutate(unstable_serialize(getChatHistoryPaginationKey));
continue;
}
const artifactDefinition = artifactDefinitions.find(
(currentArtifactDefinition) =>
currentArtifactDefinition.kind === artifact.kind
@ -77,7 +86,7 @@ export function DataStreamHandler() {
}
});
}
}, [dataStream, setArtifact, setMetadata, artifact, setDataStream]);
}, [dataStream, setArtifact, setMetadata, artifact, setDataStream, mutate]);
return null;
}