Merge pull request #139 from Descendre/feature/#138

update: custom prompt mode
Descendre · Sep 5, 2024 · c0bba59 · c0bba59
2 parents 8851ca0 + c841d32
commit c0bba59
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/src/hooks/context/useChat.tsx b/src/hooks/context/useChat.tsx
@@ -47,6 +47,8 @@ export const useChat = (): UseChatProps => {
 		setIsLogSelect,
 		setSelectedContent,
 		characters,
+		isCustom,
+		customText,
 	} = context;
 
 	const handleGetChatRooms = async (): Promise<void> => {
@@ -224,6 +226,7 @@ export const useChat = (): UseChatProps => {
 		const messages = generateMessages({
 			content: content,
 			character: currentCharacter.name,
+			...(isCustom ? { custom: customText } : {}),
 		});
 		const chatGPTResponse = await handleChatGPT({
 			model: 'gpt-4o-mini',

diff --git a/src/interfaces/utils/HenerateMessagesProps.d.ts b/src/interfaces/utils/HenerateMessagesProps.d.ts
@@ -1,6 +1,7 @@
 export interface HenerateMessagesProps {
 	content: string;
 	character: string;
+	custom?: string | null;
 }
 
 export interface GPTMessageRequestProps {

diff --git a/src/utils/generateMessages.ts b/src/utils/generateMessages.ts
@@ -3,15 +3,27 @@ import { GPTMessageRequestProps, HenerateMessagesProps } from '@/interfaces';
 export const generateMessages = ({
 	content,
 	character,
+	custom,
 }: HenerateMessagesProps): GPTMessageRequestProps[] => {
+	const customContent = custom
+		? `
+            しかし以下の設定を優先的に守ってください。
+            ${custom}
+        `
+		: '';
+
 	const messages = [
 		{
 			role: 'system',
 			content: `
+				基本設定は以下の通りです。
+
                 あなたはvoicevoxの${character}です。
                 記号や英語はあまり使わないように。
                 あまり長文にならないように。
                 喋り方も${character}のように振る舞うこと。
+				
+				${customContent}
                 `,
 		},
 		{