fix tool message not added bug

Author: RomneyDa

gui/src/pages/gui/ToolCallDiv/index.tsx

@@ -69,6 +69,7 @@ function getStatusIcon(state: ToolStatus) {
 export function ToolCallDiv(props: ToolCallDivProps) {
   const availableTools = useAppSelector((state) => state.config.config.tools);
   const tool = useMemo(() => {
+    console.log(availableTools, props.toolCall);
     return availableTools.find(
       (tool) => props.toolCall.function?.name === tool.function.name,
     );

gui/src/redux/slices/sessionSlice.ts

@@ -295,6 +295,14 @@ export const sessionSlice = createSlice({
               const newHistoryItem =
                 createThinkingChatHistoryItem(incomingMessage);
               state.history.push(newHistoryItem);
+            } else if (incomingMessage.role === "tool") {
+              state.history.push({
+                message: {
+                  ...incomingMessage,
+                  id: uuidv4(),
+                },
+                contextItems: [],
+              });
             } else {
               continue;
             }

gui/src/util/toolCallState.ts

@@ -30,8 +30,17 @@ export function addToolCallDeltaToState(
   const nameDelta = toolCallDelta.function?.name ?? "";
   const argsDelta = toolCallDelta.function?.arguments ?? "";
 
-  const mergedName =
-    currentName === nameDelta ? currentName : currentName + nameDelta; // Some models may include the name repeatedly. This doesn't account for an edge case where the name is like "dothisdothis" and it happens to stream name in chunks "dothis" and "dothis" but that's a super edge case
+  let mergedName = currentName;
+  if (nameDelta.startsWith(currentName)) {
+    // Case where model progresssively streams name but full name each time e.g. "readFi" -> "readFil" -> "readFile"
+    mergedName = nameDelta;
+  } else if (currentName.startsWith(nameDelta)) {
+    // Case where model streams in full name each time e.g. readFile -> readFile -> readFile
+    // do nothing
+  } else {
+    // Case where model streams in name in parts e.g. "read" -> "File"
+    mergedName = currentName + nameDelta;
+  }
   const mergedArgs = currentArgs + argsDelta;
 
   const [_, parsedArgs] = incrementalParseJson(mergedArgs || "{}");