edit message

2023-03-25 05:49:41 -06:00
parent e30336c00e
commit a03d8b2ba9
4 changed files with 149 additions and 10 deletions
@@ -20,10 +20,13 @@ interface Props {
  onSend: (message: Message, isResend: boolean) => void;
  onUpdateConversation: (conversation: Conversation, data: KeyValuePair) => void;
  onAcceptEnv: (accept: boolean) => void;
+  onEditMessage: (message: Message, messageIndex: number) => void;
+  onDeleteMessage: (message: Message, messageIndex: number) => void;
+  onRegenerate: () => void;
  stopConversationRef: MutableRefObject<boolean>;
 }

-export const Chat: FC<Props> = ({ conversation, models, apiKey, isUsingEnv, messageIsStreaming, modelError, messageError, loading, lightMode, onSend, onUpdateConversation, onAcceptEnv, stopConversationRef }) => {
+export const Chat: FC<Props> = ({ conversation, models, apiKey, isUsingEnv, messageIsStreaming, modelError, messageError, loading, lightMode, onSend, onUpdateConversation, onAcceptEnv, onEditMessage, onDeleteMessage, onRegenerate, stopConversationRef }) => {
  const [currentMessage, setCurrentMessage] = useState<Message>();
  const [autoScrollEnabled, setAutoScrollEnabled] = useState(true);

@@ -122,7 +125,10 @@ export const Chat: FC<Props> = ({ conversation, models, apiKey, isUsingEnv, mess
                  <ChatMessage
                    key={index}
                    message={message}
+                    messageIndex={index}
                    lightMode={lightMode}
+                    onEditMessage={onEditMessage}
+                    onDeleteMessage={onDeleteMessage}
                  />
                ))}

@@ -149,11 +155,12 @@ export const Chat: FC<Props> = ({ conversation, models, apiKey, isUsingEnv, mess
              stopConversationRef={stopConversationRef}
              textareaRef={textareaRef}
              messageIsStreaming={messageIsStreaming}
+              model={conversation.model}
              onSend={(message) => {
                setCurrentMessage(message);
                onSend(message, false);
              }}
-              model={conversation.model}
+              onRegenerate={onRegenerate}
            />
          )}
        </>