GithubBackup/geekgeekrun
1
0
Fork 0
mirror of https://github.com/geekgeekrun/geekgeekrun.git synced 2026-05-22 08:46:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

change providerApiSecretMd5 column to providerApiSecret for LlmModelUsageRecord table

Browse Source
This commit is contained in:
geekgeekrun
2025-04-22 13:54:54 +08:00
parent 3d148e8e01
commit bbd670b674
2 changed files with 5 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/sqlite-plugin/src/entity/LlmModelUsageRecord.ts
View File

@@ -2,7 +2,7 @@ import { requireTypeorm } from "../utils/module-loader";
const { Entity, PrimaryGeneratedColumn, Column, Index } = requireTypeorm()
@Entity()
@Index(["providerCompleteApiUrl", "model", "providerApiSecretMd5"])
@Index(["providerCompleteApiUrl", "model", "providerApiSecret"])
export class LlmModelUsageRecord {
@PrimaryGeneratedColumn()
id: number;
@@ -13,8 +13,10 @@ export class LlmModelUsageRecord {
@Column()
model: string
@Column()
providerApiSecretMd5: string
@Column({
nullable: true
})
providerApiSecret: string
@Column({
nullable: true

7
packages/ui/src/main/features/llm-request-log.ts
View File

@@ -5,7 +5,6 @@ export const RequestSceneEnum = {
readNoReplyAutoReminder: 2,
geekAutoStartChatWithBoss: 3
}
export const providerApiSecretToMd5Map = {}
let dbInitPromise
export const recordGptCompletionRequest = async (payload) => {
@@ -13,17 +12,11 @@ export const recordGptCompletionRequest = async (payload) => {
'@geekgeekrun/geek-auto-start-chat-with-boss/runtime-file-utils.mjs'
)
const { initDb } = await import('@geekgeekrun/sqlite-plugin')
const SparkMD5 = await import('spark-md5')
if (!dbInitPromise) {
dbInitPromise = initDb(getPublicDbFilePath())
}
const ds = await dbInitPromise
const o = { ...payload }
if (!providerApiSecretToMd5Map[o.providerApiSecret]) {
providerApiSecretToMd5Map[o.providerApiSecret] = SparkMD5.hash(o.providerApiSecret)
}
o.providerApiSecretMd5 = providerApiSecretToMd5Map[o.providerApiSecret]
delete o.providerApiSecret
await saveGptCompletionRequestRecord(ds, [o])
}