SrJuggernaut/st-randomness-helpers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: support array prompts in placeholder services

Browse Source
This commit is contained in:
SrJuggernaut
2026-01-19 18:40:16 -06:00
parent 0c243c17c2
commit 5e2d4821b9
4 changed files with 199 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

178
dist/index.js vendored
View File

File diff suppressed because one or more lines are too long

45
src/services/pick.ts
View File

@@ -1,19 +1,38 @@
import type { generationData } from '@/types/SillyTavern'
const pickPlaceholderRegex = /%%pick::([\w\W]*?)%%/g
export const setupPickPlaceholders = () => {
const { eventTypes, eventSource } = SillyTavern.getContext()
eventSource.on(eventTypes.GENERATE_AFTER_DATA, (chat: generationData) => {
chat.prompt = chat.prompt.replaceAll(
pickPlaceholderRegex,
(_, optionsString: string) => {
const options = optionsString.split('::')
if (options.length === 0) {
return ''
}
return options[Math.floor(Math.random() * options.length)] ?? ''
eventSource.on(
eventTypes.GENERATE_AFTER_DATA,
(
chat: { prompt: string } | { prompt: { role: string; content: string }[] }
) => {
if (typeof chat.prompt === 'string') {
chat.prompt = chat.prompt.replaceAll(
pickPlaceholderRegex,
(_, optionsString: string) => {
const options = optionsString.split('::')
if (options.length === 0) {
return ''
}
return options[Math.floor(Math.random() * options.length)] ?? ''
}
)
} else if (Array.isArray(chat.prompt)) {
chat.prompt = chat.prompt.map((prompt) => {
prompt.content = prompt.content.replaceAll(
pickPlaceholderRegex,
(_, optionsString: string) => {
const options = optionsString.split('::')
if (options.length === 0) {
return ''
}
return options[Math.floor(Math.random() * options.length)] ?? ''
}
)
return prompt
})
}
)
})
}
)
}

61
src/services/randomWord.ts
View File

@@ -1,5 +1,4 @@
import { useStore } from '@/store'
import type { generationData } from '@/types/SillyTavern'
const registerRandomWordMacros = () => {
const { macros } = SillyTavern.getContext()
@@ -113,16 +112,54 @@ const randomWordPlaceholderRegex = /%%randomWord::([\w\W]*?)%%/g
export const setupRandomWordPlaceholders = () => {
const { eventTypes, eventSource } = SillyTavern.getContext()
eventSource.on(eventTypes.GENERATE_AFTER_DATA, (chat: generationData) => {
chat.prompt = chat.prompt.replaceAll(
randomWordPlaceholderRegex,
(_, wordList: string) => {
const words = useStore.getState().wordLists[wordList]
if (words === undefined || words.length === 0) {
return ''
}
return words[Math.floor(Math.random() * words.length)] ?? ''
eventSource.on(
eventTypes.GENERATE_AFTER_DATA,
(
chat: { prompt: string } | { prompt: { role: string; content: string }[] }
) => {
if (typeof chat.prompt === 'string') {
chat.prompt = chat.prompt.replaceAll(
randomWordPlaceholderRegex,
(_, wordList: string) => {
const words = new Map(
Object.entries(useStore.getState().wordLists)
).get(wordList)
if (words === undefined || words.length === 0) {
console.error(
`[st-randomness-helpers] randomWordPlaceholders: No words found for ${wordList}`
)
return ''
}
const word = words[Math.floor(Math.random() * words.length)]
return word ?? ''
}
)
} else if (Array.isArray(chat.prompt)) {
const newPrompt = chat.prompt.map(
({ role, content }: { role: string; content: string }) => {
return {
role,
content: content.replaceAll(
randomWordPlaceholderRegex,
(_, wordList: string) => {
const words = new Map(
Object.entries(useStore.getState().wordLists)
).get(wordList)
if (words === undefined || words.length === 0) {
console.error(
`[st-randomness-helpers] randomWordPlaceholders: No words found for ${wordList}`
)
return ''
}
const word = words[Math.floor(Math.random() * words.length)]
return word ?? ''
}
)
}
}
)
chat.prompt = newPrompt
}
)
})
}
)
}

40
src/services/shuffle.ts
View File

@@ -1,5 +1,3 @@
import type { generationData } from '@/types/SillyTavern'
const fishersYatesShuffle = <Type>(arr: Type[]): Type[] => {
const shuffled = [...arr]
for (let index = shuffled.length - 1; index > 0; index--) {
@@ -17,15 +15,35 @@ const shufflePlaceholderRegex = /%%shuffle::([\w\W]*?)(;;[\w\W]*?)?%%/g
export const setupShufflePlaceholders = () => {
const { eventTypes, eventSource } = SillyTavern.getContext()
eventSource.on(eventTypes.GENERATE_AFTER_DATA, (chat: generationData) => {
chat.prompt = chat.prompt.replaceAll(
shufflePlaceholderRegex,
(_, toBeshuffleed: string, separator?: string) => {
const options = toBeshuffleed.split('::')
return fishersYatesShuffle(options).join(
separator !== undefined ? separator.substring(2) : ''
eventSource.on(
eventTypes.GENERATE_AFTER_DATA,
(
chat: { prompt: string } | { prompt: { role: string; content: string }[] }
) => {
if (typeof chat.prompt === 'string') {
chat.prompt = chat.prompt.replaceAll(
shufflePlaceholderRegex,
(_, toBeshuffleed: string, separator?: string) => {
const options = toBeshuffleed.split('::')
return fishersYatesShuffle(options).join(
separator !== undefined ? separator.substring(2) : ''
)
}
)
} else if (Array.isArray(chat.prompt)) {
chat.prompt = chat.prompt.map((prompt) => {
prompt.content = prompt.content.replaceAll(
shufflePlaceholderRegex,
(_, toBeshuffleed: string, separator?: string) => {
const options = toBeshuffleed.split('::')
return fishersYatesShuffle(options).join(
separator !== undefined ? separator.substring(2) : ''
)
}
)
return prompt
})
}
)
})
}
)
}