@mastra/playground-ui 5.1.21-alpha.3 → 5.1.21-alpha.4

package/dist/index.cjs.js

@@ -5394,7 +5394,8 @@ function MastraRuntimeProvider({
   threadId,
   refreshThreadList,
   settings,
-  runtimeContext
+  runtimeContext,
+  modelVersion
 }) {
   const [isRunning, setIsRunning] = React.useState(false);
   const [messages, setMessages] = React.useState([]);
@@ -5571,115 +5572,131 @@ function MastraRuntimeProvider({
           }
         }
       };
-      if (chatWithGenerateVNext) {
-        const response = await agent.generateVNext({
-          messages: [
-            {
-              role: "user",
-              content: input
+      if (modelVersion === "v2") {
+        if (chatWithGenerateVNext) {
+          const response = await agent.generateVNext({
+            messages: [
+              {
+                role: "user",
+                content: input
+              },
+              ...attachments
+            ],
+            runId: agentId,
+            modelSettings: {
+              frequencyPenalty,
+              presencePenalty,
+              maxRetries,
+              temperature,
+              topK,
+              topP,
+              maxOutputTokens: maxTokens
             },
-            ...attachments
-          ],
-          runId: agentId,
-          modelSettings: {
-            frequencyPenalty,
-            presencePenalty,
-            maxRetries,
-            temperature,
-            topK,
-            topP,
-            maxOutputTokens: maxTokens
-          },
-          providerOptions,
-          instructions,
-          runtimeContext: runtimeContextInstance,
-          ...memory ? { threadId, resourceId: agentId } : {}
-        });
-        handleGenerateResponse(response);
-        setIsRunning(false);
-        return;
-      }
-      if (chatWithStreamVNext) {
-        let updater = function() {
-          setMessages((currentConversation) => {
-            const message2 = {
-              role: "assistant",
-              content: [{ type: "text", text: content }]
-            };
-            if (!assistantMessageAdded) {
-              assistantMessageAdded = true;
+            providerOptions,
+            instructions,
+            runtimeContext: runtimeContextInstance,
+            ...memory ? { threadId, resourceId: agentId } : {}
+          });
+          handleGenerateResponse(response);
+          setIsRunning(false);
+          return;
+        } else {
+          let updater = function() {
+            setMessages((currentConversation) => {
+              const message2 = {
+                role: "assistant",
+                content: [{ type: "text", text: content }]
+              };
+              if (!assistantMessageAdded) {
+                assistantMessageAdded = true;
+                if (assistantToolCallAddedForUpdater) {
+                  assistantToolCallAddedForUpdater = false;
+                }
+                return [...currentConversation, message2];
+              }
               if (assistantToolCallAddedForUpdater) {
                 assistantToolCallAddedForUpdater = false;
+                return [...currentConversation, message2];
               }
-              return [...currentConversation, message2];
-            }
-            if (assistantToolCallAddedForUpdater) {
-              assistantToolCallAddedForUpdater = false;
-              return [...currentConversation, message2];
-            }
-            return [...currentConversation.slice(0, -1), message2];
-          });
-        };
-        const response = await agent.streamVNext({
-          messages: [
-            {
-              role: "user",
-              content: input
+              return [...currentConversation.slice(0, -1), message2];
+            });
+          };
+          const response = await agent.streamVNext({
+            messages: [
+              {
+                role: "user",
+                content: input
+              },
+              ...attachments
+            ],
+            runId: agentId,
+            modelSettings: {
+              frequencyPenalty,
+              presencePenalty,
+              maxRetries,
+              maxOutputTokens: maxTokens,
+              temperature,
+              topK,
+              topP
             },
-            ...attachments
-          ],
-          runId: agentId,
-          modelSettings: {
-            frequencyPenalty,
-            presencePenalty,
-            maxRetries,
-            maxOutputTokens: maxTokens,
-            temperature,
-            topK,
-            topP
-          },
-          instructions,
-          runtimeContext: runtimeContextInstance,
-          ...memory ? { threadId, resourceId: agentId } : {},
-          providerOptions
-        });
-        if (!response.body) {
-          throw new Error("No response body");
-        }
-        let content = "";
-        let assistantMessageAdded = false;
-        let assistantToolCallAddedForUpdater = false;
-        let assistantToolCallAddedForContent = false;
-        await response.processDataStream({
-          onChunk: async (chunk) => {
-            switch (chunk.type) {
-              case "text-delta": {
-                if (assistantToolCallAddedForContent) {
-                  assistantToolCallAddedForContent = false;
-                  content = chunk.payload.text;
-                } else {
-                  content += chunk.payload.text;
+            instructions,
+            runtimeContext: runtimeContextInstance,
+            ...memory ? { threadId, resourceId: agentId } : {},
+            providerOptions
+          });
+          if (!response.body) {
+            throw new Error("No response body");
+          }
+          let content = "";
+          let assistantMessageAdded = false;
+          let assistantToolCallAddedForUpdater = false;
+          let assistantToolCallAddedForContent = false;
+          await response.processDataStream({
+            onChunk: async (chunk) => {
+              switch (chunk.type) {
+                case "text-delta": {
+                  if (assistantToolCallAddedForContent) {
+                    assistantToolCallAddedForContent = false;
+                    content = chunk.payload.text;
+                  } else {
+                    content += chunk.payload.text;
+                  }
+                  console.log(chunk.payload.text, "VALUE");
+                  updater();
+                  break;
                 }
-                console.log(chunk.payload.text, "VALUE");
-                updater();
-                break;
-              }
-              case "tool-call": {
-                setMessages((currentConversation) => {
-                  const lastMessage = currentConversation[currentConversation.length - 1];
-                  if (lastMessage && lastMessage.role === "assistant") {
-                    const updatedMessage = {
-                      ...lastMessage,
-                      content: Array.isArray(lastMessage.content) ? [
-                        ...lastMessage.content,
-                        {
-                          type: "tool-call",
-                          toolCallId: chunk.payload.toolCallId,
-                          toolName: chunk.payload.toolName,
-                          args: chunk.payload.args
-                        }
-                      ] : [
-                        ...typeof lastMessage.content === "string" ? [{ type: "text", text: lastMessage.content }] : [],
+                case "tool-call": {
+                  setMessages((currentConversation) => {
+                    const lastMessage = currentConversation[currentConversation.length - 1];
+                    if (lastMessage && lastMessage.role === "assistant") {
+                      const updatedMessage = {
+                        ...lastMessage,
+                        content: Array.isArray(lastMessage.content) ? [
+                          ...lastMessage.content,
+                          {
+                            type: "tool-call",
+                            toolCallId: chunk.payload.toolCallId,
+                            toolName: chunk.payload.toolName,
+                            args: chunk.payload.args
+                          }
+                        ] : [
+                          ...typeof lastMessage.content === "string" ? [{ type: "text", text: lastMessage.content }] : [],
+                          {
+                            type: "tool-call",
+                            toolCallId: chunk.payload.toolCallId,
+                            toolName: chunk.payload.toolName,
+                            args: chunk.payload.args
+                          }
+                        ]
+                      };
+                      assistantToolCallAddedForUpdater = true;
+                      assistantToolCallAddedForContent = true;
+                      return [...currentConversation.slice(0, -1), updatedMessage];
+                    }
+                    const newMessage = {
+                      role: "assistant",
+                      content: [
+                        { type: "text", text: content },
                         {
                           type: "tool-call",
                           toolCallId: chunk.payload.toolCallId,
@@ -5690,277 +5707,278 @@ function MastraRuntimeProvider({
                     };
                     assistantToolCallAddedForUpdater = true;
                     assistantToolCallAddedForContent = true;
-                    return [...currentConversation.slice(0, -1), updatedMessage];
+                    return [...currentConversation, newMessage];
+                  });
+                  toolCallIdToName.current[chunk.payload.toolCallId] = chunk.payload.toolName;
+                  break;
+                }
+                case "tool-result": {
+                  setMessages((currentConversation) => {
+                    const lastMessage = currentConversation[currentConversation.length - 1];
+                    if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
+                      const updatedContent = lastMessage.content.map((part) => {
+                        if (typeof part === "object" && part.type === "tool-call" && part.toolCallId === chunk.payload.toolCallId) {
+                          return {
+                            ...part,
+                            result: chunk.payload.result
+                          };
+                        }
+                        return part;
+                      });
+                      const updatedMessage = {
+                        ...lastMessage,
+                        content: updatedContent
+                      };
+                      return [...currentConversation.slice(0, -1), updatedMessage];
+                    }
+                    return currentConversation;
+                  });
+                  try {
+                    const toolName = toolCallIdToName.current[chunk.payload.toolCallId];
+                    if (toolName === "updateWorkingMemory" && chunk.payload.result?.success) {
+                      await refreshWorkingMemory?.();
+                    }
+                  } finally {
+                    delete toolCallIdToName.current[chunk.payload.toolCallId];
                   }
-                  const newMessage = {
-                    role: "assistant",
+                  break;
+                }
+                case "error": {
+                  if (typeof chunk.payload.error === "string") {
+                    throw new Error(chunk.payload.error);
+                  }
+                  break;
+                }
+                case "finish": {
+                  handleFinishReason(chunk.payload.finishReason);
+                  break;
+                }
+                case "reasoning-delta": {
+                  setMessages((currentConversation) => {
+                    const lastMessage = currentConversation[currentConversation.length - 1];
+                    if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
+                      const updatedContent = lastMessage.content.map((part) => {
+                        if (typeof part === "object" && part.type === "reasoning") {
+                          return {
+                            ...part,
+                            text: part.text + chunk.payload.text
+                          };
+                        }
+                        return part;
+                      });
+                      const updatedMessage = {
+                        ...lastMessage,
+                        content: updatedContent
+                      };
+                      return [...currentConversation.slice(0, -1), updatedMessage];
+                    }
+                    const newMessage = {
+                      role: "assistant",
+                      content: [
+                        {
+                          type: "reasoning",
+                          text: chunk.payload.text
+                        },
+                        { type: "text", text: content }
+                      ]
+                    };
+                    return [...currentConversation, newMessage];
+                  });
+                  break;
+                }
+              }
+            }
+          });
+          setIsRunning(false);
+          return;
+        }
+      } else {
+        if (chatWithGenerate) {
+          const generateResponse = await agent.generate({
+            messages: [
+              {
+                role: "user",
+                content: input
+              },
+              ...attachments
+            ],
+            runId: agentId,
+            frequencyPenalty,
+            presencePenalty,
+            maxRetries,
+            maxSteps,
+            maxTokens,
+            temperature,
+            topK,
+            topP,
+            instructions,
+            runtimeContext: runtimeContextInstance,
+            ...memory ? { threadId, resourceId: agentId } : {},
+            providerOptions
+          });
+          if (generateResponse.response && "messages" in generateResponse.response) {
+            const latestMessage = generateResponse.response.messages.reduce(
+              (acc, message2) => {
+                const _content = Array.isArray(acc.content) ? acc.content : [];
+                if (typeof message2.content === "string") {
+                  return {
+                    ...acc,
                     content: [
-                      { type: "text", text: content },
+                      ..._content,
+                      ...generateResponse.reasoning ? [{ type: "reasoning", text: generateResponse.reasoning }] : [],
                       {
-                        type: "tool-call",
-                        toolCallId: chunk.payload.toolCallId,
-                        toolName: chunk.payload.toolName,
-                        args: chunk.payload.args
+                        type: "text",
+                        text: message2.content
                       }
                     ]
                   };
-                  assistantToolCallAddedForUpdater = true;
-                  assistantToolCallAddedForContent = true;
-                  return [...currentConversation, newMessage];
-                });
-                toolCallIdToName.current[chunk.payload.toolCallId] = chunk.payload.toolName;
-                break;
-              }
-              case "tool-result": {
-                setMessages((currentConversation) => {
-                  const lastMessage = currentConversation[currentConversation.length - 1];
-                  if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
-                    const updatedContent = lastMessage.content.map((part) => {
-                      if (typeof part === "object" && part.type === "tool-call" && part.toolCallId === chunk.payload.toolCallId) {
-                        return {
-                          ...part,
-                          result: chunk.payload.result
-                        };
+                }
+                if (message2.role === "assistant") {
+                  const toolCallContent = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "tool-call") : void 0;
+                  const reasoningContent = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "reasoning") : void 0;
+                  if (toolCallContent) {
+                    const newContent = _content.map((c) => {
+                      if (c.type === "tool-call" && c.toolCallId === toolCallContent?.toolCallId) {
+                        return { ...c, ...toolCallContent };
                       }
-                      return part;
+                      return c;
                     });
-                    const updatedMessage = {
-                      ...lastMessage,
-                      content: updatedContent
+                    const containsToolCall = newContent.some((c) => c.type === "tool-call");
+                    return {
+                      ...acc,
+                      content: containsToolCall ? [...reasoningContent ? [reasoningContent] : [], ...newContent] : [..._content, ...reasoningContent ? [reasoningContent] : [], toolCallContent]
                     };
-                    return [...currentConversation.slice(0, -1), updatedMessage];
                   }
-                  return currentConversation;
-                });
-                try {
-                  const toolName = toolCallIdToName.current[chunk.payload.toolCallId];
-                  if (toolName === "updateWorkingMemory" && chunk.payload.result?.success) {
-                    await refreshWorkingMemory?.();
+                  const textContent = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "text" && content.text) : void 0;
+                  if (textContent) {
+                    return {
+                      ...acc,
+                      content: [..._content, ...reasoningContent ? [reasoningContent] : [], textContent]
+                    };
                   }
-                } finally {
-                  delete toolCallIdToName.current[chunk.payload.toolCallId];
                 }
-                break;
-              }
-              case "error": {
-                if (typeof chunk.payload.error === "string") {
-                  throw new Error(chunk.payload.error);
-                }
-                break;
-              }
-              case "finish": {
-                handleFinishReason(chunk.payload.finishReason);
-                break;
-              }
-              case "reasoning-delta": {
-                setMessages((currentConversation) => {
-                  const lastMessage = currentConversation[currentConversation.length - 1];
-                  if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
-                    const updatedContent = lastMessage.content.map((part) => {
-                      if (typeof part === "object" && part.type === "reasoning") {
-                        return {
-                          ...part,
-                          text: part.text + chunk.payload.text
-                        };
+                if (message2.role === "tool") {
+                  const toolResult = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "tool-result") : void 0;
+                  if (toolResult) {
+                    const newContent = _content.map((c) => {
+                      if (c.type === "tool-call" && c.toolCallId === toolResult?.toolCallId) {
+                        return { ...c, result: toolResult.result };
                       }
-                      return part;
+                      return c;
                     });
-                    const updatedMessage = {
-                      ...lastMessage,
-                      content: updatedContent
+                    const containsToolCall = newContent.some((c) => c.type === "tool-call");
+                    return {
+                      ...acc,
+                      content: containsToolCall ? newContent : [
+                        ..._content,
+                        { type: "tool-result", toolCallId: toolResult.toolCallId, result: toolResult.result }
+                      ]
                     };
-                    return [...currentConversation.slice(0, -1), updatedMessage];
                   }
-                  const newMessage = {
-                    role: "assistant",
-                    content: [
-                      {
-                        type: "reasoning",
-                        text: chunk.payload.text
-                      },
-                      { type: "text", text: content }
-                    ]
-                  };
-                  return [...currentConversation, newMessage];
-                });
-                break;
-              }
-            }
-          }
-        });
-        setIsRunning(false);
-        return;
-      }
-      if (chatWithGenerate) {
-        const generateResponse = await agent.generate({
-          messages: [
-            {
-              role: "user",
-              content: input
-            },
-            ...attachments
-          ],
-          runId: agentId,
-          frequencyPenalty,
-          presencePenalty,
-          maxRetries,
-          maxSteps,
-          maxTokens,
-          temperature,
-          topK,
-          topP,
-          instructions,
-          runtimeContext: runtimeContextInstance,
-          ...memory ? { threadId, resourceId: agentId } : {},
-          providerOptions
-        });
-        if (generateResponse.response && "messages" in generateResponse.response) {
-          const latestMessage = generateResponse.response.messages.reduce(
-            (acc, message2) => {
-              const _content = Array.isArray(acc.content) ? acc.content : [];
-              if (typeof message2.content === "string") {
-                return {
-                  ...acc,
-                  content: [
-                    ..._content,
-                    ...generateResponse.reasoning ? [{ type: "reasoning", text: generateResponse.reasoning }] : [],
-                    {
-                      type: "text",
-                      text: message2.content
-                    }
-                  ]
-                };
-              }
-              if (message2.role === "assistant") {
-                const toolCallContent = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "tool-call") : void 0;
-                const reasoningContent = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "reasoning") : void 0;
-                if (toolCallContent) {
-                  const newContent = _content.map((c) => {
-                    if (c.type === "tool-call" && c.toolCallId === toolCallContent?.toolCallId) {
-                      return { ...c, ...toolCallContent };
-                    }
-                    return c;
-                  });
-                  const containsToolCall = newContent.some((c) => c.type === "tool-call");
-                  return {
-                    ...acc,
-                    content: containsToolCall ? [...reasoningContent ? [reasoningContent] : [], ...newContent] : [..._content, ...reasoningContent ? [reasoningContent] : [], toolCallContent]
-                  };
-                }
-                const textContent = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "text" && content.text) : void 0;
-                if (textContent) {
                   return {
                     ...acc,
-                    content: [..._content, ...reasoningContent ? [reasoningContent] : [], textContent]
+                    content: [..._content, toolResult]
                   };
                 }
-              }
-              if (message2.role === "tool") {
-                const toolResult = Array.isArray(message2.content) ? message2.content.find((content) => content.type === "tool-result") : void 0;
-                if (toolResult) {
-                  const newContent = _content.map((c) => {
-                    if (c.type === "tool-call" && c.toolCallId === toolResult?.toolCallId) {
-                      return { ...c, result: toolResult.result };
-                    }
-                    return c;
-                  });
-                  const containsToolCall = newContent.some((c) => c.type === "tool-call");
-                  return {
-                    ...acc,
-                    content: containsToolCall ? newContent : [
-                      ..._content,
-                      { type: "tool-result", toolCallId: toolResult.toolCallId, result: toolResult.result }
-                    ]
-                  };
+                return acc;
+              },
+              { role: "assistant", content: [] }
+            );
+            setMessages((currentConversation) => [...currentConversation, latestMessage]);
+            handleFinishReason(generateResponse.finishReason);
+          }
+        } else {
+          let updater = function() {
+            setMessages((currentConversation) => {
+              const message2 = {
+                role: "assistant",
+                content: [{ type: "text", text: content }]
+              };
+              if (!assistantMessageAdded) {
+                assistantMessageAdded = true;
+                if (assistantToolCallAddedForUpdater) {
+                  assistantToolCallAddedForUpdater = false;
                 }
-                return {
-                  ...acc,
-                  content: [..._content, toolResult]
-                };
+                return [...currentConversation, message2];
               }
-              return acc;
-            },
-            { role: "assistant", content: [] }
-          );
-          setMessages((currentConversation) => [...currentConversation, latestMessage]);
-          handleFinishReason(generateResponse.finishReason);
-        }
-      } else {
-        let updater = function() {
-          setMessages((currentConversation) => {
-            const message2 = {
-              role: "assistant",
-              content: [{ type: "text", text: content }]
-            };
-            if (!assistantMessageAdded) {
-              assistantMessageAdded = true;
               if (assistantToolCallAddedForUpdater) {
                 assistantToolCallAddedForUpdater = false;
+                return [...currentConversation, message2];
               }
-              return [...currentConversation, message2];
-            }
-            if (assistantToolCallAddedForUpdater) {
-              assistantToolCallAddedForUpdater = false;
-              return [...currentConversation, message2];
-            }
-            return [...currentConversation.slice(0, -1), message2];
+              return [...currentConversation.slice(0, -1), message2];
+            });
+          };
+          const response = await agent.stream({
+            messages: [
+              {
+                role: "user",
+                content: input
+              },
+              ...attachments
+            ],
+            runId: agentId,
+            frequencyPenalty,
+            presencePenalty,
+            maxRetries,
+            maxSteps,
+            maxTokens,
+            temperature,
+            topK,
+            topP,
+            instructions,
+            runtimeContext: runtimeContextInstance,
+            ...memory ? { threadId, resourceId: agentId } : {},
+            providerOptions
           });
-        };
-        const response = await agent.stream({
-          messages: [
-            {
-              role: "user",
-              content: input
+          if (!response.body) {
+            throw new Error("No response body");
+          }
+          let content = "";
+          let assistantMessageAdded = false;
+          let assistantToolCallAddedForUpdater = false;
+          let assistantToolCallAddedForContent = false;
+          await response.processDataStream({
+            onTextPart(value) {
+              if (assistantToolCallAddedForContent) {
+                assistantToolCallAddedForContent = false;
+                content = value;
+              } else {
+                content += value;
+              }
+              updater();
             },
-            ...attachments
-          ],
-          runId: agentId,
-          frequencyPenalty,
-          presencePenalty,
-          maxRetries,
-          maxSteps,
-          maxTokens,
-          temperature,
-          topK,
-          topP,
-          instructions,
-          runtimeContext: runtimeContextInstance,
-          ...memory ? { threadId, resourceId: agentId } : {},
-          providerOptions
-        });
-        if (!response.body) {
-          throw new Error("No response body");
-        }
-        let content = "";
-        let assistantMessageAdded = false;
-        let assistantToolCallAddedForUpdater = false;
-        let assistantToolCallAddedForContent = false;
-        await response.processDataStream({
-          onTextPart(value) {
-            if (assistantToolCallAddedForContent) {
-              assistantToolCallAddedForContent = false;
-              content = value;
-            } else {
-              content += value;
-            }
-            updater();
-          },
-          async onToolCallPart(value) {
-            setMessages((currentConversation) => {
-              const lastMessage = currentConversation[currentConversation.length - 1];
-              if (lastMessage && lastMessage.role === "assistant") {
-                const updatedMessage = {
-                  ...lastMessage,
-                  content: Array.isArray(lastMessage.content) ? [
-                    ...lastMessage.content,
-                    {
-                      type: "tool-call",
-                      toolCallId: value.toolCallId,
-                      toolName: value.toolName,
-                      args: value.args
-                    }
-                  ] : [
-                    ...typeof lastMessage.content === "string" ? [{ type: "text", text: lastMessage.content }] : [],
+            async onToolCallPart(value) {
+              setMessages((currentConversation) => {
+                const lastMessage = currentConversation[currentConversation.length - 1];
+                if (lastMessage && lastMessage.role === "assistant") {
+                  const updatedMessage = {
+                    ...lastMessage,
+                    content: Array.isArray(lastMessage.content) ? [
+                      ...lastMessage.content,
+                      {
+                        type: "tool-call",
+                        toolCallId: value.toolCallId,
+                        toolName: value.toolName,
+                        args: value.args
+                      }
+                    ] : [
+                      ...typeof lastMessage.content === "string" ? [{ type: "text", text: lastMessage.content }] : [],
+                      {
+                        type: "tool-call",
+                        toolCallId: value.toolCallId,
+                        toolName: value.toolName,
+                        args: value.args
+                      }
+                    ]
+                  };
+                  assistantToolCallAddedForUpdater = true;
+                  assistantToolCallAddedForContent = true;
+                  return [...currentConversation.slice(0, -1), updatedMessage];
+                }
+                const newMessage = {
+                  role: "assistant",
+                  content: [
+                    { type: "text", text: content },
                     {
                       type: "tool-call",
                       toolCallId: value.toolCallId,
@@ -5971,95 +5989,80 @@ function MastraRuntimeProvider({
                 };
                 assistantToolCallAddedForUpdater = true;
                 assistantToolCallAddedForContent = true;
-                return [...currentConversation.slice(0, -1), updatedMessage];
+                return [...currentConversation, newMessage];
+              });
+              toolCallIdToName.current[value.toolCallId] = value.toolName;
+            },
+            async onToolResultPart(value) {
+              setMessages((currentConversation) => {
+                const lastMessage = currentConversation[currentConversation.length - 1];
+                if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
+                  const updatedContent = lastMessage.content.map((part) => {
+                    if (typeof part === "object" && part.type === "tool-call" && part.toolCallId === value.toolCallId) {
+                      return {
+                        ...part,
+                        result: value.result
+                      };
+                    }
+                    return part;
+                  });
+                  const updatedMessage = {
+                    ...lastMessage,
+                    content: updatedContent
+                  };
+                  return [...currentConversation.slice(0, -1), updatedMessage];
+                }
+                return currentConversation;
+              });
+              try {
+                const toolName = toolCallIdToName.current[value.toolCallId];
+                if (toolName === "updateWorkingMemory" && value.result?.success) {
+                  await refreshWorkingMemory?.();
+                }
+              } finally {
+                delete toolCallIdToName.current[value.toolCallId];
               }
-              const newMessage = {
-                role: "assistant",
-                content: [
-                  { type: "text", text: content },
-                  {
-                    type: "tool-call",
-                    toolCallId: value.toolCallId,
-                    toolName: value.toolName,
-                    args: value.args
-                  }
-                ]
-              };
-              assistantToolCallAddedForUpdater = true;
-              assistantToolCallAddedForContent = true;
-              return [...currentConversation, newMessage];
-            });
-            toolCallIdToName.current[value.toolCallId] = value.toolName;
-          },
-          async onToolResultPart(value) {
-            setMessages((currentConversation) => {
-              const lastMessage = currentConversation[currentConversation.length - 1];
-              if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
-                const updatedContent = lastMessage.content.map((part) => {
-                  if (typeof part === "object" && part.type === "tool-call" && part.toolCallId === value.toolCallId) {
-                    return {
-                      ...part,
-                      result: value.result
-                    };
-                  }
-                  return part;
-                });
-                const updatedMessage = {
-                  ...lastMessage,
-                  content: updatedContent
+            },
+            onErrorPart(error) {
+              throw new Error(error);
+            },
+            onFinishMessagePart({ finishReason }) {
+              handleFinishReason(finishReason);
+            },
+            onReasoningPart(value) {
+              setMessages((currentConversation) => {
+                const lastMessage = currentConversation[currentConversation.length - 1];
+                if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
+                  const updatedContent = lastMessage.content.map((part) => {
+                    if (typeof part === "object" && part.type === "reasoning") {
+                      return {
+                        ...part,
+                        text: part.text + value
+                      };
+                    }
+                    return part;
+                  });
+                  const updatedMessage = {
+                    ...lastMessage,
+                    content: updatedContent
+                  };
+                  return [...currentConversation.slice(0, -1), updatedMessage];
+                }
+                const newMessage = {
+                  role: "assistant",
+                  content: [
+                    {
+                      type: "reasoning",
+                      text: value
+                    },
+                    { type: "text", text: content }
+                  ]
                 };
-                return [...currentConversation.slice(0, -1), updatedMessage];
-              }
-              return currentConversation;
-            });
-            try {
-              const toolName = toolCallIdToName.current[value.toolCallId];
-              if (toolName === "updateWorkingMemory" && value.result?.success) {
-                await refreshWorkingMemory?.();
-              }
-            } finally {
-              delete toolCallIdToName.current[value.toolCallId];
+                return [...currentConversation, newMessage];
+              });
             }
-          },
-          onErrorPart(error) {
-            throw new Error(error);
-          },
-          onFinishMessagePart({ finishReason }) {
-            handleFinishReason(finishReason);
-          },
-          onReasoningPart(value) {
-            setMessages((currentConversation) => {
-              const lastMessage = currentConversation[currentConversation.length - 1];
-              if (lastMessage && lastMessage.role === "assistant" && Array.isArray(lastMessage.content)) {
-                const updatedContent = lastMessage.content.map((part) => {
-                  if (typeof part === "object" && part.type === "reasoning") {
-                    return {
-                      ...part,
-                      text: part.text + value
-                    };
-                  }
-                  return part;
-                });
-                const updatedMessage = {
-                  ...lastMessage,
-                  content: updatedContent
-                };
-                return [...currentConversation.slice(0, -1), updatedMessage];
-              }
-              const newMessage = {
-                role: "assistant",
-                content: [
-                  {
-                    type: "reasoning",
-                    text: value
-                  },
-                  { type: "text", text: content }
-                ]
-              };
-              return [...currentConversation, newMessage];
-            });
-          }
-        });
+          });
+        }
       }
       setIsRunning(false);
       setTimeout(() => {
@@ -6109,7 +6112,8 @@ const defaultSettings = {
     maxSteps: 5,
     temperature: 0.5,
     topP: 1,
-    chatWithGenerate: false
+    chatWithGenerate: false,
+    chatWithGenerateVNext: false
   }
 };
 function useAgentSettingsState({ agentId }) {
@@ -6181,7 +6185,8 @@ const AgentChat = ({
   initialMessages,
   memory,
   refreshThreadList,
-  onInputChange
+  onInputChange,
+  modelVersion
 }) => {
   const { settings } = useAgentSettings();
   const { runtimeContext } = usePlaygroundStore();
@@ -6190,6 +6195,7 @@ const AgentChat = ({
     {
       agentId,
       agentName,
+      modelVersion,
       threadId,
       initialMessages,
       memory,