diff --git a/packages/sqlite-plugin/src/entity/LlmModelUsageRecord.ts b/packages/sqlite-plugin/src/entity/LlmModelUsageRecord.ts index 6aa22e0..db5c179 100644 --- a/packages/sqlite-plugin/src/entity/LlmModelUsageRecord.ts +++ b/packages/sqlite-plugin/src/entity/LlmModelUsageRecord.ts @@ -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 diff --git a/packages/ui/src/main/features/llm-request-log.ts b/packages/ui/src/main/features/llm-request-log.ts index 741a74c..e3c1f70 100644 --- a/packages/ui/src/main/features/llm-request-log.ts +++ b/packages/ui/src/main/features/llm-request-log.ts @@ -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]) }