Spaces:
Running
Running
File size: 2,444 Bytes
aa01edd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
import type { ChatMessage } from "gpt-tokenizer/GptEncoding";
import { addLogEntry } from "./logEntries";
import { generateChatResponse } from "./textGeneration";
interface FollowUpQuestionParams {
topic: string;
currentContent: string;
previousQuestions?: string[];
}
export async function generateFollowUpQuestion({
topic,
currentContent,
previousQuestions = [],
}: FollowUpQuestionParams): Promise<string> {
try {
addLogEntry("Generating a follow-up question");
const promptMessages: ChatMessage[] = [
{
role: "user",
content: topic,
},
{
role: "assistant",
content: currentContent,
},
{
role: "user",
content: `CRITICAL INSTRUCTION: You MUST use the EXACT SAME LANGUAGE as the original question and response. Also, note that the following follow-up questions were already asked. Do NOT repeat them or their meaning:
${
previousQuestions.length > 0
? previousQuestions.map((q, i) => `${i + 1}. ${q}`).join("\n")
: "(No previous follow-up questions yet)"
}
Knowing that, generate one short follow-up question that explores an important unexplored aspect of the topic.
- The question MUST be in the SAME LANGUAGE as the previous text (this is the highest priority requirement)
- The question MUST be different from all previously asked questions above
- Keep it to 1-2 sentences maximum
- End with a question mark
Respond with just the question, no additional text or explanations.`,
},
];
const response = await generateChatResponse(promptMessages, () => {});
const lines = response
.trim()
.split("\n")
.map((line) => line.trim())
.filter((line) => line.length > 0);
let questionLine = lines
.reverse()
.find(
(line) =>
line.endsWith("?") || line.endsWith('?"') || line.endsWith("?'"),
);
if (!questionLine) {
addLogEntry("No valid follow-up question generated");
return "";
}
if (questionLine.startsWith('"') || questionLine.startsWith("'")) {
questionLine = questionLine.slice(1);
}
if (questionLine.endsWith('"') || questionLine.endsWith("'")) {
questionLine = questionLine.slice(0, -1);
}
addLogEntry("Generated follow-up question successfully");
return questionLine;
} catch (error) {
addLogEntry(`Error generating follow-up question: ${error}`);
return "";
}
}
|