fix missing messages

Author: Fr4nc3

src/backend/app_kernel.py

@@ -620,9 +620,10 @@ async def get_plans(
 
         # Use get_steps_by_plan to match the original implementation
         steps = await memory_store.get_steps_by_plan(plan_id=plan.id)
-        messages = await memory_store.get_data_by_type_and_plan_id(
-            "agent_message", plan_id=plan.id
+        messages = await memory_store.get_data_by_type_and_session_id(
+            "agent_message", session_id=plan.session_id
         )
+
         plan_with_steps = PlanWithSteps(**plan.model_dump(), steps=steps)
         plan_with_steps.update_step_counts()
         return [plan_with_steps, messages]

src/backend/context/cosmos_memory_kernel.py

@@ -443,8 +443,8 @@ async def get_data_by_type(self, data_type: str) -> List[BaseDataModel]:
             logging.exception(f"Failed to query data by type from Cosmos DB: {e}")
             return []
 
-    async def get_data_by_type_and_plan_id(
-        self, data_type: str, plan_id: str
+    async def get_data_by_type_and_session_id(
+        self, data_type: str, session_id: str
     ) -> List[BaseDataModel]:
         """Query the Cosmos DB for documents with the matching data_type, session_id and user_id."""
         await self.ensure_initialized()
@@ -453,11 +453,11 @@ async def get_data_by_type_and_plan_id(
 
         model_class = self.MODEL_CLASS_MAPPING.get(data_type, BaseDataModel)
         try:
-            query = "SELECT * FROM c WHERE c.user_id=@user_id AND c.data_type=@data_type AND c.plan_id=@plan_id ORDER BY c._ts ASC"
+            query = "SELECT * FROM c WHERE c.session_id=@session_id AND c.user_id=@user_id AND c.data_type=@data_type ORDER BY c._ts ASC"
             parameters = [
+                {"name": "@session_id", "value": session_id},
                 {"name": "@data_type", "value": data_type},
                 {"name": "@user_id", "value": self.user_id},
-                {"name": "@plan_id", "value": plan_id},
             ]
             return await self.query_items(query, parameters, model_class)
         except Exception as e:

src/frontend_react/src/components/content/PlanChat.tsx

@@ -3,7 +3,7 @@ import { Copy, Send } from "@/coral/imports/bundleicons";
 import ChatInput from "@/coral/modules/ChatInput";
 import remarkGfm from "remark-gfm";
 import rehypePrism from "rehype-prism";
-import { PlanChatProps } from "@/models";
+import { AgentType, PlanChatProps, role } from "@/models";
 import {
     Body1,
     Button,
@@ -48,10 +48,10 @@ const PlanChat: React.FC<PlanChatProps> = ({
             <div className="messages" ref={messagesContainerRef}>
                 <div className="message-wrapper">
                     {messages.map((msg, index) => {
-                        const isHuman = msg.source.includes("human");
+                        const isHuman = msg.source === AgentType.HUMAN;
 
                         return (
-                            <div key={index} className={`message ${isHuman ? "user" : "assistant"}`}>
+                            <div key={index} className={`message ${isHuman ? role.user : role.assistant}`}>
                                 {!isHuman && (
                                     <div className="plan-chat-header">
                                         <div className="plan-chat-speaker">
@@ -119,7 +119,7 @@ const PlanChat: React.FC<PlanChatProps> = ({
                             appearance="transparent"
                             onClick={() => OnChatSubmit(inputValue)}
                             icon={<Send />}
-                            disabled={!planData?.enableChat}
+                            disabled={!planData?.enableChat && !inputValue.trim()}
                         />
                     </ChatInput>
                 </div>

src/frontend_react/src/models/enums.tsx

@@ -17,6 +17,10 @@ export enum AgentType {
     PLANNER = "Planner_Agent"
 }
 
+export enum role {
+    user = "user",
+    assistant = "assistant"
+}
 /**
  * Enumeration of possible statuses for a step.
  */