fix(weave): Prevent sending null message in playground (#3278)

* Filter null messages, prevent sending null message * clean
wandb · Dec 18, 2024 · 6a093a4 · 6a093a4
1 parent 5fc28f2
commit 6a093a4
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 8 deletions.
diff --git a/...s/PagePanelComponents/Home/Browse3/pages/PlaygroundPage/PlaygroundChat/PlaygroundChat.tsx b/...s/PagePanelComponents/Home/Browse3/pages/PlaygroundPage/PlaygroundChat/PlaygroundChat.tsx
@@ -41,7 +41,6 @@ export const PlaygroundChat = ({
   const {handleRetry, handleSend} = useChatCompletionFunctions(
     setPlaygroundStates,
     setIsLoading,
-    chatText,
     playgroundStates,
     entity,
     project,
@@ -168,7 +167,13 @@ export const PlaygroundChat = ({
                             content: string,
                             toolCallId?: string
                           ) => {
-                            handleSend(role, idx, content, toolCallId);
+                            handleSend(
+                              role,
+                              chatText,
+                              idx,
+                              content,
+                              toolCallId
+                            );
                           },
                           setSelectedChoiceIndex: (choiceIndex: number) =>
                             setPlaygroundStateField(

diff --git a/...ePanelComponents/Home/Browse3/pages/PlaygroundPage/PlaygroundChat/PlaygroundChatInput.tsx b/...ePanelComponents/Home/Browse3/pages/PlaygroundPage/PlaygroundChat/PlaygroundChatInput.tsx
@@ -10,8 +10,8 @@ type PlaygroundChatInputProps = {
   chatText: string;
   setChatText: (text: string) => void;
   isLoading: boolean;
-  onSend: (role: PlaygroundMessageRole) => void;
-  onAdd: (role: PlaygroundMessageRole, text: string) => void;
+  onSend: (role: PlaygroundMessageRole, chatText: string) => void;
+  onAdd: (role: PlaygroundMessageRole, chatText: string) => void;
   settingsTab: number | null;
 };
 
@@ -43,7 +43,7 @@ export const PlaygroundChatInput: React.FC<PlaygroundChatInputProps> = ({
   };
 
   const handleSend = (role: PlaygroundMessageRole) => {
-    onSend(role);
+    onSend(role, chatText);
     handleReset();
   };
 

diff --git a/...omponents/Home/Browse3/pages/PlaygroundPage/PlaygroundChat/useChatCompletionFunctions.tsx b/...omponents/Home/Browse3/pages/PlaygroundPage/PlaygroundChat/useChatCompletionFunctions.tsx
@@ -13,7 +13,6 @@ import {clearTraceCall} from './useChatFunctions';
 export const useChatCompletionFunctions = (
   setPlaygroundStates: (states: PlaygroundState[]) => void,
   setIsLoading: (isLoading: boolean) => void,
-  chatText: string,
   playgroundStates: PlaygroundState[],
   entity: string,
   project: string,
@@ -59,19 +58,22 @@ export const useChatCompletionFunctions = (
 
   const handleSend = async (
     role: PlaygroundMessageRole,
+    chatText: string,
     callIndex?: number,
     content?: string,
     toolCallId?: string
   ) => {
     try {
       setIsLoading(true);
-      const newMessage = createMessage(role, content || chatText, toolCallId);
+      const newMessageContent = content || chatText;
+      const newMessage = createMessage(role, newMessageContent, toolCallId);
       const updatedStates = playgroundStates.map((state, index) => {
         if (callIndex !== undefined && callIndex !== index) {
           return state;
         }
         const updatedState = appendChoiceToMessages(state);
-        if (updatedState.traceCall?.inputs?.messages) {
+        // If the new message is not empty, add it to the messages
+        if (newMessageContent && updatedState.traceCall?.inputs?.messages) {
           updatedState.traceCall.inputs.messages.push(newMessage);
         }
         return updatedState;