Spaces:
Sleeping
Sleeping
File size: 2,159 Bytes
05c5ed5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | "use client";
import { appStore } from "@/app/store";
import { useCompletion } from "@ai-sdk/react";
import { ChatModel } from "app-types/chat";
import { useCallback, useEffect } from "react";
import { mutate } from "swr";
import { safe } from "ts-safe";
export function useGenerateThreadTitle(option: {
threadId: string;
chatModel?: ChatModel;
}) {
const { complete, completion } = useCompletion({
api: "/api/chat/title",
});
const updateTitle = useCallback(
(title: string) => {
appStore.setState((prev) => {
if (prev.threadList.some((v) => v.id !== option.threadId)) {
return {
threadList: [
{
id: option.threadId,
title,
userId: "",
createdAt: new Date(),
},
...prev.threadList,
],
};
}
return {
threadList: prev.threadList.map((v) =>
v.id === option.threadId ? { ...v, title } : v,
),
};
});
},
[option.threadId, option.chatModel?.model, option.chatModel?.provider],
);
const generateTitle = useCallback(
(message: string) => {
const { threadId, chatModel } = option;
if (appStore.getState().generatingTitleThreadIds.includes(threadId))
return;
appStore.setState((prev) => ({
generatingTitleThreadIds: [...prev.generatingTitleThreadIds, threadId],
}));
safe(() => {
updateTitle("");
return complete("", {
body: {
message,
threadId,
chatModel: chatModel ?? appStore.getState().chatModel,
},
});
})
.ifOk(() => mutate("/api/thread"))
.watch(() => {
appStore.setState((prev) => ({
generatingTitleThreadIds: prev.generatingTitleThreadIds.filter(
(v) => v !== threadId,
),
}));
});
},
[updateTitle],
);
useEffect(() => {
const title = completion.trim();
if (title) {
updateTitle(title);
}
}, [completion, updateTitle]);
return generateTitle;
}
|