InputPredictではpastMessageではなくdialogueListを使う

yuiseki · Nov 9, 2024 · f35253c · f35253c
1 parent 50838cc
commit f35253c
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 33 deletions.
diff --git a/src/app/api/ai/suggests/route.ts b/src/app/api/ai/suggests/route.ts
@@ -5,7 +5,6 @@ import {
   loadTridentSuggestChain,
 } from "@/utils/langchain/chains/suggest";
 import { OpenAIEmbeddings } from "@langchain/openai";
-import { parsePastMessagesToLines } from "@/utils/trident/parsePastMessagesToLines";
 import { VercelPostgres } from "@langchain/community/vectorstores/vercel_postgres";
 import {
   createCheckDocumentExists,
@@ -19,12 +18,7 @@ export async function POST(request: Request) {
   const reqJson = await request.json();
   const lang = reqJson.lang;
   const location = reqJson.location;
-  const pastMessagesJsonString = reqJson.pastMessages;
-
-  const chatHistoryLines = parsePastMessagesToLines(
-    pastMessagesJsonString,
-    true
-  );
+  const chatHistoryLines = reqJson.dialogueList;
 
   let input = "";
 

diff --git a/src/app/page.tsx b/src/app/page.tsx
@@ -398,21 +398,18 @@ export default function Home() {
                     }}
                   />
                 )}
-                {pastMessages &&
-                  pastMessages.length > 1 &&
-                  !responding &&
-                  !mapping && (
-                    <InputPredict
-                      pastMessages={pastMessages}
-                      onUpdateSuggestions={() => {
-                        scrollToBottom();
-                      }}
-                      onSelect={(value: string) => {
-                        setInputText(value);
-                        onSubmit(value);
-                      }}
-                    />
-                  )}
+                {!responding && !mapping && dialogueList.length > 1 && (
+                  <InputPredict
+                    dialogueList={dialogueList}
+                    onUpdateSuggestions={() => {
+                      scrollToBottom();
+                    }}
+                    onSelect={(value: string) => {
+                      setInputText(value);
+                      onSubmit(value);
+                    }}
+                  />
+                )}
               </LocationProvider>
               <div style={{ height: "1px" }} ref={dialogueEndRef} />
             </div>

diff --git a/src/components/InputPredict/index.tsx b/src/components/InputPredict/index.tsx
@@ -3,23 +3,20 @@ import { nextPostJsonWithCache } from "@/utils/nextPostJson";
 import { useContext, useEffect, useState } from "react";
 
 import styles from "./styles.module.scss";
+import { DialogueElement } from "@/types/DialogueElement";
 
 export const InputPredict: React.FC<{
-  pastMessages?: any[];
+  dialogueList: DialogueElement[];
   onUpdateSuggestions?: () => void;
   onSelect?: (value: string) => void;
-}> = ({ pastMessages, onUpdateSuggestions, onSelect }) => {
+}> = ({ dialogueList, onUpdateSuggestions, onSelect }) => {
   const locationInfo = useContext(LocationContext);
   const [requesting, setRequesting] = useState(false);
   const [suggests, setSuggests] = useState<string[] | undefined>(undefined);
 
+  console.log("pastMessages", dialogueList);
+
   useEffect(() => {
-    if (!pastMessages) {
-      return;
-    }
-    if (pastMessages.length === 0) {
-      return;
-    }
     if (requesting) {
       return;
     }
@@ -28,7 +25,10 @@ export const InputPredict: React.FC<{
       const resJson = await nextPostJsonWithCache("/api/ai/suggests", {
         lang: window.navigator.language,
         location: locationInfo.location,
-        pastMessages: JSON.stringify(pastMessages),
+        dialogueList: dialogueList
+          .filter((d) => d.who === "user")
+          .map((d) => d.text)
+          .join("\n"),
       });
       if (!resJson.suggests) {
         return;
@@ -39,7 +39,7 @@ export const InputPredict: React.FC<{
       setRequesting(false);
     };
     thisEffect();
-  }, [locationInfo, onUpdateSuggestions, pastMessages, requesting]);
+  }, [dialogueList, locationInfo, onUpdateSuggestions, requesting]);
 
   return (
     <>

diff --git a/src/types/DialogueElement.ts b/src/types/DialogueElement.ts
@@ -1,5 +1,5 @@
 export type DialogueElement = {
-  who: string;
+  who: "user" | "assistant";
   text: string;
   textEnd?: string;
 };