opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-09-27 18:15:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: feature gate on collecting usage data

Browse Source
This commit is contained in:
Tianzhou Chen
2023-05-26 22:08:09 +08:00
parent a2ea7ce191
commit 1a7c9b70d0
4 changed files with 35 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
src/components/ConversationView/index.tsx
View File

@ -14,7 +14,7 @@ import {
useUserStore,
} from "@/store";
import { Conversation, CreatorRole, Message } from "@/types";
import { countTextTokens, generateUUID } from "@/utils";
import { countTextTokens, generateUUID, hasFeature } from "@/utils";
import getEventEmitter from "@/utils/event-emitter";
import Header from "./Header";
import EmptyView from "../EmptyView";
@ -290,36 +290,38 @@ const ConversationView = () => {
// Emit usage update event so quota widget can update.
getEventEmitter().emit("usage.update");
// Collect system prompt
// We only collect the db prompt for the system prompt. We do not collect the intermediate
// exchange to save space since those can be derived from the previous record.
usageMessageList.push({
id: generateUUID(),
createdAt: Date.now(),
creatorRole: CreatorRole.System,
content: dbPrompt,
} as Message);
if (hasFeature("collect")) {
// Collect system prompt
// We only collect the db prompt for the system prompt. We do not collect the intermediate
// exchange to save space since those can be derived from the previous record.
usageMessageList.push({
id: generateUUID(),
createdAt: Date.now(),
creatorRole: CreatorRole.System,
content: dbPrompt,
} as Message);
// Collect user message
usageMessageList.push(userMessage);
// Collect user message
usageMessageList.push(userMessage);
// Collect assistant response
usageMessageList.push(assistantMessage);
// Collect assistant response
usageMessageList.push(assistantMessage);
axios
.post<string[]>(
"/api/collect",
{
conversation: currentConversation,
messages: usageMessageList,
},
{
headers: requestHeaders,
}
)
.catch(() => {
// do nth
});
axios
.post<string[]>(
"/api/collect",
{
conversation: currentConversation,
messages: usageMessageList,
},
{
headers: requestHeaders,
}
)
.catch(() => {
// do nth
});
}
};
return (