Merge pull request #139 from kagent-dev/peterj/statefix

show new chat button in case of an error, fix duplicate messages

Author: EItanya

ui/src/components/chat/ChatInterface.tsx

@@ -90,7 +90,7 @@ export default function ChatInterface({ selectedTeamId, selectedRun }: ChatInter
   const displayMessages = selectedRun ? selectedRun.messages : messages;
   const actualRun = selectedRun || run || null;
   const runStatus = actualRun?.status;
-  const canSendMessage = status !== "thinking" && runStatus !== "complete" && runStatus !== "error" && runStatus !== "stopped";
+  const canSendMessage = status !== "thinking" && status !== "error" && runStatus !== "complete" && runStatus !== "error" && runStatus !== "stopped";
 
   // Should we show the streaming message?
   const showStreamingMessage = !selectedRun && currentStreamingContent && currentStreamingMessage;
@@ -145,7 +145,7 @@ export default function ChatInterface({ selectedTeamId, selectedRun }: ChatInter
               </Button>
             )}
 
-            {(runStatus === "complete" || runStatus === "error" || runStatus === "stopped") && (
+            {(runStatus === "complete" || runStatus === "error" || runStatus === "stopped" || status === 'error') && (
               <Button className="bg-violet-500 hover:bg-violet-600" asChild>
                 <Link href={`/agents/${selectedTeamId}/chat`}>
                 Start New Chat

ui/src/components/sidebars/SessionGroup.tsx

@@ -14,7 +14,6 @@ interface SessionGroupProps {
 
 // The sessions are grouped by today, yesterday, and older
 const SessionGroup = ({ title, sessions, onDeleteSession, agentId }: SessionGroupProps) => {
-  console.log("title", title);
   return (
     <SidebarGroup>
       <SidebarMenu>

ui/src/lib/useChatStore.ts

@@ -181,6 +181,11 @@ const useChatStore = create<ChatState>((set, get) => ({
             message_meta: {},
           };
 
+          // Check the finalMessage is not included in the messages array (avoid duplicates)
+          if (state.messages.find((m) => m.config.source === finalMessage.config.source && m.config.content === finalMessage.config.content)) {
+            return state;
+          }
+
           const finalMessages = [...state.messages, finalMessage];
 
           const updatedRun = {