Refactor data_type fields and remove unused classes

Updated Field definitions to include Literal=True for data_type attributes in multiple classes. Removed ActionRequest and HumanFeedback classes.

Author: NirajC-Microsoft

src/backend/common/models/messages_kernel.py

@@ -87,7 +87,7 @@ class BaseDataModel(KernelBaseModel):
 class AgentMessage(BaseDataModel):
     """Base class for messages sent between agents."""
 
-    data_type: Literal[DataType.agent_message] = Field(DataType.agent_message)
+    data_type: Literal[DataType.agent_message] = Field(DataType.agent_message, Literal=True)
     plan_id: str
     content: str
     source: str
@@ -98,7 +98,7 @@ class AgentMessage(BaseDataModel):
 class Session(BaseDataModel):
     """Represents a user session."""
 
-    data_type: Literal[DataType.session] = Field(DataType.session)
+    data_type: Literal[DataType.session] = Field(DataType.session, Literal=True)
     user_id: str
     current_status: str
     message_to_user: Optional[str] = None
@@ -107,16 +107,16 @@ class Session(BaseDataModel):
 class UserCurrentTeam(BaseDataModel):
     """Represents the current team of a user."""
 
-    data_type: Literal[DataType.user_current_team] = Field(DataType.user_current_team)
+    data_type: Literal[DataType.user_current_team] = Field(DataType.user_current_team, Literal=True)
     user_id: str
     team_id: str
 
 
 class Plan(BaseDataModel):
     """Represents a plan containing multiple steps."""
 
-    data_type: Literal[DataType.plan] = Field(DataType.plan)
-    plan_id: str = Field(default_factory=lambda: str(uuid.uuid4()))
+    data_type: Literal[DataType.plan] = Field(DataType.plan, Literal=True)
+    plan_id: str
     user_id: str
     initial_goal: str
     overall_status: PlanStatus = PlanStatus.in_progress
@@ -133,7 +133,7 @@ class Plan(BaseDataModel):
 class Step(BaseDataModel):
     """Represents an individual step (task) within a plan."""
 
-    data_type: Literal[DataType.step] = Field(DataType.step)
+    data_type: Literal[DataType.step] = Field(DataType.step, Literal=True)
     plan_id: str
     user_id: str
     action: str
@@ -184,7 +184,7 @@ class TeamConfiguration(BaseDataModel):
     """Represents a team configuration stored in the database."""
 
     team_id: str
-    data_type: Literal[DataType.team_config] = Field(DataType.team_config)
+    data_type: Literal[DataType.team_config] = Field(DataType.team_config, Literal=True)
     session_id: str  # Partition key
     name: str
     status: str
@@ -260,27 +260,9 @@ class AgentMessageType(str, Enum):
     AI_AGENT = "AI_Agent",
 
 
-class ActionRequest(BaseDataModel):
-    """Represents an action request for a specific step."""
-
-    step_id: str
-    plan_id: str
-    action: str
-    agent: AgentType
-
-
-class HumanFeedback(BaseDataModel):
-    """Represents human feedback for a specific step."""
-
-    step_id: str
-    plan_id: str
-    approved: bool
-    human_feedback: str
-
-
 class AgentMessageData (BaseDataModel):
 
-    data_type: Literal[DataType.m_plan_message] = Field(DataType.m_plan_message)
+    data_type: Literal[DataType.m_plan_message] = Field(DataType.m_plan_message, Literal=True)
     plan_id: str
     user_id: str
     agent: str
@@ -290,4 +272,4 @@ class AgentMessageData (BaseDataModel):
     raw_data: str
     steps: List[Any] = Field(default_factory=list)       
     next_steps: List[Any] = Field(default_factory=list)  
-    
+