Skip to content

Commit 3442cf4

Browse files
committed
Add SimpleChat agent type and factory support
1 parent 8697c6c commit 3442cf4

File tree

3 files changed

+16
-0
lines changed

3 files changed

+16
-0
lines changed

src/backend/kernel_agents/agent_factory.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from kernel_agents.procurement_agent import ProcurementAgent
2121
from kernel_agents.product_agent import ProductAgent
2222
from kernel_agents.tech_support_agent import TechSupportAgent
23+
from kernel_agents.simple_chat_agent import SimpleChatAgent
2324
from models.messages_kernel import AgentType, PlannerResponsePlan
2425
# pylint:disable=E0611
2526
from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
@@ -41,6 +42,7 @@ class AgentFactory:
4142
AgentType.HUMAN: HumanAgent,
4243
AgentType.PLANNER: PlannerAgent,
4344
AgentType.GROUP_CHAT_MANAGER: GroupChatManager, # Add GroupChatManager
45+
AgentType.SIMPLE_CHAT: SimpleChatAgent,
4446
}
4547

4648
# Mapping of agent types to their string identifiers (for automatic tool loading)
@@ -54,6 +56,7 @@ class AgentFactory:
5456
AgentType.HUMAN: AgentType.HUMAN.value,
5557
AgentType.PLANNER: AgentType.PLANNER.value,
5658
AgentType.GROUP_CHAT_MANAGER: AgentType.GROUP_CHAT_MANAGER.value,
59+
AgentType.SIMPLE_CHAT: AgentType.SIMPLE_CHAT.value,
5760
}
5861

5962
# System messages for each agent type
@@ -67,6 +70,7 @@ class AgentFactory:
6770
AgentType.HUMAN: HumanAgent.default_system_message(),
6871
AgentType.PLANNER: PlannerAgent.default_system_message(),
6972
AgentType.GROUP_CHAT_MANAGER: GroupChatManager.default_system_message(),
73+
AgentType.SIMPLE_CHAT: SimpleChatAgent.default_system_message(),
7074
}
7175

7276
# Cache of agent instances by session_id and agent_type
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
from kernel_agents.generic_agent import GenericAgent
2+
from models.messages_kernel import AgentType
3+
4+
5+
class SimpleChatAgent(GenericAgent):
6+
"""Agent for basic conversational interactions."""
7+
8+
@staticmethod
9+
def default_system_message(agent_name: str = AgentType.SIMPLE_CHAT.value) -> str:
10+
"""Get the default system message for the SimpleChatAgent."""
11+
return "You are a Simple Chat agent that engages in basic conversation and provides helpful responses."

src/backend/models/messages_kernel.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ class AgentType(str, Enum):
4848
TECH_SUPPORT = "Tech_Support_Agent"
4949
GROUP_CHAT_MANAGER = "Group_Chat_Manager"
5050
PLANNER = "Planner_Agent"
51+
SIMPLE_CHAT = "Simple_Chat_Agent"
5152

5253
# Add other agents as needed
5354

0 commit comments

Comments
 (0)