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

add logic to random pick a llm config

Browse Source
This commit is contained in:
geekgeekrun
2025-04-12 14:32:06 +08:00
parent c5ac63b030
commit f72a5c014b
1 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
packages/ui/src/main/flow/READ_NO_REPLY_AUTO_REMINDER/boss-operation.ts
View File

@@ -19,6 +19,23 @@ export const sendLookForwardReplyEmotion = async (page: Page) => {
await lookForwardReplyEmojiProxy!.click()
}
const pickLlmConfigFromList = (llmConfigList) => {
if (llmConfigList.length === 1) {
return llmConfigList[0]
}
llmConfigList = llmConfigList.filter((it) => it.enabled)
const pool: number[] = []
for (let i = 0; i < llmConfigList.length; i++) {
for (let j = 0; j < Math.floor(llmConfigList[i].serveWeight); j++) {
pool.push(i)
}
}
const index = Math.floor(pool.length * Math.random())
return llmConfigList[
pool[index]
]
}
// let _index = 0
export const sendGptContent = async (page: Page, chatRecords) => {
@@ -86,7 +103,9 @@ export const sendGptContent = async (page: Page, chatRecords) => {
})
}
console.log(chatList)
const llmConfig = await readConfigFile('llm.json')
const llmConfigList = await readConfigFile('llm.json')
const llmConfig = pickLlmConfigFromList(llmConfigList)
console.log(llmConfig.providerCompleteApiUrl)
const res = await completes(
{
baseURL: llmConfig.providerCompleteApiUrl,