iod/page-assist
10
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Enable title generation in GeneralSettings component

Browse Source
This commit is contained in:
n4ze3m
2024-08-05 00:49:27 +05:30
parent 48a4af50ee
commit eccf8f8a88
4 changed files with 99 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/hooks/chat-helper/index.ts
View File

@@ -1,5 +1,6 @@
import { saveHistory, saveMessage } from "@/db"
import { setLastUsedChatModel } from "@/services/model-settings"
import { generateTitle } from "@/services/title"
import { ChatHistory } from "@/store/option"
export const saveMessageOnError = async ({
@@ -14,7 +15,7 @@ export const saveMessageOnError = async ({
setHistoryId,
isRegenerating,
message_source = "web-ui",
message_type
message_type
}: {
e: any
setHistory: (history: ChatHistory) => void
@@ -73,7 +74,8 @@ export const saveMessageOnError = async ({
)
await setLastUsedChatModel(historyId, selectedModel)
} else {
const newHistoryId = await saveHistory(userMessage, false, message_source)
const title = await generateTitle(selectedModel, userMessage, userMessage)
const newHistoryId = await saveHistory(title, false, message_source)
if (!isRegenerating) {
await saveMessage(
newHistoryId.id,
@@ -154,7 +156,8 @@ export const saveMessageOnSuccess = async ({
)
await setLastUsedChatModel(historyId, selectedModel!)
} else {
const newHistoryId = await saveHistory(message, false, message_source)
const title = await generateTitle(selectedModel, message, message)
const newHistoryId = await saveHistory(title, false, message_source)
await saveMessage(
newHistoryId.id,
selectedModel,