tui/sucker.severian.dev
1
0
Fork 0
mirror of https://github.com/severian-dev/sucker.severian.dev.git synced 2025-10-28 20:55:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: r1 and blank user persona handling

Browse Source
This commit is contained in:
Severian
2025-02-05 07:16:46 +08:00
parent b6df5d196f
commit b5d7532a84
3 changed files with 42 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
src/app/api/proxy/route.ts
View File

@@ -52,7 +52,7 @@ function safeReplace(text: string, old: string, newStr: string): string {
function extractCardData(messages: Message[]): CardData {
const content0 = messages[0].content;
const content1 = messages[1].content;
const content1 = messages[2].content;
const userName = extractPersonaName(content0, 0);
const charName = extractPersonaName(content0, 1);
@@ -67,41 +67,41 @@ function extractCardData(messages: Message[]): CardData {
first_mes: content1,
};
if (personaMatches.length >= 2) {
const secondPersonaIdx = personaMatches[1].index!;
const startDesc = secondPersonaIdx + "'s Persona:".length;
const remaining = content0.slice(startDesc);
// blank user persona handling, or at least an attempt
let secondPersonaIdx =
personaMatches[personaMatches.length >= 2 ? 1 : 0]?.index;
const startDesc = secondPersonaIdx + "'s Persona:".length;
const remaining = content0.slice(startDesc);
const scenarioMarker = remaining.match(/Scenario of the roleplay:/);
const exampleMarker = remaining.match(/Example conversations between/);
const scenarioMarker = remaining.match(/Scenario of the roleplay:/);
const exampleMarker = remaining.match(/Example conversations between/);
let endIdx = remaining.length;
if (scenarioMarker) endIdx = Math.min(endIdx, scenarioMarker.index!);
if (exampleMarker) endIdx = Math.min(endIdx, exampleMarker.index!);
let endIdx = remaining.length;
if (scenarioMarker) endIdx = Math.min(endIdx, scenarioMarker.index!);
if (exampleMarker) endIdx = Math.min(endIdx, exampleMarker.index!);
cardData.description = remaining.slice(0, endIdx).trim();
cardData.description = remaining.slice(0, endIdx).trim();
if (scenarioMarker) {
const scenarioStart = scenarioMarker.index! + scenarioMarker[0].length;
const scenarioRemaining = remaining.slice(scenarioStart);
const exampleInScenarioMarker = scenarioRemaining.match(
/Example conversations between/
);
const scenarioEnd = exampleInScenarioMarker
? exampleInScenarioMarker.index!
: scenarioRemaining.length;
cardData.scenario = scenarioRemaining.slice(0, scenarioEnd).trim();
}
if (scenarioMarker) {
const scenarioStart = scenarioMarker.index! + scenarioMarker[0].length;
const scenarioRemaining = remaining.slice(scenarioStart);
const exampleInScenarioMarker = scenarioRemaining.match(
/Example conversations between/
);
const scenarioEnd = exampleInScenarioMarker
? exampleInScenarioMarker.index!
: scenarioRemaining.length;
cardData.scenario = scenarioRemaining.slice(0, scenarioEnd).trim();
}
if (exampleMarker) {
const exampleStart = exampleMarker.index!;
const rawExampleStr = remaining.slice(exampleStart).trim();
const colonIdx = rawExampleStr.indexOf(":");
cardData.mes_example =
colonIdx !== -1
? rawExampleStr.slice(colonIdx + 1).trim()
: rawExampleStr.trim();
}
if (exampleMarker) {
const exampleStart = exampleMarker.index!;
const rawExampleStr = remaining.slice(exampleStart).trim();
const colonIdx = rawExampleStr.indexOf(":");
cardData.mes_example =
colonIdx !== -1
? rawExampleStr.slice(colonIdx + 1).trim()
: rawExampleStr.trim();
}
for (const field in cardData) {